Update test

Author: aclark4life

django_mongodb_backend/features.py

@@ -581,9 +581,17 @@ def supports_atlas_search(self):
     @cached_property
     def supports_queryable_encryption(self):
         """
-        Queryable Encryption is supported if the server is Atlas or Enterprise.
+        Queryable Encryption is supported if the server is Atlas or Enterprise
+        and if pymongocrypt is installed.
         """
         self.connection.ensure_connection()
         client = self.connection.connection.admin
         build_info = client.command("buildInfo")
-        return "enterprise" in build_info.get("modules")
+        is_enterprise = "enterprise" in build_info.get("modules")
+        try:
+            import pymongocrypt  # noqa: F401
+
+            has_pymongocrypt = True
+        except ImportError:
+            has_pymongocrypt = False
+        return is_enterprise and has_pymongocrypt

tests/backend_/utils/test_parse_uri.py

@@ -2,7 +2,7 @@
 
 import pymongo
 from django.core.exceptions import ImproperlyConfigured
-from django.test import SimpleTestCase
+from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
 
 from django_mongodb_backend import get_auto_encryption_options, parse_uri
 
@@ -95,5 +95,10 @@ def test_no_scheme(self):
         with self.assertRaisesMessage(pymongo.errors.InvalidURI, "Invalid URI scheme"):
             parse_uri("cluster0.example.mongodb.net")
 
+
+# TODO: This can go in `test_features` once transaction support is added.
+class ParseUriOptionsTests(TestCase):
+    @skipUnlessDBFeature("supports_queryable_encryption")
     def test_queryable_encryption_config(self):
-        get_auto_encryption_options()
+        auto_encryption_options = get_auto_encryption_options()
+        self.assertEqual(auto_encryption_options._key_vault_namespace, "encryption.__keyVault")