"""Test Blacklist Words Logic."""

from unittest.mock import MagicMock

from owsresponse import response
import pytest

from blacklist_manager.constants import header
from blacklist_manager.logic import product
from blacklist_manager.models import ows_product


@pytest.mark.parametrize('expected_status', [200, 403])
def test_check_product_ownership_success(monkeypatch, expected_status):
    """Test successful retrieval of blacklist words."""
    monkeypatch.setattr(
        ows_product,
        'check_product_ownership',
        value=MagicMock(
            return_value=response.Response(status=expected_status)))
    result = product.check_product_ownership(
        100, header.GRASS_ACCOUNT_TYPE_VENDOR, 1)
    assert result.status == expected_status