supervisor/tests/resolution/fixup/test_system_create_full_backup.py
Pascal Vizeli 942b5e6150
Rework fixups & Add integrity (#3614)
* Rework fixups & Add integrity

* Fix tests

* fix test v2

* Protect ddos
2022-05-05 16:21:19 +02:00

30 lines
858 B
Python

"""Test evaluation base."""
# pylint: disable=import-error,protected-access
from unittest.mock import AsyncMock
from supervisor.coresys import CoreSys
from supervisor.resolution.const import ContextType, SuggestionType
from supervisor.resolution.data import Suggestion
from supervisor.resolution.fixups.system_create_full_backup import (
FixupSystemCreateFullBackup,
)
async def test_fixup(coresys: CoreSys):
"""Test fixup."""
create_full_backup = FixupSystemCreateFullBackup(coresys)
assert not create_full_backup.auto
coresys.resolution.suggestions = Suggestion(
SuggestionType.CREATE_FULL_BACKUP, ContextType.SYSTEM
)
mock_backups = AsyncMock()
coresys.backups.do_backup_full = mock_backups
await create_full_backup()
mock_backups.assert_called()
assert len(coresys.resolution.suggestions) == 0