GeezerVison is showing github.com — python-xlib/runtests.py at master · OpenSourceJavaProject/python-xlib

Expand file tree

runtests.py

Copy path

More file actions

Latest commit

History

executable file

134 lines (112 loc) · 3.54 KB

runtests.py

Copy path

File metadata and controls

executable file

134 lines (112 loc) · 3.54 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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()