mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-08 18:39:33 +00:00
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
"""Test fixup system reboot."""
|
|
|
|
from unittest.mock import PropertyMock, patch
|
|
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.host.const import HostFeature
|
|
from supervisor.host.manager import HostManager
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
from supervisor.resolution.data import Issue, Suggestion
|
|
from supervisor.resolution.fixups.system_execute_reboot import FixupSystemExecuteReboot
|
|
|
|
|
|
async def test_fixup(coresys: CoreSys, dbus: list[str]):
|
|
"""Test fixup."""
|
|
await coresys.dbus.logind.connect(coresys.dbus.bus)
|
|
dbus.clear()
|
|
|
|
system_execute_reboot = FixupSystemExecuteReboot(coresys)
|
|
assert system_execute_reboot.auto is False
|
|
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.EXECUTE_REBOOT, ContextType.SYSTEM
|
|
)
|
|
coresys.resolution.issues = Issue(IssueType.REBOOT_REQUIRED, ContextType.SYSTEM)
|
|
|
|
with patch.object(
|
|
HostManager, "features", new=PropertyMock(return_value=[HostFeature.REBOOT])
|
|
):
|
|
await system_execute_reboot()
|
|
|
|
assert dbus == ["/org/freedesktop/login1-org.freedesktop.login1.Manager.Reboot"]
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
assert len(coresys.resolution.issues) == 0
|