from unittest.mock import Mock

import pytest

from aki_prj23_transparenzregister.utils.news_mongo_service import MongoNewsService


@pytest.fixture()
def mock_mongo_connector(mocker) -> Mock:
    """Mock MongoConnector class.

    Args:
        mocker (any): Library mocker

    Returns:
        Mock: Mocked MongoConnector
    """
    mock = Mock()
    mocker.patch(
        "aki_prj23_transparenzregister.utils.mongo.MongoConnector", return_value=mock
    )
    return mock


@pytest.fixture()
def mock_collection() -> Mock:
    """Mock mongo collection.

    Returns:
        Mock: Mock object
    """
    return Mock()


def test_init(mock_mongo_connector, mock_collection):
    """Test CompanyMongoService constructor.

    Args:
        mock_mongo_connector (Mock): Mocked MongoConnector library
        mock_collection (Mock): Mocked pymongo collection
    """
    mock_mongo_connector.database = {"news": mock_collection}
    service = MongoNewsService(mock_mongo_connector)
    assert service.collection == mock_collection