"""Tests for ows-carveouts model."""
from backend.constants import error
from backend.constants import services
from backend.models import ows_carveouts
from tests.models.test_ows_product import _patch_request


def test_get_stores_for_instant_grats_error(monkeypatch):
    """Test get stores grats from ows-carveouts service failed."""
    message = 'Failed to get stores'
    mock_json_response = {
        'code': error.OWS_CARVEOUTS_ERROR_CODE,
        'message': message
    }
    get_request_mock = _patch_request(
        monkeypatch, 'get', mock_json_response, 500)
    result = ows_carveouts.get_stores_allowed_for_instant_grats()
    get_request_mock.assert_called_with(
        services.OWS_CARVEOUTS, '/stores/grats')

    assert not result
    assert result.errors['message'] == mock_json_response


def test_get_stores_for_instant_grats(monkeypatch,):
    """Test get stores grats from ows-carveouts service success."""
    mock_json_response = {'1': 'ITunes'}
    get_request_mock = _patch_request(
        monkeypatch, 'get', mock_json_response, 200)
    result = ows_carveouts.get_stores_allowed_for_instant_grats()
    get_request_mock.assert_called_with(
        services.OWS_CARVEOUTS, '/stores/grats')
    expected_result = [{'id': '1', 'name': 'ITunes'}]
    assert result
    assert result.message == expected_result