Django tests: create unmanaged modules using a pre_migrate signal

Author: celine-m-s

django/config/settings/base.py

@@ -171,3 +171,4 @@
 ##########################################
 
 UPSTREAM_NUXT = env.str("UPSTREAM_NUXT")
+CREATE_UNMANAGED_TABLES = False

django/config/settings/test.py

@@ -6,3 +6,4 @@
 )
 SESSION_COOKIE_SECURE = False
 SECURE_SSL_REDIRECT = False
+CREATE_UNMANAGED_TABLES = True

django/docurba/core/apps.py

@@ -1,6 +1,31 @@
+from typing import Any
+
 from django.apps import AppConfig
+from django.conf import settings
+from django.db import connection, models
 
 
 class CoreAppConfig(AppConfig):
     name = "docurba.core"
     verbose_name = "Cœur"
+
+    def ready(self) -> None:
+        super().ready()
+        models.signals.pre_migrate.connect(create_unmanaged_tables, sender=self)
+
+
+def create_unmanaged_tables(*args: list[str, Any], **kwargs: dict[str, Any]) -> None:  # noqa: ARG001
+    if not settings.CREATE_UNMANAGED_TABLES:
+        return
+
+    from docurba.core.models import (  # noqa: PLC0415
+        CommuneProcedure,
+        Event,
+        Procedure,
+    )
+
+    unmanaged_core_models = [Procedure, CommuneProcedure, Event]
+
+    for model in unmanaged_core_models:
+        with connection.schema_editor() as schema_editor:
+            schema_editor.create_model(model)

django/docurba/core/migrations/0001_initial.py

@@ -10,7 +10,9 @@
 class Migration(migrations.Migration):
     initial = True
 
-    dependencies = []
+    dependencies = [
+        ("users", "0001_initial"),
+    ]
 
     operations = [
         migrations.CreateModel(

django/docurba/core/models.py

@@ -371,6 +371,7 @@ class Procedure(models.Model):
     objects = ProcedureManager.from_queryset(ProcedureQuerySet)()
 
     class Meta:
+        # Table created by a pre_migrate signal in apps.py.
         managed = False
         db_table = "procedures"
         constraints = (
@@ -585,6 +586,7 @@ class Event(models.Model):
     visibility = models.TextField(db_default="public")
 
     class Meta:
+        # Table created by a pre_migrate signal in apps.py.
         managed = False
         db_table = "doc_frise_events"
         ordering = ("-date_evenement",)
@@ -922,6 +924,7 @@ class CommuneProcedure(models.Model):  # noqa: DJ008
     procedure = models.ForeignKey(Procedure, models.DO_NOTHING)
 
     class Meta:
+        # Table created by a pre_migrate signal in apps.py.
         managed = False
         db_table = "procedures_perimetres"
         verbose_name = "Périmètre"

django/docurba/users/apps.py

@@ -1,7 +1,31 @@
+from typing import Any
+
 from django.apps import AppConfig
+from django.conf import settings
+from django.db import connection, models
 
 
 class UsersConfig(AppConfig):
     default_auto_field = "django.db.models.BigAutoField"
     name = "docurba.users"
     verbose_name = "Utilisateurs"
+
+    def ready(self) -> None:
+        super().ready()
+        models.signals.pre_migrate.connect(create_unmanaged_tables, sender=self)
+
+
+def create_unmanaged_tables(*args: list[str, Any], **kwargs: dict[str, Any]) -> None:  # noqa: ARG001
+    if not settings.CREATE_UNMANAGED_TABLES:
+        return
+
+    from docurba.users.models import Profile, User  # noqa: PLC0415
+
+    unmanaged_user_models = [User, Profile]
+
+    with connection.schema_editor() as schema_editor:
+        schema_editor.execute("CREATE SCHEMA auth;")
+
+    for model in unmanaged_user_models:
+        with connection.schema_editor() as schema_editor:
+            schema_editor.create_model(model)

django/docurba/users/models.py

@@ -17,6 +17,7 @@ class User(models.Model):
     email = models.EmailField(verbose_name="Email", blank=True)
 
     class Meta:
+        # Table created by a pre_migrate signal in apps.py.
         managed = False
         db_table = '"auth"."users"'
         verbose_name = "utilisateur"
@@ -123,6 +124,7 @@ class Profile(models.Model):
     )
 
     class Meta:
+        # Table created by a pre_migrate signal in apps.py.
         managed = False
         db_table = "profiles"
         verbose_name = "profil"