diff options
Diffstat (limited to 'tests/unit/test_db_utils.py')
| -rw-r--r-- | tests/unit/test_db_utils.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/unit/test_db_utils.py b/tests/unit/test_db_utils.py index 7dc66c0..039bc39 100644 --- a/tests/unit/test_db_utils.py +++ b/tests/unit/test_db_utils.py @@ -18,7 +18,7 @@ class TestDatabase(TestCase): @mock.patch("gn3.db_utils.parse_db_url") def test_database_connector(self, mock_db_parser, mock_sql): """test for creating database connection""" - mock_db_parser.return_value = ("localhost", "guest", "4321", "users") + mock_db_parser.return_value = ("localhost", "guest", "4321", "users", None) callable_cursor = lambda: SimpleNamespace(execute=3) cursor_object = SimpleNamespace(cursor=callable_cursor) mock_sql.connect.return_value = cursor_object @@ -26,7 +26,7 @@ class TestDatabase(TestCase): results = database_connector() mock_sql.connect.assert_called_with( - "localhost", "guest", "4321", "users") + host="localhost", user="guest", passwd="4321", db="users") self.assertIsInstance( results, SimpleNamespace, "database not created successfully") @@ -35,6 +35,6 @@ class TestDatabase(TestCase): "mysql://username:4321@localhost/test") def test_parse_db_url(self): """test for parsing db_uri env variable""" - results = parse_db_url() - expected_results = ("localhost", "username", "4321", "test") + results = parse_db_url("mysql://username:4321@localhost/test") + expected_results = ("localhost", "username", "4321", "test", None) self.assertEqual(results, expected_results) |