mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-12 20:40:21 +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.
114 lines
3.8 KiB
Python
114 lines
3.8 KiB
Python
"""Test system fixup rename data disk."""
|
|
|
|
# pylint: disable=import-error
|
|
from dbus_fast import Variant
|
|
import pytest
|
|
|
|
from supervisor.coresys import CoreSys
|
|
from supervisor.resolution.const import ContextType, IssueType, SuggestionType
|
|
from supervisor.resolution.data import Issue, Suggestion
|
|
from supervisor.resolution.fixups.system_rename_data_disk import (
|
|
FixupSystemRenameDataDisk,
|
|
)
|
|
|
|
from tests.dbus_service_mocks.base import DBusServiceMock
|
|
from tests.dbus_service_mocks.udisks2_filesystem import Filesystem as FilesystemService
|
|
from tests.dbus_service_mocks.udisks2_manager import (
|
|
UDisks2Manager as UDisks2ManagerService,
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="sda1_filesystem_service")
|
|
async def fixture_sda1_filesystem_service(
|
|
udisks2_services: dict[str, DBusServiceMock | dict[str, DBusServiceMock]],
|
|
) -> FilesystemService:
|
|
"""Return sda1 filesystem service."""
|
|
return udisks2_services["udisks2_filesystem"][
|
|
"/org/freedesktop/UDisks2/block_devices/sda1"
|
|
]
|
|
|
|
|
|
@pytest.fixture(name="udisks2_service")
|
|
async def fixture_udisks2_service(
|
|
udisks2_services: dict[str, DBusServiceMock | dict[str, DBusServiceMock]],
|
|
) -> UDisks2ManagerService:
|
|
"""Return udisks2 manager service."""
|
|
return udisks2_services["udisks2_manager"]
|
|
|
|
|
|
async def test_fixup(coresys: CoreSys, sda1_filesystem_service: FilesystemService):
|
|
"""Test fixup."""
|
|
sda1_filesystem_service.SetLabel.calls.clear()
|
|
system_rename_data_disk = FixupSystemRenameDataDisk(coresys)
|
|
|
|
assert not system_rename_data_disk.auto
|
|
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.RENAME_DATA_DISK, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
coresys.resolution.issues = Issue(
|
|
IssueType.MULTIPLE_DATA_DISKS, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
|
|
await system_rename_data_disk()
|
|
|
|
assert sda1_filesystem_service.SetLabel.calls == [
|
|
(
|
|
"/org/freedesktop/UDisks2/block_devices/sda1",
|
|
"hassos-data-old",
|
|
{"auth.no_user_interaction": Variant("b", True)},
|
|
)
|
|
]
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
assert len(coresys.resolution.issues) == 0
|
|
|
|
|
|
async def test_fixup_device_removed(
|
|
coresys: CoreSys,
|
|
udisks2_service: UDisks2ManagerService,
|
|
caplog: pytest.LogCaptureFixture,
|
|
):
|
|
"""Test fixup when device removed."""
|
|
system_rename_data_disk = FixupSystemRenameDataDisk(coresys)
|
|
|
|
assert not system_rename_data_disk.auto
|
|
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.RENAME_DATA_DISK, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
coresys.resolution.issues = Issue(
|
|
IssueType.MULTIPLE_DATA_DISKS, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
|
|
udisks2_service.resolved_devices = []
|
|
await system_rename_data_disk()
|
|
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
assert len(coresys.resolution.issues) == 0
|
|
assert "Data disk at /dev/sda1 with name conflict was removed" in caplog.text
|
|
|
|
|
|
async def test_fixup_device_not_filesystem(
|
|
coresys: CoreSys,
|
|
udisks2_service: UDisks2ManagerService,
|
|
caplog: pytest.LogCaptureFixture,
|
|
):
|
|
"""Test fixup when device is no longer a filesystem."""
|
|
system_rename_data_disk = FixupSystemRenameDataDisk(coresys)
|
|
|
|
assert not system_rename_data_disk.auto
|
|
|
|
coresys.resolution.suggestions = Suggestion(
|
|
SuggestionType.RENAME_DATA_DISK, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
coresys.resolution.issues = Issue(
|
|
IssueType.MULTIPLE_DATA_DISKS, ContextType.SYSTEM, reference="/dev/sda1"
|
|
)
|
|
|
|
udisks2_service.resolved_devices = ["/org/freedesktop/UDisks2/block_devices/sda"]
|
|
await system_rename_data_disk()
|
|
|
|
assert len(coresys.resolution.suggestions) == 0
|
|
assert len(coresys.resolution.issues) == 0
|
|
assert "Data disk at /dev/sda1 no longer appears to be a filesystem" in caplog.text
|