core/tests/components/zwave/test_init.py

1792 lines
60 KiB
Python
Raw Normal View History

2017-02-13 16:54:11 +00:00
"""Tests for the Z-Wave init."""
import asyncio
from collections import OrderedDict
from datetime import datetime
import unittest
from unittest.mock import MagicMock, patch
import pytest
from pytz import utc
import voluptuous as vol
2017-02-13 16:54:11 +00:00
from homeassistant.bootstrap import async_setup_component
from homeassistant.components import zwave
from homeassistant.components.zwave import (
2019-07-31 19:25:30 +00:00
CONF_DEVICE_CONFIG_GLOB,
CONFIG_SCHEMA,
DATA_NETWORK,
const,
)
from homeassistant.components.zwave.binary_sensor import get_device
from homeassistant.const import ATTR_ENTITY_ID, EVENT_HOMEASSISTANT_START
from homeassistant.helpers.entity_registry import async_get_registry
2019-07-31 19:25:30 +00:00
from homeassistant.helpers.device_registry import async_get_registry as get_dev_reg
from homeassistant.setup import setup_component
from tests.common import (
2019-07-31 19:25:30 +00:00
async_fire_time_changed,
get_test_home_assistant,
mock_coro,
mock_registry,
)
from tests.mock.zwave import MockEntityValues, MockNetwork, MockNode, MockValue
2017-02-13 16:54:11 +00:00
async def test_valid_device_config(hass, mock_openzwave):
"""Test valid device config."""
2019-07-31 19:25:30 +00:00
device_config = {"light.kitchen": {"ignored": "true"}}
result = await async_setup_component(
hass, "zwave", {"zwave": {"device_config": device_config}}
)
await hass.async_block_till_done()
2017-02-13 16:54:11 +00:00
assert result
2017-02-13 16:54:11 +00:00
async def test_invalid_device_config(hass, mock_openzwave):
"""Test invalid device config."""
2019-07-31 19:25:30 +00:00
device_config = {"light.kitchen": {"some_ignored": "true"}}
result = await async_setup_component(
hass, "zwave", {"zwave": {"device_config": device_config}}
)
await hass.async_block_till_done()
2017-02-13 16:54:11 +00:00
assert not result
def test_config_access_error():
"""Test threading error accessing config values."""
node = MagicMock()
def side_effect():
raise RuntimeError
node.values.values.side_effect = side_effect
result = zwave.get_config_value(node, 1)
assert result is None
async def test_network_options(hass, mock_openzwave):
"""Test network options."""
2019-07-31 19:25:30 +00:00
result = await async_setup_component(
hass,
"zwave",
{"zwave": {"usb_path": "mock_usb_path", "config_path": "mock_config_path"}},
)
await hass.async_block_till_done()
assert result
network = hass.data[zwave.DATA_NETWORK]
2019-07-31 19:25:30 +00:00
assert network.options.device == "mock_usb_path"
assert network.options.config_path == "mock_config_path"
async def test_network_key_validation(hass, mock_openzwave):
"""Test network key validation."""
test_values = [
2019-07-31 19:25:30 +00:00
(
"0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, "
"0x0C, 0x0D, 0x0E, 0x0F, 0x10"
),
(
"0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0A,0x0B,0x0C,0x0D,"
"0x0E,0x0F,0x10"
),
]
for value in test_values:
2019-07-31 19:25:30 +00:00
result = zwave.CONFIG_SCHEMA({"zwave": {"network_key": value}})
assert result["zwave"]["network_key"] == value
async def test_erronous_network_key_fails_validation(hass, mock_openzwave):
"""Test failing erronous network key validation."""
test_values = [
2019-07-31 19:25:30 +00:00
(
"0x 01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, "
"0x0C, 0x0D, 0x0E, 0x0F, 0x10"
),
(
"0X01,0X02,0X03,0X04,0X05,0X06,0X07,0X08,0X09,0X0A,0X0B,0X0C,0X0D,"
"0X0E,0X0F,0X10"
),
"invalid",
"1234567",
1234567,
]
for value in test_values:
with pytest.raises(vol.Invalid):
2019-07-31 19:25:30 +00:00
zwave.CONFIG_SCHEMA({"zwave": {"network_key": value}})
async def test_auto_heal_midnight(hass, mock_openzwave):
"""Test network auto-heal at midnight."""
2019-07-31 19:25:30 +00:00
await async_setup_component(hass, "zwave", {"zwave": {"autoheal": True}})
await hass.async_block_till_done()
network = hass.data[zwave.DATA_NETWORK]
assert not network.heal.called
time = utc.localize(datetime(2017, 5, 6, 0, 0, 0))
async_fire_time_changed(hass, time)
await hass.async_block_till_done()
assert network.heal.called
assert len(network.heal.mock_calls) == 1
async def test_auto_heal_disabled(hass, mock_openzwave):
"""Test network auto-heal disabled."""
2019-07-31 19:25:30 +00:00
await async_setup_component(hass, "zwave", {"zwave": {"autoheal": False}})
await hass.async_block_till_done()
network = hass.data[zwave.DATA_NETWORK]
assert not network.heal.called
time = utc.localize(datetime(2017, 5, 6, 0, 0, 0))
async_fire_time_changed(hass, time)
await hass.async_block_till_done()
assert not network.heal.called
async def test_setup_platform(hass, mock_openzwave):
"""Test invalid device config."""
mock_device = MagicMock()
hass.data[DATA_NETWORK] = MagicMock()
hass.data[zwave.DATA_DEVICES] = {456: mock_device}
async_add_entities = MagicMock()
2019-07-31 19:25:30 +00:00
result = await zwave.async_setup_platform(hass, None, async_add_entities, None)
assert not result
assert not async_add_entities.called
result = await zwave.async_setup_platform(
2019-07-31 19:25:30 +00:00
hass, None, async_add_entities, {const.DISCOVERY_DEVICE: 123}
)
assert not result
assert not async_add_entities.called
result = await zwave.async_setup_platform(
2019-07-31 19:25:30 +00:00
hass, None, async_add_entities, {const.DISCOVERY_DEVICE: 456}
)
assert result
assert async_add_entities.called
assert len(async_add_entities.mock_calls) == 1
assert async_add_entities.mock_calls[0][1][0] == [mock_device]
async def test_zwave_ready_wait(hass, mock_openzwave):
"""Test that zwave continues after waiting for network ready."""
# Initialize zwave
2019-07-31 19:25:30 +00:00
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
sleeps = []
def utcnow():
return datetime.fromtimestamp(len(sleeps))
asyncio_sleep = asyncio.sleep
async def sleep(duration, loop=None):
if duration > 0:
sleeps.append(duration)
await asyncio_sleep(0)
2019-07-31 19:25:30 +00:00
with patch("homeassistant.components.zwave.dt_util.utcnow", new=utcnow):
with patch("asyncio.sleep", new=sleep):
with patch.object(zwave, "_LOGGER") as mock_logger:
hass.data[DATA_NETWORK].state = MockNetwork.STATE_STARTED
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
await hass.async_block_till_done()
assert len(sleeps) == const.NETWORK_READY_WAIT_SECS
assert mock_logger.warning.called
assert len(mock_logger.warning.mock_calls) == 1
2019-07-31 19:25:30 +00:00
assert (
mock_logger.warning.mock_calls[0][1][1]
== const.NETWORK_READY_WAIT_SECS
)
async def test_device_entity(hass, mock_openzwave):
"""Test device entity base class."""
2019-07-31 19:25:30 +00:00
node = MockNode(node_id="10", name="Mock Node")
value = MockValue(
data=False,
node=node,
instance=2,
object_id="11",
label="Sensor",
command_class=const.COMMAND_CLASS_SENSOR_BINARY,
)
power_value = MockValue(
data=50.123456, node=node, precision=3, command_class=const.COMMAND_CLASS_METER
)
values = MockEntityValues(primary=value, power=power_value)
2019-07-31 19:25:30 +00:00
device = zwave.ZWaveDeviceEntity(values, "zwave")
device.hass = hass
device.value_added()
device.update_properties()
await hass.async_block_till_done()
assert not device.should_poll
assert device.unique_id == "10-11"
2019-07-31 19:25:30 +00:00
assert device.name == "Mock Node Sensor"
assert device.device_state_attributes[zwave.ATTR_POWER] == 50.123
async def test_node_removed(hass, mock_openzwave):
"""Test node removed in base class."""
# Create a mock node & node entity
2019-07-31 19:25:30 +00:00
node = MockNode(node_id="10", name="Mock Node")
value = MockValue(
data=False,
node=node,
instance=2,
object_id="11",
label="Sensor",
command_class=const.COMMAND_CLASS_SENSOR_BINARY,
)
power_value = MockValue(
data=50.123456, node=node, precision=3, command_class=const.COMMAND_CLASS_METER
)
values = MockEntityValues(primary=value, power=power_value)
2019-07-31 19:25:30 +00:00
device = zwave.ZWaveDeviceEntity(values, "zwave")
device.hass = hass
2019-07-31 19:25:30 +00:00
device.entity_id = "zwave.mock_node"
device.value_added()
device.update_properties()
await hass.async_block_till_done()
# Save it to the entity registry
registry = mock_registry(hass)
2019-07-31 19:25:30 +00:00
registry.async_get_or_create("zwave", "zwave", device.unique_id)
device.entity_id = registry.async_get_entity_id("zwave", "zwave", device.unique_id)
# Create dummy entity registry entries for other integrations
2019-07-31 19:25:30 +00:00
hue_entity = registry.async_get_or_create("light", "hue", 1234)
zha_entity = registry.async_get_or_create("sensor", "zha", 5678)
# Verify our Z-Wave entity is registered
assert registry.async_is_registered(device.entity_id)
# Remove it
entity_id = device.entity_id
await device.node_removed()
# Verify registry entry for our Z-Wave node is gone
assert not registry.async_is_registered(entity_id)
# Verify registry entries for our other entities remain
assert registry.async_is_registered(hue_entity.entity_id)
assert registry.async_is_registered(zha_entity.entity_id)
async def test_node_discovery(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_NODE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
node = MockNode(node_id=14)
hass.async_add_job(mock_receivers[0], node)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert hass.states.get("zwave.mock_node").state == "unknown"
async def test_unparsed_node_discovery(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_NODE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
2019-07-31 19:25:30 +00:00
node = MockNode(node_id=14, manufacturer_name=None, name=None, is_ready=False)
sleeps = []
def utcnow():
return datetime.fromtimestamp(len(sleeps))
asyncio_sleep = asyncio.sleep
async def sleep(duration, loop=None):
if duration > 0:
sleeps.append(duration)
await asyncio_sleep(0)
2019-07-31 19:25:30 +00:00
with patch("homeassistant.components.zwave.dt_util.utcnow", new=utcnow):
with patch("asyncio.sleep", new=sleep):
with patch.object(zwave, "_LOGGER") as mock_logger:
hass.async_add_job(mock_receivers[0], node)
await hass.async_block_till_done()
assert len(sleeps) == const.NODE_READY_WAIT_SECS
assert mock_logger.warning.called
assert len(mock_logger.warning.mock_calls) == 1
2019-07-31 19:25:30 +00:00
assert mock_logger.warning.mock_calls[0][1][1:] == (
14,
const.NODE_READY_WAIT_SECS,
)
assert hass.states.get("zwave.unknown_node_14").state == "unknown"
async def test_node_ignored(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_NODE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(
hass,
"zwave",
{"zwave": {"device_config": {"zwave.mock_node": {"ignored": True}}}},
)
await hass.async_block_till_done()
assert len(mock_receivers) == 1
node = MockNode(node_id=14)
hass.async_add_job(mock_receivers[0], node)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert hass.states.get("zwave.mock_node") is None
async def test_value_discovery(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_VALUE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_SENSOR_BINARY)
2019-07-31 19:25:30 +00:00
value = MockValue(
data=False,
node=node,
index=12,
instance=13,
command_class=const.COMMAND_CLASS_SENSOR_BINARY,
type=const.TYPE_BOOL,
genre=const.GENRE_USER,
)
hass.async_add_job(mock_receivers[0], node, value)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert hass.states.get("binary_sensor.mock_node_mock_value").state == "off"
async def test_value_entities(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = {}
def mock_connect(receiver, signal, *args, **kwargs):
mock_receivers[signal] = receiver
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
zwave_network = hass.data[DATA_NETWORK]
zwave_network.state = MockNetwork.STATE_READY
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
await hass.async_block_till_done()
assert mock_receivers
2019-07-31 19:25:30 +00:00
hass.async_add_job(mock_receivers[MockNetwork.SIGNAL_ALL_NODES_QUERIED])
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_SENSOR_BINARY)
zwave_network.nodes = {node.node_id: node}
value = MockValue(
2019-07-31 19:25:30 +00:00
data=False,
node=node,
index=12,
instance=1,
command_class=const.COMMAND_CLASS_SENSOR_BINARY,
2019-07-31 19:25:30 +00:00
type=const.TYPE_BOOL,
genre=const.GENRE_USER,
)
node.values = {"primary": value, value.value_id: value}
value2 = MockValue(
2019-07-31 19:25:30 +00:00
data=False,
node=node,
index=12,
instance=2,
label="Mock Value B",
command_class=const.COMMAND_CLASS_SENSOR_BINARY,
2019-07-31 19:25:30 +00:00
type=const.TYPE_BOOL,
genre=const.GENRE_USER,
)
node.values[value2.value_id] = value2
2019-07-31 19:25:30 +00:00
hass.async_add_job(mock_receivers[MockNetwork.SIGNAL_NODE_ADDED], node)
hass.async_add_job(mock_receivers[MockNetwork.SIGNAL_VALUE_ADDED], node, value)
hass.async_add_job(mock_receivers[MockNetwork.SIGNAL_VALUE_ADDED], node, value2)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert hass.states.get("binary_sensor.mock_node_mock_value").state == "off"
assert hass.states.get("binary_sensor.mock_node_mock_value_b").state == "off"
ent_reg = await async_get_registry(hass)
dev_reg = await get_dev_reg(hass)
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("zwave.mock_node")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "node-{}".format(node.node_id)
node_dev_id = entry.device_id
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.mock_node_mock_value")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "{}-{}".format(node.node_id, value.object_id)
assert entry.name is None
assert entry.device_id == node_dev_id
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.mock_node_mock_value_b")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "{}-{}".format(node.node_id, value2.object_id)
assert entry.name is None
assert entry.device_id != node_dev_id
device_id_b = entry.device_id
device = dev_reg.async_get(node_dev_id)
assert device is not None
assert device.name == node.name
old_device = device
device = dev_reg.async_get(device_id_b)
assert device is not None
assert device.name == "{} ({})".format(node.name, value2.instance)
# test renaming without updating
2019-07-31 19:25:30 +00:00
await hass.services.async_call(
"zwave",
"rename_node",
{const.ATTR_NODE_ID: node.node_id, const.ATTR_NAME: "Demo Node"},
)
await hass.async_block_till_done()
assert node.name == "Demo Node"
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("zwave.mock_node")
assert entry is not None
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.mock_node_mock_value")
assert entry is not None
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.mock_node_mock_value_b")
assert entry is not None
device = dev_reg.async_get(node_dev_id)
assert device is not None
assert device.id == old_device.id
assert device.name == node.name
device = dev_reg.async_get(device_id_b)
assert device is not None
assert device.name == "{} ({})".format(node.name, value2.instance)
# test renaming
2019-07-31 19:25:30 +00:00
await hass.services.async_call(
"zwave",
"rename_node",
{
const.ATTR_NODE_ID: node.node_id,
const.ATTR_UPDATE_IDS: True,
const.ATTR_NAME: "New Node",
},
)
await hass.async_block_till_done()
assert node.name == "New Node"
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("zwave.new_node")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "node-{}".format(node.node_id)
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.new_node_mock_value")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "{}-{}".format(node.node_id, value.object_id)
device = dev_reg.async_get(node_dev_id)
assert device is not None
assert device.id == old_device.id
assert device.name == node.name
device = dev_reg.async_get(device_id_b)
assert device is not None
assert device.name == "{} ({})".format(node.name, value2.instance)
2019-07-31 19:25:30 +00:00
await hass.services.async_call(
"zwave",
"rename_value",
{
const.ATTR_NODE_ID: node.node_id,
const.ATTR_VALUE_ID: value.object_id,
const.ATTR_UPDATE_IDS: True,
const.ATTR_NAME: "New Label",
},
)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
entry = ent_reg.async_get("binary_sensor.new_node_new_label")
assert entry is not None
2019-07-31 19:25:30 +00:00
assert entry.unique_id == "{}-{}".format(node.node_id, value.object_id)
async def test_value_discovery_existing_entity(hass, mock_openzwave):
"""Test discovery of a node."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_VALUE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_THERMOSTAT)
setpoint = MockValue(
2019-07-31 19:25:30 +00:00
data=22.0,
node=node,
index=12,
instance=13,
command_class=const.COMMAND_CLASS_THERMOSTAT_SETPOINT,
2019-07-31 19:25:30 +00:00
genre=const.GENRE_USER,
units="C",
)
hass.async_add_job(mock_receivers[0], node, setpoint)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert (
hass.states.get("climate.mock_node_mock_value").attributes["temperature"]
== 22.0
)
assert (
hass.states.get("climate.mock_node_mock_value").attributes[
"current_temperature"
]
is None
)
def mock_update(self):
self.hass.add_job(self.async_update_ha_state)
2019-07-31 19:25:30 +00:00
with patch.object(
zwave.node_entity.ZWaveBaseEntity, "maybe_schedule_update", new=mock_update
):
temperature = MockValue(
2019-07-31 19:25:30 +00:00
data=23.5,
node=node,
index=1,
instance=13,
command_class=const.COMMAND_CLASS_SENSOR_MULTILEVEL,
2019-07-31 19:25:30 +00:00
genre=const.GENRE_USER,
units="C",
)
hass.async_add_job(mock_receivers[0], node, temperature)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert (
hass.states.get("climate.mock_node_mock_value").attributes["temperature"]
== 22.0
)
assert (
hass.states.get("climate.mock_node_mock_value").attributes[
"current_temperature"
]
== 23.5
)
async def test_power_schemes(hass, mock_openzwave):
"""Test power attribute."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_VALUE_ADDED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_SWITCH_BINARY)
switch = MockValue(
2019-07-31 19:25:30 +00:00
data=True,
node=node,
index=12,
instance=13,
command_class=const.COMMAND_CLASS_SWITCH_BINARY,
2019-07-31 19:25:30 +00:00
genre=const.GENRE_USER,
type=const.TYPE_BOOL,
)
hass.async_add_job(mock_receivers[0], node, switch)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert hass.states.get("switch.mock_node_mock_value").state == "on"
assert (
"power_consumption"
not in hass.states.get("switch.mock_node_mock_value").attributes
)
def mock_update(self):
self.hass.add_job(self.async_update_ha_state)
2019-07-31 19:25:30 +00:00
with patch.object(
zwave.node_entity.ZWaveBaseEntity, "maybe_schedule_update", new=mock_update
):
power = MockValue(
2019-07-31 19:25:30 +00:00
data=23.5,
node=node,
index=const.INDEX_SENSOR_MULTILEVEL_POWER,
instance=13,
command_class=const.COMMAND_CLASS_SENSOR_MULTILEVEL,
)
hass.async_add_job(mock_receivers[0], node, power)
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
assert (
hass.states.get("switch.mock_node_mock_value").attributes["power_consumption"]
== 23.5
)
async def test_network_ready(hass, mock_openzwave):
"""Test Node network ready event."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_ALL_NODES_QUERIED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
events = []
def listener(event):
events.append(event)
hass.bus.async_listen(const.EVENT_NETWORK_COMPLETE, listener)
hass.async_add_job(mock_receivers[0])
await hass.async_block_till_done()
assert len(events) == 1
async def test_network_complete(hass, mock_openzwave):
"""Test Node network complete event."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_AWAKE_NODES_QUERIED:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
events = []
def listener(event):
events.append(event)
hass.bus.async_listen(const.EVENT_NETWORK_READY, listener)
hass.async_add_job(mock_receivers[0])
await hass.async_block_till_done()
assert len(events) == 1
async def test_network_complete_some_dead(hass, mock_openzwave):
"""Test Node network complete some dead event."""
mock_receivers = []
def mock_connect(receiver, signal, *args, **kwargs):
if signal == MockNetwork.SIGNAL_ALL_NODES_QUERIED_SOME_DEAD:
mock_receivers.append(receiver)
2019-07-31 19:25:30 +00:00
with patch("pydispatch.dispatcher.connect", new=mock_connect):
await async_setup_component(hass, "zwave", {"zwave": {}})
await hass.async_block_till_done()
assert len(mock_receivers) == 1
events = []
def listener(event):
events.append(event)
hass.bus.async_listen(const.EVENT_NETWORK_COMPLETE_SOME_DEAD, listener)
hass.async_add_job(mock_receivers[0])
await hass.async_block_till_done()
assert len(events) == 1
class TestZWaveDeviceEntityValues(unittest.TestCase):
"""Tests for the ZWaveDeviceEntityValues helper."""
@pytest.fixture(autouse=True)
def set_mock_openzwave(self, mock_openzwave):
"""Use the mock_openzwave fixture for this class."""
self.mock_openzwave = mock_openzwave
def setUp(self):
"""Initialize values for this testcase class."""
self.hass = get_test_home_assistant()
self.hass.start()
self.registry = mock_registry(self.hass)
2019-07-31 19:25:30 +00:00
setup_component(self.hass, "zwave", {"zwave": {}})
self.hass.block_till_done()
self.node = MockNode()
self.mock_schema = {
2019-07-31 19:25:30 +00:00
const.DISC_COMPONENT: "mock_component",
const.DISC_VALUES: {
2019-07-31 19:25:30 +00:00
const.DISC_PRIMARY: {const.DISC_COMMAND_CLASS: ["mock_primary_class"]},
"secondary": {const.DISC_COMMAND_CLASS: ["mock_secondary_class"]},
"optional": {
const.DISC_COMMAND_CLASS: ["mock_optional_class"],
const.DISC_OPTIONAL: True,
2019-07-31 19:25:30 +00:00
},
},
}
self.primary = MockValue(
2019-07-31 19:25:30 +00:00
command_class="mock_primary_class", node=self.node, value_id=1000
)
self.secondary = MockValue(command_class="mock_secondary_class", node=self.node)
self.duplicate_secondary = MockValue(
2019-07-31 19:25:30 +00:00
command_class="mock_secondary_class", node=self.node
)
self.optional = MockValue(command_class="mock_optional_class", node=self.node)
self.no_match_value = MockValue(command_class="mock_bad_class", node=self.node)
self.entity_id = "mock_component.mock_node_mock_value"
self.zwave_config = {"zwave": {}}
self.device_config = {self.entity_id: {}}
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_discovery(self, discovery, import_module):
"""Test the creation of a new entity."""
discovery.async_load_platform.return_value = mock_coro()
mock_platform = MagicMock()
import_module.return_value = mock_platform
mock_device = MagicMock()
2019-07-31 19:25:30 +00:00
mock_device.name = "test_device"
mock_platform.get_device.return_value = mock_device
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
assert values.primary is self.primary
assert len(list(values)) == 3
2019-07-31 19:25:30 +00:00
assert sorted(list(values), key=lambda a: id(a)) == sorted(
[self.primary, None, None], key=lambda a: id(a)
)
assert not discovery.async_load_platform.called
values.check_value(self.secondary)
self.hass.block_till_done()
assert values.secondary is self.secondary
assert len(list(values)) == 3
2019-07-31 19:25:30 +00:00
assert sorted(list(values), key=lambda a: id(a)) == sorted(
[self.primary, self.secondary, None], key=lambda a: id(a)
)
assert discovery.async_load_platform.called
assert len(discovery.async_load_platform.mock_calls) == 1
args = discovery.async_load_platform.mock_calls[0][1]
assert args[0] == self.hass
2019-07-31 19:25:30 +00:00
assert args[1] == "mock_component"
assert args[2] == "zwave"
assert args[3] == {const.DISCOVERY_DEVICE: mock_device.unique_id}
assert args[4] == self.zwave_config
discovery.async_load_platform.reset_mock()
values.check_value(self.optional)
values.check_value(self.duplicate_secondary)
values.check_value(self.no_match_value)
self.hass.block_till_done()
assert values.optional is self.optional
assert len(list(values)) == 3
2019-07-31 19:25:30 +00:00
assert sorted(list(values), key=lambda a: id(a)) == sorted(
[self.primary, self.secondary, self.optional], key=lambda a: id(a)
)
assert not discovery.async_load_platform.called
assert values._entity.value_added.called
assert len(values._entity.value_added.mock_calls) == 1
assert values._entity.value_changed.called
assert len(values._entity.value_changed.mock_calls) == 1
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_existing_values(self, discovery, import_module):
"""Test the loading of already discovered values."""
discovery.async_load_platform.return_value = mock_coro()
mock_platform = MagicMock()
import_module.return_value = mock_platform
mock_device = MagicMock()
2019-07-31 19:25:30 +00:00
mock_device.name = "test_device"
mock_platform.get_device.return_value = mock_device
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
self.optional.value_id: self.optional,
self.no_match_value.value_id: self.no_match_value,
}
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
self.hass.block_till_done()
assert values.primary is self.primary
assert values.secondary is self.secondary
assert values.optional is self.optional
assert len(list(values)) == 3
2019-07-31 19:25:30 +00:00
assert sorted(list(values), key=lambda a: id(a)) == sorted(
[self.primary, self.secondary, self.optional], key=lambda a: id(a)
)
assert discovery.async_load_platform.called
assert len(discovery.async_load_platform.mock_calls) == 1
args = discovery.async_load_platform.mock_calls[0][1]
assert args[0] == self.hass
2019-07-31 19:25:30 +00:00
assert args[1] == "mock_component"
assert args[2] == "zwave"
assert args[3] == {const.DISCOVERY_DEVICE: mock_device.unique_id}
assert args[4] == self.zwave_config
assert not self.primary.enable_poll.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_node_schema_mismatch(self, discovery, import_module):
"""Test node schema mismatch."""
2019-07-31 19:25:30 +00:00
self.node.generic = "no_match"
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
}
2019-07-31 19:25:30 +00:00
self.mock_schema[const.DISC_GENERIC_DEVICE_CLASS] = ["generic_match"]
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
values._check_entity_ready()
self.hass.block_till_done()
assert not discovery.async_load_platform.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_workaround_component(self, discovery, import_module):
"""Test component workaround."""
discovery.async_load_platform.return_value = mock_coro()
mock_platform = MagicMock()
import_module.return_value = mock_platform
mock_device = MagicMock()
2019-07-31 19:25:30 +00:00
mock_device.name = "test_device"
mock_platform.get_device.return_value = mock_device
2019-07-31 19:25:30 +00:00
self.node.manufacturer_id = "010f"
self.node.product_type = "0b00"
self.primary.command_class = const.COMMAND_CLASS_SENSOR_ALARM
2019-07-31 19:25:30 +00:00
self.entity_id = "binary_sensor.mock_node_mock_value"
self.device_config = {self.entity_id: {}}
self.mock_schema = {
2019-07-31 19:25:30 +00:00
const.DISC_COMPONENT: "mock_component",
const.DISC_VALUES: {
const.DISC_PRIMARY: {
2019-07-31 19:25:30 +00:00
const.DISC_COMMAND_CLASS: [const.COMMAND_CLASS_SWITCH_BINARY]
}
},
}
2019-07-31 19:25:30 +00:00
with patch.object(zwave, "async_dispatcher_send") as mock_dispatch_send:
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
values._check_entity_ready()
self.hass.block_till_done()
assert mock_dispatch_send.called
assert len(mock_dispatch_send.mock_calls) == 1
args = mock_dispatch_send.mock_calls[0][1]
2019-07-31 19:25:30 +00:00
assert args[1] == "zwave_new_binary_sensor"
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_workaround_ignore(self, discovery, import_module):
"""Test ignore workaround."""
2019-07-31 19:25:30 +00:00
self.node.manufacturer_id = "010f"
self.node.product_type = "0301"
self.primary.command_class = const.COMMAND_CLASS_SWITCH_BINARY
self.mock_schema = {
2019-07-31 19:25:30 +00:00
const.DISC_COMPONENT: "mock_component",
const.DISC_VALUES: {
const.DISC_PRIMARY: {
2019-07-31 19:25:30 +00:00
const.DISC_COMMAND_CLASS: [const.COMMAND_CLASS_SWITCH_BINARY]
}
},
}
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
values._check_entity_ready()
self.hass.block_till_done()
assert not discovery.async_load_platform.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_config_ignore(self, discovery, import_module):
"""Test ignore config."""
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
}
2019-07-31 19:25:30 +00:00
self.device_config = {self.entity_id: {zwave.CONF_IGNORED: True}}
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
values._check_entity_ready()
self.hass.block_till_done()
assert not discovery.async_load_platform.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_config_ignore_with_registry(self, discovery, import_module):
"""Test ignore config.
The case when the device is in entity registry.
"""
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
}
2019-07-31 19:25:30 +00:00
self.device_config = {"mock_component.registry_id": {zwave.CONF_IGNORED: True}}
with patch.object(self.registry, "async_schedule_save"):
self.registry.async_get_or_create(
2019-07-31 19:25:30 +00:00
"mock_component",
zwave.DOMAIN,
"567-1000",
suggested_object_id="registry_id",
)
zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
self.hass.block_till_done()
assert not discovery.async_load_platform.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_entity_platform_ignore(self, discovery, import_module):
"""Test platform ignore device."""
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
}
platform = MagicMock()
import_module.return_value = platform
platform.get_device.return_value = None
zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
self.hass.block_till_done()
assert not discovery.async_load_platform.called
2019-07-31 19:25:30 +00:00
@patch.object(zwave, "import_module")
@patch.object(zwave, "discovery")
def test_config_polling_intensity(self, discovery, import_module):
"""Test polling intensity."""
mock_platform = MagicMock()
import_module.return_value = mock_platform
mock_device = MagicMock()
2019-07-31 19:25:30 +00:00
mock_device.name = "test_device"
mock_platform.get_device.return_value = mock_device
self.node.values = {
self.primary.value_id: self.primary,
self.secondary.value_id: self.secondary,
}
2019-07-31 19:25:30 +00:00
self.device_config = {self.entity_id: {zwave.CONF_POLLING_INTENSITY: 123}}
values = zwave.ZWaveDeviceEntityValues(
hass=self.hass,
schema=self.mock_schema,
primary_value=self.primary,
zwave_config=self.zwave_config,
device_config=self.device_config,
2019-07-31 19:25:30 +00:00
registry=self.registry,
)
values._check_entity_ready()
self.hass.block_till_done()
assert discovery.async_load_platform.called
assert self.primary.enable_poll.called
assert len(self.primary.enable_poll.mock_calls) == 1
assert self.primary.enable_poll.mock_calls[0][1][0] == 123
class TestZwave(unittest.TestCase):
"""Test zwave init."""
def test_device_config_glob_is_ordered(self):
"""Test that device_config_glob preserves order."""
2019-07-31 19:25:30 +00:00
conf = CONFIG_SCHEMA({"zwave": {CONF_DEVICE_CONFIG_GLOB: OrderedDict()}})
assert isinstance(conf["zwave"][CONF_DEVICE_CONFIG_GLOB], OrderedDict)
class TestZWaveServices(unittest.TestCase):
"""Tests for zwave services."""
@pytest.fixture(autouse=True)
def set_mock_openzwave(self, mock_openzwave):
"""Use the mock_openzwave fixture for this class."""
self.mock_openzwave = mock_openzwave
def setUp(self):
"""Initialize values for this testcase class."""
self.hass = get_test_home_assistant()
self.hass.start()
# Initialize zwave
2019-07-31 19:25:30 +00:00
setup_component(self.hass, "zwave", {"zwave": {}})
self.hass.block_till_done()
self.zwave_network = self.hass.data[DATA_NETWORK]
self.zwave_network.state = MockNetwork.STATE_READY
self.hass.bus.fire(EVENT_HOMEASSISTANT_START)
self.hass.block_till_done()
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "stop_network", {})
self.hass.block_till_done()
self.hass.stop()
def test_add_node(self):
"""Test zwave add_node service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "add_node", {})
self.hass.block_till_done()
assert self.zwave_network.controller.add_node.called
2019-07-31 19:25:30 +00:00
assert len(self.zwave_network.controller.add_node.mock_calls) == 1
assert len(self.zwave_network.controller.add_node.mock_calls[0][1]) == 0
def test_add_node_secure(self):
"""Test zwave add_node_secure service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "add_node_secure", {})
self.hass.block_till_done()
assert self.zwave_network.controller.add_node.called
assert len(self.zwave_network.controller.add_node.mock_calls) == 1
2019-07-31 19:25:30 +00:00
assert self.zwave_network.controller.add_node.mock_calls[0][1][0] is True
def test_remove_node(self):
"""Test zwave remove_node service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "remove_node", {})
self.hass.block_till_done()
assert self.zwave_network.controller.remove_node.called
assert len(self.zwave_network.controller.remove_node.mock_calls) == 1
def test_cancel_command(self):
"""Test zwave cancel_command service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "cancel_command", {})
self.hass.block_till_done()
assert self.zwave_network.controller.cancel_command.called
2019-07-31 19:25:30 +00:00
assert len(self.zwave_network.controller.cancel_command.mock_calls) == 1
def test_heal_network(self):
"""Test zwave heal_network service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "heal_network", {})
self.hass.block_till_done()
assert self.zwave_network.heal.called
assert len(self.zwave_network.heal.mock_calls) == 1
def test_soft_reset(self):
"""Test zwave soft_reset service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "soft_reset", {})
self.hass.block_till_done()
assert self.zwave_network.controller.soft_reset.called
assert len(self.zwave_network.controller.soft_reset.mock_calls) == 1
def test_test_network(self):
"""Test zwave test_network service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "test_network", {})
self.hass.block_till_done()
assert self.zwave_network.test.called
assert len(self.zwave_network.test.mock_calls) == 1
def test_stop_network(self):
"""Test zwave stop_network service."""
2019-07-31 19:25:30 +00:00
with patch.object(self.hass.bus, "fire") as mock_fire:
self.hass.services.call("zwave", "stop_network", {})
self.hass.block_till_done()
assert self.zwave_network.stop.called
assert len(self.zwave_network.stop.mock_calls) == 1
assert mock_fire.called
assert len(mock_fire.mock_calls) == 1
assert mock_fire.mock_calls[0][1][0] == const.EVENT_NETWORK_STOP
def test_rename_node(self):
"""Test zwave rename_node service."""
self.zwave_network.nodes = {11: MagicMock()}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"rename_node",
{const.ATTR_NODE_ID: 11, const.ATTR_NAME: "test_name"},
)
self.hass.block_till_done()
2019-07-31 19:25:30 +00:00
assert self.zwave_network.nodes[11].name == "test_name"
def test_rename_value(self):
"""Test zwave rename_value service."""
node = MockNode(node_id=14)
value = MockValue(index=12, value_id=123456, label="Old Label")
node.values = {123456: value}
self.zwave_network.nodes = {11: node}
assert value.label == "Old Label"
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"rename_value",
{
const.ATTR_NODE_ID: 11,
const.ATTR_VALUE_ID: 123456,
const.ATTR_NAME: "New Label",
},
)
self.hass.block_till_done()
assert value.label == "New Label"
def test_set_poll_intensity_enable(self):
"""Test zwave set_poll_intensity service, successful set."""
node = MockNode(node_id=14)
value = MockValue(index=12, value_id=123456, poll_intensity=0)
node.values = {123456: value}
self.zwave_network.nodes = {11: node}
assert value.poll_intensity == 0
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_poll_intensity",
{
const.ATTR_NODE_ID: 11,
const.ATTR_VALUE_ID: 123456,
const.ATTR_POLL_INTENSITY: 4,
},
)
self.hass.block_till_done()
enable_poll = value.enable_poll
assert value.enable_poll.called
assert len(enable_poll.mock_calls) == 2
assert enable_poll.mock_calls[0][1][0] == 4
def test_set_poll_intensity_enable_failed(self):
"""Test zwave set_poll_intensity service, failed set."""
node = MockNode(node_id=14)
value = MockValue(index=12, value_id=123456, poll_intensity=0)
value.enable_poll.return_value = False
node.values = {123456: value}
self.zwave_network.nodes = {11: node}
assert value.poll_intensity == 0
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_poll_intensity",
{
const.ATTR_NODE_ID: 11,
const.ATTR_VALUE_ID: 123456,
const.ATTR_POLL_INTENSITY: 4,
},
)
self.hass.block_till_done()
enable_poll = value.enable_poll
assert value.enable_poll.called
assert len(enable_poll.mock_calls) == 1
def test_set_poll_intensity_disable(self):
"""Test zwave set_poll_intensity service, successful disable."""
node = MockNode(node_id=14)
value = MockValue(index=12, value_id=123456, poll_intensity=4)
node.values = {123456: value}
self.zwave_network.nodes = {11: node}
assert value.poll_intensity == 4
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_poll_intensity",
{
const.ATTR_NODE_ID: 11,
const.ATTR_VALUE_ID: 123456,
const.ATTR_POLL_INTENSITY: 0,
},
)
self.hass.block_till_done()
disable_poll = value.disable_poll
assert value.disable_poll.called
assert len(disable_poll.mock_calls) == 2
def test_set_poll_intensity_disable_failed(self):
"""Test zwave set_poll_intensity service, failed disable."""
node = MockNode(node_id=14)
value = MockValue(index=12, value_id=123456, poll_intensity=4)
value.disable_poll.return_value = False
node.values = {123456: value}
self.zwave_network.nodes = {11: node}
assert value.poll_intensity == 4
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_poll_intensity",
{
const.ATTR_NODE_ID: 11,
const.ATTR_VALUE_ID: 123456,
const.ATTR_POLL_INTENSITY: 0,
},
)
self.hass.block_till_done()
disable_poll = value.disable_poll
assert value.disable_poll.called
assert len(disable_poll.mock_calls) == 1
def test_remove_failed_node(self):
"""Test zwave remove_failed_node service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "remove_failed_node", {const.ATTR_NODE_ID: 12})
self.hass.block_till_done()
remove_failed_node = self.zwave_network.controller.remove_failed_node
assert remove_failed_node.called
assert len(remove_failed_node.mock_calls) == 1
assert remove_failed_node.mock_calls[0][1][0] == 12
def test_replace_failed_node(self):
"""Test zwave replace_failed_node service."""
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave", "replace_failed_node", {const.ATTR_NODE_ID: 13}
)
self.hass.block_till_done()
replace_failed_node = self.zwave_network.controller.replace_failed_node
assert replace_failed_node.called
assert len(replace_failed_node.mock_calls) == 1
assert replace_failed_node.mock_calls[0][1][0] == 13
def test_set_config_parameter(self):
"""Test zwave set_config_parameter service."""
value_byte = MockValue(
index=12,
command_class=const.COMMAND_CLASS_CONFIGURATION,
type=const.TYPE_BYTE,
)
value_list = MockValue(
index=13,
command_class=const.COMMAND_CLASS_CONFIGURATION,
type=const.TYPE_LIST,
2019-07-31 19:25:30 +00:00
data_items=["item1", "item2", "item3"],
)
value_button = MockValue(
index=14,
command_class=const.COMMAND_CLASS_CONFIGURATION,
type=const.TYPE_BUTTON,
)
value_list_int = MockValue(
index=15,
command_class=const.COMMAND_CLASS_CONFIGURATION,
type=const.TYPE_LIST,
2019-07-31 19:25:30 +00:00
data_items=["1", "2", "3"],
)
value_bool = MockValue(
index=16,
command_class=const.COMMAND_CLASS_CONFIGURATION,
type=const.TYPE_BOOL,
)
node = MockNode(node_id=14)
node.get_values.return_value = {
12: value_byte,
13: value_list,
14: value_button,
15: value_list_int,
2019-07-31 19:25:30 +00:00
16: value_bool,
}
self.zwave_network.nodes = {14: node}
# Byte
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 12,
const.ATTR_CONFIG_VALUE: 7,
},
)
self.hass.block_till_done()
assert value_byte.data == 7
# List
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 13,
const.ATTR_CONFIG_VALUE: "item3",
},
)
self.hass.block_till_done()
2019-07-31 19:25:30 +00:00
assert value_list.data == "item3"
# Button
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 14,
const.ATTR_CONFIG_VALUE: True,
},
)
self.hass.block_till_done()
assert self.zwave_network.manager.pressButton.called
assert self.zwave_network.manager.releaseButton.called
# List of Ints
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 15,
const.ATTR_CONFIG_VALUE: 3,
},
)
self.hass.block_till_done()
2019-07-31 19:25:30 +00:00
assert value_list_int.data == "3"
# Boolean Truthy
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 16,
const.ATTR_CONFIG_VALUE: "True",
},
)
self.hass.block_till_done()
assert value_bool.data == 1
# Boolean Falsy
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 16,
const.ATTR_CONFIG_VALUE: "False",
},
)
self.hass.block_till_done()
assert value_bool.data == 0
# Different Parameter Size
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_config_parameter",
{
const.ATTR_NODE_ID: 14,
const.ATTR_CONFIG_PARAMETER: 19,
const.ATTR_CONFIG_VALUE: 0x01020304,
const.ATTR_CONFIG_SIZE: 4,
},
)
self.hass.block_till_done()
assert node.set_config_param.called
assert len(node.set_config_param.mock_calls) == 1
assert node.set_config_param.mock_calls[0][1][0] == 19
assert node.set_config_param.mock_calls[0][1][1] == 0x01020304
assert node.set_config_param.mock_calls[0][1][2] == 4
node.set_config_param.reset_mock()
def test_print_config_parameter(self):
"""Test zwave print_config_parameter service."""
value1 = MockValue(
2019-07-31 19:25:30 +00:00
index=12, command_class=const.COMMAND_CLASS_CONFIGURATION, data=1234
)
value2 = MockValue(
2019-07-31 19:25:30 +00:00
index=13, command_class=const.COMMAND_CLASS_CONFIGURATION, data=2345
)
node = MockNode(node_id=14)
node.values = {12: value1, 13: value2}
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
with patch.object(zwave, "_LOGGER") as mock_logger:
self.hass.services.call(
"zwave",
"print_config_parameter",
{const.ATTR_NODE_ID: 14, const.ATTR_CONFIG_PARAMETER: 13},
)
self.hass.block_till_done()
assert mock_logger.info.called
assert len(mock_logger.info.mock_calls) == 1
assert mock_logger.info.mock_calls[0][1][1] == 13
assert mock_logger.info.mock_calls[0][1][2] == 14
assert mock_logger.info.mock_calls[0][1][3] == 2345
def test_print_node(self):
"""Test zwave print_node_parameter service."""
node = MockNode(node_id=14)
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
with self.assertLogs(level="DEBUG") as mock_logger:
self.hass.services.call("zwave", "print_node", {const.ATTR_NODE_ID: 14})
self.hass.block_till_done()
assert "FOUND NODE " in mock_logger.output[1]
def test_set_wakeup(self):
"""Test zwave set_wakeup service."""
2019-07-31 19:25:30 +00:00
value = MockValue(index=12, command_class=const.COMMAND_CLASS_WAKE_UP)
node = MockNode(node_id=14)
node.values = {12: value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave", "set_wakeup", {const.ATTR_NODE_ID: 14, const.ATTR_CONFIG_VALUE: 15}
)
self.hass.block_till_done()
assert value.data == 15
node.can_wake_up_value = False
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave", "set_wakeup", {const.ATTR_NODE_ID: 14, const.ATTR_CONFIG_VALUE: 20}
)
self.hass.block_till_done()
assert value.data == 15
def test_reset_node_meters(self):
"""Test zwave reset_node_meters service."""
value = MockValue(
2019-07-31 19:25:30 +00:00
instance=1, index=8, data=99.5, command_class=const.COMMAND_CLASS_METER
)
reset_value = MockValue(
2019-07-31 19:25:30 +00:00
instance=1, index=33, command_class=const.COMMAND_CLASS_METER
)
node = MockNode(node_id=14)
node.values = {8: value, 33: reset_value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"reset_node_meters",
{const.ATTR_NODE_ID: 14, const.ATTR_INSTANCE: 2},
)
self.hass.block_till_done()
assert not self.zwave_network.manager.pressButton.called
assert not self.zwave_network.manager.releaseButton.called
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "reset_node_meters", {const.ATTR_NODE_ID: 14})
self.hass.block_till_done()
assert self.zwave_network.manager.pressButton.called
value_id, = self.zwave_network.manager.pressButton.mock_calls.pop(0)[1]
assert value_id == reset_value.value_id
assert self.zwave_network.manager.releaseButton.called
2019-07-31 19:25:30 +00:00
value_id, = self.zwave_network.manager.releaseButton.mock_calls.pop(0)[1]
assert value_id == reset_value.value_id
def test_add_association(self):
"""Test zwave change_association service."""
ZWaveGroup = self.mock_openzwave.group.ZWaveGroup
group = MagicMock()
ZWaveGroup.return_value = group
2019-07-31 19:25:30 +00:00
value = MockValue(index=12, command_class=const.COMMAND_CLASS_WAKE_UP)
node = MockNode(node_id=14)
node.values = {12: value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"change_association",
{
const.ATTR_ASSOCIATION: "add",
const.ATTR_NODE_ID: 14,
const.ATTR_TARGET_NODE_ID: 24,
const.ATTR_GROUP: 3,
const.ATTR_INSTANCE: 5,
},
)
self.hass.block_till_done()
assert ZWaveGroup.called
assert len(ZWaveGroup.mock_calls) == 2
assert ZWaveGroup.mock_calls[0][1][0] == 3
assert ZWaveGroup.mock_calls[0][1][2] == 14
assert group.add_association.called
assert len(group.add_association.mock_calls) == 1
assert group.add_association.mock_calls[0][1][0] == 24
assert group.add_association.mock_calls[0][1][1] == 5
def test_remove_association(self):
"""Test zwave change_association service."""
ZWaveGroup = self.mock_openzwave.group.ZWaveGroup
group = MagicMock()
ZWaveGroup.return_value = group
2019-07-31 19:25:30 +00:00
value = MockValue(index=12, command_class=const.COMMAND_CLASS_WAKE_UP)
node = MockNode(node_id=14)
node.values = {12: value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"change_association",
{
const.ATTR_ASSOCIATION: "remove",
const.ATTR_NODE_ID: 14,
const.ATTR_TARGET_NODE_ID: 24,
const.ATTR_GROUP: 3,
const.ATTR_INSTANCE: 5,
},
)
self.hass.block_till_done()
assert ZWaveGroup.called
assert len(ZWaveGroup.mock_calls) == 2
assert ZWaveGroup.mock_calls[0][1][0] == 3
assert ZWaveGroup.mock_calls[0][1][2] == 14
assert group.remove_association.called
assert len(group.remove_association.mock_calls) == 1
assert group.remove_association.mock_calls[0][1][0] == 24
assert group.remove_association.mock_calls[0][1][1] == 5
def test_refresh_entity(self):
"""Test zwave refresh_entity service."""
node = MockNode()
2019-07-31 19:25:30 +00:00
value = MockValue(
data=False, node=node, command_class=const.COMMAND_CLASS_SENSOR_BINARY
)
power_value = MockValue(
data=50, node=node, command_class=const.COMMAND_CLASS_METER
)
values = MockEntityValues(primary=value, power=power_value)
device = get_device(node=node, values=values, node_config={})
device.hass = self.hass
2019-07-31 19:25:30 +00:00
device.entity_id = "binary_sensor.mock_entity_id"
self.hass.add_job(device.async_added_to_hass())
self.hass.block_till_done()
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave", "refresh_entity", {ATTR_ENTITY_ID: "binary_sensor.mock_entity_id"}
)
self.hass.block_till_done()
assert node.refresh_value.called
assert len(node.refresh_value.mock_calls) == 2
2019-07-31 19:25:30 +00:00
assert sorted(
[
node.refresh_value.mock_calls[0][1][0],
node.refresh_value.mock_calls[1][1][0],
]
) == sorted([value.value_id, power_value.value_id])
def test_refresh_node(self):
"""Test zwave refresh_node service."""
node = MockNode(node_id=14)
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "refresh_node", {const.ATTR_NODE_ID: 14})
self.hass.block_till_done()
assert node.refresh_info.called
assert len(node.refresh_info.mock_calls) == 1
def test_set_node_value(self):
"""Test zwave set_node_value service."""
2019-07-31 19:25:30 +00:00
value = MockValue(index=12, command_class=const.COMMAND_CLASS_INDICATOR, data=4)
node = MockNode(node_id=14, command_classes=[const.COMMAND_CLASS_INDICATOR])
node.values = {12: value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"set_node_value",
{
const.ATTR_NODE_ID: 14,
const.ATTR_VALUE_ID: 12,
const.ATTR_CONFIG_VALUE: 2,
},
)
self.hass.block_till_done()
assert self.zwave_network.nodes[14].values[12].data == 2
def test_refresh_node_value(self):
"""Test zwave refresh_node_value service."""
2019-07-31 19:25:30 +00:00
node = MockNode(
node_id=14,
command_classes=[const.COMMAND_CLASS_INDICATOR],
network=self.zwave_network,
)
value = MockValue(
2019-07-31 19:25:30 +00:00
node=node, index=12, command_class=const.COMMAND_CLASS_INDICATOR, data=2
)
value.refresh = MagicMock()
node.values = {12: value}
node.get_values.return_value = node.values
self.zwave_network.nodes = {14: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call(
"zwave",
"refresh_node_value",
{const.ATTR_NODE_ID: 14, const.ATTR_VALUE_ID: 12},
)
self.hass.block_till_done()
assert value.refresh.called
def test_heal_node(self):
"""Test zwave heal_node service."""
node = MockNode(node_id=19)
self.zwave_network.nodes = {19: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "heal_node", {const.ATTR_NODE_ID: 19})
self.hass.block_till_done()
assert node.heal.called
assert len(node.heal.mock_calls) == 1
def test_test_node(self):
"""Test the zwave test_node service."""
node = MockNode(node_id=19)
self.zwave_network.nodes = {19: node}
2019-07-31 19:25:30 +00:00
self.hass.services.call("zwave", "test_node", {const.ATTR_NODE_ID: 19})
self.hass.block_till_done()
assert node.test.called
assert len(node.test.mock_calls) == 1