Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-27 06:28:31 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

460
tests/components/zwave/test_node_entity.py
View File

@@ -12,7 +12,7 @@ async def test_maybe_schedule_update(hass, mock_openzwave):
base_entity = node_entity.ZWaveBaseEntity()
base_entity.hass = hass
with patch.object(hass.loop, 'call_later') as mock_call_later:
with patch.object(hass.loop, "call_later") as mock_call_later:
base_entity._schedule_update()
assert mock_call_later.called
@@ -21,7 +21,7 @@ async def test_maybe_schedule_update(hass, mock_openzwave):
do_update = mock_call_later.mock_calls[0][1][1]
with patch.object(hass, 'async_add_job') as mock_add_job:
with patch.object(hass, "async_add_job") as mock_add_job:
do_update()
assert mock_add_job.called
@@ -39,7 +39,7 @@ async def test_node_event_activated(hass, mock_openzwave):
node = mock_zwave.MockNode(node_id=11)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
with patch("pydispatch.dispatcher.connect", new=mock_connect):
entity = node_entity.ZWaveNodeEntity(node, mock_openzwave)
assert len(mock_receivers) == 1
@@ -59,7 +59,7 @@ async def test_node_event_activated(hass, mock_openzwave):
# Add entity to hass
entity.hass = hass
entity.entity_id = 'zwave.mock_node'
entity.entity_id = "zwave.mock_node"
value = 234
hass.async_add_job(mock_receivers[0], node, value)
@@ -81,7 +81,7 @@ async def test_scene_activated(hass, mock_openzwave):
node = mock_zwave.MockNode(node_id=11)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
with patch("pydispatch.dispatcher.connect", new=mock_connect):
entity = node_entity.ZWaveNodeEntity(node, mock_openzwave)
assert len(mock_receivers) == 1
@@ -101,7 +101,7 @@ async def test_scene_activated(hass, mock_openzwave):
# Add entity to hass
entity.hass = hass
entity.entity_id = 'zwave.mock_node'
entity.entity_id = "zwave.mock_node"
scene_id = 123
hass.async_add_job(mock_receivers[0], node, scene_id)
@@ -123,7 +123,7 @@ async def test_central_scene_activated(hass, mock_openzwave):
node = mock_zwave.MockNode(node_id=11)
with patch('pydispatch.dispatcher.connect', new=mock_connect):
with patch("pydispatch.dispatcher.connect", new=mock_connect):
entity = node_entity.ZWaveNodeEntity(node, mock_openzwave)
assert len(mock_receivers) == 1
@@ -139,23 +139,21 @@ async def test_central_scene_activated(hass, mock_openzwave):
scene_id = 1
scene_data = 3
value = mock_zwave.MockValue(
command_class=const.COMMAND_CLASS_CENTRAL_SCENE,
index=scene_id,
data=scene_data)
command_class=const.COMMAND_CLASS_CENTRAL_SCENE, index=scene_id, data=scene_data
)
hass.async_add_job(mock_receivers[0], node, value)
await hass.async_block_till_done()
assert len(events) == 0
# Add entity to hass
entity.hass = hass
entity.entity_id = 'zwave.mock_node'
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)
command_class=const.COMMAND_CLASS_CENTRAL_SCENE, index=scene_id, data=scene_data
)
hass.async_add_job(mock_receivers[0], node, value)
await hass.async_block_till_done()
@@ -166,7 +164,7 @@ async def test_central_scene_activated(hass, mock_openzwave):
assert events[0].data[const.ATTR_SCENE_DATA] == scene_data
@pytest.mark.usefixtures('mock_openzwave')
@pytest.mark.usefixtures("mock_openzwave")
class TestZWaveNodeEntity(unittest.TestCase):
"""Class to test ZWaveNodeEntity."""
@@ -174,41 +172,48 @@ class TestZWaveNodeEntity(unittest.TestCase):
"""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)
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)
def test_network_node_changed_from_value(self):
"""Test for network_node_changed."""
value = mock_zwave.MockValue(node=self.node)
with patch.object(self.entity, 'maybe_schedule_update') as mock:
with patch.object(self.entity, "maybe_schedule_update") as mock:
mock_zwave.value_changed(value)
mock.assert_called_once_with()
def test_network_node_changed_from_node(self):
"""Test for network_node_changed."""
with patch.object(self.entity, 'maybe_schedule_update') as mock:
with patch.object(self.entity, "maybe_schedule_update") as mock:
mock_zwave.node_changed(self.node)
mock.assert_called_once_with()
def test_network_node_changed_from_another_node(self):
"""Test for network_node_changed."""
with patch.object(self.entity, 'maybe_schedule_update') as mock:
with patch.object(self.entity, "maybe_schedule_update") as mock:
node = mock_zwave.MockNode(node_id=1024)
mock_zwave.node_changed(node)
assert not mock.called
def test_network_node_changed_from_notification(self):
"""Test for network_node_changed."""
with patch.object(self.entity, 'maybe_schedule_update') as mock:
with patch.object(self.entity, "maybe_schedule_update") as mock:
mock_zwave.notification(node_id=self.node.node_id)
mock.assert_called_once_with()
def test_network_node_changed_from_another_notification(self):
"""Test for network_node_changed."""
with patch.object(self.entity, 'maybe_schedule_update') as mock:
with patch.object(self.entity, "maybe_schedule_update") as mock:
mock_zwave.notification(node_id=1024)
assert not mock.called
@@ -216,99 +221,330 @@ class TestZWaveNodeEntity(unittest.TestCase):
"""Test node_changed function."""
self.maxDiff = None
assert {
'node_id': self.node.node_id,
'node_name': 'Mock Node',
'manufacturer_name': 'Test Manufacturer',
'product_name': 'Test Product'
"node_id": self.node.node_id,
"node_name": "Mock Node",
"manufacturer_name": "Test Manufacturer",
"product_name": "Test Product",
} == self.entity.device_state_attributes
self.node.get_values.return_value = {
1: mock_zwave.MockValue(data=1800)
}
self.node.get_values.return_value = {1: mock_zwave.MockValue(data=1800)}
self.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}],
"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,
"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 "}
"receivedTS": "2017-03-27 15:38:19:298 ",
}
self.entity.node_changed()
assert {
'node_id': self.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 '
"node_id": self.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 ",
} == self.entity.device_state_attributes
self.node.can_wake_up_value = False
self.entity.node_changed()
assert 'wake_up_interval' not in self.entity.device_state_attributes
assert "wake_up_interval" not in self.entity.device_state_attributes
def test_name(self):
"""Test name property."""
assert 'Mock Node' == self.entity.name
assert "Mock Node" == self.entity.name
def test_state_before_update(self):
"""Test state before update was called."""
@@ -318,33 +554,33 @@ class TestZWaveNodeEntity(unittest.TestCase):
"""Test state property."""
self.node.is_ready = False
self.entity.node_changed()
assert 'initializing' == self.entity.state
assert "initializing" == self.entity.state
self.node.is_failed = True
self.node.query_stage = 'Complete'
self.node.query_stage = "Complete"
self.entity.node_changed()
assert 'dead' == self.entity.state
assert "dead" == self.entity.state
self.node.is_failed = False
self.node.is_awake = False
self.entity.node_changed()
assert 'sleeping' == self.entity.state
assert "sleeping" == self.entity.state
def test_state_ready(self):
"""Test state property."""
self.node.query_stage = 'Complete'
self.node.query_stage = "Complete"
self.node.is_ready = True
self.entity.node_changed()
assert 'ready' == self.entity.state
assert "ready" == self.entity.state
self.node.is_failed = True
self.entity.node_changed()
assert 'dead' == self.entity.state
assert "dead" == self.entity.state
self.node.is_failed = False
self.node.is_awake = False
self.entity.node_changed()
assert 'sleeping' == self.entity.state
assert "sleeping" == self.entity.state
def test_not_polled(self):
"""Test should_poll property."""
@@ -352,7 +588,7 @@ class TestZWaveNodeEntity(unittest.TestCase):
def test_unique_id(self):
"""Test unique_id."""
assert 'node-567' == self.entity.unique_id
assert "node-567" == self.entity.unique_id
def test_unique_id_missing_data(self):
"""Test unique_id."""