logind.py

"""Mock of logind dbus service."""

import os
import tempfile

from dbus_fast import DBusError
from dbus_fast.service import signal

from .base import DBusServiceMock, dbus_method

BUS_NAME = "org.freedesktop.login1"


def setup(object_path: str | None = None) -> DBusServiceMock:
    """Create dbus mock object."""
    return Logind()


class Logind(DBusServiceMock):
    """Logind mock.

    gdbus introspect --system --dest org.freedesktop.login1 --object-path /org/freedesktop/login1
    """

    object_path = "/org/freedesktop/login1"
    interface = "org.freedesktop.login1.Manager"
    side_effect_reboot: DBusError | None = None

    @dbus_method()
    def Reboot(self, interactive: "b") -> None:
        """Reboot."""
        if self.side_effect_reboot:
            raise self.side_effect_reboot  # pylint: disable=raising-bad-type

    @dbus_method()
    def PowerOff(self, interactive: "b") -> None:
        """PowerOff."""

    @dbus_method()
    def Inhibit(self, what: "s", who: "s", why: "s", mode: "s") -> "h":
        """Take an inhibitor lock. Returns a file descriptor."""
        fd, path = tempfile.mkstemp()
        os.unlink(path)
        return fd

    @signal()
    def PrepareForShutdown(self) -> "b":
        """Signal prepare for shutdown."""
        return True