Add EncryptedArrayField + test

Author: aclark4life

.github/workflows/mongodb_settings.py

@@ -34,6 +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 allow_migrate(self, db, app_label, model_name=None, **hints):
         # The encryption_ app's models are only created in the encrypted
         # database.

django_mongodb_backend/features.py

@@ -639,7 +639,7 @@ def _supports_transactions(self):
     @cached_property
     def supports_queryable_encryption(self):
         """
-        Queryable Encryption requires a MongoDB 7.0 or later replica set or sharded
+        Queryable Encryption requires a MongoDB 8.0 or later replica set or sharded
         cluster, as well as MongoDB Atlas or Enterprise.
         """
         self.connection.ensure_connection()

django_mongodb_backend/fields/__init__.py

@@ -4,6 +4,7 @@
 from .embedded_model import EmbeddedModelField
 from .embedded_model_array import EmbeddedModelArrayField
 from .encryption import (
+    EncryptedArrayField,
     EncryptedBigIntegerField,
     EncryptedBinaryField,
     EncryptedBooleanField,
@@ -36,6 +37,7 @@
     "ArrayField",
     "EmbeddedModelArrayField",
     "EmbeddedModelField",
+    "EncryptedArrayField",
     "EncryptedBigIntegerField",
     "EncryptedBinaryField",
     "EncryptedBooleanField",

django_mongodb_backend/fields/encryption.py

@@ -1,6 +1,10 @@
 from django.db import models
 
-from django_mongodb_backend.fields import EmbeddedModelArrayField, EmbeddedModelField
+from django_mongodb_backend.fields import ArrayField, EmbeddedModelArrayField, EmbeddedModelField
+
+
+class EncryptedArrayField(ArrayField):
+    encrypted = True
 
 
 class EncryptedEmbeddedModelArrayField(EmbeddedModelArrayField):

tests/encryption_/models.py

@@ -2,6 +2,7 @@
 
 from django_mongodb_backend.fields import (
     EmbeddedModelField,
+    EncryptedArrayField,
     EncryptedBigIntegerField,
     EncryptedBinaryField,
     EncryptedBooleanField,
@@ -33,6 +34,14 @@ class Meta:
         required_db_features = {"supports_queryable_encryption"}
 
 
+# Array models
+class ArrayModel(EncryptedTestModel):
+    values = EncryptedArrayField(
+        models.IntegerField(),
+        size=5,
+    )
+
+
 # Embedded models
 class Patient(EncryptedTestModel):
     patient_name = models.CharField(max_length=255)

tests/encryption_/test_fields.py

@@ -5,6 +5,7 @@
 
 from .models import (
     Actor,
+    ArrayModel,
     BigIntegerModel,
     Billing,
     BinaryModel,
@@ -32,6 +33,16 @@
 from .test_base import EncryptionTestCase
 
 
+class ArrayModelTests(EncryptionTestCase):
+    def setUp(self):
+        self.array_model = ArrayModel.objects.create(values=[1, 2, 3, 4, 5])
+
+    def test_array(self):
+        array_model = ArrayModel.objects.get(id=self.array_model.id)
+        self.assertEqual(array_model.values, [1, 2, 3, 4, 5])
+        # self.assertEncrypted(self.array_model, "values")
+
+
 class EmbeddedModelTests(EncryptionTestCase):
     def setUp(self):
         self.billing = Billing(cc_type="Visa", cc_number="4111111111111111")