"""Unit tests for ows-account model."""

import pytest

from src.model import ows_account


class TestIsDistributor:
    """Tests for get_account function."""

    @pytest.fixture
    def get_is_distributor_url_mock(self, mocker):
        """Mock _get_is_distributor_url function."""
        return mocker.patch.object(ows_account, '_get_is_distributor_url')

    @pytest.fixture
    def vendor_id(self):
        """Vendor id."""
        return 26640

    @pytest.fixture
    def requests_head_mock(self, mocker):
        """Mock requests get method."""
        return mocker.patch('requests.head')
        # mock.status_code = 200

    @pytest.mark.parametrize(
        'status_code, expected_result',
        [
            (200, True),
            (204, False),
        ],
    )
    def test_is_distributor(
        self, get_is_distributor_url_mock, requests_head_mock, vendor_id, status_code, expected_result
    ):
        """Test is_distributor function."""
        # mocks
        url_mock = get_is_distributor_url_mock.return_value
        response_mock = requests_head_mock.return_value
        response_mock.status_code = status_code
        # test call
        result = ows_account.is_distributor(vendor_id)
        # checks
        get_is_distributor_url_mock.assert_called_once_with(vendor_id)
        requests_head_mock.assert_called_once_with(url_mock)
        assert response_mock.raise_for_status.called
        assert result == expected_result