mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-12-24 20:35:55 +00:00
42e78408a7
Make sure that add-on store resets do not delete the root folder. This is important so that successive reset attempts do not fail (the directory passed to `remove_folder` must exist, otherwise find fails with an non-zero exit code). While at it, handle find errors properly and report errors as critical.
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.suggestions = Suggestion(
|
|
SuggestionType.EXECUTE_RESET, ContextType.STORE, reference="test"
|
|
)
|
|
coresys.resolution.issues = 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
|