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


from tests.flows.fixtures import ows_territories
from yt_conflict_elasticsearch.flows.elasticsearch_export.models import\
    ows_territories as ows_territories_model


def _mock_ows_territories(mocker, response):
    """Mock ows-territories helper."""
    return mocker.patch(
        'yt_conflict_elasticsearch.flows.elasticsearch_export.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