jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve tests for migration

Browse Source
This commit is contained in:
jbouwh 2025-07-15 19:49:44 +00:00
parent d86b3d8e94
commit 9ba6056935
1 changed files with 26 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/components/mqtt/test_repairs.py
View File

@ -90,6 +90,7 @@ async def test_subentry_reconfigure_export_settings(
# assert we have a device for the subentry # assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)}) device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {subentry_id}
assert device is not None assert device is not None
# assert we entity for all subentry components # assert we entity for all subentry components
@ -124,15 +125,20 @@ async def test_subentry_reconfigure_export_settings(
"url": "https://www.home-assistant.io/integrations/mqtt/" "url": "https://www.home-assistant.io/integrations/mqtt/"
} }
# Copy the suggested values for an export # Copy the exported config suggested values for an export
suggested_values_from_schema = { suggested_values_from_schema = {
field: field.description["suggested_value"] field: field.description["suggested_value"]
for field in result["data_schema"].schema for field in result["data_schema"].schema
} }
# Try to set up the exported config # Try to set up the exported config with a changed device name
events = async_capture_events(hass, ir.EVENT_REPAIRS_ISSUE_REGISTRY_UPDATED) events = async_capture_events(hass, ir.EVENT_REPAIRS_ISSUE_REGISTRY_UPDATED)
await setup_helper(hass, suggested_values_from_schema) await setup_helper(hass, suggested_values_from_schema)
# Assert the subentry device was not effected by the exported configs
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {subentry_id}
assert device is not None
# Assert a repair flow was created # Assert a repair flow was created
assert len(events) == 1 assert len(events) == 1
issue_id = events[0].data["issue_id"] issue_id = events[0].data["issue_id"]
@ -151,3 +157,21 @@ async def test_subentry_reconfigure_export_settings(
data = await process_repair_fix_flow(client, flow_id) data = await process_repair_fix_flow(client, flow_id)
assert data["type"] == "create_entry" assert data["type"] == "create_entry"
# Assert the subentry is removed and no other entity has linked the device
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
assert device is None
await hass.async_block_till_done(wait_background_tasks=True)
assert len(config_entry.subentries) == 0
# Try to set up the exported config again
events = async_capture_events(hass, ir.EVENT_REPAIRS_ISSUE_REGISTRY_UPDATED)
await setup_helper(hass, suggested_values_from_schema)
assert len(events) == 0
# The MQTT device was now set up from the new source
await hass.async_block_till_done(wait_background_tasks=True)
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {None}
assert device is not None