async def test_state_ready(opp, mock_openzwave): """Test state property.""" zwave_network = MagicMock() node = mock_zwave.MockNode( query_stage="Dynamic", is_awake=True, is_ready=False, is_failed=False, is_info_received=True, ) entity = node_entity.ZWaveNodeEntity(node, zwave_network) node.query_stage = "Complete" node.is_ready = True entity.node_changed() await opp.async_block_till_done() assert entity.state == "ready" node.is_failed = True entity.node_changed() assert entity.state == "dead" node.is_failed = False node.is_awake = False entity.node_changed() assert entity.state == "sleeping"
def test_unique_id_missing_data(self): """Test unique_id.""" self.node.manufacturer_name = None self.node.name = None entity = node_entity.ZWaveNodeEntity(self.node, self.zwave_network) assert entity.unique_id is None
async def test_application_version(opp, mock_openzwave): """Test application version.""" mock_receivers = {} signal_mocks = [ mock_zwave.MockNetwork.SIGNAL_VALUE_CHANGED, mock_zwave.MockNetwork.SIGNAL_VALUE_ADDED, ] def mock_connect(receiver, signal, *args, **kwargs): if signal in signal_mocks: mock_receivers[signal] = receiver node = mock_zwave.MockNode(node_id=11) with patch("pydispatch.dispatcher.connect", new=mock_connect): entity = node_entity.ZWaveNodeEntity(node, mock_openzwave) for signal_mock in signal_mocks: assert signal_mock in mock_receivers.keys() events = [] def listener(event): events.append(event) # Make sure application version isn't set before assert (node_entity.ATTR_APPLICATION_VERSION not in entity.device_state_attributes.keys()) # Add entity to opp entity.opp = opp entity.entity_id = "zwave.mock_node" # Fire off an added value value = mock_zwave.MockValue( command_class=const.COMMAND_CLASS_VERSION, label="Application Version", data="5.10", ) opp.async_add_job( mock_receivers[mock_zwave.MockNetwork.SIGNAL_VALUE_ADDED], node, value) await opp.async_block_till_done() assert (entity.device_state_attributes[ node_entity.ATTR_APPLICATION_VERSION] == "5.10") # Fire off a changed value = mock_zwave.MockValue( command_class=const.COMMAND_CLASS_VERSION, label="Application Version", data="4.14", ) opp.async_add_job( mock_receivers[mock_zwave.MockNetwork.SIGNAL_VALUE_CHANGED], node, value) await opp.async_block_till_done() assert (entity.device_state_attributes[ node_entity.ATTR_APPLICATION_VERSION] == "4.14")
async def test_network_node_changed_from_notification(opp, mock_openzwave): """Test for network_node_changed.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) with patch.object(entity, "maybe_schedule_update") as mock: mock_zwave.notification(node_id=node.node_id) mock.assert_called_once_with()
async def test_network_node_changed_from_another_node(opp, mock_openzwave): """Test for network_node_changed.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) with patch.object(entity, "maybe_schedule_update") as mock: another_node = mock_zwave.MockNode(node_id=1024) mock_zwave.node_changed(another_node) assert not mock.called
async def test_unique_id_missing_data(opp, mock_openzwave): """Test unique_id.""" zwave_network = MagicMock() node = mock_zwave.MockNode() node.manufacturer_name = None node.name = None node.is_ready = False entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert entity.unique_id is None
async def test_central_scene_activated(opp, mock_openzwave): """Test central scene activated event.""" mock_receivers = [] def mock_connect(receiver, signal, *args, **kwargs): if signal == mock_zwave.MockNetwork.SIGNAL_VALUE_CHANGED: mock_receivers.append(receiver) node = mock_zwave.MockNode(node_id=11) with patch("pydispatch.dispatcher.connect", new=mock_connect): entity = node_entity.ZWaveNodeEntity(node, mock_openzwave) assert len(mock_receivers) == 1 events = [] def listener(event): events.append(event) opp.bus.async_listen(const.EVENT_SCENE_ACTIVATED, listener) # Test event before entity added to opp scene_id = 1 scene_data = 3 value = mock_zwave.MockValue( command_class=const.COMMAND_CLASS_CENTRAL_SCENE, index=scene_id, data=scene_data) opp.async_add_job(mock_receivers[0], node, value) await opp.async_block_till_done() assert len(events) == 0 # Add entity to opp entity.opp = opp entity.entity_id = "zwave.mock_node" scene_id = 1 scene_data = 3 value = mock_zwave.MockValue( command_class=const.COMMAND_CLASS_CENTRAL_SCENE, index=scene_id, data=scene_data) opp.async_add_job(mock_receivers[0], node, value) await opp.async_block_till_done() assert len(events) == 1 assert events[0].data[ATTR_ENTITY_ID] == "zwave.mock_node" assert events[0].data[const.ATTR_NODE_ID] == 11 assert events[0].data[const.ATTR_SCENE_ID] == scene_id assert events[0].data[const.ATTR_SCENE_DATA] == scene_data
def setUp(self): """Initialize values for this testcase class.""" self.zwave_network = MagicMock() self.node = mock_zwave.MockNode( query_stage="Dynamic", is_awake=True, is_ready=False, is_failed=False, is_info_received=True, max_baud_rate=40000, is_zwave_plus=False, capabilities=[], neighbors=[], location=None, ) self.entity = node_entity.ZWaveNodeEntity(self.node, self.zwave_network)
async def test_scene_activated(opp, mock_openzwave): """Test scene activated event.""" mock_receivers = [] def mock_connect(receiver, signal, *args, **kwargs): if signal == mock_zwave.MockNetwork.SIGNAL_SCENE_EVENT: mock_receivers.append(receiver) node = mock_zwave.MockNode(node_id=11) with patch("pydispatch.dispatcher.connect", new=mock_connect): entity = node_entity.ZWaveNodeEntity(node, mock_openzwave) assert len(mock_receivers) == 1 events = [] def listener(event): events.append(event) opp.bus.async_listen(const.EVENT_SCENE_ACTIVATED, listener) # Test event before entity added to opp scene_id = 123 opp.async_add_job(mock_receivers[0], node, scene_id) await opp.async_block_till_done() assert len(events) == 0 # Add entity to opp entity.opp = opp entity.entity_id = "zwave.mock_node" scene_id = 123 opp.async_add_job(mock_receivers[0], node, scene_id) await opp.async_block_till_done() assert len(events) == 1 assert events[0].data[ATTR_ENTITY_ID] == "zwave.mock_node" assert events[0].data[const.ATTR_NODE_ID] == 11 assert events[0].data[const.ATTR_SCENE_ID] == scene_id
async def test_unique_id(opp, mock_openzwave): """Test unique_id.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert entity.unique_id == "node-567"
async def test_not_polled(opp, mock_openzwave): """Test should_poll property.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert not entity.should_poll
async def test_state_before_update(opp, mock_openzwave): """Test state before update was called.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert entity.state is None
async def test_name(opp, mock_openzwave): """Test name property.""" zwave_network = MagicMock() node = mock_zwave.MockNode() entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert entity.name == "Mock Node"
async def test_node_changed(opp, mock_openzwave): """Test node_changed function.""" zwave_network = MagicMock() node = mock_zwave.MockNode( query_stage="Dynamic", is_awake=True, is_ready=False, is_failed=False, is_info_received=True, max_baud_rate=40000, is_zwave_plus=False, capabilities=[], neighbors=[], location=None, ) entity = node_entity.ZWaveNodeEntity(node, zwave_network) assert { "node_id": node.node_id, "node_name": "Mock Node", "manufacturer_name": "Test Manufacturer", "product_name": "Test Product", } == entity.extra_state_attributes node.get_values.return_value = {1: mock_zwave.MockValue(data=1800)} zwave_network.manager.getNodeStatistics.return_value = { "receivedCnt": 4, "ccData": [ { "receivedCnt": 0, "commandClassId": 134, "sentCnt": 0 }, { "receivedCnt": 1, "commandClassId": 133, "sentCnt": 1 }, { "receivedCnt": 1, "commandClassId": 115, "sentCnt": 1 }, { "receivedCnt": 0, "commandClassId": 114, "sentCnt": 0 }, { "receivedCnt": 0, "commandClassId": 112, "sentCnt": 0 }, { "receivedCnt": 1, "commandClassId": 32, "sentCnt": 1 }, { "receivedCnt": 0, "commandClassId": 0, "sentCnt": 0 }, ], "receivedUnsolicited": 0, "sentTS": "2017-03-27 15:38:15:620 ", "averageRequestRTT": 2462, "lastResponseRTT": 3679, "retries": 0, "sentFailed": 1, "sentCnt": 7, "quality": 0, "lastRequestRTT": 1591, "lastReceivedMessage": [ 0, 4, 0, 15, 3, 32, 3, 0, 221, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, ], "receivedDups": 1, "averageResponseRTT": 2443, "receivedTS": "2017-03-27 15:38:19:298 ", } entity.node_changed() assert { "node_id": node.node_id, "node_name": "Mock Node", "manufacturer_name": "Test Manufacturer", "product_name": "Test Product", "query_stage": "Dynamic", "is_awake": True, "is_ready": False, "is_failed": False, "is_info_received": True, "max_baud_rate": 40000, "is_zwave_plus": False, "battery_level": 42, "wake_up_interval": 1800, "averageRequestRTT": 2462, "averageResponseRTT": 2443, "lastRequestRTT": 1591, "lastResponseRTT": 3679, "receivedCnt": 4, "receivedDups": 1, "receivedTS": "2017-03-27 15:38:19:298 ", "receivedUnsolicited": 0, "retries": 0, "sentCnt": 7, "sentFailed": 1, "sentTS": "2017-03-27 15:38:15:620 ", } == entity.extra_state_attributes node.can_wake_up_value = False entity.node_changed() assert "wake_up_interval" not in entity.extra_state_attributes