mirror of
https://github.com/jdejaegh/irm-kmi-ha.git
synced 2025-06-27 03:35:56 +02:00
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
from datetime import datetime
|
|
|
|
from freezegun import freeze_time
|
|
from homeassistant.core import HomeAssistant
|
|
from pytest_homeassistant_custom_component.common import MockConfigEntry
|
|
|
|
from custom_components.irm_kmi import IrmKmiCoordinator
|
|
from custom_components.irm_kmi.binary_sensor import IrmKmiWarning
|
|
from custom_components.irm_kmi.const import CONF_LANGUAGE_OVERRIDE
|
|
from custom_components.irm_kmi.sensor import IrmKmiNextWarning
|
|
from tests.conftest import get_api_data
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat('2024-01-12T07:55:00+01:00'))
|
|
async def test_warning_data(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
api_data = get_api_data("be_forecast_warning.json")
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
result = coordinator.warnings_from_data(api_data.get('for', {}).get('warning'))
|
|
|
|
coordinator.data = {'warnings': result}
|
|
warning = IrmKmiWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.is_on
|
|
assert len(warning.extra_state_attributes['warnings']) == 2
|
|
|
|
for w in warning.extra_state_attributes['warnings']:
|
|
assert w['is_active']
|
|
|
|
assert warning.extra_state_attributes['active_warnings_friendly_names'] == "Fog, Ice or snow"
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat('2024-01-12T07:55:00+01:00'))
|
|
async def test_warning_data_unknown_lang(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
# When language is unknown, default to english setting
|
|
hass.config.language = "foo"
|
|
|
|
api_data = get_api_data("be_forecast_warning.json")
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
result = coordinator.warnings_from_data(api_data.get('for', {}).get('warning'))
|
|
|
|
coordinator.data = {'warnings': result}
|
|
warning = IrmKmiWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.is_on
|
|
assert len(warning.extra_state_attributes['warnings']) == 2
|
|
|
|
for w in warning.extra_state_attributes['warnings']:
|
|
assert w['is_active']
|
|
|
|
assert warning.extra_state_attributes['active_warnings_friendly_names'] == "Fog, Ice or snow"
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat('2024-01-11T20:00:00+01:00'))
|
|
async def test_next_warning_when_data_available(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
api_data = get_api_data("be_forecast_warning.json")
|
|
mock_config_entry.data = mock_config_entry.data | {CONF_LANGUAGE_OVERRIDE: 'de'}
|
|
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
result = coordinator.warnings_from_data(api_data.get('for', {}).get('warning'))
|
|
|
|
coordinator.data = {'warnings': result}
|
|
warning = IrmKmiNextWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.state == "2024-01-12T06:00:00+00:00"
|
|
assert len(warning.extra_state_attributes['next_warnings']) == 2
|
|
|
|
assert warning.extra_state_attributes['next_warnings_friendly_names'] == "Nebel, Glätte"
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat('2024-01-12T07:30:00+01:00'))
|
|
async def test_next_warning_none_when_only_active_warnings(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
api_data = get_api_data("be_forecast_warning.json")
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
result = coordinator.warnings_from_data(api_data.get('for', {}).get('warning'))
|
|
|
|
coordinator.data = {'warnings': result}
|
|
warning = IrmKmiNextWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.state is None
|
|
assert len(warning.extra_state_attributes['next_warnings']) == 0
|
|
|
|
assert warning.extra_state_attributes['next_warnings_friendly_names'] == ""
|
|
|
|
|
|
@freeze_time(datetime.fromisoformat('2024-01-12T07:30:00+01:00'))
|
|
async def test_next_warning_none_when_no_warnings(
|
|
hass: HomeAssistant,
|
|
mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
coordinator = IrmKmiCoordinator(hass, mock_config_entry)
|
|
|
|
coordinator.data = {'warnings': []}
|
|
warning = IrmKmiNextWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.state is None
|
|
assert len(warning.extra_state_attributes['next_warnings']) == 0
|
|
|
|
assert warning.extra_state_attributes['next_warnings_friendly_names'] == ""
|
|
|
|
coordinator.data = dict()
|
|
warning = IrmKmiNextWarning(coordinator, mock_config_entry)
|
|
warning.hass = hass
|
|
|
|
assert warning.state is None
|
|
assert len(warning.extra_state_attributes['next_warnings']) == 0
|
|
|
|
assert warning.extra_state_attributes['next_warnings_friendly_names'] == ""
|