jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 01:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unify error reporting in onboarding backup API (#138200)

Browse Source
This commit is contained in:
Erik Montnemery 2025-02-11 09:47:09 +01:00 committed by GitHub
parent 35416189f2
commit ec0cef0611
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/onboarding/views.py
View File

@ -357,7 +357,7 @@ def with_backup_manager[_ViewT: BackupOnboardingView, **_P](
manager = async_get_backup_manager(request.app[KEY_HASS])
except HomeAssistantError:
return self.json(
{"error": "backup_disabled"},
{"code": "backup_disabled"},
status_code=HTTPStatus.INTERNAL_SERVER_ERROR,
)
@ -420,7 +420,7 @@ class RestoreBackupView(BackupOnboardingView):
)
except IncorrectPasswordError:
return self.json(
{"message": "incorrect_password"}, status_code=HTTPStatus.BAD_REQUEST
{"code": "incorrect_password"}, status_code=HTTPStatus.BAD_REQUEST
)
return web.Response(status=HTTPStatus.OK)

26
tests/components/onboarding/test_views.py
View File

@ -777,7 +777,7 @@ async def test_onboarding_backup_view_without_backup(
resp = await client.request(method, f"/api/onboarding/{view}", **kwargs)
assert resp.status == 500
assert await resp.json() == {"error": "backup_disabled"}
assert await resp.json() == {"code": "backup_disabled"}
async def test_onboarding_backup_info(
@ -920,14 +920,16 @@ async def test_onboarding_backup_restore(
@pytest.mark.parametrize(
("params", "restore_error", "expected_status", "expected_message", "restore_calls"),
("params", "restore_error", "expected_status", "expected_json", "restore_calls"),
[
# Missing agent_id
(
{"backup_id": "abc123"},
None,
400,
"Message format incorrect: required key not provided @ data['agent_id']",
{
"message": "Message format incorrect: required key not provided @ data['agent_id']"
},
0,
),
# Missing backup_id
@ -935,7 +937,9 @@ async def test_onboarding_backup_restore(
{"agent_id": "backup.local"},
None,
400,
"Message format incorrect: required key not provided @ data['backup_id']",
{
"message": "Message format incorrect: required key not provided @ data['backup_id']"
},
0,
),
# Invalid restore_database
@ -947,7 +951,9 @@ async def test_onboarding_backup_restore(
},
None,
400,
"Message format incorrect: expected bool for dictionary value @ data['restore_database']",
{
"message": "Message format incorrect: expected bool for dictionary value @ data['restore_database']"
},
0,
),
# Invalid folder
@ -959,7 +965,9 @@ async def test_onboarding_backup_restore(
},
None,
400,
"Message format incorrect: expected Folder or one of 'share', 'addons/local', 'ssl', 'media' @ data['restore_folders'][0]",
{
"message": "Message format incorrect: expected Folder or one of 'share', 'addons/local', 'ssl', 'media' @ data['restore_folders'][0]"
},
0,
),
# Wrong password
@ -967,7 +975,7 @@ async def test_onboarding_backup_restore(
{"backup_id": "abc123", "agent_id": "backup.local"},
backup.IncorrectPasswordError,
400,
"incorrect_password",
{"code": "incorrect_password"},
1,
),
],
@ -979,7 +987,7 @@ async def test_onboarding_backup_restore_error(
params: dict[str, Any],
restore_error: Exception | None,
expected_status: int,
expected_message: str,
expected_json: str,
restore_calls: int,
) -> None:
"""Test returning installation type during onboarding."""
@ -998,7 +1006,7 @@ async def test_onboarding_backup_restore_error(
resp = await client.post("/api/onboarding/backup/restore", json=params)
assert resp.status == expected_status
assert await resp.json() == {"message": expected_message}
assert await resp.json() == expected_json
assert len(mock_restore.mock_calls) == restore_calls