import datetime

import pytest

from gallery.painting.yandextv.api import YandexTVApi
from tests.common.mock import create_mock_api
from tests.data.yandextv import YANDEXTV_MOCK_SOURCE


@pytest.fixture(name="yandextv_api", scope="module")
def yandextv_api_fixture() -> YandexTVApi:
    return create_mock_api(YandexTVApi, YANDEXTV_MOCK_SOURCE)


async def test_search(yandextv_api: YandexTVApi):
    result = await yandextv_api.find_channels("матч")
    assert len(result) == 4


async def test_channel(yandextv_api: YandexTVApi):
    result = await yandextv_api.get_schedule("test", datetime.date.today())
    assert result is not None
    assert len(result.values) > 0