mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-24 23:52:34 +02:00
30 lines
989 B
Python
30 lines
989 B
Python
from unittest.mock import patch
|
|
|
|
from aki_prj23_transparenzregister.utils.mongo.connector 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
|