Re-add test_base with EncryptionTestCase

- Factor out common attributes (would be nice to do decorators too but
  looks like that requires __init_subclass_ shenanigans)

Author: aclark4life

tests/encryption_/test_base.py

@@ -0,0 +1,6 @@
+from django.test import TestCase
+
+
+class EncryptionTestCase(TestCase):
+    databases = {"default", "encrypted"}
+    maxDiff = None

tests/encryption_/test_fields.py

@@ -27,12 +27,11 @@
     Patient,
     PatientRecord,
 )
+from .test_base import EncryptionTestCase
 
 
 @skipUnlessDBFeature("supports_queryable_encryption")
-class PatientModelTests(TestCase):
-    databases = {"default", "encrypted"}
-
+class PatientModelTests(EncryptionTestCase):
     def setUp(self):
         self.billing = Billing(cc_type="Visa", cc_number="4111111111111111")
         self.patient_record = PatientRecord(ssn="123-45-6789", billing=self.billing)

tests/encryption_/test_management.py

@@ -2,15 +2,14 @@
 
 from bson import json_util
 from django.core.management import call_command
-from django.test import TestCase, modify_settings, skipUnlessDBFeature
+from django.test import modify_settings, skipUnlessDBFeature
+
+from .test_base import EncryptionTestCase
 
 
 @skipUnlessDBFeature("supports_queryable_encryption")
 @modify_settings(INSTALLED_APPS={"prepend": "django_mongodb_backend"})
-class CommandTests(TestCase):
-    databases = {"default", "encrypted"}
-    maxDiff = None
-
+class CommandTests(EncryptionTestCase):
     # Expected encrypted field maps for all Encrypted* models
     expected_maps = {
         "encryption__patientrecord": {

tests/encryption_/test_schema.py

@@ -1,13 +1,10 @@
 from django.db import connections
-from django.test import TestCase
 
 from . import models
+from .test_base import EncryptionTestCase
 
 
-class SchemaTests(TestCase):
-    databases = {"default", "encrypted"}
-    maxDiff = None
-
+class SchemaTests(EncryptionTestCase):
     # Expected encrypted fields map per model
     expected_map = {
         "PatientRecord": {