glib/gio/tests/memory-monitor-dbus.py.in

#!/usr/bin/python3

# 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.

__author__ = 'Bastien Nocera'
__email__ = 'hadess@hadess.net'
__copyright__ = '(c) 2019 Red Hat Inc.'
__license__ = 'LGPL 3+'

import unittest
import sys
import subprocess
import dbus
import dbus.mainloop.glib
import dbusmock
import fcntl
import os
import time

from gi.repository import GLib
from gi.repository import Gio

dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)

# XDG_DESKTOP_PORTAL_PATH = os.path.expanduser("~/.cache/jhbuild/build/xdg-desktop-portal/xdg-desktop-portal")
XDG_DESKTOP_PORTAL_PATH = "@libexecdir@/xdg-desktop-portal"

class TestLowMemoryMonitor(dbusmock.DBusTestCase):
    '''Test GMemoryMonitorDBus'''

    @classmethod
    def setUpClass(klass):
        klass.start_system_bus()
        klass.dbus_con = klass.get_dbus(True)

    def setUp(self):
        try:
            (self.p_mock, self.obj_lmm) = self.spawn_server_template(
                'low_memory_monitor', {}, stdout=subprocess.PIPE)
        except ModuleNotFoundError:
            raise unittest.SkipTest("Low memory monitor dbusmock template not "
                                    "found. Requires dbusmock ≥ 0.18.4.")
        # set log to nonblocking
        flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL)
        fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
        self.last_warning = -1
        self.dbusmock = dbus.Interface(self.obj_lmm, dbusmock.MOCK_IFACE)
        self.memory_monitor = Gio.MemoryMonitor.dup_default()
        self.memory_monitor.connect("low-memory-warning", self.memory_warning_cb)
        self.mainloop = GLib.MainLoop()
        self.main_context = self.mainloop.get_context()

    def tearDown(self):
        self.p_mock.terminate()
        self.p_mock.wait()

    def memory_warning_cb(self, monitor, level):
        self.last_warning = level
        self.main_context.wakeup()

    def test_low_memory_warning_signal(self):
        '''LowMemoryWarning signal'''

        # Wait 2 seconds
        timeout = 2
        while timeout > 0:
            time.sleep(0.5)
            timeout -= 0.5
            self.main_context.iteration(False)

        self.dbusmock.EmitWarning(100)
        # Wait 2 seconds or until warning
        timeout = 2
        while timeout > 0 and self.last_warning != 100:
            time.sleep(0.5)
            timeout -= 0.5
            self.main_context.iteration(False)
        self.assertEqual(self.last_warning, 100)

        self.dbusmock.EmitWarning(255)
        # Wait 2 seconds or until warning
        timeout = 2
        while timeout > 0 and self.last_warning != 255:
            time.sleep(0.5)
            timeout -= 0.5
            self.main_context.iteration(False)
        self.assertEqual(self.last_warning, 255)


if __name__ == '__main__':
    # avoid writing to stderr
    unittest.main(testRunner=unittest.TextTestRunner(stream=sys.stdout, verbosity=2))