"""Test for ows_features model."""

from src.models import ows_assets


def test_get_image_success(
    mocker,
    coverart,
    get_image_success_response,
):
    """Test get features."""
    mocker.patch("requests.get", return_value=get_image_success_response)
    result = ows_assets.get_image(product_id=123, cover_type="large_cover")
    assert result == coverart


def test_get_image_not_found(mocker, get_image_not_found_error):
    """Test get features when it encounters an error."""
    mocker.patch("requests.get", return_value=get_image_not_found_error)
    result = ows_assets.get_image(product_id=123, cover_type="large_cover")
    assert result == "No image found for product 123"


def test_fetch_covers_success(
    mocker,
    coverarts,
    fetch_covers_success_response,
):
    """Test get features."""
    mocker.patch("requests.get", return_value=fetch_covers_success_response)
    result = ows_assets.get_image(product_id=123, cover_type="large_cover")
    assert result == coverarts