307 lines
8.8 KiB
Python
307 lines
8.8 KiB
Python
"""Test the Z-Wave JS number platform."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from zwave_js_server.event import Event
|
|
from zwave_js_server.model.node import Node
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import STATE_UNKNOWN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity import EntityCategory
|
|
import homeassistant.helpers.entity_registry as er
|
|
|
|
DEFAULT_TONE_SELECT_ENTITY = "select.indoor_siren_6_default_tone_2"
|
|
PROTECTION_SELECT_ENTITY = "select.family_room_combo_local_protection_state"
|
|
MULTILEVEL_SWITCH_SELECT_ENTITY = "select.front_door_siren"
|
|
|
|
|
|
async def test_default_tone_select(
|
|
hass: HomeAssistant,
|
|
client: MagicMock,
|
|
aeotec_zw164_siren: Node,
|
|
integration: ConfigEntry,
|
|
) -> None:
|
|
"""Test the default tone select entity."""
|
|
node = aeotec_zw164_siren
|
|
state = hass.states.get(DEFAULT_TONE_SELECT_ENTITY)
|
|
|
|
assert state
|
|
assert state.state == "17ALAR~1 (35 sec)"
|
|
attr = state.attributes
|
|
assert attr["options"] == [
|
|
"01DING~1 (5 sec)",
|
|
"02DING~1 (9 sec)",
|
|
"03TRAD~1 (11 sec)",
|
|
"04ELEC~1 (2 sec)",
|
|
"05WEST~1 (13 sec)",
|
|
"06CHIM~1 (7 sec)",
|
|
"07CUCK~1 (31 sec)",
|
|
"08TRAD~1 (6 sec)",
|
|
"09SMOK~1 (11 sec)",
|
|
"10SMOK~1 (6 sec)",
|
|
"11FIRE~1 (35 sec)",
|
|
"12COSE~1 (5 sec)",
|
|
"13KLAX~1 (38 sec)",
|
|
"14DEEP~1 (41 sec)",
|
|
"15WARN~1 (37 sec)",
|
|
"16TORN~1 (46 sec)",
|
|
"17ALAR~1 (35 sec)",
|
|
"18DEEP~1 (62 sec)",
|
|
"19ALAR~1 (15 sec)",
|
|
"20ALAR~1 (7 sec)",
|
|
"21DIGI~1 (8 sec)",
|
|
"22ALER~1 (64 sec)",
|
|
"23SHIP~1 (4 sec)",
|
|
"25CHRI~1 (4 sec)",
|
|
"26GONG~1 (12 sec)",
|
|
"27SING~1 (1 sec)",
|
|
"28TONA~1 (5 sec)",
|
|
"29UPWA~1 (2 sec)",
|
|
"30DOOR~1 (27 sec)",
|
|
]
|
|
|
|
entity_registry = er.async_get(hass)
|
|
entity_entry = entity_registry.async_get(DEFAULT_TONE_SELECT_ENTITY)
|
|
|
|
assert entity_entry
|
|
assert entity_entry.entity_category is EntityCategory.CONFIG
|
|
|
|
# Test select option with string value
|
|
await hass.services.async_call(
|
|
"select",
|
|
"select_option",
|
|
{"entity_id": DEFAULT_TONE_SELECT_ENTITY, "option": "30DOOR~1 (27 sec)"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert len(client.async_send_command.call_args_list) == 1
|
|
args = client.async_send_command.call_args[0][0]
|
|
assert args["command"] == "node.set_value"
|
|
assert args["nodeId"] == node.node_id
|
|
assert args["valueId"] == {
|
|
"endpoint": 2,
|
|
"commandClass": 121,
|
|
"commandClassName": "Sound Switch",
|
|
"property": "defaultToneId",
|
|
"propertyName": "defaultToneId",
|
|
"ccVersion": 1,
|
|
"metadata": {
|
|
"type": "number",
|
|
"readable": True,
|
|
"writeable": True,
|
|
"label": "Default tone ID",
|
|
"min": 0,
|
|
"max": 254,
|
|
},
|
|
"value": 17,
|
|
}
|
|
assert args["value"] == 30
|
|
|
|
client.async_send_command.reset_mock()
|
|
|
|
# Test value update from value updated event
|
|
event = Event(
|
|
type="value updated",
|
|
data={
|
|
"source": "node",
|
|
"event": "value updated",
|
|
"nodeId": node.node_id,
|
|
"args": {
|
|
"commandClassName": "Sound Switch",
|
|
"commandClass": 121,
|
|
"endpoint": 2,
|
|
"property": "defaultToneId",
|
|
"newValue": 30,
|
|
"prevValue": 17,
|
|
"propertyName": "defaultToneId",
|
|
},
|
|
},
|
|
)
|
|
node.receive_event(event)
|
|
|
|
state = hass.states.get(DEFAULT_TONE_SELECT_ENTITY)
|
|
assert state
|
|
assert state.state == "30DOOR~1 (27 sec)"
|
|
|
|
|
|
async def test_protection_select(
|
|
hass: HomeAssistant,
|
|
client: MagicMock,
|
|
inovelli_lzw36: Node,
|
|
integration: ConfigEntry,
|
|
) -> None:
|
|
"""Test the default tone select entity."""
|
|
node = inovelli_lzw36
|
|
state = hass.states.get(PROTECTION_SELECT_ENTITY)
|
|
|
|
assert state
|
|
assert state.state == "Unprotected"
|
|
attr = state.attributes
|
|
assert attr["options"] == [
|
|
"Unprotected",
|
|
"ProtectedBySequence",
|
|
"NoOperationPossible",
|
|
]
|
|
|
|
entity_registry = er.async_get(hass)
|
|
entity_entry = entity_registry.async_get(PROTECTION_SELECT_ENTITY)
|
|
|
|
assert entity_entry
|
|
assert entity_entry.entity_category is EntityCategory.CONFIG
|
|
|
|
# Test select option with string value
|
|
await hass.services.async_call(
|
|
"select",
|
|
"select_option",
|
|
{"entity_id": PROTECTION_SELECT_ENTITY, "option": "ProtectedBySequence"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert len(client.async_send_command.call_args_list) == 1
|
|
args = client.async_send_command.call_args[0][0]
|
|
assert args["command"] == "node.set_value"
|
|
assert args["nodeId"] == node.node_id
|
|
assert args["valueId"] == {
|
|
"endpoint": 0,
|
|
"commandClass": 117,
|
|
"commandClassName": "Protection",
|
|
"property": "local",
|
|
"propertyName": "local",
|
|
"ccVersion": 2,
|
|
"metadata": {
|
|
"type": "number",
|
|
"readable": True,
|
|
"writeable": True,
|
|
"label": "Local protection state",
|
|
"states": {
|
|
"0": "Unprotected",
|
|
"1": "ProtectedBySequence",
|
|
"2": "NoOperationPossible",
|
|
},
|
|
},
|
|
"value": 0,
|
|
}
|
|
assert args["value"] == 1
|
|
|
|
client.async_send_command.reset_mock()
|
|
|
|
# Test value update from value updated event
|
|
event = Event(
|
|
type="value updated",
|
|
data={
|
|
"source": "node",
|
|
"event": "value updated",
|
|
"nodeId": node.node_id,
|
|
"args": {
|
|
"commandClassName": "Protection",
|
|
"commandClass": 117,
|
|
"endpoint": 0,
|
|
"property": "local",
|
|
"newValue": 1,
|
|
"prevValue": 0,
|
|
"propertyName": "local",
|
|
},
|
|
},
|
|
)
|
|
node.receive_event(event)
|
|
|
|
state = hass.states.get(PROTECTION_SELECT_ENTITY)
|
|
assert state
|
|
assert state.state == "ProtectedBySequence"
|
|
|
|
# Test null value
|
|
event = Event(
|
|
type="value updated",
|
|
data={
|
|
"source": "node",
|
|
"event": "value updated",
|
|
"nodeId": node.node_id,
|
|
"args": {
|
|
"commandClassName": "Protection",
|
|
"commandClass": 117,
|
|
"endpoint": 0,
|
|
"property": "local",
|
|
"newValue": None,
|
|
"prevValue": 1,
|
|
"propertyName": "local",
|
|
},
|
|
},
|
|
)
|
|
node.receive_event(event)
|
|
|
|
state = hass.states.get(PROTECTION_SELECT_ENTITY)
|
|
assert state
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
|
|
async def test_multilevel_switch_select(hass, client, fortrezz_ssa1_siren, integration):
|
|
"""Test Multilevel Switch CC based select entity."""
|
|
node = fortrezz_ssa1_siren
|
|
state = hass.states.get(MULTILEVEL_SWITCH_SELECT_ENTITY)
|
|
|
|
assert state
|
|
assert state.state == "Off"
|
|
attr = state.attributes
|
|
assert attr["options"] == [
|
|
"Off",
|
|
"Strobe ONLY",
|
|
"Siren ONLY",
|
|
"Siren & Strobe FULL Alarm",
|
|
]
|
|
|
|
# Test select option with string value
|
|
await hass.services.async_call(
|
|
"select",
|
|
"select_option",
|
|
{"entity_id": MULTILEVEL_SWITCH_SELECT_ENTITY, "option": "Strobe ONLY"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert len(client.async_send_command.call_args_list) == 1
|
|
args = client.async_send_command.call_args[0][0]
|
|
assert args["command"] == "node.set_value"
|
|
assert args["nodeId"] == node.node_id
|
|
assert args["valueId"] == {
|
|
"endpoint": 0,
|
|
"commandClass": 38,
|
|
"commandClassName": "Multilevel Switch",
|
|
"property": "targetValue",
|
|
"propertyName": "targetValue",
|
|
"ccVersion": 1,
|
|
"metadata": {
|
|
"type": "number",
|
|
"readable": True,
|
|
"writeable": True,
|
|
"label": "Target value",
|
|
"valueChangeOptions": ["transitionDuration"],
|
|
"min": 0,
|
|
"max": 99,
|
|
},
|
|
}
|
|
assert args["value"] == 33
|
|
|
|
client.async_send_command.reset_mock()
|
|
|
|
# Test value update from value updated event
|
|
event = Event(
|
|
type="value updated",
|
|
data={
|
|
"source": "node",
|
|
"event": "value updated",
|
|
"nodeId": node.node_id,
|
|
"args": {
|
|
"commandClassName": "Multilevel Switch",
|
|
"commandClass": 38,
|
|
"endpoint": 0,
|
|
"property": "currentValue",
|
|
"newValue": 33,
|
|
"prevValue": 0,
|
|
"propertyName": "currentValue",
|
|
},
|
|
},
|
|
)
|
|
node.receive_event(event)
|
|
|
|
state = hass.states.get(MULTILEVEL_SWITCH_SELECT_ENTITY)
|
|
assert state.state == "Strobe ONLY"
|