mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-12-20 18:38:59 +00:00
121 lines
4.1 KiB
Python
121 lines
4.1 KiB
Python
"""Test fixup addon execute restart."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from supervisor.addons.addon import Addon
|
|
from supervisor.const import AddonState
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.docker.addon import DockerAddon
|
|
from supervisor.docker.interface import DockerInterface
|
|
from supervisor.exceptions import DockerError
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
from supervisor.resolution.data import Issue, Suggestion
|
|
from supervisor.resolution.fixups.addon_execute_restart import FixupAddonExecuteRestart
|
|
|
|
from tests.const import TEST_ADDON_SLUG
|
|
|
|
DEVICE_ACCESS_MISSING_ISSUE = Issue(
|
|
IssueType.DEVICE_ACCESS_MISSING,
|
|
ContextType.ADDON,
|
|
reference=TEST_ADDON_SLUG,
|
|
)
|
|
EXECUTE_RESTART_SUGGESTION = Suggestion(
|
|
SuggestionType.EXECUTE_RESTART, ContextType.ADDON, reference="local_ssh"
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures("path_extern")
|
|
async def test_fixup(coresys: CoreSys, install_addon_ssh: Addon):
|
|
"""Test fixup restarts addon."""
|
|
install_addon_ssh.state = AddonState.STARTED
|
|
addon_execute_restart = FixupAddonExecuteRestart(coresys)
|
|
assert addon_execute_restart.auto is False
|
|
|
|
async def mock_stop(*args, **kwargs):
|
|
install_addon_ssh.state = AddonState.STOPPED
|
|
|
|
coresys.resolution.add_issue(
|
|
DEVICE_ACCESS_MISSING_ISSUE,
|
|
suggestions=[SuggestionType.EXECUTE_RESTART],
|
|
)
|
|
with (
|
|
patch.object(DockerInterface, "stop") as stop,
|
|
patch.object(DockerAddon, "run") as run,
|
|
patch.object(Addon, "_wait_for_startup"),
|
|
patch.object(Addon, "write_options"),
|
|
):
|
|
await addon_execute_restart()
|
|
stop.assert_called_once()
|
|
run.assert_called_once()
|
|
|
|
assert not coresys.resolution.issues
|
|
assert not coresys.resolution.suggestions
|
|
|
|
|
|
@pytest.mark.usefixtures("path_extern")
|
|
async def test_fixup_stop_error(
|
|
coresys: CoreSys, install_addon_ssh: Addon, caplog: pytest.LogCaptureFixture
|
|
):
|
|
"""Test fixup fails on stop addon failure."""
|
|
install_addon_ssh.state = AddonState.STARTED
|
|
addon_execute_start = FixupAddonExecuteRestart(coresys)
|
|
|
|
coresys.resolution.add_issue(
|
|
DEVICE_ACCESS_MISSING_ISSUE,
|
|
suggestions=[SuggestionType.EXECUTE_RESTART],
|
|
)
|
|
with (
|
|
patch.object(DockerInterface, "stop", side_effect=DockerError),
|
|
patch.object(DockerAddon, "run") as run,
|
|
):
|
|
await addon_execute_start()
|
|
run.assert_not_called()
|
|
|
|
assert DEVICE_ACCESS_MISSING_ISSUE in coresys.resolution.issues
|
|
assert EXECUTE_RESTART_SUGGESTION in coresys.resolution.suggestions
|
|
assert "Could not stop local_ssh" in caplog.text
|
|
|
|
|
|
@pytest.mark.usefixtures("path_extern")
|
|
async def test_fixup_start_error(
|
|
coresys: CoreSys, install_addon_ssh: Addon, caplog: pytest.LogCaptureFixture
|
|
):
|
|
"""Test fixup logs a start addon failure."""
|
|
install_addon_ssh.state = AddonState.STARTED
|
|
addon_execute_start = FixupAddonExecuteRestart(coresys)
|
|
|
|
coresys.resolution.add_issue(
|
|
DEVICE_ACCESS_MISSING_ISSUE,
|
|
suggestions=[SuggestionType.EXECUTE_RESTART],
|
|
)
|
|
with (
|
|
patch.object(DockerInterface, "stop") as stop,
|
|
patch.object(DockerAddon, "run", side_effect=DockerError),
|
|
patch.object(Addon, "write_options"),
|
|
):
|
|
await addon_execute_start()
|
|
stop.assert_called_once()
|
|
|
|
assert DEVICE_ACCESS_MISSING_ISSUE not in coresys.resolution.issues
|
|
assert EXECUTE_RESTART_SUGGESTION not in coresys.resolution.suggestions
|
|
assert "Could not restart local_ssh" in caplog.text
|
|
|
|
|
|
async def test_fixup_no_addon(coresys: CoreSys, caplog: pytest.LogCaptureFixture):
|
|
"""Test fixup dismisses if addon is missing."""
|
|
addon_execute_start = FixupAddonExecuteRestart(coresys)
|
|
|
|
coresys.resolution.add_issue(
|
|
DEVICE_ACCESS_MISSING_ISSUE,
|
|
suggestions=[SuggestionType.EXECUTE_RESTART],
|
|
)
|
|
with patch.object(DockerAddon, "stop") as stop:
|
|
await addon_execute_start()
|
|
stop.assert_not_called()
|
|
|
|
assert not coresys.resolution.issues
|
|
assert not coresys.resolution.suggestions
|
|
assert "Cannot restart addon local_ssh as it does not exist" in caplog.text
|