blob: 528f46a18e701aab7cd799fd898f59ff656028b0 [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
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000185
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000186# Some times __path__ and __file__ are not absolute (e.g. while running from
187# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
188# imports might fail. This affects only the modules imported before os.chdir().
189# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
190# they are found in the CWD their __file__ and __path__ will be relative (this
191# happens before the chdir). All the modules imported after the chdir, are
192# not found in the CWD, and since the other paths in sys.path[1:] are absolute
193# (site.py absolutize them), the __file__ and __path__ will be absolute too.
194# Therefore it is necessary to absolutize manually the __file__ and __path__ of
195# the packages to prevent later imports to fail when the CWD is different.
196for module in sys.modules.values():
197 if hasattr(module, '__path__'):
198 module.__path__ = [os.path.abspath(path) for path in module.__path__]
199 if hasattr(module, '__file__'):
200 module.__file__ = os.path.abspath(module.__file__)
201
Guido van Rossumdc15c272002-08-12 21:55:51 +0000202
Guido van Rossumbb484652002-12-02 09:56:21 +0000203# MacOSX (a.k.a. Darwin) has a default stack size that is too small
204# for deeply recursive regular expressions. We see this as crashes in
205# the Python test suite when running test_re.py and test_sre.py. The
206# fix is to set the stack limit to 2048.
207# This approach may also be useful for other Unixy platforms that
208# suffer from small default stack limits.
209if sys.platform == 'darwin':
210 try:
211 import resource
212 except ImportError:
213 pass
214 else:
215 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
216 newsoft = min(hard, max(soft, 1024*2048))
217 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
218
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000219# Test result constants.
220PASSED = 1
221FAILED = 0
222ENV_CHANGED = -1
223SKIPPED = -2
224RESOURCE_DENIED = -3
225INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200226CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000227
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000228from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000229
Georg Brandl1158a332009-06-04 09:30:30 +0000230RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000231 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000232
Florent Xiclunaec882212010-08-09 16:56:43 +0000233TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000234
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000235def usage(msg):
236 print(msg, file=sys.stderr)
237 print("Use --help for usage", file=sys.stderr)
238 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000239
240
Antoine Pitrou88909542009-06-29 13:54:42 +0000241def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000242 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000243 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000244 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000245 random_seed=None, use_mp=None, verbose3=False, forever=False,
Victor Stinner9bf133c2011-04-13 01:06:27 +0200246 header=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000247 """Execute a test suite.
248
Thomas Wouters7e474022000-07-16 12:04:32 +0000249 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000250 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000251
252 tests -- a list of strings containing test names (optional)
253 testdir -- the directory in which to look for tests (optional)
254
255 Users other than the Python test suite will certainly want to
256 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000257 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000258
259 If the tests argument is omitted, the tests listed on the
260 command-line will be used. If that's empty, too, then all *.py
261 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000262
Antoine Pitrou88909542009-06-29 13:54:42 +0000263 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000264 single, randomize, findleaks, use_resources, trace, coverdir,
265 print_slow, and random_seed) allow programmers calling main()
266 directly to set the values that would normally be set by flags
267 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000268 """
Victor Stinner3c18f252011-05-22 15:27:14 +0200269
Victor Stinnercb41cda2011-07-13 23:47:21 +0200270 # Display the Python traceback on fatal errors (e.g. segfault)
Victor Stinner3c18f252011-05-22 15:27:14 +0200271 faulthandler.enable(all_threads=True)
272
Victor Stinnercb41cda2011-07-13 23:47:21 +0200273 # Display the Python traceback on SIGALRM or SIGUSR1 signal
274 signals = []
275 if hasattr(signal, 'SIGALRM'):
276 signals.append(signal.SIGALRM)
277 if hasattr(signal, 'SIGUSR1'):
278 signals.append(signal.SIGUSR1)
279 for signum in signals:
280 faulthandler.register(signum, chain=True)
281
Victor Stinner1802d3f2010-05-19 17:11:19 +0000282 replace_stdout()
283
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000284 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000285 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000286 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000287 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
288 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
289 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000290 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000291 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Brian Curtinf6643452011-04-28 17:45:17 -0500292 'start=', 'nowindows', 'header', 'testdir=', 'timeout=', 'wait'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000293 except getopt.error as msg:
294 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000295
296 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000297 if random_seed is None:
298 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000299 if use_resources is None:
300 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000301 debug = False
302 start = None
Victor Stinner30196882011-06-03 12:53:26 +0200303 timeout = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000304 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000305 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000306 print(__doc__)
307 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000308 elif o in ('-v', '--verbose'):
309 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000310 elif o in ('-w', '--verbose2'):
311 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000312 elif o in ('-d', '--debug'):
313 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000314 elif o in ('-W', '--verbose3'):
315 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000316 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000317 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000318 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000319 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000320 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000321 elif o in ('-S', '--start'):
322 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000323 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000324 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000325 elif o in ('-S', '--slow'):
326 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000327 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000328 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000329 elif o == '--randseed':
330 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000331 elif o in ('-f', '--fromfile'):
332 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000333 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000334 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000335 elif o in ('-L', '--runleaks'):
336 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000337 elif o in ('-t', '--threshold'):
338 import gc
339 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000340 elif o in ('-T', '--coverage'):
341 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000342 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400343 # CWD is replaced with a temporary dir before calling main(), so we
344 # need join it with the saved CWD so it goes where the user expects.
345 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000346 elif o in ('-N', '--nocoverdir'):
347 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000348 elif o in ('-R', '--huntrleaks'):
349 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000350 if len(huntrleaks) not in (2, 3):
351 print(a, huntrleaks)
352 usage('-R takes 2 or 3 colon-separated arguments')
353 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000354 huntrleaks[0] = 5
355 else:
356 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000357 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000358 huntrleaks[1] = 4
359 else:
360 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000361 if len(huntrleaks) == 2 or not huntrleaks[2]:
362 huntrleaks[2:] = ["reflog.txt"]
363 # Avoid false positives due to the character cache in
364 # stringobject.c filling slowly with random data
365 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000366 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000367 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000368 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000369 u = [x.lower() for x in a.split(',')]
370 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000371 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000372 use_resources[:] = RESOURCE_NAMES
373 continue
374 remove = False
375 if r[0] == '-':
376 remove = True
377 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000378 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000379 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000380 if remove:
381 if r in use_resources:
382 use_resources.remove(r)
383 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000384 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000385 elif o in ('-n', '--nowindows'):
386 import msvcrt
387 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
388 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
389 msvcrt.SEM_NOGPFAULTERRORBOX|
390 msvcrt.SEM_NOOPENFILEERRORBOX)
391 try:
392 msvcrt.CrtSetReportMode
393 except AttributeError:
394 # release build
395 pass
396 else:
397 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
398 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
399 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000400 elif o in ('-F', '--forever'):
401 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000402 elif o in ('-j', '--multiprocess'):
403 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100404 if use_mp <= 0:
405 try:
406 import multiprocessing
407 # Use all cores + extras for tests that like to sleep
408 use_mp = 2 + multiprocessing.cpu_count()
409 except (ImportError, NotImplementedError):
410 use_mp = 3
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000411 elif o == '--header':
412 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000413 elif o == '--slaveargs':
414 args, kwargs = json.loads(a)
415 try:
416 result = runtest(*args, **kwargs)
417 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000418 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000419 sys.stdout.flush()
420 print() # Force a newline (just in case)
421 print(json.dumps(result))
422 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400423 elif o == '--testdir':
424 # CWD is replaced with a temporary dir before calling main(), so we
425 # join it with the saved CWD so it ends up where the user expects.
426 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200427 elif o == '--timeout':
Victor Stinner88345682011-06-06 12:14:23 +0200428 if hasattr(faulthandler, 'dump_tracebacks_later'):
429 timeout = float(a)
430 if timeout <= 0:
431 timeout = None
432 else:
433 print("Warning: The timeout option requires "
434 "faulthandler.dump_tracebacks_later")
Victor Stinner9e586c22011-06-06 12:04:36 +0200435 timeout = None
Brian Curtinf6643452011-04-28 17:45:17 -0500436 elif o == '--wait':
437 input("Press any key to continue...")
R. David Murray35768ad2009-11-15 00:23:21 +0000438 else:
439 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000440 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000441 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000442 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000443 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000444 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000445 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000446 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000447 usage("-l and -j don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000448
Guido van Rossum152494a1996-12-20 03:12:20 +0000449 good = []
450 bad = []
451 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000452 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000453 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000454 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000455
Neil Schemenauerd569f232000-09-22 15:29:28 +0000456 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000457 try:
458 import gc
459 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000460 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000461 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000462 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000463 # Uncomment the line below to report garbage that is not
464 # freeable by reference counting alone. By default only
465 # garbage that is not collectable by the GC is reported.
466 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000467 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000468
Barry Warsawe11e3de1999-01-28 19:51:51 +0000469 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000470 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000471 try:
472 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000473 next_test = fp.read().strip()
474 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000475 fp.close()
476 except IOError:
477 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000478
479 if fromfile:
480 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000481 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000482 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000483 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000484 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000485 guts = line.split() # assuming no test has whitespace in its name
486 if guts and not guts[0].startswith('#'):
487 tests.extend(guts)
488 fp.close()
489
490 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000491 removepy(args)
492 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000493
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000494 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000495 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000496 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000497 for arg in args:
498 if arg in stdtests:
499 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000500 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000501 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000502
503 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000504 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000505 # Print basic platform information
506 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000507 print("== ", platform.platform(aliased=True),
508 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000509 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000510 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000511
R David Murrayb588f8d2011-03-24 14:42:58 -0400512 # if testdir is set, then we are not running the python tests suite, so
513 # don't add default tests to be executed or skipped (pass empty values)
514 if testdir:
515 alltests = findtests(testdir, list(), set())
516 else:
517 alltests = findtests(testdir, stdtests, nottests)
518
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000519 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000520 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000521 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000522 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000523 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000524 except IndexError:
525 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000526 # Remove all the tests that precede start if it's set.
527 if start:
528 try:
529 del tests[:tests.index(start)]
530 except ValueError:
531 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000532 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000533 random.seed(random_seed)
534 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000535 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000536 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000537 import trace, tempfile
538 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
539 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000540 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000541
Christian Heimesb186d002008-03-18 15:15:01 +0000542 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000543 support.verbose = verbose # Tell tests to be moderately quiet
544 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000545 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000546
547 def accumulate_result(test, result):
548 ok, test_time = result
549 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000550 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000551 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000552 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000553 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000554 elif ok == ENV_CHANGED:
555 bad.append(test)
556 environment_changed.append(test)
557 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000558 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000559 elif ok == RESOURCE_DENIED:
560 skipped.append(test)
561 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000562
R. David Murray7dc72cc2009-11-14 16:13:02 +0000563 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000564 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000565 while True:
566 for test in tests:
567 yield test
568 if bad:
569 return
570 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000571 test_count = ''
572 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000573 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000574 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000575 test_count = '/{}'.format(len(selected))
576 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000577
Antoine Pitrou88909542009-06-29 13:54:42 +0000578 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000579 try:
580 from threading import Thread
581 except ImportError:
582 print("Multiprocess option requires thread support")
583 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000584 from queue import Queue
585 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000586 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000587 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000588 def tests_and_args():
589 for test in tests:
590 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000591 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000592 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200593 debug=debug, output_on_failure=verbose3,
594 timeout=timeout)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000595 )
596 yield (test, args_tuple)
597 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000598 opt_args = support.args_from_interpreter_flags()
599 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000600 def work():
601 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000602 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000603 while True:
604 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000605 test, args_tuple = next(pending)
606 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000607 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000608 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000609 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000610 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000611 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000612 universal_newlines=True,
613 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000614 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200615 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000616 # Strip last refcount output line if it exists, since it
617 # comes from the shutdown of the interpreter in the subcommand.
618 stderr = debug_output_pat.sub("", stderr)
619 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200620 if retcode != 0:
621 result = (CHILD_ERROR, "Exit code %s" % retcode)
622 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
623 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000624 if not result:
625 output.put((None, None, None, None))
626 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000627 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000628 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000629 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000630 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000631 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000632 workers = [Thread(target=work) for i in range(use_mp)]
633 for worker in workers:
634 worker.start()
635 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000636 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000637 try:
638 while finished < use_mp:
639 test, stdout, stderr, result = output.get()
640 if test is None:
641 finished += 1
642 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200643 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000644 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300645 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
646 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200647 test_count_width, test_index, test_count,
648 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000649 if stdout:
650 print(stdout)
651 if stderr:
652 print(stderr, file=sys.stderr)
653 if result[0] == INTERRUPTED:
654 assert result[1] == 'KeyboardInterrupt'
655 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200656 if result[0] == CHILD_ERROR:
Victor Stinner571e8fd2011-05-01 22:57:43 +0200657 raise Exception("Child error on {}: {}".format(test, result[1]))
Georg Brandldee7b852010-08-02 18:59:52 +0000658 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000659 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000660 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000661 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000662 for worker in workers:
663 worker.join()
664 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000665 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000666 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300667 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
668 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200669 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000670 sys.stdout.flush()
671 if trace:
672 # If we're tracing code coverage, then we don't exit with status
673 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200674 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000675 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000676 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000677 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100678 result = runtest(test, verbose, quiet, huntrleaks, debug,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200679 output_on_failure=verbose3, timeout=timeout)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000680 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000681 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000682 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000683 break
684 except:
685 raise
686 if findleaks:
687 gc.collect()
688 if gc.garbage:
689 print("Warning: test created", len(gc.garbage), end=' ')
690 print("uncollectable object(s).")
691 # move the uncollectable objects somewhere so we don't see
692 # them again
693 found_garbage.extend(gc.garbage)
694 del gc.garbage[:]
695 # Unload the newly imported modules (best effort finalization)
696 for module in sys.modules.keys():
697 if module not in save_modules and module.startswith("test."):
698 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000699
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000700 if interrupted:
701 # print a newline after ^C
702 print()
703 print("Test suite interrupted by signal SIGINT.")
704 omitted = set(selected) - set(good) - set(bad) - set(skipped)
705 print(count(len(omitted), "test"), "omitted:")
706 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000707 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000708 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000709 print("All", end=' ')
710 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000711 if print_slow:
712 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000713 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000714 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000715 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000716 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000717 bad = sorted(set(bad) - set(environment_changed))
718 if bad:
719 print(count(len(bad), "test"), "failed:")
720 printlist(bad)
721 if environment_changed:
722 print("{} altered the execution environment:".format(
723 count(len(environment_changed), "test")))
724 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000725 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000726 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000727 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000728
Tim Petersb5b7b782001-08-12 01:20:39 +0000729 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000730 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000731 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000732 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000733 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000734 print(count(len(surprise), "skip"), \
735 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000736 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000737 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000738 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000739 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000740 print("Ask someone to teach regrtest.py about which tests are")
741 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000742
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000743 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000744 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000745 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000746 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000747 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000748 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000749 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200750 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000751 except KeyboardInterrupt:
752 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000753 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000754 break
755 except:
756 raise
757
Barry Warsawe11e3de1999-01-28 19:51:51 +0000758 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000759 if next_single_test:
760 with open(filename, 'w') as fp:
761 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000762 else:
763 os.unlink(filename)
764
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000765 if trace:
766 r = tracer.results()
767 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
768
Skip Montanaro0179a182004-06-06 15:53:18 +0000769 if runleaks:
770 os.system("leaks %d" % os.getpid())
771
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000772 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000773
Guido van Rossum152494a1996-12-20 03:12:20 +0000774
R David Murrayb588f8d2011-03-24 14:42:58 -0400775# small set of tests to determine if we have a basically functioning interpreter
776# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000777STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000778 'test_grammar',
779 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000780 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000781 'test_builtin',
782 'test_exceptions',
783 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000784 'test_unittest',
785 'test_doctest',
786 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000787]
Guido van Rossum152494a1996-12-20 03:12:20 +0000788
R David Murrayb588f8d2011-03-24 14:42:58 -0400789# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400790NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000791
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000792def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000793 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000794 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000795 names = os.listdir(testdir)
796 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000797 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000798 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400799 mod, ext = os.path.splitext(name)
800 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
801 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000802 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000803
Victor Stinnerf58087b2010-05-02 17:24:51 +0000804def replace_stdout():
805 """Set stdout encoder error handler to backslashreplace (as stderr error
806 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000807 import atexit
808
Victor Stinnerf58087b2010-05-02 17:24:51 +0000809 stdout = sys.stdout
810 sys.stdout = open(stdout.fileno(), 'w',
811 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000812 errors="backslashreplace",
Victor Stinnerbe621032011-05-25 02:01:55 +0200813 closefd=False,
814 newline='\n')
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000815
816 def restore_stdout():
817 sys.stdout.close()
818 sys.stdout = stdout
819 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000820
Antoine Pitrou88909542009-06-29 13:54:42 +0000821def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100822 huntrleaks=False, debug=False, use_resources=None,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200823 output_on_failure=False, timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000824 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000825
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000826 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000827 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000828 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000829 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000830 huntrleaks -- run multiple times to test for leaks; requires a debug
831 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200832 output_on_failure -- if true, display test output on failure
Victor Stinner0cc8d592011-03-31 18:10:13 +0200833 timeout -- dump the traceback and exit if a test takes more than
834 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000835
836 Returns one of the test result constants:
837 INTERRUPTED KeyboardInterrupt when run under -j
838 RESOURCE_DENIED test skipped because resource denied
839 SKIPPED test skipped for some other reason
840 ENV_CHANGED test failed because it changed the execution environment
841 FAILED test failed
842 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000843 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000844
Antoine Pitrou88909542009-06-29 13:54:42 +0000845 if use_resources is not None:
846 support.use_resources = use_resources
Victor Stinner30196882011-06-03 12:53:26 +0200847 use_timeout = (timeout is not None)
Victor Stinner7d648a02011-03-31 18:27:50 +0200848 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200849 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000850 try:
Victor Stinner8313d6a2011-06-29 15:22:26 +0200851 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200852 support.verbose = True
853
854 # Reuse the same instance to all calls to runtest(). Some
855 # tests keep a reference to sys.stdout or sys.stderr
856 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200857 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200858 stream = io.StringIO()
859 runtest.stringio = stream
860 else:
861 stream = runtest.stringio
862 stream.seek(0)
863 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200864
865 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200866 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200867 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200868 sys.stdout = stream
869 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200870 result = runtest_inner(test, verbose, quiet, huntrleaks,
871 debug, display_failure=False)
872 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200873 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200874 orig_stderr.write(output)
875 orig_stderr.flush()
876 finally:
877 sys.stdout = orig_stdout
878 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200879 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200880 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200881 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
882 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100883 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000884 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200885 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200886 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000887 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200888runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000889
Nick Coghlan6ead5522009-10-18 13:19:33 +0000890# Unit tests are supposed to leave the execution environment unchanged
891# once they complete. But sometimes tests have bugs, especially when
892# tests fail, and the changes to environment go on to mess up other
893# tests. This can cause issues with buildbot stability, since tests
894# are run in random order and so problems may appear to come and go.
895# There are a few things we can save and restore to mitigate this, and
896# the following context manager handles this task.
897
898class saved_test_environment:
899 """Save bits of the test environment and restore them at block exit.
900
901 with saved_test_environment(testname, verbose, quiet):
902 #stuff
903
904 Unless quiet is True, a warning is printed to stderr if any of
905 the saved items was changed by the test. The attribute 'changed'
906 is initially False, but is set to True if a change is detected.
907
908 If verbose is more than 1, the before and after state of changed
909 items is also printed.
910 """
911
912 changed = False
913
914 def __init__(self, testname, verbose=0, quiet=False):
915 self.testname = testname
916 self.verbose = verbose
917 self.quiet = quiet
918
919 # To add things to save and restore, add a name XXX to the resources list
920 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
921 # return the value to be saved and compared against a second call to the
922 # get function when test execution completes. restore_XXX should accept
923 # the saved value and restore the resource using it. It will be called if
924 # and only if a change in the value is detected.
925 #
926 # Note: XXX will have any '.' replaced with '_' characters when determining
927 # the corresponding method names.
928
929 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000930 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000931 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200932 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Ezio Melotti0123e052011-03-20 15:09:26 +0200933 'sys.warnoptions')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000934
935 def get_sys_argv(self):
936 return id(sys.argv), sys.argv, sys.argv[:]
937 def restore_sys_argv(self, saved_argv):
938 sys.argv = saved_argv[1]
939 sys.argv[:] = saved_argv[2]
940
941 def get_cwd(self):
942 return os.getcwd()
943 def restore_cwd(self, saved_cwd):
944 os.chdir(saved_cwd)
945
946 def get_sys_stdout(self):
947 return sys.stdout
948 def restore_sys_stdout(self, saved_stdout):
949 sys.stdout = saved_stdout
950
951 def get_sys_stderr(self):
952 return sys.stderr
953 def restore_sys_stderr(self, saved_stderr):
954 sys.stderr = saved_stderr
955
956 def get_sys_stdin(self):
957 return sys.stdin
958 def restore_sys_stdin(self, saved_stdin):
959 sys.stdin = saved_stdin
960
961 def get_os_environ(self):
962 return id(os.environ), os.environ, dict(os.environ)
963 def restore_os_environ(self, saved_environ):
964 os.environ = saved_environ[1]
965 os.environ.clear()
966 os.environ.update(saved_environ[2])
967
968 def get_sys_path(self):
969 return id(sys.path), sys.path, sys.path[:]
970 def restore_sys_path(self, saved_path):
971 sys.path = saved_path[1]
972 sys.path[:] = saved_path[2]
973
Brett Cannon055470a2010-02-19 15:57:10 +0000974 def get_sys_path_hooks(self):
975 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
976 def restore_sys_path_hooks(self, saved_hooks):
977 sys.path_hooks = saved_hooks[1]
978 sys.path_hooks[:] = saved_hooks[2]
979
Brett Cannon31f59292011-02-21 19:29:56 +0000980 def get_sys_gettrace(self):
981 return sys.gettrace()
982 def restore_sys_gettrace(self, trace_fxn):
983 sys.settrace(trace_fxn)
984
Brett Cannon055470a2010-02-19 15:57:10 +0000985 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000986 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000987 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000988 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000989
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000990 def get_warnings_filters(self):
991 return id(warnings.filters), warnings.filters, warnings.filters[:]
992 def restore_warnings_filters(self, saved_filters):
993 warnings.filters = saved_filters[1]
994 warnings.filters[:] = saved_filters[2]
995
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000996 def get_asyncore_socket_map(self):
997 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000998 # XXX Making a copy keeps objects alive until __exit__ gets called.
999 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001000 def restore_asyncore_socket_map(self, saved_map):
1001 asyncore = sys.modules.get('asyncore')
1002 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +00001003 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001004 asyncore.socket_map.update(saved_map)
1005
Nick Coghlan7d819752010-12-05 06:45:03 +00001006 def get_logging__handlers(self):
1007 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001008 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001009 def restore_logging__handlers(self, saved_handlers):
1010 # Can't easily revert the logging state
1011 pass
1012
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001013 def get_logging__handlerList(self):
1014 # _handlerList is a list of weakrefs to handlers
1015 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1016 def restore_logging__handlerList(self, saved_handlerList):
1017 # Can't easily revert the logging state
1018 pass
1019
Ezio Melotti0123e052011-03-20 15:09:26 +02001020 def get_sys_warnoptions(self):
1021 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1022 def restore_sys_warnoptions(self, saved_options):
1023 sys.warnoptions = saved_options[1]
1024 sys.warnoptions[:] = saved_options[2]
1025
Nick Coghlan6ead5522009-10-18 13:19:33 +00001026 def resource_info(self):
1027 for name in self.resources:
1028 method_suffix = name.replace('.', '_')
1029 get_name = 'get_' + method_suffix
1030 restore_name = 'restore_' + method_suffix
1031 yield name, getattr(self, get_name), getattr(self, restore_name)
1032
1033 def __enter__(self):
1034 self.saved_values = dict((name, get()) for name, get, restore
1035 in self.resource_info())
1036 return self
1037
1038 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001039 saved_values = self.saved_values
1040 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001041 for name, get, restore in self.resource_info():
1042 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001043 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001044 # Check for changes to the resource's value
1045 if current != original:
1046 self.changed = True
1047 restore(original)
1048 if not self.quiet:
1049 print("Warning -- {} was modified by {}".format(
1050 name, self.testname),
1051 file=sys.stderr)
1052 if self.verbose > 1:
1053 print(" Before: {}\n After: {} ".format(
1054 original, current),
1055 file=sys.stderr)
1056 return False
1057
1058
Victor Stinnera7c33e52011-06-29 13:00:54 +02001059def runtest_inner(test, verbose, quiet,
1060 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001061 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001062
Antoine Pitrou88909542009-06-29 13:54:42 +00001063 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001064 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001065 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001066 if test.startswith('test.'):
1067 abstest = test
1068 else:
1069 # Always import it from the test package
1070 abstest = 'test.' + test
1071 with saved_test_environment(test, verbose, quiet) as environment:
1072 start_time = time.time()
1073 the_package = __import__(abstest, globals(), locals(), [])
1074 the_module = getattr(the_package, test)
1075 # Old tests run to completion simply as a side-effect of
1076 # being imported. For tests based on unittest or doctest,
1077 # explicitly invoke their test_main() function (if it exists).
1078 indirect_test = getattr(the_module, "test_main", None)
1079 if indirect_test is not None:
1080 indirect_test()
1081 if huntrleaks:
1082 refleak = dash_R(the_module, test, indirect_test,
1083 huntrleaks)
1084 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001085 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001086 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001087 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001088 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001089 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001090 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001091 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001092 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001093 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001094 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001095 except KeyboardInterrupt:
1096 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001097 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001098 if display_failure:
1099 print("test", test, "failed --", msg, file=sys.stderr)
1100 else:
1101 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001102 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001103 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001104 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001105 msg = traceback.format_exc()
1106 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001107 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001108 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001109 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001110 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001111 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001112 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001113 return ENV_CHANGED, test_time
1114 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001115
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001116def cleanup_test_droppings(testname, verbose):
1117 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001118 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001119 import gc
1120
1121 # First kill any dangling references to open files etc.
1122 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001123 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001124 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001125
1126 # Try to clean up junk commonly left behind. While tests shouldn't leave
1127 # any files or directories behind, when a test fails that can be tedious
1128 # for it to arrange. The consequences can be especially nasty on Windows,
1129 # since if a test leaves a file open, it cannot be deleted by name (while
1130 # there's nothing we can do about that here either, we can display the
1131 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001132 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001133 "db_home",
1134 ):
1135 if not os.path.exists(name):
1136 continue
1137
1138 if os.path.isdir(name):
1139 kind, nuker = "directory", shutil.rmtree
1140 elif os.path.isfile(name):
1141 kind, nuker = "file", os.unlink
1142 else:
1143 raise SystemError("os.path says %r exists but is neither "
1144 "directory nor file" % name)
1145
1146 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001147 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001148 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001149 # if we have chmod, fix possible permissions problems
1150 # that might prevent cleanup
1151 if (hasattr(os, 'chmod')):
1152 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001153 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001154 except Exception as msg:
1155 print(("%r left behind %s %r and it couldn't be "
1156 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001157
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001158def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001159 """Run a test multiple times, looking for reference leaks.
1160
1161 Returns:
1162 False if the test didn't leak references; True if we detected refleaks.
1163 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001164 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001165 import copyreg
1166 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001167
1168 if not hasattr(sys, 'gettotalrefcount'):
1169 raise Exception("Tracking reference leaks requires a debug build "
1170 "of Python")
1171
1172 # Save current values for dash_R_cleanup() to restore.
1173 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001174 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001175 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001176 try:
1177 import zipimport
1178 except ImportError:
1179 zdc = None # Run unmodified on platforms without zipimport support
1180 else:
1181 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001182 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001183 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001184 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001185 continue
1186 for obj in abc.__subclasses__() + [abc]:
1187 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001188
1189 if indirect_test:
1190 def run_the_test():
1191 indirect_test()
1192 else:
1193 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001194 del sys.modules[the_module.__name__]
1195 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001196
1197 deltas = []
1198 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001199 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001200 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001201 print("beginning", repcount, "repetitions", file=sys.stderr)
1202 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001203 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001204 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001205 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001206 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001207 run_the_test()
1208 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001209 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001210 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001211 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001212 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001213 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001214 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001215 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001216 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001217 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001218 sys.stderr.flush()
1219 with open(fname, "a") as refrep:
1220 print(msg, file=refrep)
1221 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001222 return True
1223 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001224
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001225def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001226 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001227 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001228 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001229 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001230 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001231 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001232
Christian Heimesdae2a892008-04-19 00:55:37 +00001233 # Clear the warnings registry, so they can be displayed again
1234 for mod in sys.modules.values():
1235 if hasattr(mod, '__warningregistry__'):
1236 del mod.__warningregistry__
1237
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001238 # Restore some original values.
1239 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001240 copyreg.dispatch_table.clear()
1241 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001242 sys.path_importer_cache.clear()
1243 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001244 try:
1245 import zipimport
1246 except ImportError:
1247 pass # Run unmodified on platforms without zipimport support
1248 else:
1249 zipimport._zip_directory_cache.clear()
1250 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001251
Christian Heimes26855632008-01-27 23:50:43 +00001252 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001253 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001254
Guido van Rossum3de862d2007-08-18 00:10:33 +00001255 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001256 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001257 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001258 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001259 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001260 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001261 obj._abc_cache.clear()
1262 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001263
Antoine Pitrou046467c2009-10-30 18:30:35 +00001264 # Flush standard output, so that buffered data is sent to the OS and
1265 # associated Python objects are reclaimed.
1266 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1267 if stream is not None:
1268 stream.flush()
1269
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001270 # Clear assorted module caches.
1271 _path_created.clear()
1272 re.purge()
1273 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001274 urllib.parse.clear_cache()
1275 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001276 linecache.clearcache()
1277 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001278 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001279 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001280 doctest.master = None
1281
1282 # Collect cyclic trash.
1283 gc.collect()
1284
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001285def warm_char_cache():
1286 s = bytes(range(256))
1287 for i in range(256):
1288 s[i:i+1]
1289
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001290def findtestdir(path=None):
1291 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001292
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001293def removepy(names):
1294 if not names:
1295 return
1296 for idx, name in enumerate(names):
1297 basename, ext = os.path.splitext(name)
1298 if ext == '.py':
1299 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001300
Guido van Rossum152494a1996-12-20 03:12:20 +00001301def count(n, word):
1302 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001303 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001304 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001305 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001306
Tim Petersa45da922001-08-12 03:45:50 +00001307def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001308 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001309
1310 Optional arg width (default 70) is the maximum line length.
1311 Optional arg indent (default 4) is the number of blanks with which to
1312 begin each line.
1313 """
1314
Tim Petersba78bc42002-07-04 19:45:06 +00001315 from textwrap import fill
1316 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001317 # Print the sorted list: 'x' may be a '--random' list or a set()
1318 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001319 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001320
Tim Petersde14a302002-04-01 05:04:46 +00001321# Map sys.platform to a string containing the basenames of tests
1322# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001323#
1324# Special cases:
1325# test_pep277
1326# The _ExpectedSkips constructor adds this to the set of expected
1327# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001328# test_timeout
1329# Controlled by test_timeout.skip_expected. Requires the network
1330# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001331#
1332# Tests that are expected to be skipped everywhere except on one platform
1333# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001334
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001335_expectations = {
1336 'win32':
1337 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001338 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001339 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001340 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001341 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001342 test_fcntl
1343 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001344 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001345 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001346 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001347 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001348 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001349 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001350 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001351 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001352 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001353 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001354 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001355 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001356 test_pty
1357 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001358 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001359 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001360 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001361 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001362 test_wait3
1363 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001364 """,
1365 'linux2':
1366 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001367 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001368 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001369 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001370 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001371 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001372 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001373 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001374 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001375 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001376 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001377 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001378 test_openpty
1379 test_pyexpat
1380 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001381 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001382 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001383 'openunix8':
1384 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001385 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001386 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001387 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001388 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001389 test_openpty
1390 test_pyexpat
1391 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001392 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001393 """,
1394 'sco_sv3':
1395 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001396 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001397 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001398 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001399 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001400 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001401 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001402 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001403 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001404 test_openpty
1405 test_pyexpat
1406 test_queue
1407 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001408 test_sundry
1409 test_thread
1410 test_threaded_import
1411 test_threadedtempfile
1412 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001413 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001414 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001415 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001416 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001417 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001418 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001419 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001420 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001421 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001422 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001423 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001424 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001425 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001426 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001427 'sunos5':
1428 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001429 test_curses
1430 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001431 test_epoll
1432 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001433 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001434 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001435 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001436 test_zipfile
1437 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001438 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001439 'hp-ux11':
1440 """
Skip Montanarob3230212002-03-15 02:54:03 +00001441 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001442 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001443 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001444 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001445 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001446 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001447 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001448 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001449 test_openpty
1450 test_pyexpat
1451 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001452 test_zipfile
1453 test_zlib
1454 """,
Jason Tishler25115942002-12-05 15:18:15 +00001455 'cygwin':
1456 """
Jason Tishler25115942002-12-05 15:18:15 +00001457 test_curses
1458 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001459 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001460 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001461 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001462 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001463 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001464 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001465 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001466 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001467 'os2emx':
1468 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001469 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001470 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001471 test_epoll
1472 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001473 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001474 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001475 test_openpty
1476 test_ossaudiodev
1477 test_pty
1478 test_resource
1479 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001480 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001481 'freebsd4':
1482 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001483 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001484 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001485 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001486 test_ossaudiodev
1487 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001488 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001489 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001490 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001491 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001492 test_ttk_guionly
1493 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001494 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001495 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001496 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001497 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001498 'aix5':
1499 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001500 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001501 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001502 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001503 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001504 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001505 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001506 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001507 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001508 test_ttk_guionly
1509 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001510 test_zipimport
1511 test_zlib
1512 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001513 'openbsd3':
1514 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001515 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001516 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001517 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001518 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001519 test_normalization
1520 test_ossaudiodev
1521 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001522 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001523 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001524 test_ttk_guionly
1525 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001526 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001527 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001528 'netbsd3':
1529 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001530 test_ctypes
1531 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001532 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001533 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001534 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001535 test_ossaudiodev
1536 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001537 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001538 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001539 test_ttk_guionly
1540 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001541 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001542 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001543}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001544_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001545_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001546_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001547_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001548
Tim Petersb5b7b782001-08-12 01:20:39 +00001549class _ExpectedSkips:
1550 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001551 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001552 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001553
Tim Peters7c7efe92002-08-23 17:55:54 +00001554 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001555 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001556 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001557 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001558
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001559 # These are broken tests, for now skipped on every platform.
1560 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001561 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001562
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001563 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001564 if not os.path.supports_unicode_filenames:
1565 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001566
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001567 # doctest, profile and cProfile tests fail when the codec for the
1568 # fs encoding isn't built in because PyUnicode_Decode() adds two
1569 # calls into Python.
1570 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1571 if sys.getfilesystemencoding().lower() not in encs:
1572 self.expected.add('test_profile')
1573 self.expected.add('test_cProfile')
1574 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001575
Neal Norwitz55b61d22003-02-28 19:57:03 +00001576 if test_timeout.skip_expected:
1577 self.expected.add('test_timeout')
1578
Tim Petersecd79eb2003-01-29 00:35:32 +00001579 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001580 # test_sqlite is only reliable on Windows where the library
1581 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001582 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001583 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001584 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001585 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001586
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001587 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001588 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001589
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001590 if support.python_is_optimized():
1591 self.expected.add("test_gdb")
1592
Tim Peters7c7efe92002-08-23 17:55:54 +00001593 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001594
1595 def isvalid(self):
1596 "Return true iff _ExpectedSkips knows about the current platform."
1597 return self.valid
1598
1599 def getexpected(self):
1600 """Return set of test names we expect to skip on current platform.
1601
1602 self.isvalid() must be true.
1603 """
1604
1605 assert self.isvalid()
1606 return self.expected
1607
Michael Foord3ab34cc2010-12-03 12:27:40 +00001608def _make_temp_dir_for_build(TEMPDIR):
1609 # When tests are run from the Python build directory, it is best practice
1610 # to keep the test files in a subfolder. It eases the cleanup of leftover
1611 # files using command "make distclean".
1612 if sysconfig.is_python_build():
1613 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1614 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001615 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001616 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001617 except OSError as e:
1618 if e.errno != errno.EEXIST:
1619 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001620
1621 # Define a writable temp dir that will be used as cwd while running
1622 # the tests. The name of the dir includes the pid to allow parallel
1623 # testing (see the -j option).
1624 TESTCWD = 'test_python_{}'.format(os.getpid())
1625
1626 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1627 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001628
Guido van Rossum152494a1996-12-20 03:12:20 +00001629if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001630 # Remove regrtest.py's own directory from the module search path. Despite
1631 # the elimination of implicit relative imports, this is still needed to
1632 # ensure that submodules of the test package do not inappropriately appear
1633 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1634 # directly instead of using the -m switch
1635 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1636 i = len(sys.path)
1637 while i >= 0:
1638 i -= 1
1639 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1640 del sys.path[i]
1641
Florent Xiclunadc69e722010-09-13 16:35:02 +00001642 # findtestdir() gets the dirname out of __file__, so we have to make it
1643 # absolute before changing the working directory.
1644 # For example __file__ may be relative when running trace or profile.
1645 # See issue #9323.
1646 __file__ = os.path.abspath(__file__)
1647
1648 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001649 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001650
Michael Foord3ab34cc2010-12-03 12:27:40 +00001651 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001652
1653 # Run the tests in a context manager that temporary changes the CWD to a
1654 # temporary and writable directory. If it's not possible to create or
1655 # change the CWD, the original CWD will be used. The original CWD is
1656 # available from support.SAVEDCWD.
1657 with support.temp_cwd(TESTCWD, quiet=True):
1658 main()