summaryrefslogtreecommitdiffstats
path: root/tests/x11session.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--tests/x11session.py120
1 files changed, 120 insertions, 0 deletions
diff --git a/tests/x11session.py b/tests/x11session.py
new file mode 100644
index 0000000..5678073
--- /dev/null
+++ b/tests/x11session.py
@@ -0,0 +1,120 @@
+import os
+import sys
+import subprocess
+from dbusmock import DBusTestCase
+
+# This file has been submitted for inclusion in python-dbusmock
+# https://github.com/martinpitt/python-dbusmock/pull/44
+
+if sys.version_info[0] < 3:
+ PIPE_DEVNULL = open(os.devnull, 'wb')
+
+ def Popen(*args, **kwargs):
+ if 'pass_fds' in kwargs:
+ pass_fds = kwargs['pass_fds']
+ del kwargs['pass_fds']
+ else:
+ pass_fds = []
+
+ orig_preexec_fn = kwargs.get('preexec_fn', None)
+
+ def _setup():
+ for fd in range(3, subprocess.MAXFD):
+ if fd in pass_fds:
+ continue
+
+ try:
+ os.close(fd)
+ except OSError:
+ pass
+
+ if orig_preexec_fn:
+ orig_preexec_fn()
+
+ # Don't let subprocess close FDs for us
+ kwargs['close_fds'] = False
+ kwargs['preexec_fn'] = _setup
+
+ return subprocess.Popen(*args, **kwargs)
+
+else:
+ PIPE_DEVNULL = subprocess.DEVNULL
+ Popen = subprocess.Popen
+
+
+class X11SessionTestCase(DBusTestCase):
+ #: The display the X server is running on
+ X_display = -1
+ #: The X server to start
+ Xserver = 'Xvfb'
+ #: Further parameters for the X server
+ Xserver_args = ['-screen', '0', '1280x1024x24', '+extension', 'GLX']
+ #: Where to redirect the X stdout and stderr to. Set to None for debugging
+ #: purposes if the X server is failing for some reason.
+ Xserver_output = PIPE_DEVNULL
+
+ @classmethod
+ def setUpClass(klass):
+ klass.start_xorg()
+ klass.addClassCleanup(klass.stop_xorg)
+
+ klass.start_system_bus()
+ klass.start_session_bus()
+
+ @classmethod
+ def start_xorg(klass):
+ r, w = os.pipe()
+
+ klass.xorg = Popen([klass.Xserver, '-displayfd', "%d" % w, '-noreset'] + klass.Xserver_args,
+ pass_fds=(w,),
+ stdout=klass.Xserver_output,
+ stderr=subprocess.STDOUT)
+ os.close(w)
+
+ # The X server will write "%d\n", we need to make sure to read the "\n".
+ # If the server dies we get zero bytes reads as EOF is reached.
+ display = b''
+ while True:
+ tmp = os.read(r, 1024)
+ display += tmp
+
+ # Break out if the read was empty or the line feed was read
+ if not tmp or tmp[-1] == b'\n':
+ break
+
+ os.close(r)
+
+ try:
+ display = int(display.strip())
+ except ValueError:
+ # This should never happen, the X server didn't return a proper integer.
+ # Most likely it died for some odd reason.
+ # Note: Set Xserver_output to None to debug Xvfb startup issues.
+ klass.stop_xorg()
+ raise AssertionError('X server reported back no or an invalid display number (%s)' % (display))
+
+ klass.X_display = display
+ # Export information into our environment for tests to use
+ os.environ['DISPLAY'] = ":%d" % klass.X_display
+ if 'GNOME_SETUP_DISPLAY' in os.environ:
+ del os.environ['GNOME_SETUP_DISPLAY']
+ if 'WAYLAND_DISPLAY' in os.environ:
+ del os.environ['WAYLAND_DISPLAY']
+
+ # Server should still be up and running at this point
+ assert klass.xorg.poll() is None
+
+ return klass.X_display
+
+ @classmethod
+ def stop_xorg(klass):
+ if hasattr(klass, 'xorg'):
+ klass.X_display = -1
+ klass.xorg.terminate()
+ try:
+ klass.xorg.wait(1)
+ except:
+ klass.xorg.kill()
+ klass.xorg.wait()
+ del klass.xorg
+