"""upowerd mock template
This creates the expected methods and properties of the main
org.freedesktop.UPower object, but no devices. You can specify any property
such as 'OnLowBattery' or the return value of 'SuspendAllowed',
'HibernateAllowed', and 'GetCriticalAction' in "parameters".
This provides the 1.0 D-Bus API of upower.
"""
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License as published by the Free
# Software Foundation; either version 3 of the License, or (at your option) any
# later version. See http://www.gnu.org/copyleft/lgpl.html for the full text
# of the license.
[docs]
__author__ = "Martin Pitt"
[docs]
__copyright__ = """
(c) 2012, 2013 Canonical Ltd.
(c) 2017 - 2022 Martin Pitt <martin@piware.de>
"""
import dbus
import dbusmock
from dbusmock import MOCK_IFACE, mockobject
[docs]
BUS_NAME = "org.freedesktop.UPower"
[docs]
MAIN_OBJ = "/org/freedesktop/UPower"
[docs]
MAIN_IFACE = "org.freedesktop.UPower"
[docs]
DEVICE_IFACE = "org.freedesktop.UPower.Device"
[docs]
def load(mock, parameters):
mock.AddMethods(
MAIN_IFACE,
[
(
"EnumerateDevices",
"",
"ao",
'ret = [k for k in objects.keys() if "/devices" in k and not k.endswith("/DisplayDevice")]',
),
],
)
props = dbus.Dictionary(
{
"DaemonVersion": parameters.get("DaemonVersion", "0.99"),
"OnBattery": parameters.get("OnBattery", False),
"LidIsPresent": parameters.get("LidIsPresent", True),
"LidIsClosed": parameters.get("LidIsClosed", False),
"LidForceSleep": parameters.get("LidForceSleep", True),
"IsDocked": parameters.get("IsDocked", False),
},
signature="sv",
)
mock.AddMethods(
MAIN_IFACE,
[
("GetCriticalAction", "", "s", f'ret = "{parameters.get("GetCriticalAction", "HybridSleep")}"'),
("GetDisplayDevice", "", "o", 'ret = "/org/freedesktop/UPower/devices/DisplayDevice"'),
],
)
mock.p_display_dev = "/org/freedesktop/UPower/devices/DisplayDevice"
# add Display device; for defined properties, see
# http://cgit.freedesktop.org/upower/tree/src/org.freedesktop.UPower.xml
mock.AddObject(
mock.p_display_dev,
DEVICE_IFACE,
{
"Type": dbus.UInt32(0),
"State": dbus.UInt32(0),
"Percentage": dbus.Double(0.0),
"Energy": dbus.Double(0.0),
"EnergyFull": dbus.Double(0.0),
"EnergyRate": dbus.Double(0.0),
"TimeToEmpty": dbus.Int64(0),
"TimeToFull": dbus.Int64(0),
"IsPresent": dbus.Boolean(False),
"IconName": dbus.String(""),
# LEVEL_NONE
"WarningLevel": dbus.UInt32(1),
},
[
("Refresh", "", "", ""),
],
)
mock.AddProperties(MAIN_IFACE, props)
@dbus.service.method(MOCK_IFACE, in_signature="ss", out_signature="s")
[docs]
def AddAC(self, device_name, model_name):
"""Convenience method to add an AC object
You have to specify a device name which must be a valid part of an object
path, e. g. "mock_ac", and an arbitrary model name.
Please note that this does not set any global properties such as
"on-battery".
Returns the new object path.
"""
path = "/org/freedesktop/UPower/devices/" + device_name
self.AddObject(
path,
DEVICE_IFACE,
{
"PowerSupply": dbus.Boolean(True),
"Model": dbus.String(model_name),
"Online": dbus.Boolean(True),
},
[],
)
self.EmitSignal(MAIN_IFACE, "DeviceAdded", "o", [path])
return path
@dbus.service.method(MOCK_IFACE, in_signature="ssdx", out_signature="s")
[docs]
def AddDischargingBattery(self, device_name, model_name, percentage, seconds_to_empty):
"""Convenience method to add a discharging battery object
You have to specify a device name which must be a valid part of an object
path, e. g. "mock_ac", an arbitrary model name, the charge percentage, and
the seconds until the battery is empty.
Please note that this does not set any global properties such as
"on-battery".
Returns the new object path.
"""
path = "/org/freedesktop/UPower/devices/" + device_name
self.AddObject(
path,
DEVICE_IFACE,
{
"PowerSupply": dbus.Boolean(True),
"IsPresent": dbus.Boolean(True),
"Model": dbus.String(model_name),
"Percentage": dbus.Double(percentage),
"TimeToEmpty": dbus.Int64(seconds_to_empty),
"EnergyFull": dbus.Double(100.0),
"Energy": dbus.Double(percentage),
# UP_DEVICE_STATE_DISCHARGING
"State": dbus.UInt32(2),
# UP_DEVICE_KIND_BATTERY
"Type": dbus.UInt32(2),
},
[],
)
self.EmitSignal(MAIN_IFACE, "DeviceAdded", "o", [path])
return path
@dbus.service.method(MOCK_IFACE, in_signature="ssdx", out_signature="s")
[docs]
def AddChargingBattery(self, device_name, model_name, percentage, seconds_to_full):
"""Convenience method to add a charging battery object
You have to specify a device name which must be a valid part of an object
path, e. g. "mock_ac", an arbitrary model name, the charge percentage, and
the seconds until the battery is full.
Please note that this does not set any global properties such as
"on-battery".
Returns the new object path.
"""
path = "/org/freedesktop/UPower/devices/" + device_name
self.AddObject(
path,
DEVICE_IFACE,
{
"PowerSupply": dbus.Boolean(True),
"IsPresent": dbus.Boolean(True),
"Model": dbus.String(model_name),
"Percentage": dbus.Double(percentage),
"TimeToFull": dbus.Int64(seconds_to_full),
"EnergyFull": dbus.Double(100.0),
"Energy": dbus.Double(percentage),
# UP_DEVICE_STATE_CHARGING
"State": dbus.UInt32(1),
# UP_DEVICE_KIND_BATTERY
"Type": dbus.UInt32(2),
},
[],
)
self.EmitSignal(MAIN_IFACE, "DeviceAdded", "o", [path])
return path
@dbus.service.method(MOCK_IFACE, in_signature="uuddddxxbsu", out_signature="")
[docs]
def SetupDisplayDevice(
self,
_type,
state,
percentage,
energy,
energy_full,
energy_rate,
time_to_empty,
time_to_full,
is_present,
icon_name,
warning_level,
):
"""Convenience method to configure DisplayDevice properties
This calls Set() for all properties that the DisplayDevice is defined to
have, and is shorter if you have to completely set it up instead of
changing just one or two properties.
"""
display_props = mockobject.objects[self.p_display_dev]
display_props.Set(DEVICE_IFACE, "Type", dbus.UInt32(_type))
display_props.Set(DEVICE_IFACE, "State", dbus.UInt32(state))
display_props.Set(DEVICE_IFACE, "Percentage", percentage)
display_props.Set(DEVICE_IFACE, "Energy", energy)
display_props.Set(DEVICE_IFACE, "EnergyFull", energy_full)
display_props.Set(DEVICE_IFACE, "EnergyRate", energy_rate)
display_props.Set(DEVICE_IFACE, "TimeToEmpty", dbus.Int64(time_to_empty))
display_props.Set(DEVICE_IFACE, "TimeToFull", dbus.Int64(time_to_full))
display_props.Set(DEVICE_IFACE, "IsPresent", is_present)
display_props.Set(DEVICE_IFACE, "IconName", icon_name)
display_props.Set(DEVICE_IFACE, "WarningLevel", dbus.UInt32(warning_level))
@dbus.service.method(MOCK_IFACE, in_signature="oa{sv}", out_signature="")
[docs]
def SetDeviceProperties(_self, object_path, properties):
"""Convenience method to Set a device's properties.
object_path: the device to update
properties: dictionary of keys to dbus variants.
Changing this property will trigger the device's PropertiesChanged signal.
"""
device = dbusmock.get_object(object_path)
# set the properties
for key, value in properties.items():
device.Set(DEVICE_IFACE, key, value)
@dbus.service.method(MOCK_IFACE, in_signature="o", out_signature="")
[docs]
def RemoveDevice(self, device_path):
self.RemoveObject(device_path)
self.EmitSignal(MAIN_IFACE, "DeviceRemoved", "o", [device_path])