"""Test subaccount id mapping in art_relations."""

from unittest.mock import patch

from src import update_subaccount_mapping_in_art_relations

subaccount_ids = {
    '12': ['35', '55'],
    '24': ['65']
}


@patch('src.update_subaccount_mapping_in_art_relations.db_execution')
@patch('src.app.logger')
def test_validate_vendor(mock_logger, mock_execution):
    """Test validate vendor method."""
    mock_execution.return_value = ((2,),)

    result = update_subaccount_mapping_in_art_relations.validate_vendor(
        subaccount_ids, '2', mock_logger
    )

    assert result is True


@patch('src.update_subaccount_mapping_in_art_relations.db_execution')
@patch('src.app.logger')
def test_update_mapping_in_art_relations(mock_logger, mock_execution):
    """Test update mapping in art_relations method."""
    mock_execution.return_value = ((2,),)

    update_subaccount_mapping_in_art_relations.update_mapping_in_art_relations(
        subaccount_ids, mock_logger
    )

    mock_execution.assert_called()


@patch('src.update_subaccount_mapping_in_art_relations.db_execution')
@patch('src.app.logger')
def test_validate_vendor_missmatch_vendor(mock_logger, mock_execution):
    """Test validate vendor when wrong vendor is found."""
    mock_execution.return_value = ((5,), (4,))

    result = update_subaccount_mapping_in_art_relations.validate_vendor(
        subaccount_ids, '2', mock_logger
    )

    assert result is False