From e9515b7584c44493701e4a19e9081c2de6c0b37e Mon Sep 17 00:00:00 2001 From: dougiteixeira <31328123+dougiteixeira@users.noreply.github.com> Date: Sat, 22 Jun 2024 10:46:55 -0300 Subject: [PATCH] Update `test_device_cleaning` in Utiltity Meter. (#120161) --- tests/components/utility_meter/test_init.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/components/utility_meter/test_init.py b/tests/components/utility_meter/test_init.py index 77d223454ec..cd549c77913 100644 --- a/tests/components/utility_meter/test_init.py +++ b/tests/components/utility_meter/test_init.py @@ -444,10 +444,12 @@ async def test_setup_and_remove_config_entry( assert len(entity_registry.entities) == 0 -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 Utility Meter.""" - device_registry = dr.async_get(hass) - entity_registry = er.async_get(hass) # Source entity device config entry source_config_entry = MockConfigEntry()