Add tests for DatabaseWrapper.get_connection_params()

Author: timgraham

tests/backend_/test_base.py

@@ -6,14 +6,56 @@
 from django_mongodb_backend.base import DatabaseWrapper
 
 
-class DatabaseWrapperTests(SimpleTestCase):
+class GetConnectionParamsTests(SimpleTestCase):
     def test_database_name_empty(self):
         settings = connection.settings_dict.copy()
         settings["NAME"] = ""
         msg = 'settings.DATABASES is missing the "NAME" value.'
         with self.assertRaisesMessage(ImproperlyConfigured, msg):
             DatabaseWrapper(settings).get_connection_params()
 
+    def test_host(self):
+        settings = connection.settings_dict.copy()
+        settings["HOST"] = "host"
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["host"], "host")
+
+    def test_host_empty(self):
+        settings = connection.settings_dict.copy()
+        settings["HOST"] = ""
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertIsNone(params["host"])
+
+    def test_user(self):
+        settings = connection.settings_dict.copy()
+        settings["USER"] = "user"
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["username"], "user")
+
+    def test_password(self):
+        settings = connection.settings_dict.copy()
+        settings["PASSWORD"] = "password"  # noqa: S105
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["password"], "password")
+
+    def test_port(self):
+        settings = connection.settings_dict.copy()
+        settings["PORT"] = 123
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["port"], 123)
+
+    def test_port_as_string(self):
+        settings = connection.settings_dict.copy()
+        settings["PORT"] = "123"
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["port"], 123)
+
+    def test_options(self):
+        settings = connection.settings_dict.copy()
+        settings["OPTIONS"] = {"extra": "option"}
+        params = DatabaseWrapper(settings).get_connection_params()
+        self.assertEqual(params["extra"], "option")
+
 
 class DatabaseWrapperConnectionTests(TestCase):
     def test_set_autocommit(self):