python-xlib/runtests.py at master · python-xlib/python-xlib · GitHub

History

executable file

134 lines (112 loc) · 3.54 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

#!/usr/bin/env python

# Python 2/3 compatibility.

from __future__ import print_function

import os

import signal

import subprocess

import sys

import tempfile

import textwrap

from pkg_resources import load_entry_point

class SigException(BaseException):

def __init__(self, signum):

super(SigException, self).__init__()

self.signum = signum

def xsession_sighandler(signum, frame):

raise SigException(signum)

def xserver_start(display, executable='Xvfb', authfile=None):

pid = os.fork()

if pid != 0:

return pid

if authfile is None:

authfile = os.devnull

# This will make the xserver send us a SIGUSR1 when ready.

signal.signal(signal.SIGUSR1, signal.SIG_IGN)

cmd = [

executable,

'-auth', authfile,

'-noreset',

display,

]

print('starting xserver: `{0}`'.format(' '.join(cmd)))

os.execlp(cmd[0], *cmd)

def tests_run(display, authfile=None):

pid = os.fork()

if pid != 0:

return pid

if authfile is None:

authfile = os.devnull

os.environ['DISPLAY'] = display

os.environ['XAUTHORITY'] = authfile

cmd = [

sys.executable,

'-c', textwrap.dedent(

'''

from pkg_resources import load_entry_point

sys.exit(load_entry_point(

'nose', 'console_scripts', 'nosetests',

)())

'''

).lstrip(),

'--exe', '--with-xunit', '--verbosity=3',

]

has_custom_tests = False

for arg in sys.argv[1:]:

if not arg.startswith('-'):

has_custom_tests = True

cmd.append(arg)

if not has_custom_tests:

cmd.extend(('test/', 'examples/run_examples.py'))

print('running tests: `{0}`'.format(' '.join(cmd)))

sys.argv = cmd

try:

load_entry_point('nose', 'console_scripts', 'nosetests')()

except SystemExit as err:

code = err.code

else:

code = 0

os._exit(code)

def runtests():

cleanup_funcs = []

try:

if hasattr(sys, 'pypy_version_info'):

server_display = ':8'

else:

server_display = ':9'

server_display += ''.join(str(n) for n in sys.version_info[:3])

# Setup a temporary authentication file.

cookie = subprocess.check_output('mcookie').strip()

authfile = tempfile.NamedTemporaryFile(delete=False)

cleanup_funcs.append(lambda: os.unlink(authfile.name))

authfile.close()

subprocess.check_call((

'xauth',

'-f', authfile.name,

'add', server_display, '.', cookie,

))

# Setup signal handler to wait for xserver to be ready.

signal.signal(signal.SIGUSR1, xsession_sighandler)

# Start xserver.

server_pid = xserver_start(server_display, authfile=authfile.name)

cleanup_funcs.append(lambda: os.waitpid(server_pid, 0))

cleanup_funcs.append(lambda: os.kill(server_pid, signal.SIGTERM))

# Give the server 3 seconds to start.

signal.alarm(3)

# Wait for server to be ready.

try:

signal.pause()

except SigException as err:

assert signal.SIGUSR1 == err.signum

signal.alarm(0)

# Run tests.

tests_pid = tests_run(server_display, authfile=authfile.name)

pid, status = os.waitpid(tests_pid, 0)

assert pid == tests_pid

sys.exit(status >> 8)

except KeyboardInterrupt:

sys.exit(1)

finally:

for func in reversed(cleanup_funcs):

func()

if __name__ == '__main__':

runtests()

python-xlib/runtests.py at master · python-xlib/python-xlib · GitHub

Uh oh!

Expand file tree

Breadcrumbs

runtests.py

Latest commit

History

Breadcrumbs

runtests.py

File metadata and controls