1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
# Copyright © 2018 Red Hat, Inc
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, see <http://www.gnu.org/licenses/>.
#
# Authors: Benjamin Berg <bberg@redhat.com>
import os
import sys
import subprocess
from dbusmock import DBusTestCase
# Intended to be shared across projects, submitted for inclusion into
# dbusmock but might need to live elsewhere
# The pull request contains python 2 compatibility code.
# https://github.com/martinpitt/python-dbusmock/pull/44
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 = subprocess.DEVNULL
@classmethod
def setUpClass(klass):
klass.start_xorg()
klass.start_system_bus()
klass.start_session_bus()
@classmethod
def start_xorg(klass):
r, w = os.pipe()
# Xvfb seems to randomly crash in some workloads if "-noreset" is not given.
# https://bugzilla.redhat.com/show_bug.cgi?id=1565847
klass.xorg = subprocess.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
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()
klass.xorg.wait()
del klass.xorg
@classmethod
def tearDownClass(klass):
DBusTestCase.tearDownClass()
klass.stop_xorg()
|
