test: prevent PSQL-only model from migrating to SQLite database

flamingo-run · May 7, 2024 · e0766b6 · e0766b6
1 parent 6610556
commit e0766b6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/test_app/database_routers.py b/test_app/database_routers.py
@@ -0,0 +1,5 @@
+class FeatureRouter:
+    def allow_migrate(self, db, app_label, model_name=None, **hints):
+        if model_name in {"reservation"}:  # PSQL-only
+            return db == "default"
+        return True
diff --git a/test_app/settings.py b/test_app/settings.py
@@ -88,6 +88,7 @@
         "NAME": str(BASE_DIR / "db.sqlite3"),
     },
 }
+DATABASE_ROUTERS = ["test_app.database_routers.FeatureRouter"]
 
 # Password validation
 # https://docs.djangoproject.com/en/3.1/ref/settings/#auth-password-validators