supervisor/tests/dbus_service_mocks/agent_datadisk.py
Mike Degatano a3204f4ebd
Use Udisks2 for available data disks (#4202)
* Use Udisks2 for available data disks

* pylint issues
2023-03-27 12:09:16 -04:00

41 lines
931 B
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