diff --git a/tests/helpers/snapshots/test_entity_platform.ambr b/tests/helpers/snapshots/test_entity_platform.ambr index 236fa623e8f..97d3e858b69 100644 --- a/tests/helpers/snapshots/test_entity_platform.ambr +++ b/tests/helpers/snapshots/test_entity_platform.ambr @@ -3,11 +3,7 @@ DeviceRegistryEntrySnapshot({ 'area_id': 'heliport', 'config_entries': , - 'config_subentries': dict({ - 'super-mock-id': set({ - None, - }), - }), + 'config_subentries': , 'configuration_url': 'http://192.168.0.100/config', 'connections': set({ tuple( @@ -44,11 +40,7 @@ DeviceRegistryEntrySnapshot({ 'area_id': 'heliport', 'config_entries': , - 'config_subentries': dict({ - 'super-mock-id': set({ - 'mock-subentry-id-1', - }), - }), + 'config_subentries': , 'configuration_url': 'http://192.168.0.100/config', 'connections': set({ tuple( diff --git a/tests/helpers/test_entity_platform.py b/tests/helpers/test_entity_platform.py index ed2e1b6c4c4..da59d21f170 100644 --- a/tests/helpers/test_entity_platform.py +++ b/tests/helpers/test_entity_platform.py @@ -1245,13 +1245,13 @@ async def test_device_info_called( device = device_registry.async_get_device(identifiers={("hue", "1234")}) assert device == snapshot assert device.config_entries == {config_entry.entry_id} - assert device.config_subentries == {config_entry.entry_id: {None}} + assert device.config_subentries == {config_entry.entry_id: None} assert device.primary_config_entry == config_entry.entry_id assert device.via_device_id == via.id device = device_registry.async_get_device(identifiers={("hue", "efgh")}) assert device == snapshot assert device.config_entries == {config_entry.entry_id} - assert device.config_subentries == {config_entry.entry_id: {"mock-subentry-id-1"}} + assert device.config_subentries == {config_entry.entry_id: "mock-subentry-id-1"} assert device.primary_config_entry == config_entry.entry_id assert device.via_device_id == via.id