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.
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
"""Test check for detached addons due to removal from repo."""
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import PropertyMock, patch
|
|
|
|
from supervisor.addons.addon import Addon
|
|
from supervisor.config import CoreConfig
|
|
from supervisor.const import CoreState
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.resolution.checks.detached_addon_removed import (
|
|
CheckDetachedAddonRemoved,
|
|
)
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
|
|
|
|
async def test_base(coresys: CoreSys):
|
|
"""Test check basics."""
|
|
detached_addon_removed = CheckDetachedAddonRemoved(coresys)
|
|
assert detached_addon_removed.slug == "detached_addon_removed"
|
|
assert detached_addon_removed.enabled
|
|
|
|
|
|
async def test_check(
|
|
coresys: CoreSys, install_addon_ssh: Addon, tmp_supervisor_data: Path
|
|
):
|
|
"""Test check for detached addons."""
|
|
detached_addon_removed = CheckDetachedAddonRemoved(coresys)
|
|
coresys.core.state = CoreState.SETUP
|
|
|
|
await detached_addon_removed()
|
|
assert len(coresys.resolution.issues) == 0
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
|
|
(addons_dir := tmp_supervisor_data / "addons" / "local").mkdir()
|
|
with patch.object(
|
|
CoreConfig, "path_addons_local", new=PropertyMock(return_value=addons_dir)
|
|
):
|
|
await coresys.store.load()
|
|
|
|
await detached_addon_removed()
|
|
|
|
assert len(coresys.resolution.issues) == 1
|
|
assert coresys.resolution.issues[0].type is IssueType.DETACHED_ADDON_REMOVED
|
|
assert coresys.resolution.issues[0].context is ContextType.ADDON
|
|
assert coresys.resolution.issues[0].reference == install_addon_ssh.slug
|
|
|
|
assert len(coresys.resolution.suggestions) == 1
|
|
assert coresys.resolution.suggestions[0].type is SuggestionType.EXECUTE_REMOVE
|
|
assert coresys.resolution.suggestions[0].context is ContextType.ADDON
|
|
assert coresys.resolution.suggestions[0].reference == install_addon_ssh.slug
|
|
|
|
|
|
async def test_approve(
|
|
coresys: CoreSys, install_addon_ssh: Addon, tmp_supervisor_data: Path
|
|
):
|
|
"""Test approve existing detached addon issues."""
|
|
detached_addon_removed = CheckDetachedAddonRemoved(coresys)
|
|
coresys.core.state = CoreState.SETUP
|
|
|
|
assert (
|
|
await detached_addon_removed.approve_check(reference=install_addon_ssh.slug)
|
|
is False
|
|
)
|
|
|
|
(addons_dir := tmp_supervisor_data / "addons" / "local").mkdir()
|
|
with patch.object(
|
|
CoreConfig, "path_addons_local", new=PropertyMock(return_value=addons_dir)
|
|
):
|
|
await coresys.store.load()
|
|
|
|
assert (
|
|
await detached_addon_removed.approve_check(reference=install_addon_ssh.slug)
|
|
is True
|
|
)
|
|
|
|
|
|
async def test_did_run(coresys: CoreSys):
|
|
"""Test that the check ran as expected."""
|
|
detached_addon_removed = CheckDetachedAddonRemoved(coresys)
|
|
should_run = detached_addon_removed.states
|
|
should_not_run = [state for state in CoreState if state not in should_run]
|
|
assert should_run == [CoreState.SETUP]
|
|
assert len(should_not_run) != 0
|
|
|
|
with patch.object(
|
|
CheckDetachedAddonRemoved, "run_check", return_value=None
|
|
) as check:
|
|
for state in should_run:
|
|
coresys.core.state = state
|
|
await detached_addon_removed()
|
|
check.assert_called_once()
|
|
check.reset_mock()
|
|
|
|
for state in should_not_run:
|
|
coresys.core.state = state
|
|
await detached_addon_removed()
|
|
check.assert_not_called()
|
|
check.reset_mock()
|