mirror of
https://github.com/home-assistant/supervisor.git
synced 2026-02-15 07:27:13 +00:00
909a2dda2f
* Migrate all docker container interactions to aiodocker
* Remove containers_legacy since its no longer used
* Add back remove color logic
* Revert accidental invert of conditional in setup_network
* Fix typos found by copilot
* Apply suggestions from code review
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
* Revert "Apply suggestions from code review"
This reverts commit 0a475433ea.
---------
Co-authored-by: Stefan Agner <stefan@agner.ch>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
"""Test fixup plugin execute rebuild."""
|
|
|
|
from collections.abc import Callable, Coroutine
|
|
from typing import Any
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from aiodocker.containers import DockerContainer
|
|
import pytest
|
|
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.docker.manager import DockerAPI
|
|
from supervisor.plugins.audio import PluginAudio
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
from supervisor.resolution.fixups.plugin_execute_rebuild import (
|
|
FixupPluginExecuteRebuild,
|
|
)
|
|
|
|
|
|
def make_mock_container_get(
|
|
status: str,
|
|
) -> Callable[[str], Coroutine[Any, Any, DockerContainer]]:
|
|
"""Make mock of container get."""
|
|
out = MagicMock(spec=DockerContainer)
|
|
out.status = status
|
|
out.show.return_value = {"State": {"Status": status, "ExitCode": 0}, "Mounts": []}
|
|
|
|
async def mock_container_get(name) -> DockerContainer:
|
|
return out
|
|
|
|
return mock_container_get
|
|
|
|
|
|
@pytest.mark.parametrize("status", ["running", "stopped"])
|
|
async def test_fixup(docker: DockerAPI, coresys: CoreSys, status: str):
|
|
"""Test fixup rebuilds plugin's container regardless of current state."""
|
|
docker.containers.get = make_mock_container_get(status)
|
|
|
|
plugin_execute_rebuild = FixupPluginExecuteRebuild(coresys)
|
|
|
|
assert plugin_execute_rebuild.auto is True
|
|
|
|
coresys.resolution.create_issue(
|
|
IssueType.DOCKER_CONFIG,
|
|
ContextType.PLUGIN,
|
|
reference="audio",
|
|
suggestions=[SuggestionType.EXECUTE_REBUILD],
|
|
)
|
|
with patch.object(PluginAudio, "rebuild") as rebuild:
|
|
await plugin_execute_rebuild()
|
|
rebuild.assert_called_once()
|
|
|
|
assert not coresys.resolution.issues
|
|
assert not coresys.resolution.suggestions
|