Fix zwave_js device remove test (#45864)

This commit is contained in:
Martin Hjelmare
2021-02-02 15:28:21 +01:00
committed by GitHub
parent e9b2d33ad8
commit 6e205965ee

View File

@@ -313,7 +313,7 @@ async def test_removed_device(hass, client, multiple_devices, integration):
entity_entries = entity_registry.async_entries_for_config_entry(
ent_reg, integration.entry_id
)
assert len(entity_entries) == 18
assert len(entity_entries) == 24
# Remove a node and reload the entry
old_node = nodes.pop(13)
@@ -329,5 +329,5 @@ async def test_removed_device(hass, client, multiple_devices, integration):
entity_entries = entity_registry.async_entries_for_config_entry(
ent_reg, integration.entry_id
)
assert len(entity_entries) == 9
assert len(entity_entries) == 15
assert dev_reg.async_get_device({get_device_id(client, old_node)}) is None