Test feature supports_queryable_encryption

Author: aclark4life

django_mongodb_backend/features.py

@@ -626,9 +626,9 @@ def supports_transactions(self):
         return "setName" in hello or hello.get("msg") == "isdbgrid"
 
     @cached_property
-    def supports_encryption(self):
+    def supports_queryable_encryption(self):
         """
-        Encryption is supported if the server is Atlas or Enterprise
+        Queryable Encryption is supported if the server is Atlas or Enterprise
         and is configured as a replica set or sharded cluster.
         """
         self.connection.ensure_connection()

tests/backend_/test_features.py

@@ -44,3 +44,17 @@ def mocked_command(command):
 
         with patch("pymongo.synchronous.database.Database.command", wraps=mocked_command):
             self.assertIs(connection.features.supports_transactions, False)
+
+
+class SupportsQueryableEncryptionTests(TestCase):
+    # TODO: Add setUp? `del connection.features.supports_queryable_encryption` returns
+    # `AttributeError: 'DatabasesFeatures' object has no attribute 'supports_queryable_encryption'`
+    # even though it does have it upon inspection in `pdb`.
+    def test_supports_queryable_encryption(self):
+        def mocked_command(command):
+            if command == "buildInfo":
+                return {"modules": ["enterprise"]}
+            raise Exception("Unexpected command")
+
+        with patch("pymongo.synchronous.database.Database.command", wraps=mocked_command):
+            self.assertIs(connection.features.supports_queryable_encryption, True)