import datetime

import pytest

from gallery.painting.gismeteo.api import GismeteoApi
from gallery.painting.gismeteo.mock import GISMETEO_MOCK_DATA


@pytest.fixture(name="gismeteo_api", scope="module")
def gismeteo_api_fixture() -> GismeteoApi:
    class MockSource:
        async def request(self, endpoint: str):
            return GISMETEO_MOCK_DATA.get_html(endpoint.split("/")[-1])

    api = GismeteoApi()
    api.SOURCE = MockSource()
    return api


async def test_day(gismeteo_api: GismeteoApi):
    result = await gismeteo_api.get_day("test", datetime.date.today())
    assert len(result.values) == 8


async def test_days(gismeteo_api: GismeteoApi):
    result = await gismeteo_api.get_days("test", 10)
    assert len(result.values) == 10