359 lines
12 KiB
Python
359 lines
12 KiB
Python
"""Test Matter sensors."""
|
|
|
|
from datetime import UTC, datetime, timedelta
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from matter_server.client.models.node import MatterNode
|
|
import pytest
|
|
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from .common import (
|
|
set_node_attribute,
|
|
setup_integration_with_node_fixture,
|
|
trigger_subscription_callback,
|
|
)
|
|
|
|
from tests.common import async_fire_time_changed
|
|
|
|
|
|
@pytest.fixture(name="flow_sensor_node")
|
|
async def flow_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a flow sensor node."""
|
|
return await setup_integration_with_node_fixture(hass, "flow-sensor", matter_client)
|
|
|
|
|
|
@pytest.fixture(name="humidity_sensor_node")
|
|
async def humidity_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a humidity sensor node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "humidity-sensor", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="light_sensor_node")
|
|
async def light_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a light sensor node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "light-sensor", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="pressure_sensor_node")
|
|
async def pressure_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a pressure sensor node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "pressure-sensor", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="temperature_sensor_node")
|
|
async def temperature_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a temperature sensor node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "temperature-sensor", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="eve_energy_plug_node")
|
|
async def eve_energy_plug_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a Eve Energy Plug node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "eve-energy-plug", matter_client
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="air_quality_sensor_node")
|
|
async def air_quality_sensor_node_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for an air quality sensor (LightFi AQ1) node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "air-quality-sensor", matter_client
|
|
)
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_sensor_null_value(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
flow_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test flow sensor."""
|
|
state = hass.states.get("sensor.mock_flow_sensor_flow")
|
|
assert state
|
|
assert state.state == "0.0"
|
|
|
|
set_node_attribute(flow_sensor_node, 1, 1028, 0, None)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_flow_sensor_flow")
|
|
assert state
|
|
assert state.state == "unknown"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_flow_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
flow_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test flow sensor."""
|
|
state = hass.states.get("sensor.mock_flow_sensor_flow")
|
|
assert state
|
|
assert state.state == "0.0"
|
|
|
|
set_node_attribute(flow_sensor_node, 1, 1028, 0, 20)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_flow_sensor_flow")
|
|
assert state
|
|
assert state.state == "2.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_humidity_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
humidity_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test humidity sensor."""
|
|
state = hass.states.get("sensor.mock_humidity_sensor_humidity")
|
|
assert state
|
|
assert state.state == "0.0"
|
|
|
|
set_node_attribute(humidity_sensor_node, 1, 1029, 0, 4000)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_humidity_sensor_humidity")
|
|
assert state
|
|
assert state.state == "40.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_light_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
light_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test light sensor."""
|
|
state = hass.states.get("sensor.mock_light_sensor_illuminance")
|
|
assert state
|
|
assert state.state == "1.3"
|
|
|
|
set_node_attribute(light_sensor_node, 1, 1024, 0, 3000)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_light_sensor_illuminance")
|
|
assert state
|
|
assert state.state == "2.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_pressure_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
pressure_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test pressure sensor."""
|
|
state = hass.states.get("sensor.mock_pressure_sensor_pressure")
|
|
assert state
|
|
assert state.state == "0.0"
|
|
|
|
set_node_attribute(pressure_sensor_node, 1, 1027, 0, 1010)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_pressure_sensor_pressure")
|
|
assert state
|
|
assert state.state == "101.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_temperature_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
temperature_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test temperature sensor."""
|
|
state = hass.states.get("sensor.mock_temperature_sensor_temperature")
|
|
assert state
|
|
assert state.state == "21.0"
|
|
|
|
set_node_attribute(temperature_sensor_node, 1, 1026, 0, 2500)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.mock_temperature_sensor_temperature")
|
|
assert state
|
|
assert state.state == "25.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_battery_sensor(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
matter_client: MagicMock,
|
|
eve_contact_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test battery sensor."""
|
|
entity_id = "sensor.eve_door_battery"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "100"
|
|
|
|
set_node_attribute(eve_contact_sensor_node, 1, 47, 12, 100)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "50"
|
|
|
|
entry = entity_registry.async_get(entity_id)
|
|
|
|
assert entry
|
|
assert entry.entity_category == EntityCategory.DIAGNOSTIC
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_eve_energy_sensors(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
matter_client: MagicMock,
|
|
eve_energy_plug_node: MatterNode,
|
|
) -> None:
|
|
"""Test Energy sensors created from Eve Energy custom cluster."""
|
|
# power sensor
|
|
entity_id = "sensor.eve_energy_plug_power"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "0.0"
|
|
assert state.attributes["unit_of_measurement"] == "W"
|
|
assert state.attributes["device_class"] == "power"
|
|
assert state.attributes["friendly_name"] == "Eve Energy Plug Power"
|
|
|
|
# voltage sensor
|
|
entity_id = "sensor.eve_energy_plug_voltage"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "238.800003051758"
|
|
assert state.attributes["unit_of_measurement"] == "V"
|
|
assert state.attributes["device_class"] == "voltage"
|
|
assert state.attributes["friendly_name"] == "Eve Energy Plug Voltage"
|
|
|
|
# energy sensor
|
|
entity_id = "sensor.eve_energy_plug_energy"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "0.220000028610229"
|
|
assert state.attributes["unit_of_measurement"] == "kWh"
|
|
assert state.attributes["device_class"] == "energy"
|
|
assert state.attributes["friendly_name"] == "Eve Energy Plug Energy"
|
|
assert state.attributes["state_class"] == "total_increasing"
|
|
|
|
# current sensor
|
|
entity_id = "sensor.eve_energy_plug_current"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "0.0"
|
|
assert state.attributes["unit_of_measurement"] == "A"
|
|
assert state.attributes["device_class"] == "current"
|
|
assert state.attributes["friendly_name"] == "Eve Energy Plug Current"
|
|
|
|
# test if the sensor gets polled on interval
|
|
eve_energy_plug_node.update_attribute("1/319486977/319422472", 237.0)
|
|
async_fire_time_changed(hass, datetime.now(UTC) + timedelta(seconds=31))
|
|
await hass.async_block_till_done()
|
|
entity_id = "sensor.eve_energy_plug_voltage"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "237.0"
|
|
|
|
# test extra poll triggered when secondary value (switch state) changes
|
|
set_node_attribute(eve_energy_plug_node, 1, 6, 0, True)
|
|
eve_energy_plug_node.update_attribute("1/319486977/319422474", 5.0)
|
|
with patch("homeassistant.components.matter.entity.EXTRA_POLL_DELAY", 0.0):
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
await hass.async_block_till_done()
|
|
entity_id = "sensor.eve_energy_plug_power"
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.state == "5.0"
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_air_quality_sensor(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
air_quality_sensor_node: MatterNode,
|
|
) -> None:
|
|
"""Test air quality sensor."""
|
|
# Carbon Dioxide
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_carbon_dioxide")
|
|
assert state
|
|
assert state.state == "678.0"
|
|
|
|
set_node_attribute(air_quality_sensor_node, 1, 1037, 0, 789)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_carbon_dioxide")
|
|
assert state
|
|
assert state.state == "789.0"
|
|
|
|
# PM1
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm1")
|
|
assert state
|
|
assert state.state == "3.0"
|
|
|
|
set_node_attribute(air_quality_sensor_node, 1, 1068, 0, 50)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm1")
|
|
assert state
|
|
assert state.state == "50.0"
|
|
|
|
# PM2.5
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm2_5")
|
|
assert state
|
|
assert state.state == "3.0"
|
|
|
|
set_node_attribute(air_quality_sensor_node, 1, 1066, 0, 50)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm2_5")
|
|
assert state
|
|
assert state.state == "50.0"
|
|
|
|
# PM10
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm10")
|
|
assert state
|
|
assert state.state == "3.0"
|
|
|
|
set_node_attribute(air_quality_sensor_node, 1, 1069, 0, 50)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("sensor.lightfi_aq1_air_quality_sensor_pm10")
|
|
assert state
|
|
assert state.state == "50.0"
|