"""Test connectors."""

from src import config
from src import connectors


def test_get_os_connector(mocker):
    """Test OpenSearch connector creation."""
    mocked_connector = mocker.patch(
        'content_utils.connectors.opensearch.LambdaOpensearchConnector',
        autospec=True,
    )
    mocker.patch.object(config, 'OPENSEARCH_PORT', None)

    connectors.get_os_connector.cache_clear()
    result = connectors.get_os_connector()

    mocked_connector.assert_called_once_with(config.OPENSEARCH_ENDPOINT, config.app_logger,
                                             alias_name=config.OPENSEARCH_INDEX_ALIAS)
    assert result is mocked_connector.return_value


def test_get_os_connector_with_port(mocker):
    """Test OpenSearch connector creation with port."""
    mocked_connector = mocker.patch(
        'content_utils.connectors.opensearch.LambdaOpensearchConnector',
        autospec=True,
    )
    mocker.patch.object(config, 'OPENSEARCH_PORT', '1234')

    connectors.get_os_connector.cache_clear()
    result = connectors.get_os_connector()

    mocked_connector.assert_called_with(config.OPENSEARCH_ENDPOINT, config.app_logger, port=1234,
                                        alias_name=config.OPENSEARCH_INDEX_ALIAS)
    assert result is mocked_connector.return_value


def test_get_graphql_connector(mocker):
    """Test GraphQL connector creation."""
    mocked_connector = mocker.patch(
        'content_utils.connectors.graphql.LambdaGraphQLConnector',
        autospec=True,
    )

    mocked_token_manager = mocker.patch('src.connectors.m2m_token_manager', autospec=True)

    connectors.get_graphql_connector.cache_clear()
    result = connectors.get_graphql_connector()

    mocked_connector.assert_called_once_with(
        graphql_service_name=config.GRAPHQL_SERVICE_NAME,
        application_name=config.APPLICATION_NAME,
        logger=config.app_logger,
        environment=config.ENVIRONMENT,
        m2m_token_manager=mocked_token_manager,
        raise_on_error=True
    )
    assert result is mocked_connector.return_value