Add EncryptedUUIDField

Author: aclark4life

django_mongodb_backend/fields/__init__.py

@@ -27,6 +27,7 @@
     EncryptedTextField,
     EncryptedTimeField,
     EncryptedURLField,
+    EncryptedUUIDField,
 )
 from .json import register_json_field
 from .objectid import ObjectIdField
@@ -60,6 +61,7 @@
     "EncryptedTextField",
     "EncryptedTimeField",
     "EncryptedURLField",
+    "EncryptedUUIDField",
     "ObjectIdAutoField",
     "ObjectIdField",
     "PolymorphicEmbeddedModelArrayField",

django_mongodb_backend/fields/encryption.py

@@ -113,3 +113,7 @@ class EncryptedTimeField(EncryptedFieldMixin, models.TimeField):
 
 class EncryptedURLField(EncryptedFieldMixin, models.URLField):
     pass
+
+
+class EncryptedUUIDField(EncryptedFieldMixin, models.UUIDField):
+    pass

tests/encryption_/models.py

@@ -24,6 +24,7 @@
     EncryptedTextField,
     EncryptedTimeField,
     EncryptedURLField,
+    EncryptedUUIDField,
 )
 from django_mongodb_backend.models import EmbeddedModel
 
@@ -108,6 +109,10 @@ class URLModel(EncryptedTestModel):
     value = EncryptedURLField(max_length=500, queries={"queryType": "equality"})
 
 
+class UUIDModel(EncryptedTestModel):
+    value = EncryptedUUIDField(queries={"queryType": "equality"})
+
+
 # Range-queryable field models
 class BigIntegerModel(EncryptedTestModel):
     value = EncryptedBigIntegerField(queries={"queryType": "range"})

tests/encryption_/test_fields.py

@@ -1,4 +1,5 @@
 import datetime
+import uuid
 from decimal import Decimal
 
 from django_mongodb_backend.fields import EncryptedCharField
@@ -29,6 +30,7 @@
     TextModel,
     TimeModel,
     URLModel,
+    UUIDModel,
 )
 from .test_base import EncryptionTestCase
 
@@ -216,3 +218,12 @@ def test_deconstruct_preserves_queries_and_rewrites_path(self):
         self.assertEqual(new_field.queries, field.queries)
         self.assertIsNot(new_field, field)
         self.assertEqual(new_field.max_length, field.max_length)
+
+
+class UUIDFieldTests(EncryptionTestCase):
+    def test_uuid_field(self):
+        test_uuid = uuid.uuid4()
+        UUIDModel.objects.create(value=test_uuid)
+        fetched = UUIDModel.objects.get(value=test_uuid)
+        self.assertEqual(fetched.value, test_uuid)
+        self.assertEncrypted(fetched, "value")