gio/tests/power-profile-monitor-*.py: Fix style using black

This commit is contained in:
Marco Trevisan (Treviño) 2025-02-07 00:42:14 +01:00
parent 236abac6fb
commit 71d08b707a
2 changed files with 119 additions and 54 deletions

View File

@ -6,10 +6,10 @@
# later version. See http://www.gnu.org/copyleft/lgpl.html for the full text # later version. See http://www.gnu.org/copyleft/lgpl.html for the full text
# of the license. # of the license.
__author__ = 'Bastien Nocera' __author__ = "Bastien Nocera"
__email__ = 'hadess@hadess.net' __email__ = "hadess@hadess.net"
__copyright__ = '(c) 2019, 2021 Red Hat Inc.' __copyright__ = "(c) 2019, 2021 Red Hat Inc."
__license__ = 'LGPL 3+' __license__ = "LGPL 3+"
import unittest import unittest
import sys import sys
@ -31,7 +31,7 @@ try:
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
class TestPowerProfileMonitor(dbusmock.DBusTestCase): class TestPowerProfileMonitor(dbusmock.DBusTestCase):
'''Test GPowerProfileMonitorDBus''' """Test GPowerProfileMonitorDBus"""
@classmethod @classmethod
def setUpClass(klass): def setUpClass(klass):
@ -42,23 +42,30 @@ try:
try: try:
Gio.PowerProfileMonitor Gio.PowerProfileMonitor
except AttributeError: except AttributeError:
raise unittest.SkipTest('Power Profile Monitor not in ' raise unittest.SkipTest(
'introspection data. Requires ' "Power Profile Monitor not in "
'GObject-Introspection ≥ 1.63.2') # FIXME version "introspection data. Requires "
"GObject-Introspection ≥ 1.63.2"
) # FIXME version
try: try:
(self.p_mock, self.obj_ppd) = self.spawn_server_template( (self.p_mock, self.obj_ppd) = self.spawn_server_template(
'power_profiles_daemon', {}, stdout=subprocess.PIPE) "power_profiles_daemon", {}, stdout=subprocess.PIPE
)
except ModuleNotFoundError: except ModuleNotFoundError:
raise unittest.SkipTest("power-profiles-daemon dbusmock template not " raise unittest.SkipTest(
"found. Requires dbusmock > 0.23.1.") # FIXME version "power-profiles-daemon dbusmock template not "
"found. Requires dbusmock > 0.23.1."
) # FIXME version
# set log to nonblocking # set log to nonblocking
flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL) flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL)
fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK) fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
self.power_saver_enabled = False self.power_saver_enabled = False
self.dbus_props = dbus.Interface(self.obj_ppd, dbus.PROPERTIES_IFACE) self.dbus_props = dbus.Interface(self.obj_ppd, dbus.PROPERTIES_IFACE)
self.power_profile_monitor = Gio.PowerProfileMonitor.dup_default() self.power_profile_monitor = Gio.PowerProfileMonitor.dup_default()
assert("GPowerProfileMonitorDBus" in str(self.power_profile_monitor)) assert "GPowerProfileMonitorDBus" in str(self.power_profile_monitor)
self.power_profile_monitor.connect("notify::power-saver-enabled", self.power_saver_enabled_cb) self.power_profile_monitor.connect(
"notify::power-saver-enabled", self.power_saver_enabled_cb
)
self.mainloop = GLib.MainLoop() self.mainloop = GLib.MainLoop()
self.main_context = self.mainloop.get_context() self.main_context = self.mainloop.get_context()
@ -67,13 +74,13 @@ try:
self.p_mock.wait() self.p_mock.wait()
def assertEventually(self, condition, message=None, timeout=5): def assertEventually(self, condition, message=None, timeout=5):
'''Assert that condition function eventually returns True. """Assert that condition function eventually returns True.
Timeout is in seconds, defaulting to 5 seconds. message is Timeout is in seconds, defaulting to 5 seconds. message is
printed on failure. printed on failure.
''' """
if not message: if not message:
message = 'timed out waiting for ' + str(condition) message = "timed out waiting for " + str(condition)
def timed_out_cb(message): def timed_out_cb(message):
self.fail(message) self.fail(message)
@ -89,24 +96,46 @@ try:
timeout_source.destroy() timeout_source.destroy()
def power_saver_enabled_cb(self, spec, data): def power_saver_enabled_cb(self, spec, data):
self.power_saver_enabled = self.power_profile_monitor.get_power_saver_enabled() self.power_saver_enabled = (
self.power_profile_monitor.get_power_saver_enabled()
)
self.main_context.wakeup() self.main_context.wakeup()
def test_power_profile_power_saver_enabled(self): def test_power_profile_power_saver_enabled(self):
'''power-saver-enabled property''' """power-saver-enabled property"""
self.assertEqual(self.power_profile_monitor.get_power_saver_enabled(), False) self.assertEqual(
self.dbus_props.Set('net.hadess.PowerProfiles', 'ActiveProfile', dbus.String('power-saver', variant_level=1)) self.power_profile_monitor.get_power_saver_enabled(), False
self.assertEventually(lambda: self.power_saver_enabled == True, "power-saver didn't become enabled", 1) )
self.dbus_props.Set(
"net.hadess.PowerProfiles",
"ActiveProfile",
dbus.String("power-saver", variant_level=1),
)
self.assertEventually(
lambda: self.power_saver_enabled == True,
"power-saver didn't become enabled",
1,
)
self.dbus_props.Set('net.hadess.PowerProfiles', 'ActiveProfile', dbus.String('balanced', variant_level=1)) self.dbus_props.Set(
self.assertEventually(lambda: self.power_saver_enabled == False, "power-saver didn't become disabled", 1) "net.hadess.PowerProfiles",
"ActiveProfile",
dbus.String("balanced", variant_level=1),
)
self.assertEventually(
lambda: self.power_saver_enabled == False,
"power-saver didn't become disabled",
1,
)
except ImportError as e: except ImportError as e:
@unittest.skip("Cannot import %s" % e.name) @unittest.skip("Cannot import %s" % e.name)
class TestPowerProfileMonitor(unittest.TestCase): class TestPowerProfileMonitor(unittest.TestCase):
def test_power_profile_power_saver_enabled(self): def test_power_profile_power_saver_enabled(self):
pass pass
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main(testRunner=taptestrunner.TAPTestRunner()) unittest.main(testRunner=taptestrunner.TAPTestRunner())

View File

@ -6,10 +6,10 @@
# later version. See http://www.gnu.org/copyleft/lgpl.html for the full text # later version. See http://www.gnu.org/copyleft/lgpl.html for the full text
# of the license. # of the license.
__author__ = 'Bastien Nocera' __author__ = "Bastien Nocera"
__email__ = 'hadess@hadess.net' __email__ = "hadess@hadess.net"
__copyright__ = '(c) 2021 Red Hat Inc.' __copyright__ = "(c) 2021 Red Hat Inc."
__license__ = 'LGPL 3+' __license__ = "LGPL 3+"
import unittest import unittest
import sys import sys
@ -35,7 +35,7 @@ try:
XDG_DESKTOP_PORTAL_PATH = "@libexecdir@/xdg-desktop-portal" XDG_DESKTOP_PORTAL_PATH = "@libexecdir@/xdg-desktop-portal"
class TestPowerProfileMonitorPortal(dbusmock.DBusTestCase): class TestPowerProfileMonitorPortal(dbusmock.DBusTestCase):
'''Test GPowerProfileMonitorPortal''' """Test GPowerProfileMonitorPortal"""
@classmethod @classmethod
def setUpClass(klass): def setUpClass(klass):
@ -48,15 +48,20 @@ try:
try: try:
Gio.PowerProfileMonitor Gio.PowerProfileMonitor
except AttributeError: except AttributeError:
raise unittest.SkipTest('Power Profile Monitor not in ' raise unittest.SkipTest(
'introspection data. Requires ' "Power Profile Monitor not in "
'GObject-Introspection > 1.69.0') "introspection data. Requires "
"GObject-Introspection > 1.69.0"
)
try: try:
(self.p_mock, self.obj_ppd) = self.spawn_server_template( (self.p_mock, self.obj_ppd) = self.spawn_server_template(
'power_profiles_daemon', {}, stdout=subprocess.PIPE) "power_profiles_daemon", {}, stdout=subprocess.PIPE
)
except ModuleNotFoundError: except ModuleNotFoundError:
raise unittest.SkipTest("power-profiles-daemon dbusmock template not " raise unittest.SkipTest(
"found. Requires dbusmock > 0.23.1.") "power-profiles-daemon dbusmock template not "
"found. Requires dbusmock > 0.23.1."
)
# set log to nonblocking # set log to nonblocking
flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL) flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL)
fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK) fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
@ -70,18 +75,21 @@ try:
raise unittest.SkipTest("xdg-desktop-portal not available") raise unittest.SkipTest("xdg-desktop-portal not available")
try: try:
self.wait_for_bus_object('org.freedesktop.portal.Desktop', self.wait_for_bus_object(
'/org/freedesktop/portal/desktop') "org.freedesktop.portal.Desktop", "/org/freedesktop/portal/desktop"
)
except: except:
self.p_mock.terminate() self.p_mock.terminate()
self.p_mock.wait() self.p_mock.wait()
raise raise
# subprocess.Popen(['gdbus', 'monitor', '--session', '--dest', 'org.freedesktop.portal.Desktop']) # subprocess.Popen(['gdbus', 'monitor', '--session', '--dest', 'org.freedesktop.portal.Desktop'])
os.environ['GIO_USE_PORTALS'] = "1" os.environ["GIO_USE_PORTALS"] = "1"
self.power_profile_monitor = Gio.PowerProfileMonitor.dup_default() self.power_profile_monitor = Gio.PowerProfileMonitor.dup_default()
assert("GPowerProfileMonitorPortal" in str(self.power_profile_monitor)) assert "GPowerProfileMonitorPortal" in str(self.power_profile_monitor)
self.power_profile_monitor.connect("notify::power-saver-enabled", self.power_saver_enabled_cb) self.power_profile_monitor.connect(
"notify::power-saver-enabled", self.power_saver_enabled_cb
)
self.mainloop = GLib.MainLoop() self.mainloop = GLib.MainLoop()
self.main_context = self.mainloop.get_context() self.main_context = self.mainloop.get_context()
@ -90,13 +98,13 @@ try:
self.p_mock.wait() self.p_mock.wait()
def assertEventually(self, condition, message=None, timeout=5): def assertEventually(self, condition, message=None, timeout=5):
'''Assert that condition function eventually returns True. """Assert that condition function eventually returns True.
Timeout is in seconds, defaulting to 5 seconds. message is Timeout is in seconds, defaulting to 5 seconds. message is
printed on failure. printed on failure.
''' """
if not message: if not message:
message = 'timed out waiting for ' + str(condition) message = "timed out waiting for " + str(condition)
def timed_out_cb(message): def timed_out_cb(message):
self.fail(message) self.fail(message)
@ -112,35 +120,63 @@ try:
timeout_source.destroy() timeout_source.destroy()
def power_saver_enabled_cb(self, spec, data): def power_saver_enabled_cb(self, spec, data):
self.power_saver_enabled = self.power_profile_monitor.get_power_saver_enabled() self.power_saver_enabled = (
self.power_profile_monitor.get_power_saver_enabled()
)
self.main_context.wakeup() self.main_context.wakeup()
def test_power_profile_power_saver_enabled_portal(self): def test_power_profile_power_saver_enabled_portal(self):
'''power-saver-enabled property''' """power-saver-enabled property"""
self.assertEqual(self.power_profile_monitor.get_power_saver_enabled(), False) self.assertEqual(
self.dbus_props.Set('net.hadess.PowerProfiles', 'ActiveProfile', dbus.String('power-saver', variant_level=1)) self.power_profile_monitor.get_power_saver_enabled(), False
self.assertEventually(lambda: self.power_saver_enabled == True, "power-saver didn't become enabled", 1) )
self.dbus_props.Set(
"net.hadess.PowerProfiles",
"ActiveProfile",
dbus.String("power-saver", variant_level=1),
)
self.assertEventually(
lambda: self.power_saver_enabled == True,
"power-saver didn't become enabled",
1,
)
self.dbus_props.Set('net.hadess.PowerProfiles', 'ActiveProfile', dbus.String('balanced', variant_level=1)) self.dbus_props.Set(
self.assertEventually(lambda: self.power_saver_enabled == False, "power-saver didn't become disabled", 1) "net.hadess.PowerProfiles",
"ActiveProfile",
dbus.String("balanced", variant_level=1),
)
self.assertEventually(
lambda: self.power_saver_enabled == False,
"power-saver didn't become disabled",
1,
)
def test_power_profile_power_saver_enabled_portal_default(self): def test_power_profile_power_saver_enabled_portal_default(self):
'''power-saver-enabled property default value''' """power-saver-enabled property default value"""
self.dbus_props.Set('net.hadess.PowerProfiles', 'ActiveProfile', dbus.String('power-saver', variant_level=1)) self.dbus_props.Set(
"net.hadess.PowerProfiles",
"ActiveProfile",
dbus.String("power-saver", variant_level=1),
)
# Create a new power profile monitor and check its property value is # Create a new power profile monitor and check its property value is
# correct by default. # correct by default.
new_power_profile_monitor = GObject.new(GObject.type_from_name('GPowerProfileMonitorPortal')) new_power_profile_monitor = GObject.new(
GObject.type_from_name("GPowerProfileMonitorPortal")
)
new_power_profile_monitor.init() new_power_profile_monitor.init()
self.assertTrue(new_power_profile_monitor.get_power_saver_enabled()) self.assertTrue(new_power_profile_monitor.get_power_saver_enabled())
except ImportError as e: except ImportError as e:
@unittest.skip("Cannot import %s" % e.name) @unittest.skip("Cannot import %s" % e.name)
class TestPowerProfileMonitorPortal(unittest.TestCase): class TestPowerProfileMonitorPortal(unittest.TestCase):
def test_power_profile_power_saver_enabled_portal(self): def test_power_profile_power_saver_enabled_portal(self):
pass pass
if __name__ == '__main__':
if __name__ == "__main__":
unittest.main(testRunner=taptestrunner.TAPTestRunner()) unittest.main(testRunner=taptestrunner.TAPTestRunner())