mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-24 10:52:33 +02:00
27 lines
966 B
Python
27 lines
966 B
Python
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
|