From 06b33e5589b2936c51696f4b89f6dde512f614ae Mon Sep 17 00:00:00 2001 From: Bram Kragten Date: Tue, 31 Dec 2024 15:01:06 +0100 Subject: [PATCH] Set backup manager state to completed when restore is finished (#134283) --- homeassistant/components/backup/manager.py | 3 +++ tests/components/hassio/test_backup.py | 11 ++++++++++- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/backup/manager.py b/homeassistant/components/backup/manager.py index 9b20c82d709..9515ab89cd2 100644 --- a/homeassistant/components/backup/manager.py +++ b/homeassistant/components/backup/manager.py @@ -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) diff --git a/tests/components/hassio/test_backup.py b/tests/components/hassio/test_backup.py index c39574fd941..3c9440c41ff 100644 --- a/tests/components/hassio/test_backup.py +++ b/tests/components/hassio/test_backup.py @@ -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"}