24 lines
648 B
Python
24 lines
648 B
Python
import datetime
|
|
|
|
import pytest
|
|
|
|
from gallery.painting.yandextv.api import CHANNELS_MAP, YandexTvApi
|
|
from gallery.sketch.schedule.model import ChannelId
|
|
from tests.data.yandextv import YANDEXTV_MOCK_SOURCE
|
|
|
|
|
|
@pytest.fixture(name="yandextv_api", scope="module")
|
|
def yandextv_api_fixture() -> YandexTvApi:
|
|
api = YandexTvApi()
|
|
api.SOURCE = YANDEXTV_MOCK_SOURCE
|
|
CHANNELS_MAP[ChannelId("test")] = "test"
|
|
return api
|
|
|
|
|
|
async def test_channel(yandextv_api: YandexTvApi):
|
|
result = await yandextv_api.get_channel_schedule(
|
|
ChannelId.TEST, datetime.date.today()
|
|
)
|
|
assert result is not None
|
|
assert len(result.values) > 0
|