"""Tests for ows_track model."""

from oto import response

from conflict_manager.constants import services
from conflict_manager.models import ows_track
from tests import testutils


TUID = 1


def test_successful_response(mocker, ows_track_succesful_response):
    """Test get track from ows-track."""
    testutils.patch_get_from_ows_service(mocker, ows_track_succesful_response)
    track_response = ows_track.get_track_by_tuid(TUID)
    assert track_response == ows_track_succesful_response


def test_not_found_response(mocker, ows_not_found_response):
    """Test get track from ows-track not found."""
    testutils.patch_get_from_ows_service(mocker, ows_not_found_response)
    track_response = ows_track.get_track_by_tuid(TUID)
    assert track_response == ows_not_found_response


def test_server_error_response(mocker):
    """Test get track from ows-track error."""
    error_response = response.create_error_response(
        code=services.SERVICE_ERROR_CODES[services.OWS_TRACK],
        message='Server error.'
    )
    testutils.patch_get_from_ows_service(mocker, error_response)
    track_response = ows_track.get_track_by_tuid(TUID)
    assert track_response == error_response