jeans/supervisor
1
0
Fork 0
mirror of https://github.com/home-assistant/supervisor.git synced 2025-11-02 15:39:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add safe mode option to core rebuild (#5120)

Browse Source 
* Add safe mode option to core rebuild

* Adding logging for increased traceability
This commit is contained in:
Mike Degatano
2024-06-05 15:44:07 -04:00
committed by GitHub
parent 09166e3867
commit ab78d87304
3 changed files with 42 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tests/api/test_homeassistant.py
View File

@@ -4,6 +4,7 @@ from pathlib import Path
from unittest.mock import MagicMock, patch
from aiohttp.test_utils import TestClient
from awesomeversion import AwesomeVersion
import pytest
from supervisor.coresys import CoreSys
@@ -115,3 +116,29 @@ async def test_api_restart(
assert container.restart.call_count == 2
assert safe_mode_marker.exists()
async def test_api_rebuild(
api_client: TestClient,
coresys: CoreSys,
container: MagicMock,
tmp_supervisor_data: Path,
path_extern,
):
"""Test rebuilding homeassistant."""
coresys.homeassistant.version = AwesomeVersion("2023.09.0")
safe_mode_marker = tmp_supervisor_data / "homeassistant" / "safe-mode"
with patch.object(HomeAssistantCore, "_block_till_run"):
await api_client.post("/homeassistant/rebuild")
assert container.remove.call_count == 2
container.start.assert_called_once()
assert not safe_mode_marker.exists()
with patch.object(HomeAssistantCore, "_block_till_run"):
await api_client.post("/homeassistant/rebuild", json={"safe_mode": True})
assert container.remove.call_count == 4
assert container.start.call_count == 2
assert safe_mode_marker.exists()