Home-Assistant/supervisor
1
0
Fork 0
mirror of https://github.com/home-assistant/supervisor.git synced 2025-12-24 04:19:03 +00:00
Code Activity

Add an admin only device wipe API (#4934)

Browse Source 
* Add an admin only device wipe API

* Fix pylint issue
This commit is contained in:
Mike Degatano
2024-02-29 10:29:52 -05:00
committed by GitHub
parent 5126820619
commit 9d4848ee77
11 changed files with 114 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/api/test_os.py
View File

@@ -17,6 +17,7 @@ from tests.dbus_service_mocks.agent_boards import Boards as BoardsService
from tests.dbus_service_mocks.agent_boards_green import Green as GreenService
from tests.dbus_service_mocks.agent_boards_yellow import Yellow as YellowService
from tests.dbus_service_mocks.agent_datadisk import DataDisk as DataDiskService
from tests.dbus_service_mocks.agent_system import System as SystemService
from tests.dbus_service_mocks.base import DBusServiceMock
@@ -113,6 +114,23 @@ async def test_api_os_datadisk_migrate(
reboot.assert_called_once()
async def test_api_os_datadisk_wipe(
api_client: TestClient,
os_agent_services: dict[str, DBusServiceMock],
os_available,
):
"""Test datadisk wipe."""
system_service: SystemService = os_agent_services["agent_system"]
system_service.ScheduleWipeDevice.calls.clear()
with patch.object(SystemControl, "reboot") as reboot:
resp = await api_client.post("/os/datadisk/wipe")
assert resp.status == 200
assert system_service.ScheduleWipeDevice.calls == [()]
reboot.assert_called_once()
async def test_api_board_yellow_info(api_client: TestClient, coresys: CoreSys):
"""Test yellow board info."""
resp = await api_client.get("/os/boards/yellow")