mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-12-20 18:38:59 +00:00
f6faa18409
It seems that the codebase is not formatted with the latest ruff version. This PR reformats the codebase with ruff 0.5.7.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""Test check."""
|
|
|
|
# pylint: disable=import-error, protected-access
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from supervisor.const import CoreState
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.resolution.const import ContextType, SuggestionType
|
|
from supervisor.resolution.data import Suggestion
|
|
from supervisor.resolution.validate import get_valid_modules
|
|
|
|
|
|
async def test_check_autofix(coresys: CoreSys):
|
|
"""Test check for setup."""
|
|
coresys.core.state = CoreState.RUNNING
|
|
|
|
coresys.resolution.fixup._fixups[
|
|
"system_create_full_backup"
|
|
].process_fixup = AsyncMock()
|
|
|
|
with patch(
|
|
"supervisor.resolution.fixups.system_create_full_backup.FixupSystemCreateFullBackup.auto",
|
|
return_value=True,
|
|
):
|
|
await coresys.resolution.fixup.run_autofix()
|
|
|
|
coresys.resolution.fixup._fixups[
|
|
"system_create_full_backup"
|
|
].process_fixup.assert_not_called()
|
|
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.CREATE_FULL_BACKUP, ContextType.SYSTEM
|
|
)
|
|
with patch(
|
|
"supervisor.resolution.fixups.system_create_full_backup.FixupSystemCreateFullBackup.auto",
|
|
return_value=True,
|
|
):
|
|
await coresys.resolution.fixup.run_autofix()
|
|
|
|
coresys.resolution.fixup._fixups[
|
|
"system_create_full_backup"
|
|
].process_fixup.assert_called_once()
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
|
|
|
|
def test_dynamic_fixup_loader(coresys: CoreSys):
|
|
"""Test dynamic fixup loader, this ensures that all fixups have defined a setup function."""
|
|
for fixup in get_valid_modules("fixups"):
|
|
assert fixup in coresys.resolution.fixup._fixups
|