diff --git a/tests/components/utility_meter/test_config_flow.py b/tests/components/utility_meter/test_config_flow.py index 88a77407c07..262dbf36306 100644 --- a/tests/components/utility_meter/test_config_flow.py +++ b/tests/components/utility_meter/test_config_flow.py @@ -277,6 +277,7 @@ async def test_change_device_source(hass: HomeAssistant) -> None: # Configure source entity 1 (with a linked device) source_config_entry_1 = MockConfigEntry() + source_config_entry_1.add_to_hass(hass) source_device_entry_1 = device_registry.async_get_or_create( config_entry_id=source_config_entry_1.entry_id, identifiers={("sensor", "identifier_test1")}, @@ -292,6 +293,7 @@ async def test_change_device_source(hass: HomeAssistant) -> None: # Configure source entity 2 (with a linked device) source_config_entry_2 = MockConfigEntry() + source_config_entry_2.add_to_hass(hass) source_device_entry_2 = device_registry.async_get_or_create( config_entry_id=source_config_entry_2.entry_id, identifiers={("sensor", "identifier_test2")}, diff --git a/tests/components/utility_meter/test_sensor.py b/tests/components/utility_meter/test_sensor.py index 3d2d95fd26f..b8f197a4dee 100644 --- a/tests/components/utility_meter/test_sensor.py +++ b/tests/components/utility_meter/test_sensor.py @@ -1466,6 +1466,7 @@ async def test_device_id(hass: HomeAssistant) -> None: entity_registry = er.async_get(hass) source_config_entry = MockConfigEntry() + source_config_entry.add_to_hass(hass) source_device_entry = device_registry.async_get_or_create( config_entry_id=source_config_entry.entry_id, identifiers={("sensor", "identifier_test")},