mirror of
https://github.com/fhswf/aki_prj23_transparenzregister.git
synced 2025-04-22 22:02:54 +02:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
"""Tests for connecting to the mongodb."""
|
|
from unittest.mock import patch
|
|
|
|
from aki_prj23_transparenzregister.utils.mongo.connector import (
|
|
MongoConnection,
|
|
MongoConnector,
|
|
)
|
|
|
|
|
|
def test_get_conn_string_no_credentials() -> None:
|
|
"""Tests the mongo connection string generation."""
|
|
conn = MongoConnection("localhost", "", 27017, None, None)
|
|
assert conn.get_conn_string() == "mongodb://localhost:27017"
|
|
|
|
|
|
def test_get_conn_string_no_port_but_credentials() -> None:
|
|
"""Tests the mongo connection string generation."""
|
|
conn = MongoConnection("localhost", "", None, "admin", "password")
|
|
assert conn.get_conn_string() == "mongodb+srv://admin:password@localhost"
|
|
|
|
|
|
def test_get_conn_simple() -> None:
|
|
"""Tests the mongo connection string generation."""
|
|
conn = MongoConnection("localhost", "", None, None, None)
|
|
assert conn.get_conn_string() == "mongodb+srv://localhost"
|
|
|
|
|
|
def test_mongo_connector() -> None:
|
|
"""Tests the MongoConnector."""
|
|
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
|