Skip to content

Commit ef49313

Browse files
committed
Add encrypted model stubs
1 parent 7d5e39b commit ef49313

File tree

3 files changed

+18
-1
lines changed

3 files changed

+18
-1
lines changed

django_mongodb_backend/models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,8 @@ def delete(self, *args, **kwargs):
1414

1515
def save(self, *args, **kwargs):
1616
raise NotSupportedError("EmbeddedModels cannot be saved.")
17+
18+
19+
class EncryptedModel(models.Model):
20+
class Meta:
21+
abstract = True

tests/model_fields_/models.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
EmbeddedModelField,
99
ObjectIdField,
1010
)
11-
from django_mongodb_backend.models import EmbeddedModel
11+
from django_mongodb_backend.models import EmbeddedModel, EncryptedModel
1212

1313

1414
# ObjectIdField
@@ -136,6 +136,10 @@ class Author(EmbeddedModel):
136136
skills = ArrayField(models.CharField(max_length=100), null=True, blank=True)
137137

138138

139+
class EncryptedData(EncryptedModel):
140+
pass
141+
142+
139143
class Book(models.Model):
140144
name = models.CharField(max_length=100)
141145
author = EmbeddedModelField(Author)
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from django.test import TestCase
2+
3+
from .models import EncryptedData
4+
5+
6+
class ModelTests(TestCase):
7+
def test_save_load(self):
8+
EncryptedData.objects.create()

0 commit comments

Comments
 (0)