"""Test ows-territories interface."""


from feed_ingestion.flows.youtube_asset_conflict.models import ows_territories\
    as ows_territories_model
from tests.flows.youtube_asset_conflict.fixtures import ows_territories


def _mock_ows_territories(mocker, response):
    """Mock ows-territories helper."""
    return mocker.patch(
        'feed_ingestion.flows.youtube_asset_conflict.models.'
        'ows_territories.request.process', return_value=response)


def test_get_territories_success(mocker, territories_success_response):
    """Test get territories from ows-territories."""
    _mock_ows_territories(mocker, territories_success_response)
    result = ows_territories_model.get_territories()
    assert result
    assert result.message['items'] == ows_territories.TEST_TERRITORIES


def test_get_territories_error(mocker, ows_error_response):
    """Test get territories from ows-territories error."""
    _mock_ows_territories(mocker, ows_error_response)
    result = ows_territories_model.get_territories()
    assert not result
    assert result.status == 500


def test_get_territories_not_found(mocker, ows_not_found_response):
    """Test get territories from ows-territories not found."""
    _mock_ows_territories(mocker, ows_not_found_response)
    result = ows_territories_model.get_territories('INCORRECT_STANDARD')
    assert not result
    assert result.status == 404