From d19037a36bb5b252f5aebff0dbbaa04b2e91f4bf Mon Sep 17 00:00:00 2001 From: Raman Gupta <7243222+raman325@users.noreply.github.com> Date: Sun, 7 Jan 2024 06:26:08 -0500 Subject: [PATCH] Clean up zwave_js test_removed_device test (#107346) --- tests/components/zwave_js/test_init.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/tests/components/zwave_js/test_init.py b/tests/components/zwave_js/test_init.py index 75a7397cc4e..77b1fcb8b3a 100644 --- a/tests/components/zwave_js/test_init.py +++ b/tests/components/zwave_js/test_init.py @@ -964,22 +964,14 @@ async def test_removed_device( device_entries = dr.async_entries_for_config_entry(dev_reg, integration.entry_id) assert len(device_entries) == 3 - # Check how many entities there are - ent_reg = er.async_get(hass) - entity_entries = er.async_entries_for_config_entry(ent_reg, integration.entry_id) - assert len(entity_entries) == 93 - # Remove a node and reload the entry old_node = driver.controller.nodes.pop(13) await hass.config_entries.async_reload(integration.entry_id) await hass.async_block_till_done() - # Assert that the node and all of it's entities were removed from the device and - # entity registry + # Assert that the node was removed from the device registry device_entries = dr.async_entries_for_config_entry(dev_reg, integration.entry_id) assert len(device_entries) == 2 - entity_entries = er.async_entries_for_config_entry(ent_reg, integration.entry_id) - assert len(entity_entries) == 62 assert ( dev_reg.async_get_device(identifiers={get_device_id(driver, old_node)}) is None )