Skip to content

Commit 38fb110

Browse files
committed
Fix test for unencrypted field not in field map
1 parent bc52c8e commit 38fb110

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/encryption_/models.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,12 @@
1+
from django.db import models
2+
13
from django_mongodb_backend.fields import EncryptedCharField
24
from django_mongodb_backend.models import EncryptedModel
35

46

57
class Person(EncryptedModel):
68
ssn = EncryptedCharField("ssn", max_length=11)
9+
name = models.CharField("name", max_length=100)
710

811
def __str__(self):
912
return self.ssn

tests/encryption_/tests.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,5 @@ def test_encrypted_fields_map_on_instance(self):
2626
self.assertEqual(instance.encrypted_fields_map, expected)
2727

2828
def test_non_encrypted_fields_not_included(self):
29-
encrypted_field_names = Person.encrypted_fields_map.keys()
30-
self.assertNotIn("ssn", encrypted_field_names)
29+
encrypted_field_names = Person.encrypted_fields_map.get("fields").keys()
30+
self.assertNotIn("name", encrypted_field_names)

0 commit comments

Comments
 (0)