test: utils.mongo

tests/utils/mongo_test.py

@@ -0,0 +1,26 @@
+from unittest.mock import patch
+
+from aki_prj23_transparenzregister.utils.mongo import MongoConnection, MongoConnector
+
+
+def test_get_conn_string_no_credentials():
+    conn = MongoConnection("localhost", "", 27017, None, None)
+    assert conn.get_conn_string() == "mongodb://localhost:27017"
+
+
+def test_get_conn_string_no_port_but_credentials():
+    conn = MongoConnection("localhost", "", None, "admin", "password")
+    assert conn.get_conn_string() == "mongodb+srv://admin:password@localhost"
+
+
+def test_get_conn_simple():
+    conn = MongoConnection("localhost", "", None, None, None)
+    assert conn.get_conn_string() == "mongodb+srv://localhost"
+
+
+def test_mongo_connector():
+    with patch("pymongo.MongoClient") as mock_mongo_client:
+        expected_result = 42
+        mock_mongo_client.return_value = {"db": expected_result}
+        temp = MongoConnector(MongoConnection("localhost", "db", None, None, None))
+        assert temp.database == expected_result