"""Tests for update ownership logic layer."""

from unittest.mock import Mock
from unittest.mock import patch

from masters_registry.constant import error
from masters_registry.logic import masters_registry
from masters_registry.models import ows_territories
from tests.helpers import patches


@patch(
    'masters_registry.models.ownership.get_ownership',
    new=patches.ownership_get_ownership_not_found_patch)
@patch(
    'masters_registry.models.ows_territories.get_territories',
    new=Mock())
def test_get_ownership_not_found():
    """Expect to get empty 200 response when ownership info wasn't found."""
    ownership_response = masters_registry.get_ownership('test_isrc')
    assert ownership_response.status == 200
    assert ownership_response.message == {}
    assert not ows_territories.get_territories.called


@patch(
    'masters_registry.models.ownership.get_ownership',
    new=patches.ownership_get_ownership_single_owner_patch)
@patch(
    'masters_registry.models.ows_territories.get_complement_territories',
    new=patches.ows_territories_get_error_response)
def test_get_ownership_ows_territories_error():
    """
    Expect error from ows-territories when looking for unclaimed territories.
    """
    ownership_response = masters_registry.get_ownership('test_isrc')
    assert ownership_response.status == 404
    assert ownership_response.errors['code'] == error.OWS_TERRITORIES_ERROR