aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/test_db_utils.py
diff options
context:
space:
mode:
authorAlexander Kabui2021-04-12 16:53:48 +0300
committerAlexander Kabui2021-04-12 16:53:48 +0300
commit35f5ac0335f44923184ffe0f0a3380a9cf1859ef (patch)
tree8f1fcb15ace3574eb19bd0eafdc5b5bb0822ed09 /tests/unit/test_db_utils.py
parent8ce82f5b6cccc015c38a728864c63c026fe6a3a0 (diff)
parent31ac939f58bf7b6d353ced995ca395376203b25f (diff)
downloadgenenetwork3-35f5ac0335f44923184ffe0f0a3380a9cf1859ef.tar.gz
fix merge conflict
Diffstat (limited to 'tests/unit/test_db_utils.py')
-rw-r--r--tests/unit/test_db_utils.py37
1 files changed, 37 insertions, 0 deletions
diff --git a/tests/unit/test_db_utils.py b/tests/unit/test_db_utils.py
new file mode 100644
index 0000000..0f2de9e
--- /dev/null
+++ b/tests/unit/test_db_utils.py
@@ -0,0 +1,37 @@
+"""module contains test for db_utils"""
+
+from unittest import TestCase
+from unittest import mock
+
+from types import SimpleNamespace
+
+from gn3.db_utils import database_connector
+from gn3.db_utils import parse_db_url
+
+
+class TestDatabase(TestCase):
+ """class contains testd for db connection functions"""
+
+ @mock.patch("gn3.db_utils.mdb")
+ @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")
+ callable_cursor = lambda: SimpleNamespace(execute=3)
+ cursor_object = SimpleNamespace(cursor=callable_cursor)
+ mock_sql.connect.return_value = cursor_object
+ mock_sql.close.return_value = None
+ results = database_connector()
+
+ mock_sql.connect.assert_called_with(
+ "localhost", "guest", "4321", "users")
+ self.assertIsInstance(
+ results, tuple, "database not created successfully")
+
+ @mock.patch("gn3.db_utils.SQL_URI",
+ "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")
+ self.assertEqual(results, expected_results)