Tests for zwave discovery logic (#7566)

* Tests for zwave discovery logic

* Simplify patching

* Test ignored node
This commit is contained in:
Adam Mills 2017-05-12 20:18:20 -07:00 committed by Paulus Schoutsen
parent 11a3dc268f
commit c118be6639

View file

@ -138,6 +138,122 @@ def test_device_entity(hass, mock_openzwave):
assert device.device_state_attributes[zwave.ATTR_POWER] == 50.123
@asyncio.coroutine
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)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
yield from async_setup_component(hass, 'zwave', {'zwave': {}})
assert len(mock_receivers) == 1
node = MockNode(node_id=14)
hass.async_add_job(mock_receivers[0], node)
yield from hass.async_block_till_done()
assert hass.states.get('zwave.mock_node_14').state is 'unknown'
@asyncio.coroutine
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)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
yield from async_setup_component(hass, 'zwave', {'zwave': {
'device_config': {
'zwave.mock_node_14': {
'ignored': True,
}}}})
assert len(mock_receivers) == 1
node = MockNode(node_id=14)
hass.async_add_job(mock_receivers[0], node)
yield from hass.async_block_till_done()
assert hass.states.get('zwave.mock_node_14') is None
@asyncio.coroutine
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)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
yield from async_setup_component(hass, 'zwave', {'zwave': {}})
assert len(mock_receivers) == 1
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_SENSOR_BINARY)
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)
yield from hass.async_block_till_done()
assert hass.states.get(
'binary_sensor.mock_node_mock_value_11_12_13').state is 'off'
@asyncio.coroutine
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)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
yield from async_setup_component(hass, 'zwave', {'zwave': {}})
assert len(mock_receivers) == 1
node = MockNode(node_id=11, generic=const.GENERIC_TYPE_THERMOSTAT)
setpoint = MockValue(
data=22.0, node=node, index=12, instance=13,
command_class=const.COMMAND_CLASS_THERMOSTAT_SETPOINT,
genre=const.GENRE_USER, units='C')
hass.async_add_job(mock_receivers[0], node, setpoint)
yield from hass.async_block_till_done()
assert hass.states.get('climate.mock_node_mock_value_11_12_13').attributes[
'temperature'] == 22.0
assert hass.states.get('climate.mock_node_mock_value_11_12_13').attributes[
'current_temperature'] is None
def mock_update(self):
self.hass.async_add_job(self.async_update_ha_state)
with patch.object(zwave.node_entity.ZWaveBaseEntity,
'maybe_schedule_update', new=mock_update):
temperature = MockValue(
data=23.5, node=node, index=12, instance=13,
command_class=const.COMMAND_CLASS_SENSOR_MULTILEVEL,
label='Temperature', genre=const.GENRE_USER, units='C')
hass.async_add_job(mock_receivers[0], node, temperature)
yield from hass.async_block_till_done()
assert hass.states.get('climate.mock_node_mock_value_11_12_13').attributes[
'temperature'] == 22.0
assert hass.states.get('climate.mock_node_mock_value_11_12_13').attributes[
'current_temperature'] == 23.5
class TestZWaveDeviceEntityValues(unittest.TestCase):
"""Tests for the ZWaveDeviceEntityValues helper."""