diff --git a/tests/components/config/test_device_registry.py b/tests/components/config/test_device_registry.py index 25b465192cf495eeac2bfcb1a9a68110478ff7a9..a92b2a353ef6eb07aa3c635de3e17dc9289f7684 100644 --- a/tests/components/config/test_device_registry.py +++ b/tests/components/config/test_device_registry.py @@ -26,15 +26,17 @@ async def test_list_devices( hass: HomeAssistant, client, device_registry: dr.DeviceRegistry ) -> None: """Test list entries.""" + entry = MockConfigEntry(title=None) + entry.add_to_hass(hass) device1 = device_registry.async_get_or_create( - config_entry_id="1234", + config_entry_id=entry.entry_id, connections={("ethernet", "12:34:56:78:90:AB:CD:EF")}, identifiers={("bridgeid", "0123")}, manufacturer="manufacturer", model="model", ) device2 = device_registry.async_get_or_create( - config_entry_id="1234", + config_entry_id=entry.entry_id, identifiers={("bridgeid", "1234")}, manufacturer="manufacturer", model="model", @@ -50,7 +52,7 @@ async def test_list_devices( assert msg["result"] == [ { "area_id": None, - "config_entries": ["1234"], + "config_entries": [entry.entry_id], "configuration_url": None, "connections": [["ethernet", "12:34:56:78:90:AB:CD:EF"]], "disabled_by": None, @@ -66,7 +68,7 @@ async def test_list_devices( }, { "area_id": None, - "config_entries": ["1234"], + "config_entries": [entry.entry_id], "configuration_url": None, "connections": [], "disabled_by": None, @@ -94,7 +96,7 @@ async def test_list_devices( assert msg["result"] == [ { "area_id": None, - "config_entries": ["1234"], + "config_entries": [entry.entry_id], "configuration_url": None, "connections": [["ethernet", "12:34:56:78:90:AB:CD:EF"]], "disabled_by": None, @@ -135,8 +137,10 @@ async def test_update_device( payload_value, ) -> None: """Test update entry.""" + entry = MockConfigEntry(title=None) + entry.add_to_hass(hass) device = device_registry.async_get_or_create( - config_entry_id="1234", + config_entry_id=entry.entry_id, connections={("ethernet", "12:34:56:78:90:AB:CD:EF")}, identifiers={("bridgeid", "0123")}, manufacturer="manufacturer", diff --git a/tests/components/config/test_entity_registry.py b/tests/components/config/test_entity_registry.py index 9cad68c9c992464e94599c1c56ee6f3f92a6d240..a002f2c2d50dd50ed34bf3e76b9d863b1c802b17 100644 --- a/tests/components/config/test_entity_registry.py +++ b/tests/components/config/test_entity_registry.py @@ -721,7 +721,7 @@ async def test_enable_entity_disabled_device( config_entry.add_to_hass(hass) device = device_registry.async_get_or_create( - config_entry_id="1234", + config_entry_id=config_entry.entry_id, connections={("ethernet", "12:34:56:78:90:AB:CD:EF")}, identifiers={("bridgeid", "0123")}, manufacturer="manufacturer",