mirror of https://github.com/home-assistant/core
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""The sensor tests for the IPMA platform."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import ENTRY_CONFIG, MockLocation
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_ipma_fire_risk_create_sensors(hass: HomeAssistant) -> None:
|
|
"""Test creation of fire risk sensors."""
|
|
|
|
with patch("pyipma.location.Location.get", return_value=MockLocation()):
|
|
entry = MockConfigEntry(domain="ipma", data=ENTRY_CONFIG)
|
|
entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get("sensor.hometown_fire_risk")
|
|
|
|
assert state.state == "3"
|
|
|
|
|
|
async def test_ipma_uv_index_create_sensors(hass: HomeAssistant) -> None:
|
|
"""Test creation of uv index sensors."""
|
|
|
|
with patch("pyipma.location.Location.get", return_value=MockLocation()):
|
|
entry = MockConfigEntry(domain="ipma", data=ENTRY_CONFIG)
|
|
entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get("sensor.hometown_uv_index")
|
|
|
|
assert state.state == "6"
|