Update node statistics for zwave_js device diagnostics dump (#72509)

This commit is contained in:
Raman Gupta 2022-05-26 03:03:43 -04:00 committed by Paulus Schoutsen
parent 180b5cd2bb
commit f038d0892a
2 changed files with 13 additions and 2 deletions

View File

@ -155,6 +155,8 @@ async def async_get_device_diagnostics(
node = driver.controller.nodes[node_id] node = driver.controller.nodes[node_id]
entities = get_device_entities(hass, node, device) entities = get_device_entities(hass, node, device)
assert client.version assert client.version
node_state = redact_node_state(async_redact_data(node.data, KEYS_TO_REDACT))
node_state["statistics"] = node.statistics.data
return { return {
"versionInfo": { "versionInfo": {
"driverVersion": client.version.driver_version, "driverVersion": client.version.driver_version,
@ -163,5 +165,5 @@ async def async_get_device_diagnostics(
"maxSchemaVersion": client.version.max_schema_version, "maxSchemaVersion": client.version.max_schema_version,
}, },
"entities": entities, "entities": entities,
"state": redact_node_state(async_redact_data(node.data, KEYS_TO_REDACT)), "state": node_state,
} }

View File

@ -92,7 +92,16 @@ async def test_device_diagnostics(
assert len(diagnostics_data["entities"]) == len( assert len(diagnostics_data["entities"]) == len(
list(async_discover_node_values(multisensor_6, device, {device.id: set()})) list(async_discover_node_values(multisensor_6, device, {device.id: set()}))
) )
assert diagnostics_data["state"] == multisensor_6.data assert diagnostics_data["state"] == {
**multisensor_6.data,
"statistics": {
"commandsDroppedRX": 0,
"commandsDroppedTX": 0,
"commandsRX": 0,
"commandsTX": 0,
"timeoutResponse": 0,
},
}
async def test_device_diagnostics_error(hass, integration): async def test_device_diagnostics_error(hass, integration):