Tweaks

Author: palewire

tests/conftest.py

@@ -1,6 +1,9 @@
 import os
 from pathlib import Path
+import pytest
 from django.conf import settings
+from django.db import connections
+import django
 
 ROOT_DIR = Path(__file__).parent.parent
 PG_USER = os.environ.get("PG_USER", "postgres")
@@ -12,22 +15,22 @@ def pytest_configure():
             "default": {
                 "HOST": "localhost",
                 "PORT": 5432,
-                "NAME": "test",
+                "NAME": "postgres",
                 "USER": PG_USER,
                 "ENGINE": "django.db.backends.postgresql_psycopg2",
             },
             "other": {
                 "HOST": "localhost",
                 "PORT": 5432,
-                "NAME": "test_alternative",
+                "NAME": "postgres",
                 "USER": PG_USER,
                 "ENGINE": "django.db.backends.postgresql_psycopg2",
             },
             "sqlite": {"NAME": "sqlite", "ENGINE": "django.db.backends.sqlite3"},
             "secondary": {
                 "HOST": "localhost",
                 "PORT": 5432,
-                "NAME": "test_secondary",
+                "NAME": "postgres",
                 "USER": PG_USER,
                 "ENGINE": "django.db.backends.postgresql_psycopg2",
             },
@@ -60,3 +63,64 @@ def pytest_configure():
             },
         },
     )
+
+    # Initialize Django
+    django.setup()
+
+
+@pytest.fixture(scope="session")
+def django_db_setup(django_db_setup, django_db_blocker):
+    """
+    Fixture to set up the test database with the required tables.
+    This extends the built-in django_db_setup fixture.
+    """
+
+    # Import all test models
+    from tests.test_models import (
+        MockObject,
+        MockFKObject,
+        MockBlankObject,
+        ExtendedMockObject,
+        LimitedMockObject,
+        OverloadMockObject,
+        SecondaryMockObject,
+        UniqueMockObject,
+    )
+
+    with django_db_blocker.unblock():
+        # Check if tables exist and create them if they don't
+        for alias in connections:
+            connection = connections[alias]
+
+            # Get a list of all tables in the database
+            with connection.cursor() as cursor:
+                if connection.vendor == "postgresql":
+                    cursor.execute(
+                        "SELECT tablename FROM pg_tables WHERE schemaname = 'public'"
+                    )
+                    existing_tables = {row[0] for row in cursor.fetchall()}
+                elif connection.vendor == "sqlite":
+                    cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
+                    existing_tables = {row[0] for row in cursor.fetchall()}
+                else:
+                    existing_tables = set()
+
+                # Create tables that don't exist
+                with connection.schema_editor() as schema_editor:
+                    for model in [
+                        MockObject,
+                        MockFKObject,
+                        MockBlankObject,
+                        ExtendedMockObject,
+                        LimitedMockObject,
+                        OverloadMockObject,
+                        SecondaryMockObject,
+                        UniqueMockObject,
+                    ]:
+                        table_name = model._meta.db_table
+                        if table_name not in existing_tables:
+                            try:
+                                schema_editor.create_model(model)
+                                print(f"Created table {table_name} in {alias}")
+                            except Exception as e:
+                                print(f"Error creating {table_name} in {alias}: {e}")