Add conn_* kwargs and tests

Author: aclark4life

django_mongodb/utils.py

@@ -26,7 +26,7 @@ def check_django_compatability():
         )
 
 
-def parse(uri, engine=None):
+def parse(uri, engine=None, conn_max_age=None, conn_health_checks=None):
     """
     Parse a MongoDB URI and return a dictionary of Django database
     settings. This function is a wrapper around PyMongo's
@@ -51,7 +51,7 @@ def parse(uri, engine=None):
         elif len(nodelist) > 1:
             host = ",".join([f"{host}:{port}" for host, port in nodelist])
 
-    return {
+    settings_dict = {
         "ENGINE": engine,
         "NAME": uri["database"],
         "HOST": host,
@@ -61,6 +61,14 @@ def parse(uri, engine=None):
         "OPTIONS": uri.get("options"),
     }
 
+    if conn_max_age is not None:
+        settings_dict["CONN_MAX_AGE"] = conn_max_age
+
+    if conn_health_checks is not None:
+        settings_dict["CONN_HEALTH_CHECKS"] = conn_health_checks
+
+    return settings_dict
+
 
 def set_wrapped_methods(cls):
     """Initialize the wrapped methods on cls."""

tests/utils_/test_parse.py

@@ -33,3 +33,13 @@ def test_parse(self, mock_resolver):
     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")
+
+    @patch("dns.resolver.resolve")
+    def test_conn_max_age_kwarg(self, mock_resolver):
+        settings_dict = django_mongodb.parse(URI, conn_max_age=600)
+        self.assertEqual(settings_dict["CONN_MAX_AGE"], 600)
+
+    @patch("dns.resolver.resolve")
+    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)