Fix Router.db_for_read()/write()

Author: timgraham

.github/workflows/mongodb_settings.py

@@ -34,11 +34,12 @@
 
 
 class EncryptedRouter:
-    # def db_for_read(self, model, **hints):
-    #     if model_has_encrypted_fields(model):
-    #         return "encrypted"
-    #     return "default"
-    # db_for_write = db_for_read
+    def db_for_read(self, model, **hints):
+        if model._meta.app_label == "encryption_":
+            return "encrypted"
+        return None
+
+    db_for_write = db_for_read
 
     def allow_migrate(self, db, app_label, model_name=None, **hints):
         # The encryption_ app's models are only created in the encrypted

tests/encryption_/test_fields.py

@@ -77,7 +77,7 @@ def test_array(self):
         self.assertEqual(len(movie.cast), 2)
         self.assertEqual(movie.cast[0].name, "Actor One")
         self.assertEqual(movie.cast[1].name, "Actor Two")
-        # self.assertEncrypted(self.movie, "cast")
+        self.assertEncrypted(self.movie, "cast")
 
 
 class FieldTests(EncryptionTestCase):