415 lines
14 KiB
Python
415 lines
14 KiB
Python
"""Test Matter locks."""
|
|
|
|
from unittest.mock import MagicMock, call
|
|
|
|
from chip.clusters import Objects as clusters
|
|
from matter_server.client.models.node import MatterNode
|
|
from matter_server.common.helpers.util import create_attribute_path_from_attribute
|
|
import pytest
|
|
|
|
from homeassistant.components.climate import HVACAction, HVACMode
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import (
|
|
set_node_attribute,
|
|
setup_integration_with_node_fixture,
|
|
trigger_subscription_callback,
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="thermostat")
|
|
async def thermostat_fixture(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a thermostat node."""
|
|
return await setup_integration_with_node_fixture(hass, "thermostat", matter_client)
|
|
|
|
|
|
@pytest.fixture(name="room_airconditioner")
|
|
async def room_airconditioner(
|
|
hass: HomeAssistant, matter_client: MagicMock
|
|
) -> MatterNode:
|
|
"""Fixture for a room air conditioner node."""
|
|
return await setup_integration_with_node_fixture(
|
|
hass, "room-airconditioner", matter_client
|
|
)
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_thermostat(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
thermostat: MatterNode,
|
|
) -> None:
|
|
"""Test thermostat."""
|
|
# test default temp range
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["min_temp"] == 7
|
|
assert state.attributes["max_temp"] == 35
|
|
|
|
# test set temperature when target temp is None
|
|
assert state.attributes["temperature"] is None
|
|
assert state.state == HVACMode.COOL
|
|
with pytest.raises(
|
|
ValueError, match="Current target_temperature should not be None"
|
|
):
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"temperature": 22.5,
|
|
},
|
|
blocking=True,
|
|
)
|
|
with pytest.raises(ValueError, match="Temperature must be provided"):
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"target_temp_low": 18,
|
|
"target_temp_high": 26,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
# change system mode to heat_cool
|
|
set_node_attribute(thermostat, 1, 513, 28, 1)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
with pytest.raises(
|
|
ValueError,
|
|
match="current target_temperature_low and target_temperature_high should not be None",
|
|
):
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.HEAT_COOL
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"target_temp_low": 18,
|
|
"target_temp_high": 26,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
# initial state
|
|
set_node_attribute(thermostat, 1, 513, 3, 1600)
|
|
set_node_attribute(thermostat, 1, 513, 4, 3000)
|
|
set_node_attribute(thermostat, 1, 513, 5, 1600)
|
|
set_node_attribute(thermostat, 1, 513, 6, 3000)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["min_temp"] == 16
|
|
assert state.attributes["max_temp"] == 30
|
|
assert state.attributes["hvac_modes"] == [
|
|
HVACMode.OFF,
|
|
HVACMode.HEAT,
|
|
HVACMode.COOL,
|
|
HVACMode.HEAT_COOL,
|
|
]
|
|
|
|
# test system mode update from device
|
|
set_node_attribute(thermostat, 1, 513, 28, 0)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.OFF
|
|
|
|
set_node_attribute(thermostat, 1, 513, 28, 7)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.FAN_ONLY
|
|
|
|
set_node_attribute(thermostat, 1, 513, 28, 8)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.DRY
|
|
|
|
# test running state update from device
|
|
set_node_attribute(thermostat, 1, 513, 41, 1)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.HEATING
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 8)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.HEATING
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 2)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.COOLING
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 16)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.COOLING
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 4)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.FAN
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 32)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.FAN
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 64)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.FAN
|
|
|
|
set_node_attribute(thermostat, 1, 513, 41, 66)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["hvac_action"] == HVACAction.OFF
|
|
|
|
# change system mode to heat
|
|
set_node_attribute(thermostat, 1, 513, 28, 4)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.HEAT
|
|
|
|
# change occupied heating setpoint to 20
|
|
set_node_attribute(thermostat, 1, 513, 18, 2000)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["temperature"] == 20
|
|
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"temperature": 25,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert matter_client.send_device_command.call_count == 1
|
|
assert matter_client.send_device_command.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
endpoint_id=1,
|
|
command=clusters.Thermostat.Commands.SetpointRaiseLower(
|
|
clusters.Thermostat.Enums.SetpointAdjustMode.kHeat,
|
|
50,
|
|
),
|
|
)
|
|
matter_client.send_device_command.reset_mock()
|
|
|
|
# change system mode to cool
|
|
set_node_attribute(thermostat, 1, 513, 28, 3)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.COOL
|
|
|
|
# change occupied cooling setpoint to 18
|
|
set_node_attribute(thermostat, 1, 513, 17, 1800)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["temperature"] == 18
|
|
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"temperature": 16,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert matter_client.send_device_command.call_count == 1
|
|
assert matter_client.send_device_command.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
endpoint_id=1,
|
|
command=clusters.Thermostat.Commands.SetpointRaiseLower(
|
|
clusters.Thermostat.Enums.SetpointAdjustMode.kCool, -20
|
|
),
|
|
)
|
|
matter_client.send_device_command.reset_mock()
|
|
|
|
# change system mode to heat_cool
|
|
set_node_attribute(thermostat, 1, 513, 28, 1)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
with pytest.raises(
|
|
ValueError, match="temperature_low and temperature_high must be provided"
|
|
):
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"temperature": 18,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.state == HVACMode.HEAT_COOL
|
|
|
|
# change occupied cooling setpoint to 18
|
|
set_node_attribute(thermostat, 1, 513, 17, 2500)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
# change occupied heating setpoint to 18
|
|
set_node_attribute(thermostat, 1, 513, 18, 1700)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
state = hass.states.get("climate.longan_link_hvac")
|
|
assert state
|
|
assert state.attributes["target_temp_low"] == 17
|
|
assert state.attributes["target_temp_high"] == 25
|
|
|
|
# change target_temp_low to 18
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"target_temp_low": 18,
|
|
"target_temp_high": 25,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert matter_client.send_device_command.call_count == 1
|
|
assert matter_client.send_device_command.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
endpoint_id=1,
|
|
command=clusters.Thermostat.Commands.SetpointRaiseLower(
|
|
clusters.Thermostat.Enums.SetpointAdjustMode.kHeat, 10
|
|
),
|
|
)
|
|
matter_client.send_device_command.reset_mock()
|
|
set_node_attribute(thermostat, 1, 513, 18, 1800)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
# change target_temp_high to 26
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"target_temp_low": 18,
|
|
"target_temp_high": 26,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert matter_client.send_device_command.call_count == 1
|
|
assert matter_client.send_device_command.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
endpoint_id=1,
|
|
command=clusters.Thermostat.Commands.SetpointRaiseLower(
|
|
clusters.Thermostat.Enums.SetpointAdjustMode.kCool, 10
|
|
),
|
|
)
|
|
matter_client.send_device_command.reset_mock()
|
|
set_node_attribute(thermostat, 1, 513, 17, 2600)
|
|
await trigger_subscription_callback(hass, matter_client)
|
|
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_hvac_mode",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"hvac_mode": HVACMode.HEAT,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert matter_client.write_attribute.call_count == 1
|
|
assert matter_client.write_attribute.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
attribute_path=create_attribute_path_from_attribute(
|
|
endpoint_id=1,
|
|
attribute=clusters.Thermostat.Attributes.SystemMode,
|
|
),
|
|
value=4,
|
|
)
|
|
matter_client.send_device_command.reset_mock()
|
|
|
|
with pytest.raises(ValueError, match="Unsupported hvac mode dry in Matter"):
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_hvac_mode",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"hvac_mode": HVACMode.DRY,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
# change target_temp and hvac_mode in the same call
|
|
matter_client.send_device_command.reset_mock()
|
|
matter_client.write_attribute.reset_mock()
|
|
await hass.services.async_call(
|
|
"climate",
|
|
"set_temperature",
|
|
{
|
|
"entity_id": "climate.longan_link_hvac",
|
|
"temperature": 22,
|
|
"hvac_mode": HVACMode.COOL,
|
|
},
|
|
blocking=True,
|
|
)
|
|
assert matter_client.write_attribute.call_count == 1
|
|
assert matter_client.write_attribute.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
attribute_path=create_attribute_path_from_attribute(
|
|
endpoint_id=1,
|
|
attribute=clusters.Thermostat.Attributes.SystemMode,
|
|
),
|
|
value=3,
|
|
)
|
|
assert matter_client.send_device_command.call_count == 1
|
|
assert matter_client.send_device_command.call_args == call(
|
|
node_id=thermostat.node_id,
|
|
endpoint_id=1,
|
|
command=clusters.Thermostat.Commands.SetpointRaiseLower(
|
|
clusters.Thermostat.Enums.SetpointAdjustMode.kCool, -40
|
|
),
|
|
)
|
|
|
|
|
|
# This tests needs to be adjusted to remove lingering tasks
|
|
@pytest.mark.parametrize("expected_lingering_tasks", [True])
|
|
async def test_room_airconditioner(
|
|
hass: HomeAssistant,
|
|
matter_client: MagicMock,
|
|
room_airconditioner: MatterNode,
|
|
) -> None:
|
|
"""Test if a climate entity is created for a Room Airconditioner device."""
|
|
state = hass.states.get("climate.room_airconditioner")
|
|
assert state
|
|
assert state.attributes["current_temperature"] == 20
|
|
assert state.attributes["min_temp"] == 16
|
|
assert state.attributes["max_temp"] == 32
|