mirror of
https://github.com/home-assistant/core.git
synced 2025-06-02 12:17:07 +00:00
b72d64a3dc
* aemet: disable legacy options This enables proper timezone handling: - Atlantic/Canary for the Canary Islands. - Europe/Madrid for the Iberian Peninsula. Also provides daily data for the current day after AEMET stops providing the full day interval, which is normally after midday (12:00). This is a breaking change because with the previous behaviour the daily data for the current day wasn't available after midday and now it will be. What the integration library does to workaround this is to fallback to the 12-24 interval data if the the 00-24 is no longer provided by the API. Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com> * Fix AEMET tests with v0.4.8 Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com> --------- Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
214 lines
6.6 KiB
Python
214 lines
6.6 KiB
Python
"""The sensor tests for the AEMET OpenData platform."""
|
|
import datetime
|
|
from unittest.mock import patch
|
|
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.aemet.const import ATTRIBUTION, DOMAIN
|
|
from homeassistant.components.aemet.coordinator import WEATHER_UPDATE_INTERVAL
|
|
from homeassistant.components.weather import (
|
|
ATTR_CONDITION_SNOWY,
|
|
ATTR_FORECAST,
|
|
ATTR_FORECAST_CONDITION,
|
|
ATTR_FORECAST_PRECIPITATION,
|
|
ATTR_FORECAST_PRECIPITATION_PROBABILITY,
|
|
ATTR_FORECAST_TEMP,
|
|
ATTR_FORECAST_TEMP_LOW,
|
|
ATTR_FORECAST_TIME,
|
|
ATTR_FORECAST_WIND_BEARING,
|
|
ATTR_FORECAST_WIND_SPEED,
|
|
ATTR_WEATHER_HUMIDITY,
|
|
ATTR_WEATHER_PRESSURE,
|
|
ATTR_WEATHER_TEMPERATURE,
|
|
ATTR_WEATHER_WIND_BEARING,
|
|
ATTR_WEATHER_WIND_GUST_SPEED,
|
|
ATTR_WEATHER_WIND_SPEED,
|
|
DOMAIN as WEATHER_DOMAIN,
|
|
LEGACY_SERVICE_GET_FORECAST,
|
|
SERVICE_GET_FORECASTS,
|
|
)
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from .util import async_init_integration, mock_api_call
|
|
|
|
from tests.typing import WebSocketGenerator
|
|
|
|
|
|
async def test_aemet_weather(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
) -> None:
|
|
"""Test states of the weather."""
|
|
|
|
hass.config.set_time_zone("UTC")
|
|
freezer.move_to("2021-01-09 12:00:00+00:00")
|
|
await async_init_integration(hass)
|
|
|
|
state = hass.states.get("weather.aemet")
|
|
assert state
|
|
assert state.state == ATTR_CONDITION_SNOWY
|
|
assert state.attributes[ATTR_ATTRIBUTION] == ATTRIBUTION
|
|
assert state.attributes[ATTR_WEATHER_HUMIDITY] == 99.0
|
|
assert state.attributes[ATTR_WEATHER_PRESSURE] == 1004.4 # 100440.0 Pa -> hPa
|
|
assert state.attributes[ATTR_WEATHER_TEMPERATURE] == -0.7
|
|
assert state.attributes[ATTR_WEATHER_WIND_BEARING] == 122.0
|
|
assert state.attributes[ATTR_WEATHER_WIND_GUST_SPEED] == 12.2
|
|
assert state.attributes[ATTR_WEATHER_WIND_SPEED] == 3.2
|
|
forecast = state.attributes[ATTR_FORECAST][0]
|
|
assert forecast[ATTR_FORECAST_CONDITION] == ATTR_CONDITION_SNOWY
|
|
assert ATTR_FORECAST_PRECIPITATION not in forecast
|
|
assert forecast[ATTR_FORECAST_PRECIPITATION_PROBABILITY] == 0
|
|
assert forecast[ATTR_FORECAST_TEMP] == 2
|
|
assert forecast[ATTR_FORECAST_TEMP_LOW] == -1
|
|
assert (
|
|
forecast[ATTR_FORECAST_TIME]
|
|
== dt_util.parse_datetime("2021-01-08 23:00:00+00:00").isoformat()
|
|
)
|
|
assert forecast[ATTR_FORECAST_WIND_BEARING] == 90.0
|
|
assert forecast[ATTR_FORECAST_WIND_SPEED] == 0.0
|
|
|
|
state = hass.states.get("weather.aemet_hourly")
|
|
assert state is None
|
|
|
|
|
|
async def test_aemet_weather_legacy(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
entity_registry: er.EntityRegistry,
|
|
) -> None:
|
|
"""Test states of legacy weather."""
|
|
|
|
entity_registry.async_get_or_create(
|
|
WEATHER_DOMAIN,
|
|
DOMAIN,
|
|
"None hourly",
|
|
)
|
|
|
|
hass.config.set_time_zone("UTC")
|
|
freezer.move_to("2021-01-09 12:00:00+00:00")
|
|
await async_init_integration(hass)
|
|
|
|
state = hass.states.get("weather.aemet_daily")
|
|
assert state
|
|
assert state.state == ATTR_CONDITION_SNOWY
|
|
assert state.attributes[ATTR_ATTRIBUTION] == ATTRIBUTION
|
|
assert state.attributes[ATTR_WEATHER_HUMIDITY] == 99.0
|
|
assert state.attributes[ATTR_WEATHER_PRESSURE] == 1004.4 # 100440.0 Pa -> hPa
|
|
assert state.attributes[ATTR_WEATHER_TEMPERATURE] == -0.7
|
|
assert state.attributes[ATTR_WEATHER_WIND_BEARING] == 122.0
|
|
assert state.attributes[ATTR_WEATHER_WIND_GUST_SPEED] == 12.2
|
|
assert state.attributes[ATTR_WEATHER_WIND_SPEED] == 3.2
|
|
forecast = state.attributes[ATTR_FORECAST][0]
|
|
assert forecast[ATTR_FORECAST_CONDITION] == ATTR_CONDITION_SNOWY
|
|
assert ATTR_FORECAST_PRECIPITATION not in forecast
|
|
assert forecast[ATTR_FORECAST_PRECIPITATION_PROBABILITY] == 0
|
|
assert forecast[ATTR_FORECAST_TEMP] == 2
|
|
assert forecast[ATTR_FORECAST_TEMP_LOW] == -1
|
|
assert (
|
|
forecast[ATTR_FORECAST_TIME]
|
|
== dt_util.parse_datetime("2021-01-08 23:00:00+00:00").isoformat()
|
|
)
|
|
assert forecast[ATTR_FORECAST_WIND_BEARING] == 90.0
|
|
assert forecast[ATTR_FORECAST_WIND_SPEED] == 0.0
|
|
|
|
state = hass.states.get("weather.aemet_hourly")
|
|
assert state is None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("service"),
|
|
[
|
|
SERVICE_GET_FORECASTS,
|
|
LEGACY_SERVICE_GET_FORECAST,
|
|
],
|
|
)
|
|
async def test_forecast_service(
|
|
hass: HomeAssistant,
|
|
freezer: FrozenDateTimeFactory,
|
|
snapshot: SnapshotAssertion,
|
|
service: str,
|
|
) -> None:
|
|
"""Test multiple forecast."""
|
|
|
|
hass.config.set_time_zone("UTC")
|
|
freezer.move_to("2021-01-09 12:00:00+00:00")
|
|
await async_init_integration(hass)
|
|
|
|
response = await hass.services.async_call(
|
|
WEATHER_DOMAIN,
|
|
service,
|
|
{
|
|
"entity_id": "weather.aemet",
|
|
"type": "daily",
|
|
},
|
|
blocking=True,
|
|
return_response=True,
|
|
)
|
|
assert response == snapshot
|
|
|
|
response = await hass.services.async_call(
|
|
WEATHER_DOMAIN,
|
|
service,
|
|
{
|
|
"entity_id": "weather.aemet",
|
|
"type": "hourly",
|
|
},
|
|
blocking=True,
|
|
return_response=True,
|
|
)
|
|
assert response == snapshot
|
|
|
|
|
|
@pytest.mark.parametrize("forecast_type", ["daily", "hourly"])
|
|
async def test_forecast_subscription(
|
|
hass: HomeAssistant,
|
|
hass_ws_client: WebSocketGenerator,
|
|
freezer: FrozenDateTimeFactory,
|
|
snapshot: SnapshotAssertion,
|
|
forecast_type: str,
|
|
) -> None:
|
|
"""Test multiple forecast."""
|
|
client = await hass_ws_client(hass)
|
|
|
|
hass.config.set_time_zone("UTC")
|
|
freezer.move_to("2021-01-09 12:00:00+00:00")
|
|
await async_init_integration(hass)
|
|
|
|
await client.send_json_auto_id(
|
|
{
|
|
"type": "weather/subscribe_forecast",
|
|
"forecast_type": forecast_type,
|
|
"entity_id": "weather.aemet",
|
|
}
|
|
)
|
|
msg = await client.receive_json()
|
|
assert msg["success"]
|
|
assert msg["result"] is None
|
|
subscription_id = msg["id"]
|
|
|
|
msg = await client.receive_json()
|
|
assert msg["id"] == subscription_id
|
|
assert msg["type"] == "event"
|
|
forecast1 = msg["event"]["forecast"]
|
|
|
|
assert forecast1 == snapshot
|
|
|
|
with patch(
|
|
"homeassistant.components.aemet.AEMET.api_call",
|
|
side_effect=mock_api_call,
|
|
):
|
|
freezer.tick(WEATHER_UPDATE_INTERVAL + datetime.timedelta(seconds=1))
|
|
await hass.async_block_till_done()
|
|
msg = await client.receive_json()
|
|
|
|
assert msg["id"] == subscription_id
|
|
assert msg["type"] == "event"
|
|
forecast2 = msg["event"]["forecast"]
|
|
|
|
assert forecast2 == snapshot
|