Re-add abstract base with required db features

Author: aclark4life

tests/encryption_/models.py

@@ -26,7 +26,13 @@
 from django_mongodb_backend.models import EmbeddedModel
 
 
-class Patient(models.Model):
+class EncryptedTestModel(models.Model):
+    class Meta:
+        abstract = True
+        required_db_features = {"supports_query_encryption"}
+
+
+class Patient(EncryptedTestModel):
     patient_name = models.CharField(max_length=255)
     patient_id = models.BigIntegerField()
     patient_record = EmbeddedModelField("PatientRecord")
@@ -47,78 +53,78 @@ class Billing(EmbeddedModel):
 
 
 # Equality-queryable fields
-class EncryptedBinaryTest(models.Model):
+class EncryptedBinaryTest(EncryptedTestModel):
     value = EncryptedBinaryField(queries={"queryType": "equality"})
 
 
-class EncryptedBooleanTest(models.Model):
+class EncryptedBooleanTest(EncryptedTestModel):
     value = EncryptedBooleanField(queries={"queryType": "equality"})
 
 
-class EncryptedCharTest(models.Model):
+class EncryptedCharTest(EncryptedTestModel):
     value = EncryptedCharField(max_length=255, queries={"queryType": "equality"})
 
 
-class EncryptedEmailTest(models.Model):
+class EncryptedEmailTest(EncryptedTestModel):
     value = EncryptedEmailField(max_length=255, queries={"queryType": "equality"})
 
 
-class EncryptedGenericIPAddressTest(models.Model):
+class EncryptedGenericIPAddressTest(EncryptedTestModel):
     value = EncryptedGenericIPAddressField(queries={"queryType": "equality"})
 
 
-class EncryptedTextTest(models.Model):
+class EncryptedTextTest(EncryptedTestModel):
     value = EncryptedTextField(queries={"queryType": "equality"})
 
 
-class EncryptedURLTest(models.Model):
+class EncryptedURLTest(EncryptedTestModel):
     value = EncryptedURLField(max_length=500, queries={"queryType": "equality"})
 
 
 # Range-queryable fields (also support equality)
-class EncryptedBigIntegerTest(models.Model):
+class EncryptedBigIntegerTest(EncryptedTestModel):
     value = EncryptedBigIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedDateTest(models.Model):
+class EncryptedDateTest(EncryptedTestModel):
     value = EncryptedDateField(queries={"queryType": "range"})
 
 
-class EncryptedDateTimeTest(models.Model):
+class EncryptedDateTimeTest(EncryptedTestModel):
     value = EncryptedDateTimeField(queries={"queryType": "range"})
 
 
-class EncryptedDecimalTest(models.Model):
+class EncryptedDecimalTest(EncryptedTestModel):
     value = EncryptedDecimalField(max_digits=10, decimal_places=2, queries={"queryType": "range"})
 
 
-class EncryptedDurationTest(models.Model):
+class EncryptedDurationTest(EncryptedTestModel):
     value = EncryptedDurationField(queries={"queryType": "range"})
 
 
-class EncryptedFloatTest(models.Model):
+class EncryptedFloatTest(EncryptedTestModel):
     value = EncryptedFloatField(queries={"queryType": "range"})
 
 
-class EncryptedIntegerTest(models.Model):
+class EncryptedIntegerTest(EncryptedTestModel):
     value = EncryptedIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedPositiveBigIntegerTest(models.Model):
+class EncryptedPositiveBigIntegerTest(EncryptedTestModel):
     value = EncryptedPositiveBigIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedPositiveIntegerTest(models.Model):
+class EncryptedPositiveIntegerTest(EncryptedTestModel):
     value = EncryptedPositiveIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedPositiveSmallIntegerTest(models.Model):
+class EncryptedPositiveSmallIntegerTest(EncryptedTestModel):
     value = EncryptedPositiveSmallIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedSmallIntegerTest(models.Model):
+class EncryptedSmallIntegerTest(EncryptedTestModel):
     value = EncryptedSmallIntegerField(queries={"queryType": "range"})
 
 
-class EncryptedTimeTest(models.Model):
+class EncryptedTimeTest(EncryptedTestModel):
     value = EncryptedTimeField(queries={"queryType": "range"})