Add test

Author: aclark4life

django_mongodb/utils.py

@@ -27,7 +27,7 @@ def check_django_compatability():
 
 
 def parse(uri):
-    uri = parse_uri(uri)
+    uri = parse_uri(str(uri))
 
     host = None
     port = None

tests/utils_/test_parse.py

@@ -1,10 +1,26 @@
+from unittest.mock import MagicMock, patch
+
 from django.test import SimpleTestCase
 
+import django_mongodb
+
+MONGODB_URI = "mongodb+srv://myDatabaseUser:D1fficultP%[email protected]/myDatabase?retryWrites=true&w=majority"
+
 
-class MongoURIParseTests(SimpleTestCase):
+class MongoParseURITests(SimpleTestCase):
     """
-    Test parse function
+    Test django_mongodb.parse(uri) function
     """
 
-    def test_hello(self):
-        pass
+    @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)
+        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%40ssw0rd")
+        self.assertEqual(settings_dict["PORT"], 27017)