blob: 832228175e660b306b7f0bbe2124f84547c5a423 [file] [log] [blame]
Guido van Rossum152494a1996-12-20 03:12:20 +00001#! /usr/bin/env python
2
3"""Regression test.
4
5This will find all modules whose name is "test_*" in the test
6directory, and run them. Various command line options provide
7additional facilities.
8
9Command line options:
10
Georg Brandl96f78972009-10-27 21:02:11 +000011-h/--help -- print this text and exit
12
13Verbosity
14
15-v/--verbose -- run tests in verbose mode with output to stdout
16-w/--verbose2 -- re-run failed tests in verbose mode
17-W/--verbose3 -- re-run failed tests in verbose mode immediately
18-d/--debug -- print traceback for failed tests
19-q/--quiet -- don't print anything except if a test fails
20-S/--slow -- print the slowest 10 tests
21
22Selecting tests
23
24-r/--random -- randomize test execution order
25-f/--fromfile -- read names of tests to run from a file (see below)
26-x/--exclude -- arguments are tests to *exclude*
27-s/--single -- run only a single test (see below)
28-u/--use RES1,RES2,...
29 -- specify which special resource intensive tests to run
30-M/--memlimit LIMIT
31 -- run very large memory-consuming tests
32
33Special runs
34
35-l/--findleaks -- if GC is available detect tests that leak memory
36-L/--runleaks -- run the leaks(1) command just before exit
37-R/--huntrleaks RUNCOUNTS
38 -- search for reference leaks (needs debug build, v. slow)
39-j/--multiprocess PROCESSES
40 -- run PROCESSES processes at once
41-T/--coverage -- turn on code coverage using the trace module
42-D/--coverdir DIRECTORY
43 -- Directory where coverage files are put
44-N/--nocoverdir -- Put coverage files alongside modules
45-t/--threshold THRESHOLD
46 -- call gc.set_threshold(THRESHOLD)
47-n/--nowindows -- suppress error message boxes on Windows
R. David Murray7dc72cc2009-11-14 16:13:02 +000048-F/--forever -- run the selected tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000049
50If non-option arguments are present, they are names for tests to run,
51unless -x is given, in which case they are names for tests not to run.
52If no test names are given, all tests are run.
Guido van Rossumf58ed251997-03-07 21:04:33 +000053
Collin Winterfd12f492009-03-29 04:05:05 +000054-r randomizes test execution order. You can use --randseed=int to provide a
55int seed value for the randomizer; this is useful for reproducing troublesome
56test orders.
57
Barry Warsaw3b6d0252004-02-07 22:43:03 +000058-T turns on code coverage tracing with the trace module.
59
Walter Dörwaldaee4da62004-11-12 18:51:27 +000060-D specifies the directory where coverage files are put.
61
62-N Put coverage files alongside modules.
63
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000064-s means to run only a single test and exit. This is useful when
65doing memory analysis on the Python interpreter (which tend to consume
66too many resources to run the full regression test non-stop). The
67file /tmp/pynexttest is read to find the next test to run. If this
68file is missing, the first test_*.py file in testdir or on the command
69line is used. (actually tempfile.gettempdir() is used instead of
70/tmp).
Barry Warsawe11e3de1999-01-28 19:51:51 +000071
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000072-S is used to continue running tests after an aborted run. It will
73maintain the order a standard run (ie, this assumes -r is not used).
74This is useful after the tests have prematurely stopped for some external
75reason and you want to start running from where you left off rather
76than starting from the beginning.
77
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000078-f reads the names of tests from the file given as f's argument, one
79or more test names per line. Whitespace is ignored. Blank lines and
80lines beginning with '#' are ignored. This is especially useful for
81whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000082
Skip Montanaro0179a182004-06-06 15:53:18 +000083-L causes the leaks(1) command to be run just before exit if it exists.
84leaks(1) is available on Mac OS X and presumably on some other
85FreeBSD-derived systems.
86
Michael W. Hudson61147f62004-08-03 11:33:28 +000087-R runs each test several times and examines sys.gettotalrefcount() to
88see if the test appears to be leaking references. The argument should
89be of the form stab:run:fname where 'stab' is the number of times the
90test is run to let gettotalrefcount settle down, 'run' is the number
91of times further it is run and 'fname' is the name of the file the
92reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000093"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +000094
Thomas Wouters477c8d52006-05-27 19:21:47 +000095-M runs tests that require an exorbitant amount of memory. These tests
96typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000972 billion objects, which only works on 64-bit systems. There are also some
98tests that try to exhaust the address space of the process, which only makes
99sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000100which is a string in the form of '2.5Gb', determines howmuch memory the
101tests will limit themselves to (but they may go slightly over.) The number
102shouldn't be more memory than the machine has (including swap memory). You
103should also keep in mind that swap memory is generally much, much slower
104than RAM, and setting memlimit to all available RAM or higher will heavily
105tax the machine. On the other hand, it is no use running these tests with a
106limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
107to use more than memlimit memory will be skipped. The big-memory tests
108generally run very, very long.
109
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000110-u is used to specify which special resource intensive tests to run,
111such as those requiring large file support or network connectivity.
112The argument is a comma-separated list of words indicating the
113resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000114
Fred Drake3a15dac2002-04-11 16:39:16 +0000115 all - Enable all special resources.
116
Guido van Rossum315aa362003-03-11 14:46:48 +0000117 audio - Tests that use the audio device. (There are known
118 cases of broken audio drivers that can crash Python or
119 even the Linux kernel.)
120
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000121 curses - Tests that use curses and will modify the terminal's
122 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000123
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000124 largefile - It is okay to run some test that may create huge
125 files. These tests can take a long time and may
126 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000127
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000128 network - It is okay to run tests that use external network
129 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000130
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000131 decimal - Test the decimal module against a large suite that
132 verifies compliance with standards.
133
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000134 compiler - Allow test_tokenize to verify round-trip lexing on
135 every file in the test library.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000136
Tim Peterseba28be2005-03-28 01:08:02 +0000137 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000138
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000139 urlfetch - It is okay to download files required on testing.
140
Guilherme Polo9de29af2009-01-28 20:40:48 +0000141 gui - Run tests that require a running GUI.
142
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000143To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000144example, to run all the tests except for the gui tests, give the
145option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000146"""
147
Guido van Rossum152494a1996-12-20 03:12:20 +0000148import getopt
R. David Murray7dc72cc2009-11-14 16:13:02 +0000149import itertools
Antoine Pitrou88909542009-06-29 13:54:42 +0000150import json
Christian Heimesb186d002008-03-18 15:15:01 +0000151import os
Skip Montanaroab1c7912000-06-30 16:39:27 +0000152import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000153import re
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000154import io
Christian Heimesb186d002008-03-18 15:15:01 +0000155import sys
156import time
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000157import traceback
Christian Heimesb186d002008-03-18 15:15:01 +0000158import warnings
Benjamin Petersone549ead2009-03-28 21:42:05 +0000159import unittest
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000160from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000161import tempfile
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000162import platform
163import sysconfig
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000164
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000165
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000166# Some times __path__ and __file__ are not absolute (e.g. while running from
167# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
168# imports might fail. This affects only the modules imported before os.chdir().
169# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
170# they are found in the CWD their __file__ and __path__ will be relative (this
171# happens before the chdir). All the modules imported after the chdir, are
172# not found in the CWD, and since the other paths in sys.path[1:] are absolute
173# (site.py absolutize them), the __file__ and __path__ will be absolute too.
174# Therefore it is necessary to absolutize manually the __file__ and __path__ of
175# the packages to prevent later imports to fail when the CWD is different.
176for module in sys.modules.values():
177 if hasattr(module, '__path__'):
178 module.__path__ = [os.path.abspath(path) for path in module.__path__]
179 if hasattr(module, '__file__'):
180 module.__file__ = os.path.abspath(module.__file__)
181
Guido van Rossumdc15c272002-08-12 21:55:51 +0000182
Thomas Wouters477c8d52006-05-27 19:21:47 +0000183# Ignore ImportWarnings that only occur in the source tree,
184# (because of modules with the same name as source-directories in Modules/)
185for mod in ("ctypes", "gzip", "zipfile", "tarfile", "encodings.zlib_codec",
186 "test.test_zipimport", "test.test_zlib", "test.test_zipfile",
187 "test.test_codecs", "test.string_tests"):
188 warnings.filterwarnings(module=".*%s$" % (mod,),
189 action="ignore", category=ImportWarning)
190
Guido van Rossumbb484652002-12-02 09:56:21 +0000191# MacOSX (a.k.a. Darwin) has a default stack size that is too small
192# for deeply recursive regular expressions. We see this as crashes in
193# the Python test suite when running test_re.py and test_sre.py. The
194# fix is to set the stack limit to 2048.
195# This approach may also be useful for other Unixy platforms that
196# suffer from small default stack limits.
197if sys.platform == 'darwin':
198 try:
199 import resource
200 except ImportError:
201 pass
202 else:
203 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
204 newsoft = min(hard, max(soft, 1024*2048))
205 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
206
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000207# Test result constants.
208PASSED = 1
209FAILED = 0
210ENV_CHANGED = -1
211SKIPPED = -2
212RESOURCE_DENIED = -3
213INTERRUPTED = -4
214
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000215from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000216
Georg Brandl1158a332009-06-04 09:30:30 +0000217RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Guilherme Polo9de29af2009-01-28 20:40:48 +0000218 'decimal', 'compiler', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000219
220
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000221def usage(msg):
222 print(msg, file=sys.stderr)
223 print("Use --help for usage", file=sys.stderr)
224 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000225
226
Antoine Pitrou88909542009-06-29 13:54:42 +0000227def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000228 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000229 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000230 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
R. David Murray7dc72cc2009-11-14 16:13:02 +0000231 random_seed=None, use_mp=None, verbose3=False, forever=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000232 """Execute a test suite.
233
Thomas Wouters7e474022000-07-16 12:04:32 +0000234 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000235 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000236
237 tests -- a list of strings containing test names (optional)
238 testdir -- the directory in which to look for tests (optional)
239
240 Users other than the Python test suite will certainly want to
241 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000242 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000243
244 If the tests argument is omitted, the tests listed on the
245 command-line will be used. If that's empty, too, then all *.py
246 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000247
Antoine Pitrou88909542009-06-29 13:54:42 +0000248 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000249 single, randomize, findleaks, use_resources, trace, coverdir,
250 print_slow, and random_seed) allow programmers calling main()
251 directly to set the values that would normally be set by flags
252 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000253 """
Fred Drake004d5e62000-10-23 17:22:08 +0000254
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000255 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000256 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000257 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000258 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
259 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
260 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000261 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
262 'multiprocess=', 'slaveargs=', 'forever', 'debug', 'start=',
263 'nowindows'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000264 except getopt.error as msg:
265 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000266
267 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000268 if random_seed is None:
269 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000270 if use_resources is None:
271 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000272 debug = False
273 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000274 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000275 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000276 print(__doc__)
277 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000278 elif o in ('-v', '--verbose'):
279 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000280 elif o in ('-w', '--verbose2'):
281 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000282 elif o in ('-d', '--debug'):
283 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000284 elif o in ('-W', '--verbose3'):
285 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000286 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000287 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000288 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000289 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000290 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000291 elif o in ('-S', '--start'):
292 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000293 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000294 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000295 elif o in ('-S', '--slow'):
296 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000297 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000298 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000299 elif o == '--randseed':
300 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000301 elif o in ('-f', '--fromfile'):
302 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000303 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000304 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000305 elif o in ('-L', '--runleaks'):
306 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000307 elif o in ('-t', '--threshold'):
308 import gc
309 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000310 elif o in ('-T', '--coverage'):
311 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000312 elif o in ('-D', '--coverdir'):
313 coverdir = os.path.join(os.getcwd(), a)
314 elif o in ('-N', '--nocoverdir'):
315 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000316 elif o in ('-R', '--huntrleaks'):
317 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000318 if len(huntrleaks) not in (2, 3):
319 print(a, huntrleaks)
320 usage('-R takes 2 or 3 colon-separated arguments')
321 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000322 huntrleaks[0] = 5
323 else:
324 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000325 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000326 huntrleaks[1] = 4
327 else:
328 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000329 if len(huntrleaks) == 2 or not huntrleaks[2]:
330 huntrleaks[2:] = ["reflog.txt"]
331 # Avoid false positives due to the character cache in
332 # stringobject.c filling slowly with random data
333 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000334 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000335 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000336 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000337 u = [x.lower() for x in a.split(',')]
338 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000339 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000340 use_resources[:] = RESOURCE_NAMES
341 continue
342 remove = False
343 if r[0] == '-':
344 remove = True
345 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000346 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000347 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000348 if remove:
349 if r in use_resources:
350 use_resources.remove(r)
351 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000352 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000353 elif o in ('-n', '--nowindows'):
354 import msvcrt
355 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
356 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
357 msvcrt.SEM_NOGPFAULTERRORBOX|
358 msvcrt.SEM_NOOPENFILEERRORBOX)
359 try:
360 msvcrt.CrtSetReportMode
361 except AttributeError:
362 # release build
363 pass
364 else:
365 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
366 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
367 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000368 elif o in ('-F', '--forever'):
369 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000370 elif o in ('-j', '--multiprocess'):
371 use_mp = int(a)
372 elif o == '--slaveargs':
373 args, kwargs = json.loads(a)
374 try:
375 result = runtest(*args, **kwargs)
376 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000377 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000378 sys.stdout.flush()
379 print() # Force a newline (just in case)
380 print(json.dumps(result))
381 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000382 else:
383 print(("No handler for option {}. Please report this as a bug "
384 "at http://bugs.python.org.").format(o), file=sys.stderr)
385 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000386 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000387 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000388 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000389 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000390 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000391 usage("-l and -j don't go together!")
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000392 if use_mp and max(sys.flags):
393 # TODO: inherit the environment and the flags
394 print("Warning: flags and environment variables are ignored with -j option")
Barry Warsaw08fca522001-08-20 22:33:46 +0000395
Guido van Rossum152494a1996-12-20 03:12:20 +0000396 good = []
397 bad = []
398 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000399 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000400 environment_changed = []
Barry Warsawe11e3de1999-01-28 19:51:51 +0000401
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000402 if not quiet:
403 # Print basic platform information
404 print("==", platform.python_implementation(), *sys.version.split())
405 print("== ", platform.platform(aliased=True))
406 print("== ", os.getcwd())
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000407
Neil Schemenauerd569f232000-09-22 15:29:28 +0000408 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000409 try:
410 import gc
411 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000412 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000413 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000414 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000415 # Uncomment the line below to report garbage that is not
416 # freeable by reference counting alone. By default only
417 # garbage that is not collectable by the GC is reported.
418 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000419 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000420
Barry Warsawe11e3de1999-01-28 19:51:51 +0000421 if single:
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000422 filename = 'pynexttest'
Barry Warsawe11e3de1999-01-28 19:51:51 +0000423 try:
424 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000425 next_test = fp.read().strip()
426 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000427 fp.close()
428 except IOError:
429 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000430
431 if fromfile:
432 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000433 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Tim Petersc5000df2002-06-02 21:42:01 +0000434 for line in fp:
435 guts = line.split() # assuming no test has whitespace in its name
436 if guts and not guts[0].startswith('#'):
437 tests.extend(guts)
438 fp.close()
439
440 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000441 removepy(args)
442 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000443
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000444 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000445 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000446 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000447 for arg in args:
448 if arg in stdtests:
449 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000450 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000451 args = []
R. David Murrayef1992b2009-12-16 15:19:27 +0000452 alltests = findtests(testdir, stdtests, nottests)
453 tests = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000454 if single:
455 tests = tests[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000456 try:
457 next_single_test = alltests[alltests.index(tests[0])+1]
458 except IndexError:
459 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000460 # Remove all the tests that precede start if it's set.
461 if start:
462 try:
463 del tests[:tests.index(start)]
464 except ValueError:
465 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000466 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000467 random.seed(random_seed)
468 print("Using random seed", random_seed)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000469 random.shuffle(tests)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000470 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000471 import trace, tempfile
472 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
473 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000474 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000475
Christian Heimesb186d002008-03-18 15:15:01 +0000476 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000477 support.verbose = verbose # Tell tests to be moderately quiet
478 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000479 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000480
481 def accumulate_result(test, result):
482 ok, test_time = result
483 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000484 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000485 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000486 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000487 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000488 elif ok == ENV_CHANGED:
489 bad.append(test)
490 environment_changed.append(test)
491 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000492 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000493 elif ok == RESOURCE_DENIED:
494 skipped.append(test)
495 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000496
R. David Murray7dc72cc2009-11-14 16:13:02 +0000497 if forever:
498 def test_forever(tests=list(tests)):
499 while True:
500 for test in tests:
501 yield test
502 if bad:
503 return
504 tests = test_forever()
505 else:
506 tests = iter(tests)
507
Antoine Pitrou88909542009-06-29 13:54:42 +0000508 if use_mp:
R. David Murray9da81972009-10-19 16:53:55 +0000509 from threading import Thread
Antoine Pitrou88909542009-06-29 13:54:42 +0000510 from queue import Queue, Empty
511 from subprocess import Popen, PIPE, STDOUT
512 from collections import deque
Antoine Pitrou88909542009-06-29 13:54:42 +0000513 debug_output_pat = re.compile(r"\[\d+ refs\]$")
514 pending = deque()
515 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000516 def tests_and_args():
517 for test in tests:
518 args_tuple = (
519 (test, verbose, quiet, testdir),
520 dict(huntrleaks=huntrleaks, use_resources=use_resources,
521 debug=debug)
522 )
523 yield (test, args_tuple)
524 pending = tests_and_args()
Antoine Pitrou88909542009-06-29 13:54:42 +0000525 def work():
526 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000527 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000528 while True:
529 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000530 test, args_tuple = next(pending)
531 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000532 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000533 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000534 # -E is needed by some tests, e.g. test_import
535 popen = Popen([sys.executable, '-E', '-m', 'test.regrtest',
536 '--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000537 stdout=PIPE, stderr=PIPE,
Antoine Pitrou88909542009-06-29 13:54:42 +0000538 universal_newlines=True, close_fds=True)
R. David Murray27144602009-10-19 15:26:16 +0000539 stdout, stderr = popen.communicate()
540 # Strip last refcount output line if it exists, since it
541 # comes from the shutdown of the interpreter in the subcommand.
542 stderr = debug_output_pat.sub("", stderr)
543 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000544 if not result:
545 output.put((None, None, None, None))
546 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000547 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000548 if not quiet:
549 stdout = test+'\n'+stdout
550 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000551 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000552 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000553 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000554 workers = [Thread(target=work) for i in range(use_mp)]
555 for worker in workers:
556 worker.start()
557 finished = 0
R. David Murray7dc72cc2009-11-14 16:13:02 +0000558 try:
559 while finished < use_mp:
560 test, stdout, stderr, result = output.get()
561 if test is None:
562 finished += 1
563 continue
564 if stdout:
565 print(stdout)
566 if stderr:
567 print(stderr, file=sys.stderr)
568 if result[0] == INTERRUPTED:
569 assert result[1] == 'KeyboardInterrupt'
570 raise KeyboardInterrupt # What else?
571 accumulate_result(test, result)
572 except KeyboardInterrupt:
573 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000574 for worker in workers:
575 worker.join()
576 else:
577 for test in tests:
578 if not quiet:
579 print(test)
580 sys.stdout.flush()
581 if trace:
582 # If we're tracing code coverage, then we don't exit with status
583 # if on a false return value from main.
584 tracer.runctx('runtest(test, verbose, quiet, testdir)',
585 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000586 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000587 try:
588 result = runtest(test, verbose, quiet,
589 testdir, huntrleaks, debug)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000590 accumulate_result(test, result)
591 if verbose3 and result[0] == FAILED:
Georg Brandl96f78972009-10-27 21:02:11 +0000592 print("Re-running test {} in verbose mode".format(test))
593 runtest(test, True, quiet, testdir, huntrleaks, debug)
Antoine Pitrou88909542009-06-29 13:54:42 +0000594 except KeyboardInterrupt:
595 # print a newline separate from the ^C
596 print()
597 break
598 except:
599 raise
600 if findleaks:
601 gc.collect()
602 if gc.garbage:
603 print("Warning: test created", len(gc.garbage), end=' ')
604 print("uncollectable object(s).")
605 # move the uncollectable objects somewhere so we don't see
606 # them again
607 found_garbage.extend(gc.garbage)
608 del gc.garbage[:]
609 # Unload the newly imported modules (best effort finalization)
610 for module in sys.modules.keys():
611 if module not in save_modules and module.startswith("test."):
612 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000613
614 # The lists won't be sorted if running with -r
615 good.sort()
616 bad.sort()
617 skipped.sort()
Nick Coghlan6ead5522009-10-18 13:19:33 +0000618 environment_changed.sort()
Tim Peterse0c446b2001-10-18 21:57:37 +0000619
Guido van Rossum152494a1996-12-20 03:12:20 +0000620 if good and not quiet:
Guido van Rossum41360a41998-03-26 19:42:58 +0000621 if not bad and not skipped and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000622 print("All", end=' ')
623 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000624 if print_slow:
625 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000626 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000627 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000628 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000629 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000630 bad = sorted(set(bad) - set(environment_changed))
631 if bad:
632 print(count(len(bad), "test"), "failed:")
633 printlist(bad)
634 if environment_changed:
635 print("{} altered the execution environment:".format(
636 count(len(environment_changed), "test")))
637 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000638 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000639 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000640 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000641
Tim Petersb5b7b782001-08-12 01:20:39 +0000642 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000643 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000644 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000645 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000646 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000647 print(count(len(surprise), "skip"), \
648 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000649 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000650 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000651 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000652 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000653 print("Ask someone to teach regrtest.py about which tests are")
654 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000655
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000656 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000657 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000658 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000659 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000660 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000661 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000662 verbose = True
663 ok = runtest(test, True, quiet, testdir,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000664 huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000665 except KeyboardInterrupt:
666 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000667 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000668 break
669 except:
670 raise
671
Barry Warsawe11e3de1999-01-28 19:51:51 +0000672 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000673 if next_single_test:
674 with open(filename, 'w') as fp:
675 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000676 else:
677 os.unlink(filename)
678
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000679 if trace:
680 r = tracer.results()
681 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
682
Skip Montanaro0179a182004-06-06 15:53:18 +0000683 if runleaks:
684 os.system("leaks %d" % os.getpid())
685
Tim Peters5943b4a2003-07-23 00:30:39 +0000686 sys.exit(len(bad) > 0)
Barry Warsaw08fca522001-08-20 22:33:46 +0000687
Guido van Rossum152494a1996-12-20 03:12:20 +0000688
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000689STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000690 'test_grammar',
691 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000692 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000693 'test_builtin',
694 'test_exceptions',
695 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000696 'test_unittest',
697 'test_doctest',
698 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000699]
Guido van Rossum152494a1996-12-20 03:12:20 +0000700
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000701NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000702 'test_future1',
703 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000704}
Guido van Rossum152494a1996-12-20 03:12:20 +0000705
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000706def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000707 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000708 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000709 names = os.listdir(testdir)
710 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000711 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000712 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000713 modname, ext = os.path.splitext(name)
714 if modname[:5] == "test_" and ext == ".py" and modname not in others:
715 tests.append(modname)
716 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000717
Antoine Pitrou88909542009-06-29 13:54:42 +0000718def runtest(test, verbose, quiet,
719 testdir=None, huntrleaks=False, debug=False, use_resources=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000720 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000721
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000722 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000723 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000724 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000725 test_times -- a list of (time, test_name) pairs
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000726 testdir -- test directory
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000727 huntrleaks -- run multiple times to test for leaks; requires a debug
728 build; a triple corresponding to -R's three arguments
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000729
730 Returns one of the test result constants:
731 INTERRUPTED KeyboardInterrupt when run under -j
732 RESOURCE_DENIED test skipped because resource denied
733 SKIPPED test skipped for some other reason
734 ENV_CHANGED test failed because it changed the execution environment
735 FAILED test failed
736 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000737 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000738
Antoine Pitrou88909542009-06-29 13:54:42 +0000739 support.verbose = verbose # Tell tests to be moderately quiet
740 if use_resources is not None:
741 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000742 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000743 return runtest_inner(test, verbose, quiet,
744 testdir, huntrleaks, debug)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000745 finally:
746 cleanup_test_droppings(test, verbose)
747
Nick Coghlan6ead5522009-10-18 13:19:33 +0000748# Unit tests are supposed to leave the execution environment unchanged
749# once they complete. But sometimes tests have bugs, especially when
750# tests fail, and the changes to environment go on to mess up other
751# tests. This can cause issues with buildbot stability, since tests
752# are run in random order and so problems may appear to come and go.
753# There are a few things we can save and restore to mitigate this, and
754# the following context manager handles this task.
755
756class saved_test_environment:
757 """Save bits of the test environment and restore them at block exit.
758
759 with saved_test_environment(testname, verbose, quiet):
760 #stuff
761
762 Unless quiet is True, a warning is printed to stderr if any of
763 the saved items was changed by the test. The attribute 'changed'
764 is initially False, but is set to True if a change is detected.
765
766 If verbose is more than 1, the before and after state of changed
767 items is also printed.
768 """
769
770 changed = False
771
772 def __init__(self, testname, verbose=0, quiet=False):
773 self.testname = testname
774 self.verbose = verbose
775 self.quiet = quiet
776
777 # To add things to save and restore, add a name XXX to the resources list
778 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
779 # return the value to be saved and compared against a second call to the
780 # get function when test execution completes. restore_XXX should accept
781 # the saved value and restore the resource using it. It will be called if
782 # and only if a change in the value is detected.
783 #
784 # Note: XXX will have any '.' replaced with '_' characters when determining
785 # the corresponding method names.
786
787 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon055470a2010-02-19 15:57:10 +0000788 'os.environ', 'sys.path', 'sys.path_hooks', '__import__')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000789
790 def get_sys_argv(self):
791 return id(sys.argv), sys.argv, sys.argv[:]
792 def restore_sys_argv(self, saved_argv):
793 sys.argv = saved_argv[1]
794 sys.argv[:] = saved_argv[2]
795
796 def get_cwd(self):
797 return os.getcwd()
798 def restore_cwd(self, saved_cwd):
799 os.chdir(saved_cwd)
800
801 def get_sys_stdout(self):
802 return sys.stdout
803 def restore_sys_stdout(self, saved_stdout):
804 sys.stdout = saved_stdout
805
806 def get_sys_stderr(self):
807 return sys.stderr
808 def restore_sys_stderr(self, saved_stderr):
809 sys.stderr = saved_stderr
810
811 def get_sys_stdin(self):
812 return sys.stdin
813 def restore_sys_stdin(self, saved_stdin):
814 sys.stdin = saved_stdin
815
816 def get_os_environ(self):
817 return id(os.environ), os.environ, dict(os.environ)
818 def restore_os_environ(self, saved_environ):
819 os.environ = saved_environ[1]
820 os.environ.clear()
821 os.environ.update(saved_environ[2])
822
823 def get_sys_path(self):
824 return id(sys.path), sys.path, sys.path[:]
825 def restore_sys_path(self, saved_path):
826 sys.path = saved_path[1]
827 sys.path[:] = saved_path[2]
828
Brett Cannon055470a2010-02-19 15:57:10 +0000829 def get_sys_path_hooks(self):
830 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
831 def restore_sys_path_hooks(self, saved_hooks):
832 sys.path_hooks = saved_hooks[1]
833 sys.path_hooks[:] = saved_hooks[2]
834
835 def get___import__(self):
836 return __builtins__.__import__
837 def restore___import__(self, import_):
838 __builtins__.__import__ = import_
839
Nick Coghlan6ead5522009-10-18 13:19:33 +0000840 def resource_info(self):
841 for name in self.resources:
842 method_suffix = name.replace('.', '_')
843 get_name = 'get_' + method_suffix
844 restore_name = 'restore_' + method_suffix
845 yield name, getattr(self, get_name), getattr(self, restore_name)
846
847 def __enter__(self):
848 self.saved_values = dict((name, get()) for name, get, restore
849 in self.resource_info())
850 return self
851
852 def __exit__(self, exc_type, exc_val, exc_tb):
853 for name, get, restore in self.resource_info():
854 current = get()
855 original = self.saved_values[name]
856 # Check for changes to the resource's value
857 if current != original:
858 self.changed = True
859 restore(original)
860 if not self.quiet:
861 print("Warning -- {} was modified by {}".format(
862 name, self.testname),
863 file=sys.stderr)
864 if self.verbose > 1:
865 print(" Before: {}\n After: {} ".format(
866 original, current),
867 file=sys.stderr)
868 return False
869
870
Antoine Pitrou88909542009-06-29 13:54:42 +0000871def runtest_inner(test, verbose, quiet,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000872 testdir=None, huntrleaks=False, debug=False):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000873 support.unload(test)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000874 testdir = findtestdir(testdir)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000875
Antoine Pitrou88909542009-06-29 13:54:42 +0000876 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +0000877 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +0000878 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +0000879 if test.startswith('test.'):
880 abstest = test
881 else:
882 # Always import it from the test package
883 abstest = 'test.' + test
884 with saved_test_environment(test, verbose, quiet) as environment:
885 start_time = time.time()
886 the_package = __import__(abstest, globals(), locals(), [])
887 the_module = getattr(the_package, test)
888 # Old tests run to completion simply as a side-effect of
889 # being imported. For tests based on unittest or doctest,
890 # explicitly invoke their test_main() function (if it exists).
891 indirect_test = getattr(the_module, "test_main", None)
892 if indirect_test is not None:
893 indirect_test()
894 if huntrleaks:
895 refleak = dash_R(the_module, test, indirect_test,
896 huntrleaks)
897 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000898 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +0000899 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000900 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +0000901 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000902 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +0000903 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +0000904 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000905 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000906 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000907 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +0000908 except KeyboardInterrupt:
909 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000910 except support.TestFailed as msg:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000911 print("test", test, "failed --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000912 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000913 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +0000914 except:
Guido van Rossum41360a41998-03-26 19:42:58 +0000915 type, value = sys.exc_info()[:2]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000916 print("test", test, "crashed --", str(type) + ":", value)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000917 sys.stdout.flush()
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000918 if verbose or debug:
Guido van Rossum41360a41998-03-26 19:42:58 +0000919 traceback.print_exc(file=sys.stdout)
Guido van Rossum3cda93e2002-09-13 21:28:03 +0000920 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000921 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +0000922 else:
Collin Wintera5503d52009-05-15 01:20:21 +0000923 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000924 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +0000925 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000926 return ENV_CHANGED, test_time
927 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +0000928
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000929def cleanup_test_droppings(testname, verbose):
930 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000931 import stat
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000932
933 # Try to clean up junk commonly left behind. While tests shouldn't leave
934 # any files or directories behind, when a test fails that can be tedious
935 # for it to arrange. The consequences can be especially nasty on Windows,
936 # since if a test leaves a file open, it cannot be deleted by name (while
937 # there's nothing we can do about that here either, we can display the
938 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000939 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000940 "db_home",
941 ):
942 if not os.path.exists(name):
943 continue
944
945 if os.path.isdir(name):
946 kind, nuker = "directory", shutil.rmtree
947 elif os.path.isfile(name):
948 kind, nuker = "file", os.unlink
949 else:
950 raise SystemError("os.path says %r exists but is neither "
951 "directory nor file" % name)
952
953 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000954 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000955 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +0000956 # if we have chmod, fix possible permissions problems
957 # that might prevent cleanup
958 if (hasattr(os, 'chmod')):
959 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000960 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000961 except Exception as msg:
962 print(("%r left behind %s %r and it couldn't be "
963 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000964
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000965def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +0000966 """Run a test multiple times, looking for reference leaks.
967
968 Returns:
969 False if the test didn't leak references; True if we detected refleaks.
970 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000971 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000972 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000973
974 if not hasattr(sys, 'gettotalrefcount'):
975 raise Exception("Tracking reference leaks requires a debug build "
976 "of Python")
977
978 # Save current values for dash_R_cleanup() to restore.
979 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +0000980 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000981 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +0000982 try:
983 import zipimport
984 except ImportError:
985 zdc = None # Run unmodified on platforms without zipimport support
986 else:
987 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +0000988 abcs = {}
989 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000990 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +0000991 continue
992 for obj in abc.__subclasses__() + [abc]:
993 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000994
995 if indirect_test:
996 def run_the_test():
997 indirect_test()
998 else:
999 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001000 del sys.modules[the_module.__name__]
1001 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001002
1003 deltas = []
1004 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001005 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001006 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001007 print("beginning", repcount, "repetitions", file=sys.stderr)
1008 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001009 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001010 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001011 for i in range(repcount):
1012 rc = sys.gettotalrefcount()
1013 run_the_test()
1014 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001015 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001016 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001017 if i >= nwarmup:
1018 deltas.append(sys.gettotalrefcount() - rc - 2)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001019 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001020 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001021 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001022 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001023 sys.stderr.flush()
1024 with open(fname, "a") as refrep:
1025 print(msg, file=refrep)
1026 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001027 return True
1028 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001029
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001030def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001031 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001032 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001033 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001034 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001035 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001036 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001037
Christian Heimesdae2a892008-04-19 00:55:37 +00001038 # Clear the warnings registry, so they can be displayed again
1039 for mod in sys.modules.values():
1040 if hasattr(mod, '__warningregistry__'):
1041 del mod.__warningregistry__
1042
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001043 # Restore some original values.
1044 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001045 copyreg.dispatch_table.clear()
1046 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001047 sys.path_importer_cache.clear()
1048 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001049 try:
1050 import zipimport
1051 except ImportError:
1052 pass # Run unmodified on platforms without zipimport support
1053 else:
1054 zipimport._zip_directory_cache.clear()
1055 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001056
Christian Heimes26855632008-01-27 23:50:43 +00001057 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001058 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001059
Guido van Rossum3de862d2007-08-18 00:10:33 +00001060 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001061 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001062 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001063 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001064 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001065 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001066 obj._abc_cache.clear()
1067 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001068
Antoine Pitrou046467c2009-10-30 18:30:35 +00001069 # Flush standard output, so that buffered data is sent to the OS and
1070 # associated Python objects are reclaimed.
1071 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1072 if stream is not None:
1073 stream.flush()
1074
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001075 # Clear assorted module caches.
1076 _path_created.clear()
1077 re.purge()
1078 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001079 urllib.parse.clear_cache()
1080 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001081 linecache.clearcache()
1082 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001083 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001084 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001085 doctest.master = None
1086
1087 # Collect cyclic trash.
1088 gc.collect()
1089
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001090def warm_char_cache():
1091 s = bytes(range(256))
1092 for i in range(256):
1093 s[i:i+1]
1094
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001095def findtestdir(path=None):
1096 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001097
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001098def removepy(names):
1099 if not names:
1100 return
1101 for idx, name in enumerate(names):
1102 basename, ext = os.path.splitext(name)
1103 if ext == '.py':
1104 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001105
Guido van Rossum152494a1996-12-20 03:12:20 +00001106def count(n, word):
1107 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001108 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001109 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001110 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001111
Tim Petersa45da922001-08-12 03:45:50 +00001112def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001113 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001114
1115 Optional arg width (default 70) is the maximum line length.
1116 Optional arg indent (default 4) is the number of blanks with which to
1117 begin each line.
1118 """
1119
Tim Petersba78bc42002-07-04 19:45:06 +00001120 from textwrap import fill
1121 blanks = ' ' * indent
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001122 print(fill(' '.join(str(elt) for elt in x), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001123 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001124
Tim Petersde14a302002-04-01 05:04:46 +00001125# Map sys.platform to a string containing the basenames of tests
1126# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001127#
1128# Special cases:
1129# test_pep277
1130# The _ExpectedSkips constructor adds this to the set of expected
1131# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001132# test_timeout
1133# Controlled by test_timeout.skip_expected. Requires the network
1134# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001135#
1136# Tests that are expected to be skipped everywhere except on one platform
1137# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001138
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001139_expectations = {
1140 'win32':
1141 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001142 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001143 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001144 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001145 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001146 test_fcntl
1147 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001148 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001149 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001150 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001151 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001152 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001153 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001154 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001155 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001156 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001157 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001158 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001159 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001160 test_pty
1161 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001162 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001163 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001164 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001165 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001166 test_wait3
1167 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001168 """,
1169 'linux2':
1170 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001171 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001172 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001173 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001174 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001175 """,
Jack Jansen49a806e2001-08-28 14:49:00 +00001176 'mac':
Guido van Rossumaa782362001-09-02 03:58:41 +00001177 """
Jack Jansen67975142003-01-08 16:31:11 +00001178 test_atexit
Jack Jansen67975142003-01-08 16:31:11 +00001179 test_bz2
Guido van Rossumaa782362001-09-02 03:58:41 +00001180 test_crypt
Jack Jansenb3be2162001-11-30 14:16:36 +00001181 test_curses
Guido van Rossumaa782362001-09-02 03:58:41 +00001182 test_dbm
Guido van Rossumaa782362001-09-02 03:58:41 +00001183 test_fcntl
1184 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001185 test_epoll
Guido van Rossumaa782362001-09-02 03:58:41 +00001186 test_grp
Jack Jansenc4d6bdd2003-03-07 15:38:11 +00001187 test_ioctl
Guido van Rossumaa782362001-09-02 03:58:41 +00001188 test_largefile
Guido van Rossumaa782362001-09-02 03:58:41 +00001189 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001190 test_kqueue
Guido van Rossumaa782362001-09-02 03:58:41 +00001191 test_mmap
Guido van Rossumaa782362001-09-02 03:58:41 +00001192 test_openpty
Jack Jansen67975142003-01-08 16:31:11 +00001193 test_ossaudiodev
Guido van Rossumaa782362001-09-02 03:58:41 +00001194 test_poll
Jack Jansen67975142003-01-08 16:31:11 +00001195 test_popen
Jack Jansen5bb97e62003-02-21 22:33:55 +00001196 test_posix
Guido van Rossumaa782362001-09-02 03:58:41 +00001197 test_pty
1198 test_pwd
Jack Jansen67975142003-01-08 16:31:11 +00001199 test_resource
Guido van Rossumaa782362001-09-02 03:58:41 +00001200 test_signal
Guido van Rossumaa782362001-09-02 03:58:41 +00001201 test_sundry
Jack Jansenc4d6bdd2003-03-07 15:38:11 +00001202 test_tarfile
Guido van Rossumaa782362001-09-02 03:58:41 +00001203 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001204 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001205 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001206 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001207 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001208 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001209 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001210 test_openpty
1211 test_pyexpat
1212 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001213 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001214 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001215 'openunix8':
1216 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001217 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001218 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001219 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001220 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001221 test_openpty
1222 test_pyexpat
1223 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001224 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001225 """,
1226 'sco_sv3':
1227 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001228 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001229 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001230 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001231 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001232 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001233 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001234 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001235 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001236 test_openpty
1237 test_pyexpat
1238 test_queue
1239 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001240 test_sundry
1241 test_thread
1242 test_threaded_import
1243 test_threadedtempfile
1244 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001245 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001246 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001247 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001248 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001249 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001250 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001251 test_dbm_gnu
Jack Jansen398c2362001-12-02 21:41:36 +00001252 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001253 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001254 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001255 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001256 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001257 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001258 'sunos5':
1259 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001260 test_curses
1261 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001262 test_epoll
1263 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001264 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001265 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001266 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001267 test_zipfile
1268 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001269 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001270 'hp-ux11':
1271 """
Skip Montanarob3230212002-03-15 02:54:03 +00001272 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001273 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001274 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001275 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001276 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001277 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001278 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001279 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001280 test_openpty
1281 test_pyexpat
1282 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001283 test_zipfile
1284 test_zlib
1285 """,
Jason Tishler25115942002-12-05 15:18:15 +00001286 'cygwin':
1287 """
Jason Tishler25115942002-12-05 15:18:15 +00001288 test_curses
1289 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001290 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001291 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001292 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001293 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001294 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001295 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001296 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001297 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001298 'os2emx':
1299 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001300 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001301 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001302 test_epoll
1303 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001304 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001305 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001306 test_openpty
1307 test_ossaudiodev
1308 test_pty
1309 test_resource
1310 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001311 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001312 'freebsd4':
1313 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001314 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001315 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001316 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001317 test_ossaudiodev
1318 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001319 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001320 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001321 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001322 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001323 test_ttk_guionly
1324 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001325 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001326 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001327 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001328 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001329 'aix5':
1330 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001331 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001332 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001333 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001334 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001335 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001336 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001337 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001338 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001339 test_ttk_guionly
1340 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001341 test_zipimport
1342 test_zlib
1343 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001344 'openbsd3':
1345 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001346 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001347 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001348 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001349 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001350 test_normalization
1351 test_ossaudiodev
1352 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001353 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001354 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001355 test_ttk_guionly
1356 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001357 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001358 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001359 'netbsd3':
1360 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001361 test_ctypes
1362 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001363 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001364 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001365 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001366 test_ossaudiodev
1367 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001368 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001369 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001370 test_ttk_guionly
1371 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001372 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001373 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001374}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001375_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001376_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001377_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001378_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001379
Tim Petersb5b7b782001-08-12 01:20:39 +00001380class _ExpectedSkips:
1381 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001382 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001383 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001384
Tim Peters7c7efe92002-08-23 17:55:54 +00001385 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001386 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001387 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001388 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001389
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001390 # These are broken tests, for now skipped on every platform.
1391 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001392 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001393
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001394 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001395 if not os.path.supports_unicode_filenames:
1396 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001397
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001398 # doctest, profile and cProfile tests fail when the codec for the
1399 # fs encoding isn't built in because PyUnicode_Decode() adds two
1400 # calls into Python.
1401 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1402 if sys.getfilesystemencoding().lower() not in encs:
1403 self.expected.add('test_profile')
1404 self.expected.add('test_cProfile')
1405 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001406
Neal Norwitz55b61d22003-02-28 19:57:03 +00001407 if test_timeout.skip_expected:
1408 self.expected.add('test_timeout')
1409
Tim Petersecd79eb2003-01-29 00:35:32 +00001410 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001411 # test_sqlite is only reliable on Windows where the library
1412 # is distributed with Python
Neal Norwitz7035c982003-03-29 22:01:17 +00001413 WIN_ONLY = ["test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001414 "test_winsound", "test_startfile",
1415 "test_sqlite"]
Neal Norwitz7035c982003-03-29 22:01:17 +00001416 for skip in WIN_ONLY:
1417 self.expected.add(skip)
Tim Petersf2715e02003-02-19 02:35:07 +00001418
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001419 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001420 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001421
Tim Peters7c7efe92002-08-23 17:55:54 +00001422 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001423
1424 def isvalid(self):
1425 "Return true iff _ExpectedSkips knows about the current platform."
1426 return self.valid
1427
1428 def getexpected(self):
1429 """Return set of test names we expect to skip on current platform.
1430
1431 self.isvalid() must be true.
1432 """
1433
1434 assert self.isvalid()
1435 return self.expected
1436
Guido van Rossum152494a1996-12-20 03:12:20 +00001437if __name__ == '__main__':
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001438 # Simplification for findtestdir().
1439 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001440
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001441 # When tests are run from the Python build directory, it is best practice
1442 # to keep the test files in a subfolder. It eases the cleanup of leftover
1443 # files using command "make distclean".
1444 if sysconfig.is_python_build():
1445 parent_dir = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1446 if not os.path.exists(parent_dir):
1447 os.mkdir(parent_dir)
1448 else:
1449 parent_dir = os.path.abspath(tempfile.gettempdir())
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001450
1451 # Define a writable temp dir that will be used as cwd while running
1452 # the tests. The name of the dir includes the pid to allow parallel
1453 # testing (see the -j option).
1454 TESTCWD = 'test_python_{}'.format(os.getpid())
1455
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001456 TESTCWD = os.path.join(parent_dir, TESTCWD)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001457
1458 # Run the tests in a context manager that temporary changes the CWD to a
1459 # temporary and writable directory. If it's not possible to create or
1460 # change the CWD, the original CWD will be used. The original CWD is
1461 # available from support.SAVEDCWD.
1462 with support.temp_cwd(TESTCWD, quiet=True):
1463 main()