Add engine kwarg and test

Author: aclark4life

django_mongodb/utils.py

@@ -26,7 +26,7 @@ def check_django_compatability():
         )
 
 
-def parse(uri):
+def parse(uri, engine=None):
     """
     Parse a MongoDB URI and return a dictionary of Django database
     settings. This function is a wrapper around PyMongo's
@@ -38,6 +38,9 @@ def parse(uri):
     host = None
     port = None
 
+    if not engine:
+        engine = "django_mongodb"
+
     if uri["fqdn"] is not None:
         # If the fqdn is present, this is a SRV URI and the host is the fqdn.
         host = f"mongodb+srv://{uri['fqdn']}"
@@ -49,7 +52,7 @@ def parse(uri):
             host = ",".join([f"{host}:{port}" for host, port in nodelist])
 
     return {
-        "ENGINE": "django_mongodb",
+        "ENGINE": engine,
         "NAME": uri["database"],
         "HOST": host,
         "PORT": port,

tests/utils_/test_parse.py

@@ -4,23 +4,32 @@
 
 import django_mongodb
 
-MONGODB_URI = "mongodb+srv://myDatabaseUser:D1fficultP%[email protected]/myDatabase?retryWrites=true&w=majority"
+URI = "mongodb+srv://myDatabaseUser:D1fficultP%[email protected]/myDatabase?retryWrites=true&w=majority"
 
 
 class MongoParseURITests(SimpleTestCase):
     """
     Test django_mongodb.parse(uri) function
     """
 
+    def setUp(self):
+        self.srv_record = MagicMock()
+        self.srv_record.target.to_text.return_value = "cluster0.example.mongodb.net"
+        self.patcher = patch("dns.resolver.resolve", return_value=[self.srv_record])
+        self.mock_resolver = self.patcher.start()
+        self.addCleanup(self.patcher.stop)
+
     @patch("dns.resolver.resolve")
     def test_parse(self, mock_resolver):
-        srv_record = MagicMock()
-        srv_record.target.to_text.return_value = "cluster0.example.mongodb.net"
-        mock_resolver.return_value = [srv_record]
-        settings_dict = django_mongodb.parse(MONGODB_URI)
+        settings_dict = django_mongodb.parse(URI)
         self.assertEqual(settings_dict["ENGINE"], "django_mongodb")
         self.assertEqual(settings_dict["NAME"], "myDatabase")
         self.assertEqual(settings_dict["HOST"], "mongodb+srv://cluster0.example.mongodb.net")
         self.assertEqual(settings_dict["USER"], "myDatabaseUser")
         self.assertEqual(settings_dict["PASSWORD"], "D1fficultP@ssw0rd")
         self.assertEqual(settings_dict["PORT"], None)
+
+    @patch("dns.resolver.resolve")
+    def test_engine_kwarg(self, mock_resolver):
+        settings_dict = django_mongodb.parse(URI, engine="some_other_engine")
+        self.assertEqual(settings_dict["ENGINE"], "some_other_engine")