aboutsummaryrefslogtreecommitdiff
path: root/tests/test_postgres.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_postgres.py')
-rw-r--r--tests/test_postgres.py26
1 files changed, 25 insertions, 1 deletions
diff --git a/tests/test_postgres.py b/tests/test_postgres.py
index 92eed71..3b3ddd0 100644
--- a/tests/test_postgres.py
+++ b/tests/test_postgres.py
@@ -1,9 +1,10 @@
import socket
import subprocess
+from typing import Callable
from pydantic import PostgresDsn
-from generalresearch.models.custom_types import InternalHostname
+from generalresearch.models.custom_types import InternalHostname, PostgresDict
from generalresearch.pg_helper import PostgresConfig
@@ -42,3 +43,26 @@ class TestPostgresDSN:
)
res = config.execute_sql_query(query="SELECT 1;")
assert len(res) == 1
+
+
+class TestPostgresDjangoCreation:
+
+ def test_ping(self, postgres_instance_dict: PostgresDict):
+ assert can_ping(host=postgres_instance_dict["host"])
+
+ def test_django_creation(
+ self,
+ django_db_factory: Callable[..., None],
+ ):
+
+ dsn = django_db_factory()
+ assert isinstance(dsn, PostgresDsn)
+
+ def test_django_tables(self, thl_web_rw: PostgresConfig):
+ res = thl_web_rw.execute_sql_query(query="""
+ SELECT COUNT(*)
+ FROM information_schema.tables
+ WHERE table_schema = 'public';
+ """)
+ assert len(res) == 1
+ assert res[0]["count"] == 56