Update test_device_cleaning in Derivative (#120162)

This commit is contained in:
dougiteixeira 2024-06-22 10:54:35 -03:00 committed by GitHub
parent e9515b7584
commit 02f0050819
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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()