"""Tests for ows-territories Model."""

from collections import namedtuple

from flexmock import flexmock

from owsrequest import request as requests

from sales_goals import api
from sales_goals.models import ows_territories


Response = namedtuple('Response', ['json'])


def test_get_territories_success(
        valid_get_header, context):
    """Test calls ows territories successfully."""
    with context, api.app.test_request_context(headers=valid_get_header):
        (flexmock(requests)
         .should_receive('get')
         .with_args('ows-territories', '/territory/Orch_1_2016')
         .and_return(Response(lambda: {'items': []})))

        territory_response = ows_territories.get_territories()
        assert territory_response.status == 200


def test_caches(
        monkeypatch, valid_get_header, context):
    """Test it caches."""
    with context, api.app.test_request_context(headers=valid_get_header):
        (flexmock(requests)
         .should_receive('get')
         .with_args('ows-territories', '/territory/Orch_1_2016')
         .and_return(Response(lambda: {'items': []}))
         .never())

        territory_response = ows_territories.get_territories()
        assert territory_response.status == 200