Fix router tests

Author: aclark4life

tests/encryption_/routers.py

@@ -1,26 +1,13 @@
-from django.db import connection
-
-
 class EncryptedRouter:
     """
     Routes database operations for encrypted models to the encrypted DB.
     """
 
     def db_for_read(self, model, **hints):
-        with connection.schema_editor() as editor:
-            if model and getattr(editor._get_encrypted_fields_map(model), False):
-                return "encrypted"
-        return None
+        return "encrypted"
 
     def db_for_write(self, model, **hints):
-        with connection.schema_editor() as editor:
-            if model and getattr(editor._get_encrypted_fields_map(model), False):
-                return "encrypted"
-        return None
+        return "encrypted"
 
     def allow_migrate(self, db, app_label, model_name=None, **hints):
-        model = hints.get("model")
-        with connection.schema_editor() as editor:
-            if model and getattr(editor._get_encrypted_fields_map(model), False):
-                return db == "encrypted"
-        return None
+        return "encrypted"

tests/encryption_/tests.py

@@ -5,8 +5,10 @@
 from .routers import EncryptedRouter
 
 
-@override_settings(DATABASE_ROUTERS=[EncryptedRouter])
+@override_settings(DATABASE_ROUTERS=[EncryptedRouter()])
 class EncryptedModelTests(TestCase):
+    databases = {"default", "encrypted"}
+
     @classmethod
     def setUpTestData(cls):
         cls.person = Person(ssn="123-45-6789")