mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-05-08 03:48:38 +00:00
120 lines
4.0 KiB
Python
120 lines
4.0 KiB
Python
"""Test Resolution API."""
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from supervisor.const import (
|
|
ATTR_ISSUES,
|
|
ATTR_SUGGESTIONS,
|
|
ATTR_UNHEALTHY,
|
|
ATTR_UNSUPPORTED,
|
|
)
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.exceptions import ResolutionError
|
|
from supervisor.resolution.const import (
|
|
ContextType,
|
|
IssueType,
|
|
SuggestionType,
|
|
UnhealthyReason,
|
|
UnsupportedReason,
|
|
)
|
|
from supervisor.resolution.data import Issue, Suggestion
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_base(coresys: CoreSys, api_client):
|
|
"""Test resolution manager api."""
|
|
coresys.resolution.unsupported = UnsupportedReason.OS
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.CLEAR_FULL_SNAPSHOT, ContextType.SYSTEM
|
|
)
|
|
coresys.resolution.create_issue(IssueType.FREE_SPACE, ContextType.SYSTEM)
|
|
|
|
resp = await api_client.get("/resolution/info")
|
|
result = await resp.json()
|
|
assert UnsupportedReason.OS in result["data"][ATTR_UNSUPPORTED]
|
|
assert (
|
|
SuggestionType.CLEAR_FULL_SNAPSHOT
|
|
== result["data"][ATTR_SUGGESTIONS][-1]["type"]
|
|
)
|
|
assert IssueType.FREE_SPACE == result["data"][ATTR_ISSUES][-1]["type"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_dismiss_suggestion(coresys: CoreSys, api_client):
|
|
"""Test resolution manager suggestion apply api."""
|
|
coresys.resolution.suggestions = clear_snapshot = Suggestion(
|
|
SuggestionType.CLEAR_FULL_SNAPSHOT, ContextType.SYSTEM
|
|
)
|
|
|
|
assert SuggestionType.CLEAR_FULL_SNAPSHOT == coresys.resolution.suggestions[-1].type
|
|
await api_client.delete(f"/resolution/suggestion/{clear_snapshot.uuid}")
|
|
assert clear_snapshot not in coresys.resolution.suggestions
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_apply_suggestion(coresys: CoreSys, api_client):
|
|
"""Test resolution manager suggestion apply api."""
|
|
coresys.resolution.suggestions = clear_snapshot = Suggestion(
|
|
SuggestionType.CLEAR_FULL_SNAPSHOT, ContextType.SYSTEM
|
|
)
|
|
coresys.resolution.suggestions = create_snapshot = Suggestion(
|
|
SuggestionType.CREATE_FULL_SNAPSHOT, ContextType.SYSTEM
|
|
)
|
|
|
|
mock_snapshots = AsyncMock()
|
|
mock_health = AsyncMock()
|
|
coresys.snapshots.do_snapshot_full = mock_snapshots
|
|
coresys.resolution.healthcheck = mock_health
|
|
|
|
await api_client.post(f"/resolution/suggestion/{clear_snapshot.uuid}")
|
|
await api_client.post(f"/resolution/suggestion/{create_snapshot.uuid}")
|
|
|
|
assert clear_snapshot not in coresys.resolution.suggestions
|
|
assert create_snapshot not in coresys.resolution.suggestions
|
|
|
|
assert mock_snapshots.called
|
|
assert mock_health.called
|
|
|
|
with pytest.raises(ResolutionError):
|
|
await coresys.resolution.apply_suggestion(clear_snapshot)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_dismiss_issue(coresys: CoreSys, api_client):
|
|
"""Test resolution manager issue apply api."""
|
|
coresys.resolution.issues = updated_failed = Issue(
|
|
IssueType.UPDATE_FAILED, ContextType.SYSTEM
|
|
)
|
|
|
|
assert IssueType.UPDATE_FAILED == coresys.resolution.issues[-1].type
|
|
await api_client.delete(f"/resolution/issue/{updated_failed.uuid}")
|
|
assert updated_failed not in coresys.resolution.issues
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_unhealthy(coresys: CoreSys, api_client):
|
|
"""Test resolution manager api."""
|
|
coresys.resolution.unhealthy = UnhealthyReason.DOCKER
|
|
|
|
resp = await api_client.get("/resolution/info")
|
|
result = await resp.json()
|
|
assert UnhealthyReason.DOCKER == result["data"][ATTR_UNHEALTHY][-1]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_api_resolution_check_options(coresys: CoreSys, api_client):
|
|
"""Test client API with checks options."""
|
|
free_space = coresys.resolution.check.get("free_space")
|
|
|
|
assert free_space.enabled
|
|
await api_client.post(
|
|
f"/resolution/check/{free_space.slug}/options", json={"enabled": False}
|
|
)
|
|
assert not free_space.enabled
|
|
|
|
await api_client.post(
|
|
f"/resolution/check/{free_space.slug}/options", json={"enabled": True}
|
|
)
|
|
assert free_space.enabled
|