"""Tests for ows-sales-goals Model."""

from unittest.mock import MagicMock
from unittest.mock import Mock

from owsrequest import request as requests

from ows_product_physical.models import ows_salesgoals


def test_get_goals(monkeypatch):
    """Test getting goals."""
    mock_response = Mock()
    mock_response.status_code = 200
    monkeypatch.setattr(requests, 'get', MagicMock(
        return_value=mock_response))

    salesgoals_response = ows_salesgoals.get_goals(234567)

    assert requests.get.called
    assert salesgoals_response.status == 200


def test_get_goals_failure(monkeypatch):
    """Test getting goals failure."""
    mock_response = Mock()
    mock_response.status_code = 500
    monkeypatch.setattr(requests, 'get', MagicMock(
        return_value=mock_response))

    salesgoals_response = ows_salesgoals.get_goals(234567)

    assert requests.get.called
    assert salesgoals_response.status == 500