From 5aaa1f840493ae8f28cd85d64bad12f69aca8b2f Mon Sep 17 00:00:00 2001
From: John Arild Berentsen <turbokongen@hotmail.com>
Date: Thu, 16 Mar 2017 11:05:51 +0100
Subject: [PATCH] Add test for Z-wave switch (#6619)

* Add test for Z-wave switch

* Changes for new tests
---
 tests/components/switch/test_zwave.py | 53 +++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 tests/components/switch/test_zwave.py

diff --git a/tests/components/switch/test_zwave.py b/tests/components/switch/test_zwave.py
new file mode 100644
index 00000000000..e537eef1375
--- /dev/null
+++ b/tests/components/switch/test_zwave.py
@@ -0,0 +1,53 @@
+"""Test Z-Wave switches."""
+from homeassistant.components.switch import zwave
+
+from tests.mock.zwave import (
+   MockNode, MockValue, MockEntityValues, value_changed)
+
+
+def test_get_device_detects_switch(mock_openzwave):
+    """Test get_device returns a Z-Wave switch."""
+    node = MockNode()
+    value = MockValue(data=0, node=node)
+    values = MockEntityValues(primary=value)
+
+    device = zwave.get_device(node=node, values=values, node_config={})
+    assert isinstance(device, zwave.ZwaveSwitch)
+
+
+def test_switch_turn_on_and_off(mock_openzwave):
+    """Test turning on a Z-Wave switch."""
+    node = MockNode()
+    value = MockValue(data=0, node=node)
+    values = MockEntityValues(primary=value)
+    device = zwave.get_device(node=node, values=values, node_config={})
+
+    device.turn_on()
+
+    assert node.set_switch.called
+    value_id, state = node.set_switch.mock_calls[0][1]
+    assert value_id == value.value_id
+    assert state is True
+    node.reset_mock()
+
+    device.turn_off()
+
+    assert node.set_switch.called
+    value_id, state = node.set_switch.mock_calls[0][1]
+    assert value_id == value.value_id
+    assert state is False
+
+
+def test_switch_value_changed(mock_openzwave):
+    """Test value changed for Z-Wave switch."""
+    node = MockNode()
+    value = MockValue(data=False, node=node)
+    values = MockEntityValues(primary=value)
+    device = zwave.get_device(node=node, values=values, node_config={})
+
+    assert not device.is_on
+
+    value.data = True
+    value_changed(value)
+
+    assert device.is_on