From 6e205965eea87e5af6698b41856d84812a3cd5c7 Mon Sep 17 00:00:00 2001 From: Martin Hjelmare Date: Tue, 2 Feb 2021 15:28:21 +0100 Subject: [PATCH] Fix zwave_js device remove test (#45864) --- tests/components/zwave_js/test_init.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/components/zwave_js/test_init.py b/tests/components/zwave_js/test_init.py index 86fbf27ab4f..fa61b3deb27 100644 --- a/tests/components/zwave_js/test_init.py +++ b/tests/components/zwave_js/test_init.py @@ -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