"""Mocks for ows_territories."""

from oto.response import Response

from conflict_manager.exceptions import ServiceResponseError
from conflict_manager.models import ows_territories


def get_territories_map(mocker, data):
    """Mock OWS territories get_territories_map."""
    res = Response(ows_territories._make_territory_map(data))

    return mocker.patch(
        'conflict_manager.models.ows_territories.get_territories_map',
        return_value=res)


def get_territories_map_server_down(mocker):
    """Mock OWS territories get_territories_map when server is down."""
    e = ServiceResponseError(
        message='Error', error_code='ows_territories_error', status_code=500)

    return mocker.patch(
        'conflict_manager.models.ows_territories.get_territories_map',
        side_effect=e)