mirror of
https://github.com/jdejaegh/irm-kmi-ha.git
synced 2025-06-27 11:39:26 +02:00
146 lines
6.1 KiB
Python
146 lines
6.1 KiB
Python
import os
|
|
from datetime import datetime
|
|
from typing import List
|
|
from unittest.mock import AsyncMock
|
|
|
|
from freezegun import freeze_time
|
|
from homeassistant.components.weather import Forecast
|
|
from homeassistant.core import HomeAssistant
|
|
from pytest_homeassistant_custom_component.common import MockConfigEntry
|
|
|
|
from custom_components.irm_kmi import IrmKmiCoordinator, IrmKmiWeather
|
|
from custom_components.irm_kmi.data import ProcessedCoordinatorData, IrmKmiRadarForecast
|
|
from tests.conftest import get_api_data
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat("2023-12-28T15:30:00+01:00"))
|
|
async def test_weather_nl(
|
|
hass: HomeAssistant,
|
|
mock_image_and_nl_forecast_irm_kmi_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
hass.states.async_set(
|
|
"zone.home",
|
|
0,
|
|
{"latitude": 50.738681639, "longitude": 4.054077148},
|
|
)
|
|
hass.config.config_dir = os.getcwd()
|
|
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
weather = IrmKmiWeather(coordinator, mock_config_entry)
|
|
result = await weather.async_forecast_daily()
|
|
|
|
assert isinstance(result, list)
|
|
assert len(result) == 7
|
|
|
|
# When getting daily forecast, the min temperature of the current day
|
|
# should be the min temperature of the coming night
|
|
assert result[0]['native_templow'] == 9
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat("2024-01-21T14:15:00+01:00"))
|
|
async def test_weather_higher_temp_at_night(
|
|
hass: HomeAssistant,
|
|
mock_image_and_high_temp_irm_kmi_api: AsyncMock,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
# Test case for https://github.com/jdejaegh/irm-kmi-ha/issues/8
|
|
hass.states.async_set(
|
|
"zone.home",
|
|
0,
|
|
{"latitude": 50.738681639, "longitude": 4.054077148},
|
|
)
|
|
hass.config.config_dir = os.getcwd()
|
|
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
weather = IrmKmiWeather(coordinator, mock_config_entry)
|
|
result: List[Forecast] = await weather.async_forecast_daily()
|
|
|
|
for f in result:
|
|
if f['native_temperature'] is not None and f['native_templow'] is not None:
|
|
assert f['native_temperature'] >= f['native_templow']
|
|
|
|
result: List[Forecast] = await weather.async_forecast_twice_daily()
|
|
|
|
for f in result:
|
|
if f['native_temperature'] is not None and f['native_templow'] is not None:
|
|
assert f['native_temperature'] >= f['native_templow']
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat("2023-12-26T18:30:00+01:00"))
|
|
async def test_forecast_attribute_same_as_service_call(
|
|
hass: HomeAssistant,
|
|
mock_image_and_simple_forecast_irm_kmi_api: AsyncMock,
|
|
mock_config_entry_with_deprecated: MockConfigEntry
|
|
) -> None:
|
|
hass.states.async_set(
|
|
"zone.home",
|
|
0,
|
|
{"latitude": 50.738681639, "longitude": 4.054077148},
|
|
)
|
|
hass.config.config_dir = os.getcwd()
|
|
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry_with_deprecated)
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
weather = IrmKmiWeather(coordinator, mock_config_entry_with_deprecated)
|
|
|
|
result_service: List[Forecast] = await weather.async_forecast_twice_daily()
|
|
result_forecast: List[Forecast] = weather.extra_state_attributes['forecast']
|
|
|
|
assert result_service == result_forecast
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat("2023-12-26T17:58:03+01:00"))
|
|
async def test_radar_forecast_service(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
):
|
|
hass.config.time_zone = 'Europe/Brussels'
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
api_data = get_api_data("forecast.json")
|
|
data = IrmKmiCoordinator.radar_list_to_forecast(api_data.get('animation'))
|
|
|
|
coordinator.data = ProcessedCoordinatorData(
|
|
radar_forecast=data
|
|
)
|
|
|
|
weather = IrmKmiWeather(coordinator, mock_config_entry)
|
|
|
|
result_service: List[Forecast] = weather.get_forecasts_radar_service(False)
|
|
|
|
expected = [
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:00:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:10:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:20:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:30:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:40:00+01:00", native_precipitation=0.1, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T17:50:00+01:00", native_precipitation=0.01, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T18:00:00+01:00", native_precipitation=0.12, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T18:10:00+01:00", native_precipitation=1.2, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T18:20:00+01:00", native_precipitation=2, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T18:30:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0),
|
|
IrmKmiRadarForecast(datetime="2023-12-26T18:40:00+01:00", native_precipitation=0, might_rain=False,
|
|
rain_forecast_max=0, rain_forecast_min=0)
|
|
]
|
|
|
|
assert result_service == expected[5:]
|
|
|
|
result_service: List[Forecast] = weather.get_forecasts_radar_service(True)
|
|
|
|
assert result_service == expected
|