blob: 34b743e33f47611580aaceddd4e04eca01f6bad8 [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
Guido van Rossum152494a1996-12-20 03:12:20 +00002
R. David Murray0ba81e02010-04-26 17:02:32 +00003"""
4Usage:
Guido van Rossum152494a1996-12-20 03:12:20 +00005
R. David Murray0ba81e02010-04-26 17:02:32 +00006python -m test.regrtest [options] [test_name1 [test_name2 ...]]
7python path/to/Lib/test/regrtest.py [options] [test_name1 [test_name2 ...]]
Guido van Rossum152494a1996-12-20 03:12:20 +00008
R. David Murray0ba81e02010-04-26 17:02:32 +00009
10If no arguments or options are provided, finds all files matching
11the pattern "test_*" in the Lib/test subdirectory and runs
12them in alphabetical order (but see -M and -u, below, for exceptions).
13
14For more rigorous testing, it is useful to use the following
15command line:
16
17python -E -tt -Wd -3 -m test.regrtest [options] [test_name1 ...]
18
19
20Options:
Guido van Rossum152494a1996-12-20 03:12:20 +000021
Georg Brandl96f78972009-10-27 21:02:11 +000022-h/--help -- print this text and exit
23
24Verbosity
25
26-v/--verbose -- run tests in verbose mode with output to stdout
27-w/--verbose2 -- re-run failed tests in verbose mode
28-W/--verbose3 -- re-run failed tests in verbose mode immediately
29-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000030-q/--quiet -- no output unless one or more tests fail
Georg Brandl96f78972009-10-27 21:02:11 +000031-S/--slow -- print the slowest 10 tests
32
33Selecting tests
34
R. David Murray0ba81e02010-04-26 17:02:32 +000035-r/--random -- randomize test execution order (see below)
Georg Brandl96f78972009-10-27 21:02:11 +000036-f/--fromfile -- read names of tests to run from a file (see below)
37-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000038-s/--single -- single step through a set of tests (see below)
Georg Brandl96f78972009-10-27 21:02:11 +000039-u/--use RES1,RES2,...
40 -- specify which special resource intensive tests to run
41-M/--memlimit LIMIT
42 -- run very large memory-consuming tests
43
44Special runs
45
46-l/--findleaks -- if GC is available detect tests that leak memory
47-L/--runleaks -- run the leaks(1) command just before exit
48-R/--huntrleaks RUNCOUNTS
49 -- search for reference leaks (needs debug build, v. slow)
50-j/--multiprocess PROCESSES
51 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000052-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000053-D/--coverdir DIRECTORY
54 -- Directory where coverage files are put
55-N/--nocoverdir -- Put coverage files alongside modules
56-t/--threshold THRESHOLD
57 -- call gc.set_threshold(THRESHOLD)
58-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000059-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000060
R. David Murray0ba81e02010-04-26 17:02:32 +000061
62Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000063
Collin Winterfd12f492009-03-29 04:05:05 +000064-r randomizes test execution order. You can use --randseed=int to provide a
65int seed value for the randomizer; this is useful for reproducing troublesome
66test orders.
67
R. David Murray0ba81e02010-04-26 17:02:32 +000068-s On the first invocation of regrtest using -s, the first test file found
69or the first test file given on the command line is run, and the name of
70the next test is recorded in a file named pynexttest. If run from the
71Python build directory, pynexttest is located in the 'build' subdirectory,
72otherwise it is located in tempfile.gettempdir(). On subsequent runs,
73the test in pynexttest is run, and the next test is written to pynexttest.
74When the last test has been run, pynexttest is deleted. In this way it
75is possible to single step through the test files. This is useful when
76doing memory analysis on the Python interpreter, which process tends to
77consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000078
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000079-S is used to continue running tests after an aborted run. It will
80maintain the order a standard run (ie, this assumes -r is not used).
81This is useful after the tests have prematurely stopped for some external
82reason and you want to start running from where you left off rather
83than starting from the beginning.
84
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000085-f reads the names of tests from the file given as f's argument, one
86or more test names per line. Whitespace is ignored. Blank lines and
87lines beginning with '#' are ignored. This is especially useful for
88whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000089
Skip Montanaro0179a182004-06-06 15:53:18 +000090-L causes the leaks(1) command to be run just before exit if it exists.
91leaks(1) is available on Mac OS X and presumably on some other
92FreeBSD-derived systems.
93
Michael W. Hudson61147f62004-08-03 11:33:28 +000094-R runs each test several times and examines sys.gettotalrefcount() to
95see if the test appears to be leaking references. The argument should
96be of the form stab:run:fname where 'stab' is the number of times the
97test is run to let gettotalrefcount settle down, 'run' is the number
98of times further it is run and 'fname' is the name of the file the
99reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000100"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000101
Thomas Wouters477c8d52006-05-27 19:21:47 +0000102-M runs tests that require an exorbitant amount of memory. These tests
103typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001042 billion objects, which only works on 64-bit systems. There are also some
105tests that try to exhaust the address space of the process, which only makes
106sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000107which is a string in the form of '2.5Gb', determines howmuch memory the
108tests will limit themselves to (but they may go slightly over.) The number
109shouldn't be more memory than the machine has (including swap memory). You
110should also keep in mind that swap memory is generally much, much slower
111than RAM, and setting memlimit to all available RAM or higher will heavily
112tax the machine. On the other hand, it is no use running these tests with a
113limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
114to use more than memlimit memory will be skipped. The big-memory tests
115generally run very, very long.
116
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000117-u is used to specify which special resource intensive tests to run,
118such as those requiring large file support or network connectivity.
119The argument is a comma-separated list of words indicating the
120resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000121
Fred Drake3a15dac2002-04-11 16:39:16 +0000122 all - Enable all special resources.
123
Guido van Rossum315aa362003-03-11 14:46:48 +0000124 audio - Tests that use the audio device. (There are known
125 cases of broken audio drivers that can crash Python or
126 even the Linux kernel.)
127
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000128 curses - Tests that use curses and will modify the terminal's
129 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000130
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000131 largefile - It is okay to run some test that may create huge
132 files. These tests can take a long time and may
133 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000134
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000135 network - It is okay to run tests that use external network
136 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000137
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000138 decimal - Test the decimal module against a large suite that
139 verifies compliance with standards.
140
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000141 compiler - Allow test_tokenize to verify round-trip lexing on
142 every file in the test library.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000143
Tim Peterseba28be2005-03-28 01:08:02 +0000144 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000145
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000146 urlfetch - It is okay to download files required on testing.
147
Guilherme Polo9de29af2009-01-28 20:40:48 +0000148 gui - Run tests that require a running GUI.
149
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000150To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000151example, to run all the tests except for the gui tests, give the
152option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000153"""
154
Guido van Rossum152494a1996-12-20 03:12:20 +0000155import getopt
Antoine Pitrou88909542009-06-29 13:54:42 +0000156import json
Christian Heimesb186d002008-03-18 15:15:01 +0000157import os
Skip Montanaroab1c7912000-06-30 16:39:27 +0000158import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000159import re
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000160import io
Christian Heimesb186d002008-03-18 15:15:01 +0000161import sys
162import time
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000163import traceback
Christian Heimesb186d002008-03-18 15:15:01 +0000164import warnings
Benjamin Petersone549ead2009-03-28 21:42:05 +0000165import unittest
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000166from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000167import tempfile
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000168import platform
169import sysconfig
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000170
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000171
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000172# Some times __path__ and __file__ are not absolute (e.g. while running from
173# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
174# imports might fail. This affects only the modules imported before os.chdir().
175# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
176# they are found in the CWD their __file__ and __path__ will be relative (this
177# happens before the chdir). All the modules imported after the chdir, are
178# not found in the CWD, and since the other paths in sys.path[1:] are absolute
179# (site.py absolutize them), the __file__ and __path__ will be absolute too.
180# Therefore it is necessary to absolutize manually the __file__ and __path__ of
181# the packages to prevent later imports to fail when the CWD is different.
182for module in sys.modules.values():
183 if hasattr(module, '__path__'):
184 module.__path__ = [os.path.abspath(path) for path in module.__path__]
185 if hasattr(module, '__file__'):
186 module.__file__ = os.path.abspath(module.__file__)
187
Guido van Rossumdc15c272002-08-12 21:55:51 +0000188
Thomas Wouters477c8d52006-05-27 19:21:47 +0000189# Ignore ImportWarnings that only occur in the source tree,
190# (because of modules with the same name as source-directories in Modules/)
191for mod in ("ctypes", "gzip", "zipfile", "tarfile", "encodings.zlib_codec",
192 "test.test_zipimport", "test.test_zlib", "test.test_zipfile",
193 "test.test_codecs", "test.string_tests"):
194 warnings.filterwarnings(module=".*%s$" % (mod,),
195 action="ignore", category=ImportWarning)
196
Guido van Rossumbb484652002-12-02 09:56:21 +0000197# MacOSX (a.k.a. Darwin) has a default stack size that is too small
198# for deeply recursive regular expressions. We see this as crashes in
199# the Python test suite when running test_re.py and test_sre.py. The
200# fix is to set the stack limit to 2048.
201# This approach may also be useful for other Unixy platforms that
202# suffer from small default stack limits.
203if sys.platform == 'darwin':
204 try:
205 import resource
206 except ImportError:
207 pass
208 else:
209 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
210 newsoft = min(hard, max(soft, 1024*2048))
211 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
212
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000213# Test result constants.
214PASSED = 1
215FAILED = 0
216ENV_CHANGED = -1
217SKIPPED = -2
218RESOURCE_DENIED = -3
219INTERRUPTED = -4
220
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000221from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000222
Georg Brandl1158a332009-06-04 09:30:30 +0000223RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Guilherme Polo9de29af2009-01-28 20:40:48 +0000224 'decimal', 'compiler', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000225
226
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000227def usage(msg):
228 print(msg, file=sys.stderr)
229 print("Use --help for usage", file=sys.stderr)
230 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000231
232
Antoine Pitrou88909542009-06-29 13:54:42 +0000233def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000234 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000235 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000236 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
R. David Murray7dc72cc2009-11-14 16:13:02 +0000237 random_seed=None, use_mp=None, verbose3=False, forever=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000238 """Execute a test suite.
239
Thomas Wouters7e474022000-07-16 12:04:32 +0000240 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000241 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000242
243 tests -- a list of strings containing test names (optional)
244 testdir -- the directory in which to look for tests (optional)
245
246 Users other than the Python test suite will certainly want to
247 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000248 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000249
250 If the tests argument is omitted, the tests listed on the
251 command-line will be used. If that's empty, too, then all *.py
252 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000253
Antoine Pitrou88909542009-06-29 13:54:42 +0000254 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000255 single, randomize, findleaks, use_resources, trace, coverdir,
256 print_slow, and random_seed) allow programmers calling main()
257 directly to set the values that would normally be set by flags
258 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000259 """
Fred Drake004d5e62000-10-23 17:22:08 +0000260
Victor Stinner1802d3f2010-05-19 17:11:19 +0000261 replace_stdout()
262
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000263 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000264 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000265 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000266 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
267 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
268 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000269 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
270 'multiprocess=', 'slaveargs=', 'forever', 'debug', 'start=',
271 'nowindows'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000272 except getopt.error as msg:
273 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000274
275 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000276 if random_seed is None:
277 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000278 if use_resources is None:
279 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000280 debug = False
281 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000282 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000283 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000284 print(__doc__)
285 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000286 elif o in ('-v', '--verbose'):
287 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000288 elif o in ('-w', '--verbose2'):
289 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000290 elif o in ('-d', '--debug'):
291 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000292 elif o in ('-W', '--verbose3'):
293 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000294 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000295 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000296 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000297 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000298 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000299 elif o in ('-S', '--start'):
300 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000301 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000302 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000303 elif o in ('-S', '--slow'):
304 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000305 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000306 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000307 elif o == '--randseed':
308 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000309 elif o in ('-f', '--fromfile'):
310 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000311 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000312 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000313 elif o in ('-L', '--runleaks'):
314 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000315 elif o in ('-t', '--threshold'):
316 import gc
317 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000318 elif o in ('-T', '--coverage'):
319 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000320 elif o in ('-D', '--coverdir'):
321 coverdir = os.path.join(os.getcwd(), a)
322 elif o in ('-N', '--nocoverdir'):
323 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000324 elif o in ('-R', '--huntrleaks'):
325 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000326 if len(huntrleaks) not in (2, 3):
327 print(a, huntrleaks)
328 usage('-R takes 2 or 3 colon-separated arguments')
329 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000330 huntrleaks[0] = 5
331 else:
332 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000333 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000334 huntrleaks[1] = 4
335 else:
336 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000337 if len(huntrleaks) == 2 or not huntrleaks[2]:
338 huntrleaks[2:] = ["reflog.txt"]
339 # Avoid false positives due to the character cache in
340 # stringobject.c filling slowly with random data
341 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000342 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000343 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000344 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000345 u = [x.lower() for x in a.split(',')]
346 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000347 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000348 use_resources[:] = RESOURCE_NAMES
349 continue
350 remove = False
351 if r[0] == '-':
352 remove = True
353 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000354 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000355 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000356 if remove:
357 if r in use_resources:
358 use_resources.remove(r)
359 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000360 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000361 elif o in ('-n', '--nowindows'):
362 import msvcrt
363 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
364 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
365 msvcrt.SEM_NOGPFAULTERRORBOX|
366 msvcrt.SEM_NOOPENFILEERRORBOX)
367 try:
368 msvcrt.CrtSetReportMode
369 except AttributeError:
370 # release build
371 pass
372 else:
373 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
374 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
375 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000376 elif o in ('-F', '--forever'):
377 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000378 elif o in ('-j', '--multiprocess'):
379 use_mp = int(a)
380 elif o == '--slaveargs':
381 args, kwargs = json.loads(a)
382 try:
383 result = runtest(*args, **kwargs)
384 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000385 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000386 sys.stdout.flush()
387 print() # Force a newline (just in case)
388 print(json.dumps(result))
389 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000390 else:
391 print(("No handler for option {}. Please report this as a bug "
392 "at http://bugs.python.org.").format(o), file=sys.stderr)
393 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000394 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000395 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000396 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000397 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000398 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000399 usage("-l and -j don't go together!")
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000400 if use_mp and max(sys.flags):
401 # TODO: inherit the environment and the flags
402 print("Warning: flags and environment variables are ignored with -j option")
Barry Warsaw08fca522001-08-20 22:33:46 +0000403
Guido van Rossum152494a1996-12-20 03:12:20 +0000404 good = []
405 bad = []
406 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000407 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000408 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000409 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000410
Neil Schemenauerd569f232000-09-22 15:29:28 +0000411 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000412 try:
413 import gc
414 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000415 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000416 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000417 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000418 # Uncomment the line below to report garbage that is not
419 # freeable by reference counting alone. By default only
420 # garbage that is not collectable by the GC is reported.
421 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000422 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000423
Barry Warsawe11e3de1999-01-28 19:51:51 +0000424 if single:
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000425 filename = 'pynexttest'
Barry Warsawe11e3de1999-01-28 19:51:51 +0000426 try:
427 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000428 next_test = fp.read().strip()
429 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000430 fp.close()
431 except IOError:
432 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000433
434 if fromfile:
435 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000436 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Tim Petersc5000df2002-06-02 21:42:01 +0000437 for line in fp:
438 guts = line.split() # assuming no test has whitespace in its name
439 if guts and not guts[0].startswith('#'):
440 tests.extend(guts)
441 fp.close()
442
443 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000444 removepy(args)
445 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000446
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000447 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000448 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000449 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000450 for arg in args:
451 if arg in stdtests:
452 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000453 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000454 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000455
456 # For a partial run, we do not need to clutter the output.
457 if verbose or not (quiet or tests or args):
458 # Print basic platform information
459 print("==", platform.python_implementation(), *sys.version.split())
460 print("== ", platform.platform(aliased=True))
461 print("== ", os.getcwd())
462
R. David Murrayef1992b2009-12-16 15:19:27 +0000463 alltests = findtests(testdir, stdtests, nottests)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000464 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000465 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000466 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000467 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000468 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000469 except IndexError:
470 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000471 # Remove all the tests that precede start if it's set.
472 if start:
473 try:
474 del tests[:tests.index(start)]
475 except ValueError:
476 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000477 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000478 random.seed(random_seed)
479 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000480 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000481 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000482 import trace, tempfile
483 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
484 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000485 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000486
Christian Heimesb186d002008-03-18 15:15:01 +0000487 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000488 support.verbose = verbose # Tell tests to be moderately quiet
489 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000490 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000491
492 def accumulate_result(test, result):
493 ok, test_time = result
494 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000495 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000496 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000497 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000498 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000499 elif ok == ENV_CHANGED:
500 bad.append(test)
501 environment_changed.append(test)
502 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000503 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000504 elif ok == RESOURCE_DENIED:
505 skipped.append(test)
506 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000507
R. David Murray7dc72cc2009-11-14 16:13:02 +0000508 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000509 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000510 while True:
511 for test in tests:
512 yield test
513 if bad:
514 return
515 tests = test_forever()
516 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000517 tests = iter(selected)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000518
Antoine Pitrou88909542009-06-29 13:54:42 +0000519 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000520 try:
521 from threading import Thread
522 except ImportError:
523 print("Multiprocess option requires thread support")
524 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000525 from queue import Queue
526 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000527 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000528 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000529 def tests_and_args():
530 for test in tests:
531 args_tuple = (
532 (test, verbose, quiet, testdir),
533 dict(huntrleaks=huntrleaks, use_resources=use_resources,
534 debug=debug)
535 )
536 yield (test, args_tuple)
537 pending = tests_and_args()
Antoine Pitrou88909542009-06-29 13:54:42 +0000538 def work():
539 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000540 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000541 while True:
542 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000543 test, args_tuple = next(pending)
544 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000545 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000546 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000547 # -E is needed by some tests, e.g. test_import
548 popen = Popen([sys.executable, '-E', '-m', 'test.regrtest',
549 '--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000550 stdout=PIPE, stderr=PIPE,
Antoine Pitrou88909542009-06-29 13:54:42 +0000551 universal_newlines=True, close_fds=True)
R. David Murray27144602009-10-19 15:26:16 +0000552 stdout, stderr = popen.communicate()
553 # Strip last refcount output line if it exists, since it
554 # comes from the shutdown of the interpreter in the subcommand.
555 stderr = debug_output_pat.sub("", stderr)
556 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000557 if not result:
558 output.put((None, None, None, None))
559 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000560 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000561 if not quiet:
562 stdout = test+'\n'+stdout
563 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000564 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000565 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000566 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000567 workers = [Thread(target=work) for i in range(use_mp)]
568 for worker in workers:
569 worker.start()
570 finished = 0
R. David Murray7dc72cc2009-11-14 16:13:02 +0000571 try:
572 while finished < use_mp:
573 test, stdout, stderr, result = output.get()
574 if test is None:
575 finished += 1
576 continue
577 if stdout:
578 print(stdout)
579 if stderr:
580 print(stderr, file=sys.stderr)
581 if result[0] == INTERRUPTED:
582 assert result[1] == 'KeyboardInterrupt'
583 raise KeyboardInterrupt # What else?
584 accumulate_result(test, result)
585 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000586 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000587 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000588 for worker in workers:
589 worker.join()
590 else:
591 for test in tests:
592 if not quiet:
593 print(test)
594 sys.stdout.flush()
595 if trace:
596 # If we're tracing code coverage, then we don't exit with status
597 # if on a false return value from main.
598 tracer.runctx('runtest(test, verbose, quiet, testdir)',
599 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000600 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000601 try:
602 result = runtest(test, verbose, quiet,
603 testdir, huntrleaks, debug)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000604 accumulate_result(test, result)
605 if verbose3 and result[0] == FAILED:
Georg Brandl96f78972009-10-27 21:02:11 +0000606 print("Re-running test {} in verbose mode".format(test))
607 runtest(test, True, quiet, testdir, huntrleaks, debug)
Antoine Pitrou88909542009-06-29 13:54:42 +0000608 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000609 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000610 break
611 except:
612 raise
613 if findleaks:
614 gc.collect()
615 if gc.garbage:
616 print("Warning: test created", len(gc.garbage), end=' ')
617 print("uncollectable object(s).")
618 # move the uncollectable objects somewhere so we don't see
619 # them again
620 found_garbage.extend(gc.garbage)
621 del gc.garbage[:]
622 # Unload the newly imported modules (best effort finalization)
623 for module in sys.modules.keys():
624 if module not in save_modules and module.startswith("test."):
625 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000626
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000627 if interrupted:
628 # print a newline after ^C
629 print()
630 print("Test suite interrupted by signal SIGINT.")
631 omitted = set(selected) - set(good) - set(bad) - set(skipped)
632 print(count(len(omitted), "test"), "omitted:")
633 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000634 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000635 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000636 print("All", end=' ')
637 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000638 if print_slow:
639 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000640 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000641 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000642 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000643 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000644 bad = sorted(set(bad) - set(environment_changed))
645 if bad:
646 print(count(len(bad), "test"), "failed:")
647 printlist(bad)
648 if environment_changed:
649 print("{} altered the execution environment:".format(
650 count(len(environment_changed), "test")))
651 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000652 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000653 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000654 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000655
Tim Petersb5b7b782001-08-12 01:20:39 +0000656 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000657 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000658 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000659 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000660 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000661 print(count(len(surprise), "skip"), \
662 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000663 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000664 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000665 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000666 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000667 print("Ask someone to teach regrtest.py about which tests are")
668 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000669
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000670 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000671 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000672 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000673 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000674 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000675 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000676 verbose = True
677 ok = runtest(test, True, quiet, testdir,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000678 huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000679 except KeyboardInterrupt:
680 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000681 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000682 break
683 except:
684 raise
685
Barry Warsawe11e3de1999-01-28 19:51:51 +0000686 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000687 if next_single_test:
688 with open(filename, 'w') as fp:
689 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000690 else:
691 os.unlink(filename)
692
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000693 if trace:
694 r = tracer.results()
695 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
696
Skip Montanaro0179a182004-06-06 15:53:18 +0000697 if runleaks:
698 os.system("leaks %d" % os.getpid())
699
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000700 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000701
Guido van Rossum152494a1996-12-20 03:12:20 +0000702
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000703STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000704 'test_grammar',
705 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000706 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000707 'test_builtin',
708 'test_exceptions',
709 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000710 'test_unittest',
711 'test_doctest',
712 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000713]
Guido van Rossum152494a1996-12-20 03:12:20 +0000714
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000715NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000716 'test_future1',
717 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000718}
Guido van Rossum152494a1996-12-20 03:12:20 +0000719
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000720def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000721 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000722 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000723 names = os.listdir(testdir)
724 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000725 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000726 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000727 modname, ext = os.path.splitext(name)
728 if modname[:5] == "test_" and ext == ".py" and modname not in others:
729 tests.append(modname)
730 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000731
Victor Stinnerf58087b2010-05-02 17:24:51 +0000732def replace_stdout():
733 """Set stdout encoder error handler to backslashreplace (as stderr error
734 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000735 if os.name == "nt":
736 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
737 return
Victor Stinnerf58087b2010-05-02 17:24:51 +0000738 stdout = sys.stdout
739 sys.stdout = open(stdout.fileno(), 'w',
740 encoding=stdout.encoding,
741 errors="backslashreplace")
742
Antoine Pitrou88909542009-06-29 13:54:42 +0000743def runtest(test, verbose, quiet,
744 testdir=None, huntrleaks=False, debug=False, use_resources=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000745 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000746
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000747 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000748 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000749 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000750 test_times -- a list of (time, test_name) pairs
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000751 testdir -- test directory
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000752 huntrleaks -- run multiple times to test for leaks; requires a debug
753 build; a triple corresponding to -R's three arguments
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000754
755 Returns one of the test result constants:
756 INTERRUPTED KeyboardInterrupt when run under -j
757 RESOURCE_DENIED test skipped because resource denied
758 SKIPPED test skipped for some other reason
759 ENV_CHANGED test failed because it changed the execution environment
760 FAILED test failed
761 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000762 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000763
Antoine Pitrou88909542009-06-29 13:54:42 +0000764 support.verbose = verbose # Tell tests to be moderately quiet
765 if use_resources is not None:
766 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000767 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000768 return runtest_inner(test, verbose, quiet,
769 testdir, huntrleaks, debug)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000770 finally:
771 cleanup_test_droppings(test, verbose)
772
Nick Coghlan6ead5522009-10-18 13:19:33 +0000773# Unit tests are supposed to leave the execution environment unchanged
774# once they complete. But sometimes tests have bugs, especially when
775# tests fail, and the changes to environment go on to mess up other
776# tests. This can cause issues with buildbot stability, since tests
777# are run in random order and so problems may appear to come and go.
778# There are a few things we can save and restore to mitigate this, and
779# the following context manager handles this task.
780
781class saved_test_environment:
782 """Save bits of the test environment and restore them at block exit.
783
784 with saved_test_environment(testname, verbose, quiet):
785 #stuff
786
787 Unless quiet is True, a warning is printed to stderr if any of
788 the saved items was changed by the test. The attribute 'changed'
789 is initially False, but is set to True if a change is detected.
790
791 If verbose is more than 1, the before and after state of changed
792 items is also printed.
793 """
794
795 changed = False
796
797 def __init__(self, testname, verbose=0, quiet=False):
798 self.testname = testname
799 self.verbose = verbose
800 self.quiet = quiet
801
802 # To add things to save and restore, add a name XXX to the resources list
803 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
804 # return the value to be saved and compared against a second call to the
805 # get function when test execution completes. restore_XXX should accept
806 # the saved value and restore the resource using it. It will be called if
807 # and only if a change in the value is detected.
808 #
809 # Note: XXX will have any '.' replaced with '_' characters when determining
810 # the corresponding method names.
811
812 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000813 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
814 'warnings.filters')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000815
816 def get_sys_argv(self):
817 return id(sys.argv), sys.argv, sys.argv[:]
818 def restore_sys_argv(self, saved_argv):
819 sys.argv = saved_argv[1]
820 sys.argv[:] = saved_argv[2]
821
822 def get_cwd(self):
823 return os.getcwd()
824 def restore_cwd(self, saved_cwd):
825 os.chdir(saved_cwd)
826
827 def get_sys_stdout(self):
828 return sys.stdout
829 def restore_sys_stdout(self, saved_stdout):
830 sys.stdout = saved_stdout
831
832 def get_sys_stderr(self):
833 return sys.stderr
834 def restore_sys_stderr(self, saved_stderr):
835 sys.stderr = saved_stderr
836
837 def get_sys_stdin(self):
838 return sys.stdin
839 def restore_sys_stdin(self, saved_stdin):
840 sys.stdin = saved_stdin
841
842 def get_os_environ(self):
843 return id(os.environ), os.environ, dict(os.environ)
844 def restore_os_environ(self, saved_environ):
845 os.environ = saved_environ[1]
846 os.environ.clear()
847 os.environ.update(saved_environ[2])
848
849 def get_sys_path(self):
850 return id(sys.path), sys.path, sys.path[:]
851 def restore_sys_path(self, saved_path):
852 sys.path = saved_path[1]
853 sys.path[:] = saved_path[2]
854
Brett Cannon055470a2010-02-19 15:57:10 +0000855 def get_sys_path_hooks(self):
856 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
857 def restore_sys_path_hooks(self, saved_hooks):
858 sys.path_hooks = saved_hooks[1]
859 sys.path_hooks[:] = saved_hooks[2]
860
861 def get___import__(self):
862 return __builtins__.__import__
863 def restore___import__(self, import_):
864 __builtins__.__import__ = import_
865
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000866 def get_warnings_filters(self):
867 return id(warnings.filters), warnings.filters, warnings.filters[:]
868 def restore_warnings_filters(self, saved_filters):
869 warnings.filters = saved_filters[1]
870 warnings.filters[:] = saved_filters[2]
871
Nick Coghlan6ead5522009-10-18 13:19:33 +0000872 def resource_info(self):
873 for name in self.resources:
874 method_suffix = name.replace('.', '_')
875 get_name = 'get_' + method_suffix
876 restore_name = 'restore_' + method_suffix
877 yield name, getattr(self, get_name), getattr(self, restore_name)
878
879 def __enter__(self):
880 self.saved_values = dict((name, get()) for name, get, restore
881 in self.resource_info())
882 return self
883
884 def __exit__(self, exc_type, exc_val, exc_tb):
885 for name, get, restore in self.resource_info():
886 current = get()
887 original = self.saved_values[name]
888 # Check for changes to the resource's value
889 if current != original:
890 self.changed = True
891 restore(original)
892 if not self.quiet:
893 print("Warning -- {} was modified by {}".format(
894 name, self.testname),
895 file=sys.stderr)
896 if self.verbose > 1:
897 print(" Before: {}\n After: {} ".format(
898 original, current),
899 file=sys.stderr)
900 return False
901
902
Antoine Pitrou88909542009-06-29 13:54:42 +0000903def runtest_inner(test, verbose, quiet,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000904 testdir=None, huntrleaks=False, debug=False):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000905 support.unload(test)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000906 testdir = findtestdir(testdir)
Benjamin Petersonb48af542010-04-11 20:43:16 +0000907 if verbose:
908 capture_stdout = None
909 else:
910 capture_stdout = io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000911
Antoine Pitrou88909542009-06-29 13:54:42 +0000912 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +0000913 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +0000914 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +0000915 if test.startswith('test.'):
916 abstest = test
917 else:
918 # Always import it from the test package
919 abstest = 'test.' + test
920 with saved_test_environment(test, verbose, quiet) as environment:
921 start_time = time.time()
922 the_package = __import__(abstest, globals(), locals(), [])
923 the_module = getattr(the_package, test)
924 # Old tests run to completion simply as a side-effect of
925 # being imported. For tests based on unittest or doctest,
926 # explicitly invoke their test_main() function (if it exists).
927 indirect_test = getattr(the_module, "test_main", None)
928 if indirect_test is not None:
929 indirect_test()
930 if huntrleaks:
931 refleak = dash_R(the_module, test, indirect_test,
932 huntrleaks)
933 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000934 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +0000935 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000936 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +0000937 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000938 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +0000939 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +0000940 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000941 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000942 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000943 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +0000944 except KeyboardInterrupt:
945 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000946 except support.TestFailed as msg:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000947 print("test", test, "failed --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000948 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000949 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +0000950 except:
Guido van Rossum41360a41998-03-26 19:42:58 +0000951 type, value = sys.exc_info()[:2]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000952 print("test", test, "crashed --", str(type) + ":", value)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000953 sys.stdout.flush()
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000954 if verbose or debug:
Victor Stinnerf58087b2010-05-02 17:24:51 +0000955 traceback.print_exc(file=sys.stdout)
956 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000957 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +0000958 else:
Collin Wintera5503d52009-05-15 01:20:21 +0000959 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000960 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +0000961 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000962 return ENV_CHANGED, test_time
963 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000964
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000965def cleanup_test_droppings(testname, verbose):
966 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000967 import stat
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000968
969 # Try to clean up junk commonly left behind. While tests shouldn't leave
970 # any files or directories behind, when a test fails that can be tedious
971 # for it to arrange. The consequences can be especially nasty on Windows,
972 # since if a test leaves a file open, it cannot be deleted by name (while
973 # there's nothing we can do about that here either, we can display the
974 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000975 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000976 "db_home",
977 ):
978 if not os.path.exists(name):
979 continue
980
981 if os.path.isdir(name):
982 kind, nuker = "directory", shutil.rmtree
983 elif os.path.isfile(name):
984 kind, nuker = "file", os.unlink
985 else:
986 raise SystemError("os.path says %r exists but is neither "
987 "directory nor file" % name)
988
989 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000990 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000991 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000992 # if we have chmod, fix possible permissions problems
993 # that might prevent cleanup
994 if (hasattr(os, 'chmod')):
995 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000996 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000997 except Exception as msg:
998 print(("%r left behind %s %r and it couldn't be "
999 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001000
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001001def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001002 """Run a test multiple times, looking for reference leaks.
1003
1004 Returns:
1005 False if the test didn't leak references; True if we detected refleaks.
1006 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001007 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001008 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001009
1010 if not hasattr(sys, 'gettotalrefcount'):
1011 raise Exception("Tracking reference leaks requires a debug build "
1012 "of Python")
1013
1014 # Save current values for dash_R_cleanup() to restore.
1015 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001016 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001017 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001018 try:
1019 import zipimport
1020 except ImportError:
1021 zdc = None # Run unmodified on platforms without zipimport support
1022 else:
1023 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001024 abcs = {}
1025 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001026 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001027 continue
1028 for obj in abc.__subclasses__() + [abc]:
1029 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001030
1031 if indirect_test:
1032 def run_the_test():
1033 indirect_test()
1034 else:
1035 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001036 del sys.modules[the_module.__name__]
1037 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001038
1039 deltas = []
1040 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001041 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001042 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001043 print("beginning", repcount, "repetitions", file=sys.stderr)
1044 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001045 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001046 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001047 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001048 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001049 run_the_test()
1050 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001051 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001052 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001053 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001054 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001055 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001056 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001057 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001058 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001059 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001060 sys.stderr.flush()
1061 with open(fname, "a") as refrep:
1062 print(msg, file=refrep)
1063 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001064 return True
1065 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001066
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001067def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001068 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001069 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001070 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001071 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001072 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001073 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001074
Christian Heimesdae2a892008-04-19 00:55:37 +00001075 # Clear the warnings registry, so they can be displayed again
1076 for mod in sys.modules.values():
1077 if hasattr(mod, '__warningregistry__'):
1078 del mod.__warningregistry__
1079
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001080 # Restore some original values.
1081 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001082 copyreg.dispatch_table.clear()
1083 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001084 sys.path_importer_cache.clear()
1085 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001086 try:
1087 import zipimport
1088 except ImportError:
1089 pass # Run unmodified on platforms without zipimport support
1090 else:
1091 zipimport._zip_directory_cache.clear()
1092 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001093
Christian Heimes26855632008-01-27 23:50:43 +00001094 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001095 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001096
Guido van Rossum3de862d2007-08-18 00:10:33 +00001097 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001098 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001099 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001100 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001101 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001102 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001103 obj._abc_cache.clear()
1104 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001105
Antoine Pitrou046467c2009-10-30 18:30:35 +00001106 # Flush standard output, so that buffered data is sent to the OS and
1107 # associated Python objects are reclaimed.
1108 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1109 if stream is not None:
1110 stream.flush()
1111
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001112 # Clear assorted module caches.
1113 _path_created.clear()
1114 re.purge()
1115 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001116 urllib.parse.clear_cache()
1117 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001118 linecache.clearcache()
1119 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001120 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001121 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001122 doctest.master = None
1123
1124 # Collect cyclic trash.
1125 gc.collect()
1126
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001127def warm_char_cache():
1128 s = bytes(range(256))
1129 for i in range(256):
1130 s[i:i+1]
1131
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001132def findtestdir(path=None):
1133 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001134
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001135def removepy(names):
1136 if not names:
1137 return
1138 for idx, name in enumerate(names):
1139 basename, ext = os.path.splitext(name)
1140 if ext == '.py':
1141 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001142
Guido van Rossum152494a1996-12-20 03:12:20 +00001143def count(n, word):
1144 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001145 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001146 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001147 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001148
Tim Petersa45da922001-08-12 03:45:50 +00001149def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001150 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001151
1152 Optional arg width (default 70) is the maximum line length.
1153 Optional arg indent (default 4) is the number of blanks with which to
1154 begin each line.
1155 """
1156
Tim Petersba78bc42002-07-04 19:45:06 +00001157 from textwrap import fill
1158 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001159 # Print the sorted list: 'x' may be a '--random' list or a set()
1160 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001161 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001162
Tim Petersde14a302002-04-01 05:04:46 +00001163# Map sys.platform to a string containing the basenames of tests
1164# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001165#
1166# Special cases:
1167# test_pep277
1168# The _ExpectedSkips constructor adds this to the set of expected
1169# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001170# test_timeout
1171# Controlled by test_timeout.skip_expected. Requires the network
1172# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001173#
1174# Tests that are expected to be skipped everywhere except on one platform
1175# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001176
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001177_expectations = {
1178 'win32':
1179 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001180 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001181 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001182 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001183 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001184 test_fcntl
1185 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001186 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001187 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001188 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001189 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001190 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001191 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001192 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001193 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001194 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001195 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001196 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001197 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001198 test_pty
1199 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001200 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001201 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001202 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001203 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001204 test_wait3
1205 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001206 """,
1207 'linux2':
1208 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001209 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001210 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001211 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001212 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001213 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001214 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001215 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001216 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001217 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001218 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001219 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001220 test_openpty
1221 test_pyexpat
1222 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001223 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001224 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001225 'openunix8':
1226 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001227 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001228 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001229 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001230 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001231 test_openpty
1232 test_pyexpat
1233 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001234 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001235 """,
1236 'sco_sv3':
1237 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001238 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001239 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001240 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001241 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001242 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001243 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001244 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001245 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001246 test_openpty
1247 test_pyexpat
1248 test_queue
1249 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001250 test_sundry
1251 test_thread
1252 test_threaded_import
1253 test_threadedtempfile
1254 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001255 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001256 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001257 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001258 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001259 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001260 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001261 test_dbm_gnu
Jack Jansen398c2362001-12-02 21:41:36 +00001262 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001263 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001264 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001265 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001266 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001267 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001268 'sunos5':
1269 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001270 test_curses
1271 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001272 test_epoll
1273 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001274 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001275 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001276 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001277 test_zipfile
1278 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001279 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001280 'hp-ux11':
1281 """
Skip Montanarob3230212002-03-15 02:54:03 +00001282 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001283 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001284 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001285 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001286 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001287 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001288 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001289 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001290 test_openpty
1291 test_pyexpat
1292 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001293 test_zipfile
1294 test_zlib
1295 """,
Jason Tishler25115942002-12-05 15:18:15 +00001296 'cygwin':
1297 """
Jason Tishler25115942002-12-05 15:18:15 +00001298 test_curses
1299 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001300 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001301 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001302 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001303 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001304 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001305 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001306 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001307 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001308 'os2emx':
1309 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001310 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001311 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001312 test_epoll
1313 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001314 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001315 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001316 test_openpty
1317 test_ossaudiodev
1318 test_pty
1319 test_resource
1320 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001321 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001322 'freebsd4':
1323 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001324 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001325 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001326 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001327 test_ossaudiodev
1328 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001329 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001330 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001331 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001332 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001333 test_ttk_guionly
1334 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001335 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001336 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001337 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001338 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001339 'aix5':
1340 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001341 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001342 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001343 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001344 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001345 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001346 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001347 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001348 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001349 test_ttk_guionly
1350 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001351 test_zipimport
1352 test_zlib
1353 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001354 'openbsd3':
1355 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001356 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001357 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001358 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001359 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001360 test_normalization
1361 test_ossaudiodev
1362 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001363 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001364 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001365 test_ttk_guionly
1366 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001367 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001368 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001369 'netbsd3':
1370 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001371 test_ctypes
1372 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001373 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001374 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001375 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001376 test_ossaudiodev
1377 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001378 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001379 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001380 test_ttk_guionly
1381 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001382 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001383 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001384}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001385_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001386_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001387_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001388_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001389
Tim Petersb5b7b782001-08-12 01:20:39 +00001390class _ExpectedSkips:
1391 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001392 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001393 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001394
Tim Peters7c7efe92002-08-23 17:55:54 +00001395 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001396 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001397 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001398 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001399
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001400 # These are broken tests, for now skipped on every platform.
1401 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001402 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001403
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001404 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001405 if not os.path.supports_unicode_filenames:
1406 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001407
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001408 # doctest, profile and cProfile tests fail when the codec for the
1409 # fs encoding isn't built in because PyUnicode_Decode() adds two
1410 # calls into Python.
1411 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1412 if sys.getfilesystemencoding().lower() not in encs:
1413 self.expected.add('test_profile')
1414 self.expected.add('test_cProfile')
1415 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001416
Neal Norwitz55b61d22003-02-28 19:57:03 +00001417 if test_timeout.skip_expected:
1418 self.expected.add('test_timeout')
1419
Tim Petersecd79eb2003-01-29 00:35:32 +00001420 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001421 # test_sqlite is only reliable on Windows where the library
1422 # is distributed with Python
Neal Norwitz7035c982003-03-29 22:01:17 +00001423 WIN_ONLY = ["test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001424 "test_winsound", "test_startfile",
1425 "test_sqlite"]
Neal Norwitz7035c982003-03-29 22:01:17 +00001426 for skip in WIN_ONLY:
1427 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001428
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001429 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001430 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001431
Tim Peters7c7efe92002-08-23 17:55:54 +00001432 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001433
1434 def isvalid(self):
1435 "Return true iff _ExpectedSkips knows about the current platform."
1436 return self.valid
1437
1438 def getexpected(self):
1439 """Return set of test names we expect to skip on current platform.
1440
1441 self.isvalid() must be true.
1442 """
1443
1444 assert self.isvalid()
1445 return self.expected
1446
Guido van Rossum152494a1996-12-20 03:12:20 +00001447if __name__ == '__main__':
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001448 # Simplification for findtestdir().
1449 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001450
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001451 # When tests are run from the Python build directory, it is best practice
1452 # to keep the test files in a subfolder. It eases the cleanup of leftover
1453 # files using command "make distclean".
1454 if sysconfig.is_python_build():
1455 parent_dir = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1456 if not os.path.exists(parent_dir):
1457 os.mkdir(parent_dir)
1458 else:
1459 parent_dir = os.path.abspath(tempfile.gettempdir())
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001460
1461 # Define a writable temp dir that will be used as cwd while running
1462 # the tests. The name of the dir includes the pid to allow parallel
1463 # testing (see the -j option).
1464 TESTCWD = 'test_python_{}'.format(os.getpid())
1465
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001466 TESTCWD = os.path.join(parent_dir, TESTCWD)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001467
1468 # Run the tests in a context manager that temporary changes the CWD to a
1469 # temporary and writable directory. If it's not possible to create or
1470 # change the CWD, the original CWD will be used. The original CWD is
1471 # available from support.SAVEDCWD.
1472 with support.temp_cwd(TESTCWD, quiet=True):
1473 main()