77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
"""Tests for the Freedompro sensor."""
|
|
from datetime import timedelta
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.helpers import entity_registry as er
|
|
from homeassistant.util.dt import utcnow
|
|
|
|
from tests.common import async_fire_time_changed
|
|
from tests.components.freedompro.const import DEVICES_STATE
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"entity_id, uid, name",
|
|
[
|
|
(
|
|
"sensor.garden_humidity_sensor",
|
|
"3WRRJR6RCZQZSND8VP0YTO3YXCSOFPKBMW8T51TU-LQ*QN-DDFMPEPRDOQV7W7JQG3NL0NPZGTLIBYT3HFSPNEY",
|
|
"Garden humidity sensor",
|
|
),
|
|
(
|
|
"sensor.living_room_temperature_sensor",
|
|
"3WRRJR6RCZQZSND8VP0YTO3YXCSOFPKBMW8T51TU-LQ*LWPVY7X1AX0DRWLYUUNZ3ZSTHMYNDDBQTPZCZQUUASA",
|
|
"Living room temperature sensor",
|
|
),
|
|
(
|
|
"sensor.garden_light_sensors",
|
|
"3WRRJR6RCZQZSND8VP0YTO3YXCSOFPKBMW8T51TU-LQ*JVRAR_6WVL1Y0PJ5GFWGPMFV7FLVD4MZKBWXC_UFWYM",
|
|
"Garden light sensors",
|
|
),
|
|
],
|
|
)
|
|
async def test_sensor_get_state(
|
|
hass, init_integration, entity_id: str, uid: str, name: str
|
|
):
|
|
"""Test states of the sensor."""
|
|
init_integration
|
|
registry = er.async_get(hass)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes.get("friendly_name") == name
|
|
|
|
entry = registry.async_get(entity_id)
|
|
assert entry
|
|
assert entry.unique_id == uid
|
|
|
|
assert state.state == "0"
|
|
|
|
get_states_response = list(DEVICES_STATE)
|
|
for state_response in get_states_response:
|
|
if state_response["uid"] == uid:
|
|
if state_response["type"] == "lightSensor":
|
|
state_response["state"]["currentAmbientLightLevel"] = "1"
|
|
if state_response["type"] == "temperatureSensor":
|
|
state_response["state"]["currentTemperature"] = "1"
|
|
if state_response["type"] == "humiditySensor":
|
|
state_response["state"]["currentRelativeHumidity"] = "1"
|
|
with patch(
|
|
"homeassistant.components.freedompro.get_states",
|
|
return_value=get_states_response,
|
|
):
|
|
|
|
async_fire_time_changed(hass, utcnow() + timedelta(hours=2))
|
|
await hass.async_block_till_done()
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes.get("friendly_name") == name
|
|
|
|
entry = registry.async_get(entity_id)
|
|
assert entry
|
|
assert entry.unique_id == uid
|
|
|
|
assert state.state == "1"
|