from unittest.mock import patch

from aki_prj23_transparenzregister.utils.mongo.connector import (
    MongoConnection,
    MongoConnector,
)


def test_get_conn_string_no_credentials() -> None:
    conn = MongoConnection("localhost", "", 27017, None, None)
    assert conn.get_conn_string() == "mongodb://localhost:27017"


def test_get_conn_string_no_port_but_credentials() -> None:
    conn = MongoConnection("localhost", "", None, "admin", "password")
    assert conn.get_conn_string() == "mongodb+srv://admin:password@localhost"


def test_get_conn_simple() -> None:
    conn = MongoConnection("localhost", "", None, None, None)
    assert conn.get_conn_string() == "mongodb+srv://localhost"


def test_mongo_connector() -> None:
    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