diff options
Diffstat (limited to 'tests/unit/test_db_utils.py')
| -rw-r--r-- | tests/unit/test_db_utils.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/unit/test_db_utils.py b/tests/unit/test_db_utils.py index 0f2de9e..7dc66c0 100644 --- a/tests/unit/test_db_utils.py +++ b/tests/unit/test_db_utils.py @@ -2,9 +2,10 @@ from unittest import TestCase from unittest import mock - from types import SimpleNamespace +import pytest + from gn3.db_utils import database_connector from gn3.db_utils import parse_db_url @@ -12,6 +13,7 @@ from gn3.db_utils import parse_db_url class TestDatabase(TestCase): """class contains testd for db connection functions""" + @pytest.mark.unit_test @mock.patch("gn3.db_utils.mdb") @mock.patch("gn3.db_utils.parse_db_url") def test_database_connector(self, mock_db_parser, mock_sql): @@ -26,8 +28,9 @@ class TestDatabase(TestCase): mock_sql.connect.assert_called_with( "localhost", "guest", "4321", "users") self.assertIsInstance( - results, tuple, "database not created successfully") + results, SimpleNamespace, "database not created successfully") + @pytest.mark.unit_test @mock.patch("gn3.db_utils.SQL_URI", "mysql://username:4321@localhost/test") def test_parse_db_url(self): |