From c118be6639cebf59219ad78cd99f16b813479fbf Mon Sep 17 00:00:00 2001
From: Adam Mills <adam@armills.info>
Date: Fri, 12 May 2017 20:18:20 -0700
Subject: [PATCH] Tests for zwave discovery logic (#7566)

* Tests for zwave discovery logic

* Simplify patching

* Test ignored node
---
 tests/components/zwave/test_init.py | 116 ++++++++++++++++++++++++++++
 1 file changed, 116 insertions(+)

diff --git a/tests/components/zwave/test_init.py b/tests/components/zwave/test_init.py
index 91902f9c4a8..a3327d6d558 100644
--- a/tests/components/zwave/test_init.py
+++ b/tests/components/zwave/test_init.py
@@ -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."""