Add test kwarg and test

Author: aclark4life

django_mongodb/utils.py

@@ -26,7 +26,9 @@ def check_django_compatability():
         )
 
 
-def parse(uri, engine=None, conn_max_age=None, conn_health_checks=None):
+def parse(
+    uri, engine=None, conn_max_age=None, conn_health_checks=None, ssl_require=None, test=None
+):
     """
     Parse a MongoDB URI and return a dictionary of Django database
     settings. This function is a wrapper around PyMongo's
@@ -67,6 +69,12 @@ def parse(uri, engine=None, conn_max_age=None, conn_health_checks=None):
     if conn_health_checks is not None:
         settings_dict["CONN_HEALTH_CHECKS"] = conn_health_checks
 
+    if ssl_require is not None:
+        settings_dict["SSL_REQUIRE"] = ssl_require
+
+    if test is not None:
+        settings_dict["TEST"] = test
+
     return settings_dict
 
 

tests/utils_/test_parse.py

@@ -43,3 +43,13 @@ def test_conn_max_age_kwarg(self, mock_resolver):
     def test_conn_health_checks_kwarg(self, mock_resolver):
         settings_dict = django_mongodb.parse(URI, conn_health_checks=True)
         self.assertEqual(settings_dict["CONN_HEALTH_CHECKS"], True)
+
+    @patch("dns.resolver.resolve")
+    def test_ssl_require_kwarg(self, mock_resolver):
+        settings_dict = django_mongodb.parse(URI, ssl_require=True)
+        self.assertEqual(settings_dict["SSL_REQUIRE"], True)
+
+    @patch("dns.resolver.resolve")
+    def test_test_kwarg(self, mock_resolver):
+        settings_dict = django_mongodb.parse(URI, test={"NAME": "test_db"})
+        self.assertEqual(settings_dict["TEST"]["NAME"], "test_db")