mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-09 02:49:43 +00:00
0636e49fe2
* Fix mypy issues in addons * Fix mypy issues in api * fix docstring * Brackets instead of get with default
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""Test evaluation base."""
|
|
|
|
# pylint: disable=import-error,protected-access
|
|
from os import listdir
|
|
from pathlib import Path
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
from supervisor.resolution.data import Issue, Suggestion
|
|
from supervisor.resolution.fixups.store_execute_reset import FixupStoreExecuteReset
|
|
|
|
|
|
async def test_fixup(coresys: CoreSys, tmp_path):
|
|
"""Test fixup."""
|
|
store_execute_reset = FixupStoreExecuteReset(coresys)
|
|
test_repo = Path(tmp_path, "test_repo")
|
|
|
|
assert store_execute_reset.auto
|
|
|
|
coresys.resolution.add_suggestion(
|
|
Suggestion(SuggestionType.EXECUTE_RESET, ContextType.STORE, reference="test")
|
|
)
|
|
coresys.resolution.add_issue(
|
|
Issue(IssueType.CORRUPT_REPOSITORY, ContextType.STORE, reference="test")
|
|
)
|
|
|
|
test_repo.mkdir()
|
|
(test_repo / ".git").mkdir()
|
|
assert test_repo.exists()
|
|
|
|
mock_repositorie = AsyncMock()
|
|
mock_repositorie.git.path = test_repo
|
|
coresys.store.repositories["test"] = mock_repositorie
|
|
assert len(listdir(test_repo)) > 0
|
|
|
|
with patch("shutil.disk_usage", return_value=(42, 42, 2 * (1024.0**3))):
|
|
await store_execute_reset()
|
|
|
|
assert len(listdir(test_repo)) == 0
|
|
assert mock_repositorie.load.called
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
assert len(coresys.resolution.issues) == 0
|