Files
supervisor/tests/dbus_service_mocks/agent_system.py
Jan Čermák cb16a34401 Remove WipeDevice method from OS Agent DBus mock (#5744)
WipeDevice method was dropped from OS Agent code in [1]. Remove it from
the mock class to sync with the current API. There is no usage of
WipeDevice in the Supervisor codebase, only ScheduleWipeDevice is
called.

[1] https://github.com/home-assistant/os-agent/pull/225
2025-03-24 15:09:01 +01:00

31 lines
853 B
Python

"""Mock of OS Agent System dbus service."""
from dbus_fast import DBusError
from .base import DBusServiceMock, dbus_method
BUS_NAME = "io.hass.os"
def setup(object_path: str | None = None) -> DBusServiceMock:
"""Create dbus mock object."""
return System()
class System(DBusServiceMock):
"""System mock.
gdbus introspect --system --dest io.hass.os --object-path /io/hass/os/System
"""
object_path = "/io/hass/os/System"
interface = "io.hass.os.System"
response_schedule_wipe_device: bool | DBusError = True
@dbus_method()
def ScheduleWipeDevice(self) -> "b":
"""Schedule wipe device."""
if isinstance(self.response_schedule_wipe_device, DBusError):
raise self.response_schedule_wipe_device # pylint: disable=raising-bad-type
return self.response_schedule_wipe_device