Set backup manager state to completed when restore is finished (#134283)

pull/134242/head^2
Bram Kragten 2024-12-31 15:01:06 +01:00 committed by GitHub
parent 9348569f90
commit 06b33e5589
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 1 deletions

View File

@ -829,6 +829,9 @@ class BackupManager:
restore_folders=restore_folders,
restore_homeassistant=restore_homeassistant,
)
self.async_on_backup_event(
RestoreBackupEvent(stage=None, state=RestoreBackupState.COMPLETED)
)
except Exception:
self.async_on_backup_event(
RestoreBackupEvent(stage=None, state=RestoreBackupState.FAILED)

View File

@ -942,7 +942,9 @@ async def test_reader_writer_restore(
await client.send_json_auto_id({"type": "backup/subscribe_events"})
response = await client.receive_json()
assert response["event"] == {"manager_state": "idle"}
assert response["event"] == {
"manager_state": "idle",
}
response = await client.receive_json()
assert response["success"]
@ -980,6 +982,13 @@ async def test_reader_writer_restore(
response = await client.receive_json()
assert response["success"]
response = await client.receive_json()
assert response["event"] == {
"manager_state": "restore_backup",
"stage": None,
"state": "completed",
}
response = await client.receive_json()
assert response["event"] == {"manager_state": "idle"}