Ugh, face palm (subclassing a class with a test)

Author: aclark4life

tests/encryption_/test_fields.py

@@ -1,11 +1,6 @@
 import datetime
 from decimal import Decimal
 
-import pymongo
-from bson.binary import Binary
-from django.conf import settings
-from django.db import connections
-
 from django_mongodb_backend.fields import EncryptedCharField
 
 from .models import (
@@ -36,20 +31,20 @@
 )
 from .test_base import EncryptionTestCase
 
-
-class EncryptedFieldTests(EncryptionTestCase):
-    def assertEncrypted(self, model, field):
-        # Unencrypted connection to encrypted database
-        conn_params = connections["default"].get_connection_params()
-        db_name = settings.DATABASES["encrypted"]["NAME"]
-        with pymongo.MongoClient(**conn_params) as new_connection:
-            db = new_connection[db_name]
-            collection_name = model._meta.db_table
-            data = db[collection_name].find()[0]
-            self.assertIsInstance(data[field], Binary)
+# class EncryptedFieldTests(EncryptionTestCase):
+#     def assertEncrypted(self, model, field):
+#         # Unencrypted connection to encrypted database
+#         conn_params = connections["default"].get_connection_params()
+#         db_name = settings.DATABASES["encrypted"]["NAME"]
+#         with pymongo.MongoClient(**conn_params) as new_connection:
+#             db = new_connection[db_name]
+#             collection_name = model._meta.db_table
+#             data = db[collection_name].find()[0]
+#             self.assertIsInstance(data[field], Binary)
+#
 
 
-class EncryptedEmbeddedModelTests(EncryptedFieldTests):
+class EncryptedEmbeddedModelTests(EncryptionTestCase):
     def setUp(self):
         self.billing = Billing(cc_type="Visa", cc_number="4111111111111111")
         self.patient_record = PatientRecord(ssn="123-45-6789", billing=self.billing)
@@ -65,7 +60,7 @@ def test_object(self):
         # self.assertEncrypted(patient, ["patient_record", "ssn"])
 
 
-class EncryptedEmbeddedModelArrayTests(EncryptedFieldTests):
+class EncryptedEmbeddedModelArrayTests(EncryptionTestCase):
     def setUp(self):
         self.actor1 = Actor(name="Actor One")
         self.actor2 = Actor(name="Actor Two")
@@ -81,7 +76,7 @@ def test_array(self):
         # self.assertEncrypted(self.movie, "cast")
 
 
-class EncryptedFieldTests(EncryptedFieldTests):
+class EncryptionTestCase(EncryptionTestCase):
     def assertEquality(self, model_cls, val):
         model_cls.objects.create(value=val)
         fetched = model_cls.objects.get(value=val)
@@ -200,7 +195,7 @@ def test_time(self):
         # self.assertEncrypted(TimeModel, "value")
 
 
-class EncryptedFieldMixinTests(EncryptedFieldTests):
+class EncryptedFieldMixinTests(EncryptionTestCase):
     def test_null_true_raises_error(self):
         with self.assertRaisesMessage(
             ValueError, "'null=True' is not supported for encrypted fields."