supervisor/tests/dbus_service_mocks/agent_datadisk.py
Mike Degatano c0b75edfb7
Format data disk in Supervisor instead of OS Agent ()
* Supervisor formats data disk instead of os agent

* Fix issues occurring during tests

* Can't migrate if target is too small
2023-03-30 14:15:07 -04:00

45 lines
1.0 KiB
Python

"""Mock of OS Agent DataDisk dbus service."""
from dbus_fast.service import PropertyAccess, dbus_property
from .base import DBusServiceMock, dbus_method
BUS_NAME = "io.hass.os"
def setup(object_path: str | None = None) -> DBusServiceMock:
"""Create dbus mock object."""
return DataDisk()
# pylint: disable=invalid-name
class DataDisk(DBusServiceMock):
"""DataDisk mock.
gdbus introspect --system --dest io.hass.os --object-path /io/hass/os/DataDisk
"""
object_path = "/io/hass/os/DataDisk"
interface = "io.hass.os.DataDisk"
@dbus_property(access=PropertyAccess.READ)
def CurrentDevice(self) -> "s":
"""Get Current Device."""
return "/dev/mmcblk1"
@dbus_method()
def ChangeDevice(self, arg_0: "s") -> "b":
"""Change device."""
return True
@dbus_method()
def ReloadDevice(self) -> "b":
"""Reload device."""
return True
@dbus_method()
def MarkDataMove(self) -> None:
"""Do MarkDataMove method."""