blob: 4444f572aab61b3339764380f810e3fdd4f2b5ab [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
Georg Brandlbe41a482011-01-05 21:47:47 +00006python -m test [options] [test_name1 [test_name2 ...]]
R. David Murray0ba81e02010-04-26 17:02:32 +00007python 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
Georg Brandlbe41a482011-01-05 21:47:47 +000017python -E -Wd -m test [options] [test_name1 ...]
R. David Murray0ba81e02010-04-26 17:02:32 +000018
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
Victor Stinner0cc8d592011-03-31 18:10:13 +020023--timeout TIMEOUT
24 -- dump the traceback and exit if a test takes more
Victor Stinner9e586c22011-06-06 12:04:36 +020025 than TIMEOUT seconds; disabled if TIMEOUT is negative
26 or equals to zero
Brian Curtinf6643452011-04-28 17:45:17 -050027--wait -- wait for user input, e.g., allow a debugger to be attached
Georg Brandl96f78972009-10-27 21:02:11 +000028
29Verbosity
30
31-v/--verbose -- run tests in verbose mode with output to stdout
32-w/--verbose2 -- re-run failed tests in verbose mode
Victor Stinnera7c33e52011-06-29 13:00:54 +020033-W/--verbose3 -- display test output on failure
Georg Brandl96f78972009-10-27 21:02:11 +000034-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000035-q/--quiet -- no output unless one or more tests fail
Georg Brandl96f78972009-10-27 21:02:11 +000036-S/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000037 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000038
39Selecting tests
40
R. David Murray0ba81e02010-04-26 17:02:32 +000041-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000042 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000043-f/--fromfile -- read names of tests to run from a file (see below)
44-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000045-s/--single -- single step through a set of tests (see below)
Georg Brandl96f78972009-10-27 21:02:11 +000046-u/--use RES1,RES2,...
47 -- specify which special resource intensive tests to run
48-M/--memlimit LIMIT
49 -- run very large memory-consuming tests
R David Murrayb588f8d2011-03-24 14:42:58 -040050 --testdir DIR
51 -- execute test files in the specified directory (instead
52 of the Python stdlib test suite)
Georg Brandl96f78972009-10-27 21:02:11 +000053
54Special runs
55
56-l/--findleaks -- if GC is available detect tests that leak memory
57-L/--runleaks -- run the leaks(1) command just before exit
58-R/--huntrleaks RUNCOUNTS
59 -- search for reference leaks (needs debug build, v. slow)
60-j/--multiprocess PROCESSES
61 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000062-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000063-D/--coverdir DIRECTORY
64 -- Directory where coverage files are put
65-N/--nocoverdir -- Put coverage files alongside modules
66-t/--threshold THRESHOLD
67 -- call gc.set_threshold(THRESHOLD)
68-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000069-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000070
R. David Murray0ba81e02010-04-26 17:02:32 +000071
72Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000073
Collin Winterfd12f492009-03-29 04:05:05 +000074-r randomizes test execution order. You can use --randseed=int to provide a
75int seed value for the randomizer; this is useful for reproducing troublesome
76test orders.
77
R. David Murray0ba81e02010-04-26 17:02:32 +000078-s On the first invocation of regrtest using -s, the first test file found
79or the first test file given on the command line is run, and the name of
80the next test is recorded in a file named pynexttest. If run from the
81Python build directory, pynexttest is located in the 'build' subdirectory,
82otherwise it is located in tempfile.gettempdir(). On subsequent runs,
83the test in pynexttest is run, and the next test is written to pynexttest.
84When the last test has been run, pynexttest is deleted. In this way it
85is possible to single step through the test files. This is useful when
86doing memory analysis on the Python interpreter, which process tends to
87consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000088
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000089-S is used to continue running tests after an aborted run. It will
90maintain the order a standard run (ie, this assumes -r is not used).
91This is useful after the tests have prematurely stopped for some external
92reason and you want to start running from where you left off rather
93than starting from the beginning.
94
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000095-f reads the names of tests from the file given as f's argument, one
96or more test names per line. Whitespace is ignored. Blank lines and
97lines beginning with '#' are ignored. This is especially useful for
98whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000099
Skip Montanaro0179a182004-06-06 15:53:18 +0000100-L causes the leaks(1) command to be run just before exit if it exists.
101leaks(1) is available on Mac OS X and presumably on some other
102FreeBSD-derived systems.
103
Michael W. Hudson61147f62004-08-03 11:33:28 +0000104-R runs each test several times and examines sys.gettotalrefcount() to
105see if the test appears to be leaking references. The argument should
106be of the form stab:run:fname where 'stab' is the number of times the
107test is run to let gettotalrefcount settle down, 'run' is the number
108of times further it is run and 'fname' is the name of the file the
109reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000110"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000111
Thomas Wouters477c8d52006-05-27 19:21:47 +0000112-M runs tests that require an exorbitant amount of memory. These tests
113typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001142 billion objects, which only works on 64-bit systems. There are also some
115tests that try to exhaust the address space of the process, which only makes
116sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000117which is a string in the form of '2.5Gb', determines howmuch memory the
118tests will limit themselves to (but they may go slightly over.) The number
119shouldn't be more memory than the machine has (including swap memory). You
120should also keep in mind that swap memory is generally much, much slower
121than RAM, and setting memlimit to all available RAM or higher will heavily
122tax the machine. On the other hand, it is no use running these tests with a
123limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
124to use more than memlimit memory will be skipped. The big-memory tests
125generally run very, very long.
126
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000127-u is used to specify which special resource intensive tests to run,
128such as those requiring large file support or network connectivity.
129The argument is a comma-separated list of words indicating the
130resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000131
Fred Drake3a15dac2002-04-11 16:39:16 +0000132 all - Enable all special resources.
133
Guido van Rossum315aa362003-03-11 14:46:48 +0000134 audio - Tests that use the audio device. (There are known
135 cases of broken audio drivers that can crash Python or
136 even the Linux kernel.)
137
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000138 curses - Tests that use curses and will modify the terminal's
139 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000140
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000141 largefile - It is okay to run some test that may create huge
142 files. These tests can take a long time and may
143 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000144
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000145 network - It is okay to run tests that use external network
146 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000147
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000148 decimal - Test the decimal module against a large suite that
149 verifies compliance with standards.
150
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000151 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000152
Tim Peterseba28be2005-03-28 01:08:02 +0000153 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000154
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000155 urlfetch - It is okay to download files required on testing.
156
Guilherme Polo9de29af2009-01-28 20:40:48 +0000157 gui - Run tests that require a running GUI.
158
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000159To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000160example, to run all the tests except for the gui tests, give the
161option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000162"""
163
Brett Cannon45071902010-06-14 22:22:54 +0000164import builtins
Victor Stinner8313d6a2011-06-29 15:22:26 +0200165import errno
Victor Stinner024e37a2011-03-31 01:31:06 +0200166import faulthandler
Guido van Rossum152494a1996-12-20 03:12:20 +0000167import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200168import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000169import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200170import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000171import os
Victor Stinner8313d6a2011-06-29 15:22:26 +0200172import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000173import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000174import re
Victor Stinnercb41cda2011-07-13 23:47:21 +0200175import signal
Christian Heimesb186d002008-03-18 15:15:01 +0000176import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000177import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200178import tempfile
179import time
180import traceback
181import unittest
182import warnings
183from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000184
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200185try:
186 import threading
187except ImportError:
188 threading = None
189try:
190 import multiprocessing.process
191except ImportError:
192 multiprocessing = None
193
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000194
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000195# Some times __path__ and __file__ are not absolute (e.g. while running from
196# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
197# imports might fail. This affects only the modules imported before os.chdir().
198# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
199# they are found in the CWD their __file__ and __path__ will be relative (this
200# happens before the chdir). All the modules imported after the chdir, are
201# not found in the CWD, and since the other paths in sys.path[1:] are absolute
202# (site.py absolutize them), the __file__ and __path__ will be absolute too.
203# Therefore it is necessary to absolutize manually the __file__ and __path__ of
204# the packages to prevent later imports to fail when the CWD is different.
205for module in sys.modules.values():
206 if hasattr(module, '__path__'):
207 module.__path__ = [os.path.abspath(path) for path in module.__path__]
208 if hasattr(module, '__file__'):
209 module.__file__ = os.path.abspath(module.__file__)
210
Guido van Rossumdc15c272002-08-12 21:55:51 +0000211
Guido van Rossumbb484652002-12-02 09:56:21 +0000212# MacOSX (a.k.a. Darwin) has a default stack size that is too small
213# for deeply recursive regular expressions. We see this as crashes in
214# the Python test suite when running test_re.py and test_sre.py. The
215# fix is to set the stack limit to 2048.
216# This approach may also be useful for other Unixy platforms that
217# suffer from small default stack limits.
218if sys.platform == 'darwin':
219 try:
220 import resource
221 except ImportError:
222 pass
223 else:
224 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
225 newsoft = min(hard, max(soft, 1024*2048))
226 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
227
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000228# Test result constants.
229PASSED = 1
230FAILED = 0
231ENV_CHANGED = -1
232SKIPPED = -2
233RESOURCE_DENIED = -3
234INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200235CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000236
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000237from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000238
Georg Brandl1158a332009-06-04 09:30:30 +0000239RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000240 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000241
Florent Xiclunaec882212010-08-09 16:56:43 +0000242TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000243
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000244def usage(msg):
245 print(msg, file=sys.stderr)
246 print("Use --help for usage", file=sys.stderr)
247 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000248
249
Antoine Pitrou88909542009-06-29 13:54:42 +0000250def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000251 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000252 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000253 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000254 random_seed=None, use_mp=None, verbose3=False, forever=False,
Victor Stinner9bf133c2011-04-13 01:06:27 +0200255 header=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000256 """Execute a test suite.
257
Thomas Wouters7e474022000-07-16 12:04:32 +0000258 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000259 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000260
261 tests -- a list of strings containing test names (optional)
262 testdir -- the directory in which to look for tests (optional)
263
264 Users other than the Python test suite will certainly want to
265 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000266 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000267
268 If the tests argument is omitted, the tests listed on the
269 command-line will be used. If that's empty, too, then all *.py
270 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000271
Antoine Pitrou88909542009-06-29 13:54:42 +0000272 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000273 single, randomize, findleaks, use_resources, trace, coverdir,
274 print_slow, and random_seed) allow programmers calling main()
275 directly to set the values that would normally be set by flags
276 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000277 """
Victor Stinner3c18f252011-05-22 15:27:14 +0200278
Victor Stinnercb41cda2011-07-13 23:47:21 +0200279 # Display the Python traceback on fatal errors (e.g. segfault)
Victor Stinner3c18f252011-05-22 15:27:14 +0200280 faulthandler.enable(all_threads=True)
281
Victor Stinnercb41cda2011-07-13 23:47:21 +0200282 # Display the Python traceback on SIGALRM or SIGUSR1 signal
283 signals = []
284 if hasattr(signal, 'SIGALRM'):
285 signals.append(signal.SIGALRM)
286 if hasattr(signal, 'SIGUSR1'):
287 signals.append(signal.SIGUSR1)
288 for signum in signals:
289 faulthandler.register(signum, chain=True)
290
Victor Stinner1802d3f2010-05-19 17:11:19 +0000291 replace_stdout()
292
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000293 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000294 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000295 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000296 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
297 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
298 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000299 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000300 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Brian Curtinf6643452011-04-28 17:45:17 -0500301 'start=', 'nowindows', 'header', 'testdir=', 'timeout=', 'wait'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000302 except getopt.error as msg:
303 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000304
305 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000306 if random_seed is None:
307 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000308 if use_resources is None:
309 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000310 debug = False
311 start = None
Victor Stinner30196882011-06-03 12:53:26 +0200312 timeout = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000313 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000314 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000315 print(__doc__)
316 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000317 elif o in ('-v', '--verbose'):
318 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000319 elif o in ('-w', '--verbose2'):
320 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000321 elif o in ('-d', '--debug'):
322 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000323 elif o in ('-W', '--verbose3'):
324 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000325 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000326 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000327 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000328 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000329 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000330 elif o in ('-S', '--start'):
331 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000332 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000333 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000334 elif o in ('-S', '--slow'):
335 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000336 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000337 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000338 elif o == '--randseed':
339 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000340 elif o in ('-f', '--fromfile'):
341 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000342 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000343 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000344 elif o in ('-L', '--runleaks'):
345 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000346 elif o in ('-t', '--threshold'):
347 import gc
348 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000349 elif o in ('-T', '--coverage'):
350 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000351 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400352 # CWD is replaced with a temporary dir before calling main(), so we
353 # need join it with the saved CWD so it goes where the user expects.
354 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000355 elif o in ('-N', '--nocoverdir'):
356 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000357 elif o in ('-R', '--huntrleaks'):
358 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000359 if len(huntrleaks) not in (2, 3):
360 print(a, huntrleaks)
361 usage('-R takes 2 or 3 colon-separated arguments')
362 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000363 huntrleaks[0] = 5
364 else:
365 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000366 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000367 huntrleaks[1] = 4
368 else:
369 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000370 if len(huntrleaks) == 2 or not huntrleaks[2]:
371 huntrleaks[2:] = ["reflog.txt"]
372 # Avoid false positives due to the character cache in
373 # stringobject.c filling slowly with random data
374 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000375 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000376 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000377 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000378 u = [x.lower() for x in a.split(',')]
379 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000380 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000381 use_resources[:] = RESOURCE_NAMES
382 continue
383 remove = False
384 if r[0] == '-':
385 remove = True
386 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000387 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000388 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000389 if remove:
390 if r in use_resources:
391 use_resources.remove(r)
392 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000393 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000394 elif o in ('-n', '--nowindows'):
395 import msvcrt
396 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
397 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
398 msvcrt.SEM_NOGPFAULTERRORBOX|
399 msvcrt.SEM_NOOPENFILEERRORBOX)
400 try:
401 msvcrt.CrtSetReportMode
402 except AttributeError:
403 # release build
404 pass
405 else:
406 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
407 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
408 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000409 elif o in ('-F', '--forever'):
410 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000411 elif o in ('-j', '--multiprocess'):
412 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100413 if use_mp <= 0:
414 try:
415 import multiprocessing
416 # Use all cores + extras for tests that like to sleep
417 use_mp = 2 + multiprocessing.cpu_count()
418 except (ImportError, NotImplementedError):
419 use_mp = 3
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000420 elif o == '--header':
421 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000422 elif o == '--slaveargs':
423 args, kwargs = json.loads(a)
424 try:
425 result = runtest(*args, **kwargs)
426 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000427 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000428 sys.stdout.flush()
429 print() # Force a newline (just in case)
430 print(json.dumps(result))
431 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400432 elif o == '--testdir':
433 # CWD is replaced with a temporary dir before calling main(), so we
434 # join it with the saved CWD so it ends up where the user expects.
435 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200436 elif o == '--timeout':
Victor Stinner88345682011-06-06 12:14:23 +0200437 if hasattr(faulthandler, 'dump_tracebacks_later'):
438 timeout = float(a)
439 if timeout <= 0:
440 timeout = None
441 else:
442 print("Warning: The timeout option requires "
443 "faulthandler.dump_tracebacks_later")
Victor Stinner9e586c22011-06-06 12:04:36 +0200444 timeout = None
Brian Curtinf6643452011-04-28 17:45:17 -0500445 elif o == '--wait':
446 input("Press any key to continue...")
R. David Murray35768ad2009-11-15 00:23:21 +0000447 else:
448 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000449 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000450 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000451 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000452 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000453 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000454 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000455 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000456 usage("-l and -j don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000457
Guido van Rossum152494a1996-12-20 03:12:20 +0000458 good = []
459 bad = []
460 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000461 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000462 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000463 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000464
Neil Schemenauerd569f232000-09-22 15:29:28 +0000465 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000466 try:
467 import gc
468 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000469 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000470 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000471 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000472 # Uncomment the line below to report garbage that is not
473 # freeable by reference counting alone. By default only
474 # garbage that is not collectable by the GC is reported.
475 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000476 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000477
Barry Warsawe11e3de1999-01-28 19:51:51 +0000478 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000479 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000480 try:
481 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000482 next_test = fp.read().strip()
483 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000484 fp.close()
485 except IOError:
486 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000487
488 if fromfile:
489 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000490 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000491 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000492 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000493 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000494 guts = line.split() # assuming no test has whitespace in its name
495 if guts and not guts[0].startswith('#'):
496 tests.extend(guts)
497 fp.close()
498
499 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000500 removepy(args)
501 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000502
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000503 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000504 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000505 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000506 for arg in args:
507 if arg in stdtests:
508 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000509 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000510 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000511
512 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000513 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000514 # Print basic platform information
515 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000516 print("== ", platform.platform(aliased=True),
517 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000518 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000519 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000520
R David Murrayb588f8d2011-03-24 14:42:58 -0400521 # if testdir is set, then we are not running the python tests suite, so
522 # don't add default tests to be executed or skipped (pass empty values)
523 if testdir:
524 alltests = findtests(testdir, list(), set())
525 else:
526 alltests = findtests(testdir, stdtests, nottests)
527
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000528 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000529 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000530 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000531 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000532 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000533 except IndexError:
534 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000535 # Remove all the tests that precede start if it's set.
536 if start:
537 try:
538 del tests[:tests.index(start)]
539 except ValueError:
540 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000541 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000542 random.seed(random_seed)
543 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000544 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000545 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000546 import trace, tempfile
547 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
548 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000549 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000550
Christian Heimesb186d002008-03-18 15:15:01 +0000551 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000552 support.verbose = verbose # Tell tests to be moderately quiet
553 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000554 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000555
556 def accumulate_result(test, result):
557 ok, test_time = result
558 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000559 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000560 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000561 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000562 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000563 elif ok == ENV_CHANGED:
564 bad.append(test)
565 environment_changed.append(test)
566 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000567 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000568 elif ok == RESOURCE_DENIED:
569 skipped.append(test)
570 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000571
R. David Murray7dc72cc2009-11-14 16:13:02 +0000572 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000573 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000574 while True:
575 for test in tests:
576 yield test
577 if bad:
578 return
579 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000580 test_count = ''
581 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000582 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000583 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000584 test_count = '/{}'.format(len(selected))
585 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000586
Antoine Pitrou88909542009-06-29 13:54:42 +0000587 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000588 try:
589 from threading import Thread
590 except ImportError:
591 print("Multiprocess option requires thread support")
592 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000593 from queue import Queue
594 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000595 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000596 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000597 def tests_and_args():
598 for test in tests:
599 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000600 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000601 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200602 debug=debug, output_on_failure=verbose3,
603 timeout=timeout)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000604 )
605 yield (test, args_tuple)
606 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000607 opt_args = support.args_from_interpreter_flags()
608 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000609 def work():
610 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000611 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000612 while True:
613 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000614 test, args_tuple = next(pending)
615 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000616 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000617 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000618 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000619 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000620 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000621 universal_newlines=True,
622 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000623 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200624 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000625 # Strip last refcount output line if it exists, since it
626 # comes from the shutdown of the interpreter in the subcommand.
627 stderr = debug_output_pat.sub("", stderr)
628 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200629 if retcode != 0:
630 result = (CHILD_ERROR, "Exit code %s" % retcode)
631 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
632 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000633 if not result:
634 output.put((None, None, None, None))
635 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000636 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000637 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000638 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000639 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000640 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000641 workers = [Thread(target=work) for i in range(use_mp)]
642 for worker in workers:
643 worker.start()
644 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000645 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000646 try:
647 while finished < use_mp:
648 test, stdout, stderr, result = output.get()
649 if test is None:
650 finished += 1
651 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200652 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000653 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300654 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
655 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200656 test_count_width, test_index, test_count,
657 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000658 if stdout:
659 print(stdout)
660 if stderr:
661 print(stderr, file=sys.stderr)
662 if result[0] == INTERRUPTED:
663 assert result[1] == 'KeyboardInterrupt'
664 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200665 if result[0] == CHILD_ERROR:
Victor Stinner571e8fd2011-05-01 22:57:43 +0200666 raise Exception("Child error on {}: {}".format(test, result[1]))
Georg Brandldee7b852010-08-02 18:59:52 +0000667 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000668 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000669 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000670 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000671 for worker in workers:
672 worker.join()
673 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000674 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000675 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300676 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
677 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200678 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000679 sys.stdout.flush()
680 if trace:
681 # If we're tracing code coverage, then we don't exit with status
682 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200683 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000684 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000685 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000686 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100687 result = runtest(test, verbose, quiet, huntrleaks, debug,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200688 output_on_failure=verbose3, timeout=timeout)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000689 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000690 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000691 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000692 break
693 except:
694 raise
695 if findleaks:
696 gc.collect()
697 if gc.garbage:
698 print("Warning: test created", len(gc.garbage), end=' ')
699 print("uncollectable object(s).")
700 # move the uncollectable objects somewhere so we don't see
701 # them again
702 found_garbage.extend(gc.garbage)
703 del gc.garbage[:]
704 # Unload the newly imported modules (best effort finalization)
705 for module in sys.modules.keys():
706 if module not in save_modules and module.startswith("test."):
707 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000708
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000709 if interrupted:
710 # print a newline after ^C
711 print()
712 print("Test suite interrupted by signal SIGINT.")
713 omitted = set(selected) - set(good) - set(bad) - set(skipped)
714 print(count(len(omitted), "test"), "omitted:")
715 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000716 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000717 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000718 print("All", end=' ')
719 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000720 if print_slow:
721 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000722 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000723 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000724 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000725 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000726 bad = sorted(set(bad) - set(environment_changed))
727 if bad:
728 print(count(len(bad), "test"), "failed:")
729 printlist(bad)
730 if environment_changed:
731 print("{} altered the execution environment:".format(
732 count(len(environment_changed), "test")))
733 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000734 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000735 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000736 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000737
Tim Petersb5b7b782001-08-12 01:20:39 +0000738 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000739 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000740 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000741 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000742 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000743 print(count(len(surprise), "skip"), \
744 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000745 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000746 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000747 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000748 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000749 print("Ask someone to teach regrtest.py about which tests are")
750 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000751
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000752 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000753 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000754 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000755 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000756 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000757 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000758 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200759 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000760 except KeyboardInterrupt:
761 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000762 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000763 break
764 except:
765 raise
766
Barry Warsawe11e3de1999-01-28 19:51:51 +0000767 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000768 if next_single_test:
769 with open(filename, 'w') as fp:
770 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000771 else:
772 os.unlink(filename)
773
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000774 if trace:
775 r = tracer.results()
776 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
777
Skip Montanaro0179a182004-06-06 15:53:18 +0000778 if runleaks:
779 os.system("leaks %d" % os.getpid())
780
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000781 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000782
Guido van Rossum152494a1996-12-20 03:12:20 +0000783
R David Murrayb588f8d2011-03-24 14:42:58 -0400784# small set of tests to determine if we have a basically functioning interpreter
785# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000786STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000787 'test_grammar',
788 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000789 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000790 'test_builtin',
791 'test_exceptions',
792 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000793 'test_unittest',
794 'test_doctest',
795 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000796]
Guido van Rossum152494a1996-12-20 03:12:20 +0000797
R David Murrayb588f8d2011-03-24 14:42:58 -0400798# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400799NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000800
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000801def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000802 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000803 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000804 names = os.listdir(testdir)
805 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000806 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000807 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400808 mod, ext = os.path.splitext(name)
809 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
810 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000811 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000812
Victor Stinnerf58087b2010-05-02 17:24:51 +0000813def replace_stdout():
814 """Set stdout encoder error handler to backslashreplace (as stderr error
815 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000816 import atexit
817
Victor Stinnerf58087b2010-05-02 17:24:51 +0000818 stdout = sys.stdout
819 sys.stdout = open(stdout.fileno(), 'w',
820 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000821 errors="backslashreplace",
Victor Stinnerbe621032011-05-25 02:01:55 +0200822 closefd=False,
823 newline='\n')
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000824
825 def restore_stdout():
826 sys.stdout.close()
827 sys.stdout = stdout
828 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000829
Antoine Pitrou88909542009-06-29 13:54:42 +0000830def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100831 huntrleaks=False, debug=False, use_resources=None,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200832 output_on_failure=False, timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000833 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000834
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000835 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000836 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000837 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000838 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000839 huntrleaks -- run multiple times to test for leaks; requires a debug
840 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200841 output_on_failure -- if true, display test output on failure
Victor Stinner0cc8d592011-03-31 18:10:13 +0200842 timeout -- dump the traceback and exit if a test takes more than
843 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000844
845 Returns one of the test result constants:
846 INTERRUPTED KeyboardInterrupt when run under -j
847 RESOURCE_DENIED test skipped because resource denied
848 SKIPPED test skipped for some other reason
849 ENV_CHANGED test failed because it changed the execution environment
850 FAILED test failed
851 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000852 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000853
Antoine Pitrou88909542009-06-29 13:54:42 +0000854 if use_resources is not None:
855 support.use_resources = use_resources
Victor Stinner30196882011-06-03 12:53:26 +0200856 use_timeout = (timeout is not None)
Victor Stinner7d648a02011-03-31 18:27:50 +0200857 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200858 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000859 try:
Victor Stinner8313d6a2011-06-29 15:22:26 +0200860 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200861 support.verbose = True
862
863 # Reuse the same instance to all calls to runtest(). Some
864 # tests keep a reference to sys.stdout or sys.stderr
865 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200866 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200867 stream = io.StringIO()
868 runtest.stringio = stream
869 else:
870 stream = runtest.stringio
871 stream.seek(0)
872 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200873
874 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200875 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200876 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200877 sys.stdout = stream
878 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200879 result = runtest_inner(test, verbose, quiet, huntrleaks,
880 debug, display_failure=False)
881 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200882 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200883 orig_stderr.write(output)
884 orig_stderr.flush()
885 finally:
886 sys.stdout = orig_stdout
887 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200888 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200889 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200890 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
891 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100892 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000893 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200894 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200895 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000896 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200897runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000898
Nick Coghlan6ead5522009-10-18 13:19:33 +0000899# Unit tests are supposed to leave the execution environment unchanged
900# once they complete. But sometimes tests have bugs, especially when
901# tests fail, and the changes to environment go on to mess up other
902# tests. This can cause issues with buildbot stability, since tests
903# are run in random order and so problems may appear to come and go.
904# There are a few things we can save and restore to mitigate this, and
905# the following context manager handles this task.
906
907class saved_test_environment:
908 """Save bits of the test environment and restore them at block exit.
909
910 with saved_test_environment(testname, verbose, quiet):
911 #stuff
912
913 Unless quiet is True, a warning is printed to stderr if any of
914 the saved items was changed by the test. The attribute 'changed'
915 is initially False, but is set to True if a change is detected.
916
917 If verbose is more than 1, the before and after state of changed
918 items is also printed.
919 """
920
921 changed = False
922
923 def __init__(self, testname, verbose=0, quiet=False):
924 self.testname = testname
925 self.verbose = verbose
926 self.quiet = quiet
927
928 # To add things to save and restore, add a name XXX to the resources list
929 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
930 # return the value to be saved and compared against a second call to the
931 # get function when test execution completes. restore_XXX should accept
932 # the saved value and restore the resource using it. It will be called if
933 # and only if a change in the value is detected.
934 #
935 # Note: XXX will have any '.' replaced with '_' characters when determining
936 # the corresponding method names.
937
938 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000939 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000940 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200941 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200942 'sys.warnoptions', 'threading._dangling',
943 'multiprocessing.process._dangling')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000944
945 def get_sys_argv(self):
946 return id(sys.argv), sys.argv, sys.argv[:]
947 def restore_sys_argv(self, saved_argv):
948 sys.argv = saved_argv[1]
949 sys.argv[:] = saved_argv[2]
950
951 def get_cwd(self):
952 return os.getcwd()
953 def restore_cwd(self, saved_cwd):
954 os.chdir(saved_cwd)
955
956 def get_sys_stdout(self):
957 return sys.stdout
958 def restore_sys_stdout(self, saved_stdout):
959 sys.stdout = saved_stdout
960
961 def get_sys_stderr(self):
962 return sys.stderr
963 def restore_sys_stderr(self, saved_stderr):
964 sys.stderr = saved_stderr
965
966 def get_sys_stdin(self):
967 return sys.stdin
968 def restore_sys_stdin(self, saved_stdin):
969 sys.stdin = saved_stdin
970
971 def get_os_environ(self):
972 return id(os.environ), os.environ, dict(os.environ)
973 def restore_os_environ(self, saved_environ):
974 os.environ = saved_environ[1]
975 os.environ.clear()
976 os.environ.update(saved_environ[2])
977
978 def get_sys_path(self):
979 return id(sys.path), sys.path, sys.path[:]
980 def restore_sys_path(self, saved_path):
981 sys.path = saved_path[1]
982 sys.path[:] = saved_path[2]
983
Brett Cannon055470a2010-02-19 15:57:10 +0000984 def get_sys_path_hooks(self):
985 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
986 def restore_sys_path_hooks(self, saved_hooks):
987 sys.path_hooks = saved_hooks[1]
988 sys.path_hooks[:] = saved_hooks[2]
989
Brett Cannon31f59292011-02-21 19:29:56 +0000990 def get_sys_gettrace(self):
991 return sys.gettrace()
992 def restore_sys_gettrace(self, trace_fxn):
993 sys.settrace(trace_fxn)
994
Brett Cannon055470a2010-02-19 15:57:10 +0000995 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000996 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000997 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000998 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000999
Brett Cannon29c0e4f2010-03-20 22:22:57 +00001000 def get_warnings_filters(self):
1001 return id(warnings.filters), warnings.filters, warnings.filters[:]
1002 def restore_warnings_filters(self, saved_filters):
1003 warnings.filters = saved_filters[1]
1004 warnings.filters[:] = saved_filters[2]
1005
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001006 def get_asyncore_socket_map(self):
1007 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +00001008 # XXX Making a copy keeps objects alive until __exit__ gets called.
1009 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001010 def restore_asyncore_socket_map(self, saved_map):
1011 asyncore = sys.modules.get('asyncore')
1012 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +00001013 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001014 asyncore.socket_map.update(saved_map)
1015
Nick Coghlan7d819752010-12-05 06:45:03 +00001016 def get_logging__handlers(self):
1017 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001018 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001019 def restore_logging__handlers(self, saved_handlers):
1020 # Can't easily revert the logging state
1021 pass
1022
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001023 def get_logging__handlerList(self):
1024 # _handlerList is a list of weakrefs to handlers
1025 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1026 def restore_logging__handlerList(self, saved_handlerList):
1027 # Can't easily revert the logging state
1028 pass
1029
Ezio Melotti0123e052011-03-20 15:09:26 +02001030 def get_sys_warnoptions(self):
1031 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1032 def restore_sys_warnoptions(self, saved_options):
1033 sys.warnoptions = saved_options[1]
1034 sys.warnoptions[:] = saved_options[2]
1035
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02001036 # Controlling dangling references to Thread objects can make it easier
1037 # to track reference leaks.
1038 def get_threading__dangling(self):
1039 if not threading:
1040 return None
1041 # This copies the weakrefs without making any strong reference
1042 return threading._dangling.copy()
1043 def restore_threading__dangling(self, saved):
1044 if not threading:
1045 return
1046 threading._dangling.clear()
1047 threading._dangling.update(saved)
1048
1049 # Same for Process objects
1050 def get_multiprocessing_process__dangling(self):
1051 if not multiprocessing:
1052 return None
1053 # This copies the weakrefs without making any strong reference
1054 return multiprocessing.process._dangling.copy()
1055 def restore_multiprocessing_process__dangling(self, saved):
1056 if not multiprocessing:
1057 return
1058 multiprocessing.process._dangling.clear()
1059 multiprocessing.process._dangling.update(saved)
1060
Nick Coghlan6ead5522009-10-18 13:19:33 +00001061 def resource_info(self):
1062 for name in self.resources:
1063 method_suffix = name.replace('.', '_')
1064 get_name = 'get_' + method_suffix
1065 restore_name = 'restore_' + method_suffix
1066 yield name, getattr(self, get_name), getattr(self, restore_name)
1067
1068 def __enter__(self):
1069 self.saved_values = dict((name, get()) for name, get, restore
1070 in self.resource_info())
1071 return self
1072
1073 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001074 saved_values = self.saved_values
1075 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001076 for name, get, restore in self.resource_info():
1077 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001078 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001079 # Check for changes to the resource's value
1080 if current != original:
1081 self.changed = True
1082 restore(original)
1083 if not self.quiet:
1084 print("Warning -- {} was modified by {}".format(
1085 name, self.testname),
1086 file=sys.stderr)
1087 if self.verbose > 1:
1088 print(" Before: {}\n After: {} ".format(
1089 original, current),
1090 file=sys.stderr)
1091 return False
1092
1093
Victor Stinnera7c33e52011-06-29 13:00:54 +02001094def runtest_inner(test, verbose, quiet,
1095 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001096 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001097
Antoine Pitrou88909542009-06-29 13:54:42 +00001098 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001099 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001100 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001101 if test.startswith('test.'):
1102 abstest = test
1103 else:
1104 # Always import it from the test package
1105 abstest = 'test.' + test
1106 with saved_test_environment(test, verbose, quiet) as environment:
1107 start_time = time.time()
1108 the_package = __import__(abstest, globals(), locals(), [])
1109 the_module = getattr(the_package, test)
1110 # Old tests run to completion simply as a side-effect of
1111 # being imported. For tests based on unittest or doctest,
1112 # explicitly invoke their test_main() function (if it exists).
1113 indirect_test = getattr(the_module, "test_main", None)
1114 if indirect_test is not None:
1115 indirect_test()
1116 if huntrleaks:
1117 refleak = dash_R(the_module, test, indirect_test,
1118 huntrleaks)
1119 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001120 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001121 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001122 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001123 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001124 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001125 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001126 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001127 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001128 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001129 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001130 except KeyboardInterrupt:
1131 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001132 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001133 if display_failure:
1134 print("test", test, "failed --", msg, file=sys.stderr)
1135 else:
1136 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001137 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001138 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001139 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001140 msg = traceback.format_exc()
1141 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001142 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001143 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001144 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001145 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001146 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001147 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001148 return ENV_CHANGED, test_time
1149 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001150
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001151def cleanup_test_droppings(testname, verbose):
1152 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001153 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001154 import gc
1155
1156 # First kill any dangling references to open files etc.
1157 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001158 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001159 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001160
1161 # Try to clean up junk commonly left behind. While tests shouldn't leave
1162 # any files or directories behind, when a test fails that can be tedious
1163 # for it to arrange. The consequences can be especially nasty on Windows,
1164 # since if a test leaves a file open, it cannot be deleted by name (while
1165 # there's nothing we can do about that here either, we can display the
1166 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001167 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001168 "db_home",
1169 ):
1170 if not os.path.exists(name):
1171 continue
1172
1173 if os.path.isdir(name):
1174 kind, nuker = "directory", shutil.rmtree
1175 elif os.path.isfile(name):
1176 kind, nuker = "file", os.unlink
1177 else:
1178 raise SystemError("os.path says %r exists but is neither "
1179 "directory nor file" % name)
1180
1181 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001182 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001183 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001184 # if we have chmod, fix possible permissions problems
1185 # that might prevent cleanup
1186 if (hasattr(os, 'chmod')):
1187 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001188 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001189 except Exception as msg:
1190 print(("%r left behind %s %r and it couldn't be "
1191 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001192
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001193def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001194 """Run a test multiple times, looking for reference leaks.
1195
1196 Returns:
1197 False if the test didn't leak references; True if we detected refleaks.
1198 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001199 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001200 import copyreg
1201 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001202
1203 if not hasattr(sys, 'gettotalrefcount'):
1204 raise Exception("Tracking reference leaks requires a debug build "
1205 "of Python")
1206
1207 # Save current values for dash_R_cleanup() to restore.
1208 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001209 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001210 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001211 try:
1212 import zipimport
1213 except ImportError:
1214 zdc = None # Run unmodified on platforms without zipimport support
1215 else:
1216 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001217 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001218 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001219 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001220 continue
1221 for obj in abc.__subclasses__() + [abc]:
1222 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001223
1224 if indirect_test:
1225 def run_the_test():
1226 indirect_test()
1227 else:
1228 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001229 del sys.modules[the_module.__name__]
1230 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001231
1232 deltas = []
1233 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001234 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001235 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001236 print("beginning", repcount, "repetitions", file=sys.stderr)
1237 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001238 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001239 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001240 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001241 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001242 run_the_test()
1243 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001244 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001245 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001246 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001247 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001248 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001249 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001250 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001251 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001252 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001253 sys.stderr.flush()
1254 with open(fname, "a") as refrep:
1255 print(msg, file=refrep)
1256 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001257 return True
1258 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001259
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001260def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001261 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001262 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001263 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001264 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001265 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001266 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001267
Christian Heimesdae2a892008-04-19 00:55:37 +00001268 # Clear the warnings registry, so they can be displayed again
1269 for mod in sys.modules.values():
1270 if hasattr(mod, '__warningregistry__'):
1271 del mod.__warningregistry__
1272
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001273 # Restore some original values.
1274 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001275 copyreg.dispatch_table.clear()
1276 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001277 sys.path_importer_cache.clear()
1278 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001279 try:
1280 import zipimport
1281 except ImportError:
1282 pass # Run unmodified on platforms without zipimport support
1283 else:
1284 zipimport._zip_directory_cache.clear()
1285 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001286
Christian Heimes26855632008-01-27 23:50:43 +00001287 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001288 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001289
Guido van Rossum3de862d2007-08-18 00:10:33 +00001290 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001291 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001292 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001293 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001294 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001295 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001296 obj._abc_cache.clear()
1297 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001298
Antoine Pitrou046467c2009-10-30 18:30:35 +00001299 # Flush standard output, so that buffered data is sent to the OS and
1300 # associated Python objects are reclaimed.
1301 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1302 if stream is not None:
1303 stream.flush()
1304
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001305 # Clear assorted module caches.
1306 _path_created.clear()
1307 re.purge()
1308 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001309 urllib.parse.clear_cache()
1310 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001311 linecache.clearcache()
1312 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001313 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001314 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001315 doctest.master = None
1316
1317 # Collect cyclic trash.
1318 gc.collect()
1319
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001320def warm_char_cache():
1321 s = bytes(range(256))
1322 for i in range(256):
1323 s[i:i+1]
1324
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001325def findtestdir(path=None):
1326 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001327
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001328def removepy(names):
1329 if not names:
1330 return
1331 for idx, name in enumerate(names):
1332 basename, ext = os.path.splitext(name)
1333 if ext == '.py':
1334 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001335
Guido van Rossum152494a1996-12-20 03:12:20 +00001336def count(n, word):
1337 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001338 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001339 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001340 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001341
Tim Petersa45da922001-08-12 03:45:50 +00001342def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001343 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001344
1345 Optional arg width (default 70) is the maximum line length.
1346 Optional arg indent (default 4) is the number of blanks with which to
1347 begin each line.
1348 """
1349
Tim Petersba78bc42002-07-04 19:45:06 +00001350 from textwrap import fill
1351 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001352 # Print the sorted list: 'x' may be a '--random' list or a set()
1353 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001354 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001355
Tim Petersde14a302002-04-01 05:04:46 +00001356# Map sys.platform to a string containing the basenames of tests
1357# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001358#
1359# Special cases:
1360# test_pep277
1361# The _ExpectedSkips constructor adds this to the set of expected
1362# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001363# test_timeout
1364# Controlled by test_timeout.skip_expected. Requires the network
1365# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001366#
1367# Tests that are expected to be skipped everywhere except on one platform
1368# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001369
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001370_expectations = {
1371 'win32':
1372 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001373 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001374 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001375 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001376 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001377 test_fcntl
1378 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001379 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001380 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001381 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001382 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001383 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001384 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001385 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001386 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001387 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001388 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001389 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001390 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001391 test_pty
1392 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001393 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001394 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001395 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001396 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001397 test_wait3
1398 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001399 """,
1400 'linux2':
1401 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001402 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001403 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001404 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001405 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001406 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001407 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001408 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001409 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001410 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001411 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001412 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001413 test_openpty
1414 test_pyexpat
1415 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001416 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001417 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001418 'openunix8':
1419 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001420 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001421 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001422 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001423 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001424 test_openpty
1425 test_pyexpat
1426 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001427 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001428 """,
1429 'sco_sv3':
1430 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001431 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001432 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001433 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001434 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001435 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001436 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001437 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001438 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001439 test_openpty
1440 test_pyexpat
1441 test_queue
1442 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001443 test_sundry
1444 test_thread
1445 test_threaded_import
1446 test_threadedtempfile
1447 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001448 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001449 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001450 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001451 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001452 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001453 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001454 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001455 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001456 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001457 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001458 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001459 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001460 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001461 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001462 'sunos5':
1463 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001464 test_curses
1465 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001466 test_epoll
1467 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001468 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001469 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001470 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001471 test_zipfile
1472 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001473 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001474 'hp-ux11':
1475 """
Skip Montanarob3230212002-03-15 02:54:03 +00001476 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001477 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001478 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001479 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001480 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001481 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001482 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001483 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001484 test_openpty
1485 test_pyexpat
1486 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001487 test_zipfile
1488 test_zlib
1489 """,
Jason Tishler25115942002-12-05 15:18:15 +00001490 'cygwin':
1491 """
Jason Tishler25115942002-12-05 15:18:15 +00001492 test_curses
1493 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001494 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001495 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001496 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001497 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001498 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001499 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001500 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001501 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001502 'os2emx':
1503 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001504 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001505 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001506 test_epoll
1507 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001508 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001509 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001510 test_openpty
1511 test_ossaudiodev
1512 test_pty
1513 test_resource
1514 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001515 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001516 'freebsd4':
1517 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001518 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001519 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001520 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001521 test_ossaudiodev
1522 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001523 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001524 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001525 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001526 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001527 test_ttk_guionly
1528 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001529 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001530 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001531 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001532 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001533 'aix5':
1534 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001535 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001536 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001537 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001538 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001539 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001540 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001541 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001542 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001543 test_ttk_guionly
1544 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001545 test_zipimport
1546 test_zlib
1547 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001548 'openbsd3':
1549 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001550 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001551 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001552 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001553 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001554 test_normalization
1555 test_ossaudiodev
1556 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001557 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001558 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001559 test_ttk_guionly
1560 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001561 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001562 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001563 'netbsd3':
1564 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001565 test_ctypes
1566 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001567 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001568 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001569 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001570 test_ossaudiodev
1571 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001572 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001573 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001574 test_ttk_guionly
1575 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001576 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001577 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001578}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001579_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001580_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001581_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001582_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001583
Tim Petersb5b7b782001-08-12 01:20:39 +00001584class _ExpectedSkips:
1585 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001586 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001587 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001588
Tim Peters7c7efe92002-08-23 17:55:54 +00001589 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001590 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001591 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001592 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001593
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001594 # These are broken tests, for now skipped on every platform.
1595 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001596 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001597
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001598 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001599 if not os.path.supports_unicode_filenames:
1600 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001601
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001602 # doctest, profile and cProfile tests fail when the codec for the
1603 # fs encoding isn't built in because PyUnicode_Decode() adds two
1604 # calls into Python.
1605 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1606 if sys.getfilesystemencoding().lower() not in encs:
1607 self.expected.add('test_profile')
1608 self.expected.add('test_cProfile')
1609 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001610
Neal Norwitz55b61d22003-02-28 19:57:03 +00001611 if test_timeout.skip_expected:
1612 self.expected.add('test_timeout')
1613
Tim Petersecd79eb2003-01-29 00:35:32 +00001614 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001615 # test_sqlite is only reliable on Windows where the library
1616 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001617 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001618 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001619 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001620 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001621
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001622 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001623 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001624
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001625 if support.python_is_optimized():
1626 self.expected.add("test_gdb")
1627
Tim Peters7c7efe92002-08-23 17:55:54 +00001628 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001629
1630 def isvalid(self):
1631 "Return true iff _ExpectedSkips knows about the current platform."
1632 return self.valid
1633
1634 def getexpected(self):
1635 """Return set of test names we expect to skip on current platform.
1636
1637 self.isvalid() must be true.
1638 """
1639
1640 assert self.isvalid()
1641 return self.expected
1642
Michael Foord3ab34cc2010-12-03 12:27:40 +00001643def _make_temp_dir_for_build(TEMPDIR):
1644 # When tests are run from the Python build directory, it is best practice
1645 # to keep the test files in a subfolder. It eases the cleanup of leftover
1646 # files using command "make distclean".
1647 if sysconfig.is_python_build():
1648 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1649 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001650 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001651 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001652 except OSError as e:
1653 if e.errno != errno.EEXIST:
1654 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001655
1656 # Define a writable temp dir that will be used as cwd while running
1657 # the tests. The name of the dir includes the pid to allow parallel
1658 # testing (see the -j option).
1659 TESTCWD = 'test_python_{}'.format(os.getpid())
1660
1661 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1662 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001663
Guido van Rossum152494a1996-12-20 03:12:20 +00001664if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001665 # Remove regrtest.py's own directory from the module search path. Despite
1666 # the elimination of implicit relative imports, this is still needed to
1667 # ensure that submodules of the test package do not inappropriately appear
1668 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1669 # directly instead of using the -m switch
1670 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1671 i = len(sys.path)
1672 while i >= 0:
1673 i -= 1
1674 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1675 del sys.path[i]
1676
Florent Xiclunadc69e722010-09-13 16:35:02 +00001677 # findtestdir() gets the dirname out of __file__, so we have to make it
1678 # absolute before changing the working directory.
1679 # For example __file__ may be relative when running trace or profile.
1680 # See issue #9323.
1681 __file__ = os.path.abspath(__file__)
1682
1683 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001684 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001685
Michael Foord3ab34cc2010-12-03 12:27:40 +00001686 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001687
1688 # Run the tests in a context manager that temporary changes the CWD to a
1689 # temporary and writable directory. If it's not possible to create or
1690 # change the CWD, the original CWD will be used. The original CWD is
1691 # available from support.SAVEDCWD.
1692 with support.temp_cwd(TESTCWD, quiet=True):
1693 main()