Add assertEncrypted to verify field data is binary

Author: aclark4life

tests/encryption_/test_base.py

@@ -1,7 +1,21 @@
+import pymongo
+from bson.binary import Binary
+from django.conf import settings
+from django.db import connections
 from django.test import TestCase, skipUnlessDBFeature
 
 
 @skipUnlessDBFeature("supports_queryable_encryption")
 class EncryptionTestCase(TestCase):
     databases = {"default", "encrypted"}
     maxDiff = None
+
+    def assertEncrypted(self, model, field):
+        # Access encrypted database from an unencrypted connection
+        conn_params = connections["default"].get_connection_params()
+        db_name = settings.DATABASES["encrypted"]["NAME"]
+        with pymongo.MongoClient(**conn_params) as new_connection:
+            db = new_connection[db_name]
+            collection = db[model._meta.db_table]
+            data = collection.find_one({}, {field: 1, "_id": 0})
+            self.assertIsInstance(data[field], Binary)

tests/encryption_/test_fields.py

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