Add EncryptedArrayField + test

Author: aclark4life

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")