From 02f00508195f96f611bfaca9527afa89ea1a5856 Mon Sep 17 00:00:00 2001 From: dougiteixeira <31328123+dougiteixeira@users.noreply.github.com> Date: Sat, 22 Jun 2024 10:54:35 -0300 Subject: [PATCH] Update `test_device_cleaning` in Derivative (#120162) --- tests/components/derivative/test_init.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/components/derivative/test_init.py b/tests/components/derivative/test_init.py index 32b763ee84d..0081ab97580 100644 --- a/tests/components/derivative/test_init.py +++ b/tests/components/derivative/test_init.py @@ -62,10 +62,12 @@ async def test_setup_and_remove_config_entry( assert entity_registry.async_get(derivative_entity_id) is None -async def test_device_cleaning(hass: HomeAssistant) -> None: +async def test_device_cleaning( + hass: HomeAssistant, + device_registry: dr.DeviceRegistry, + entity_registry: er.EntityRegistry, +) -> None: """Test for source entity device for Derivative.""" - device_registry = dr.async_get(hass) - entity_registry = er.async_get(hass) # Source entity device config entry source_config_entry = MockConfigEntry()