mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-04-24 21:27:15 +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.
121 lines
3.5 KiB
Python
121 lines
3.5 KiB
Python
"""Test rauc dbus interface."""
|
|
|
|
# pylint: disable=import-error
|
|
from dbus_fast.aio.message_bus import MessageBus
|
|
import pytest
|
|
|
|
from supervisor.dbus.const import RaucState
|
|
from supervisor.dbus.rauc import Rauc
|
|
from supervisor.exceptions import DBusNotConnectedError
|
|
|
|
from tests.common import mock_dbus_services
|
|
from tests.dbus_service_mocks.rauc import Rauc as RaucService
|
|
|
|
|
|
@pytest.fixture(name="rauc_service")
|
|
async def fixture_rauc_service(dbus_session_bus: MessageBus) -> RaucService:
|
|
"""Mock rauc dbus service."""
|
|
yield (await mock_dbus_services({"rauc": None}, dbus_session_bus))["rauc"]
|
|
|
|
|
|
async def test_rauc_info(rauc_service: RaucService, dbus_session_bus: MessageBus):
|
|
"""Test rauc properties."""
|
|
rauc = Rauc()
|
|
|
|
assert rauc.boot_slot is None
|
|
assert rauc.operation is None
|
|
assert rauc.last_error is None
|
|
|
|
await rauc.connect(dbus_session_bus)
|
|
|
|
assert rauc.boot_slot == "B"
|
|
assert rauc.operation == "idle"
|
|
assert rauc.last_error == ""
|
|
|
|
rauc_service.emit_properties_changed({"LastError": "Error!"})
|
|
await rauc_service.ping()
|
|
assert rauc.last_error == "Error!"
|
|
|
|
rauc_service.emit_properties_changed({}, ["LastError"])
|
|
await rauc_service.ping()
|
|
await rauc_service.ping() # To process the follow-up get all properties call
|
|
assert rauc.last_error == ""
|
|
|
|
|
|
async def test_install(rauc_service: RaucService, dbus_session_bus: MessageBus):
|
|
"""Test install."""
|
|
rauc = Rauc()
|
|
|
|
with pytest.raises(DBusNotConnectedError):
|
|
await rauc.install("rauc_file")
|
|
|
|
await rauc.connect(dbus_session_bus)
|
|
|
|
async with rauc.signal_completed() as signal:
|
|
assert await rauc.install("rauc_file") is None
|
|
assert await signal.wait_for_signal() == [0]
|
|
|
|
|
|
async def test_get_slot_status(rauc_service: RaucService, dbus_session_bus: MessageBus):
|
|
"""Test get slot status."""
|
|
rauc = Rauc()
|
|
|
|
with pytest.raises(DBusNotConnectedError):
|
|
await rauc.get_slot_status()
|
|
|
|
await rauc.connect(dbus_session_bus)
|
|
|
|
slot_status = await rauc.get_slot_status()
|
|
assert len(slot_status) == 6
|
|
assert slot_status[0][0] == "kernel.0"
|
|
assert slot_status[0][1]["boot-status"] == "good"
|
|
assert slot_status[0][1]["device"] == "/dev/disk/by-partlabel/hassos-kernel0"
|
|
assert slot_status[0][1]["bootname"] == "A"
|
|
assert slot_status[4][0] == "kernel.1"
|
|
assert slot_status[4][1]["boot-status"] == "good"
|
|
assert slot_status[4][1]["device"] == "/dev/disk/by-partlabel/hassos-kernel1"
|
|
assert slot_status[4][1]["bootname"] == "B"
|
|
|
|
|
|
async def test_mark(rauc_service: RaucService, dbus_session_bus: MessageBus):
|
|
"""Test mark."""
|
|
rauc = Rauc()
|
|
|
|
with pytest.raises(DBusNotConnectedError):
|
|
await rauc.mark(RaucState.GOOD, "booted")
|
|
|
|
await rauc.connect(dbus_session_bus)
|
|
|
|
mark = await rauc.mark(RaucState.GOOD, "booted")
|
|
assert mark[0] == "kernel.1"
|
|
assert mark[1] == "marked slot kernel.1 as good"
|
|
|
|
|
|
async def test_dbus_rauc_connect_error(
|
|
dbus_session_bus: MessageBus, caplog: pytest.LogCaptureFixture
|
|
):
|
|
"""Test connecting to rauc error."""
|
|
rauc = Rauc()
|
|
await rauc.connect(dbus_session_bus)
|
|
assert "Host has no rauc support" in caplog.text
|
|
|
|
|
|
async def test_test_slot_status(
|
|
rauc_service: RaucService, dbus_session_bus: MessageBus
|
|
):
|
|
"""Test get slot status."""
|
|
rauc = Rauc()
|
|
|
|
await rauc.connect(dbus_session_bus)
|
|
|
|
slot_status = await rauc.get_slot_status()
|
|
out = {}
|
|
for slot in slot_status:
|
|
for k in slot[1]:
|
|
if k in out:
|
|
out[k] += 1
|
|
else:
|
|
out[k] = 1
|
|
|
|
assert out
|