blob: 8e9fe656c2abf1c7c13c0313f07ec5ae3a511197 [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
25 than TIMEOUT seconds
Georg Brandl96f78972009-10-27 21:02:11 +000026
27Verbosity
28
29-v/--verbose -- run tests in verbose mode with output to stdout
30-w/--verbose2 -- re-run failed tests in verbose mode
31-W/--verbose3 -- re-run failed tests in verbose mode immediately
32-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000033-q/--quiet -- no output unless one or more tests fail
Georg Brandl96f78972009-10-27 21:02:11 +000034-S/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000035 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000036
37Selecting tests
38
R. David Murray0ba81e02010-04-26 17:02:32 +000039-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000040 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000041-f/--fromfile -- read names of tests to run from a file (see below)
42-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000043-s/--single -- single step through a set of tests (see below)
Georg Brandl96f78972009-10-27 21:02:11 +000044-u/--use RES1,RES2,...
45 -- specify which special resource intensive tests to run
46-M/--memlimit LIMIT
47 -- run very large memory-consuming tests
R David Murrayb588f8d2011-03-24 14:42:58 -040048 --testdir DIR
49 -- execute test files in the specified directory (instead
50 of the Python stdlib test suite)
Georg Brandl96f78972009-10-27 21:02:11 +000051
52Special runs
53
54-l/--findleaks -- if GC is available detect tests that leak memory
55-L/--runleaks -- run the leaks(1) command just before exit
56-R/--huntrleaks RUNCOUNTS
57 -- search for reference leaks (needs debug build, v. slow)
58-j/--multiprocess PROCESSES
59 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000060-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000061-D/--coverdir DIRECTORY
62 -- Directory where coverage files are put
63-N/--nocoverdir -- Put coverage files alongside modules
64-t/--threshold THRESHOLD
65 -- call gc.set_threshold(THRESHOLD)
66-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000067-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000068
R. David Murray0ba81e02010-04-26 17:02:32 +000069
70Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000071
Collin Winterfd12f492009-03-29 04:05:05 +000072-r randomizes test execution order. You can use --randseed=int to provide a
73int seed value for the randomizer; this is useful for reproducing troublesome
74test orders.
75
R. David Murray0ba81e02010-04-26 17:02:32 +000076-s On the first invocation of regrtest using -s, the first test file found
77or the first test file given on the command line is run, and the name of
78the next test is recorded in a file named pynexttest. If run from the
79Python build directory, pynexttest is located in the 'build' subdirectory,
80otherwise it is located in tempfile.gettempdir(). On subsequent runs,
81the test in pynexttest is run, and the next test is written to pynexttest.
82When the last test has been run, pynexttest is deleted. In this way it
83is possible to single step through the test files. This is useful when
84doing memory analysis on the Python interpreter, which process tends to
85consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000086
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000087-S is used to continue running tests after an aborted run. It will
88maintain the order a standard run (ie, this assumes -r is not used).
89This is useful after the tests have prematurely stopped for some external
90reason and you want to start running from where you left off rather
91than starting from the beginning.
92
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000093-f reads the names of tests from the file given as f's argument, one
94or more test names per line. Whitespace is ignored. Blank lines and
95lines beginning with '#' are ignored. This is especially useful for
96whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000097
Skip Montanaro0179a182004-06-06 15:53:18 +000098-L causes the leaks(1) command to be run just before exit if it exists.
99leaks(1) is available on Mac OS X and presumably on some other
100FreeBSD-derived systems.
101
Michael W. Hudson61147f62004-08-03 11:33:28 +0000102-R runs each test several times and examines sys.gettotalrefcount() to
103see if the test appears to be leaking references. The argument should
104be of the form stab:run:fname where 'stab' is the number of times the
105test is run to let gettotalrefcount settle down, 'run' is the number
106of times further it is run and 'fname' is the name of the file the
107reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000108"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000109
Thomas Wouters477c8d52006-05-27 19:21:47 +0000110-M runs tests that require an exorbitant amount of memory. These tests
111typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001122 billion objects, which only works on 64-bit systems. There are also some
113tests that try to exhaust the address space of the process, which only makes
114sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000115which is a string in the form of '2.5Gb', determines howmuch memory the
116tests will limit themselves to (but they may go slightly over.) The number
117shouldn't be more memory than the machine has (including swap memory). You
118should also keep in mind that swap memory is generally much, much slower
119than RAM, and setting memlimit to all available RAM or higher will heavily
120tax the machine. On the other hand, it is no use running these tests with a
121limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
122to use more than memlimit memory will be skipped. The big-memory tests
123generally run very, very long.
124
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000125-u is used to specify which special resource intensive tests to run,
126such as those requiring large file support or network connectivity.
127The argument is a comma-separated list of words indicating the
128resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000129
Fred Drake3a15dac2002-04-11 16:39:16 +0000130 all - Enable all special resources.
131
Guido van Rossum315aa362003-03-11 14:46:48 +0000132 audio - Tests that use the audio device. (There are known
133 cases of broken audio drivers that can crash Python or
134 even the Linux kernel.)
135
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000136 curses - Tests that use curses and will modify the terminal's
137 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000138
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000139 largefile - It is okay to run some test that may create huge
140 files. These tests can take a long time and may
141 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000142
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000143 network - It is okay to run tests that use external network
144 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000145
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000146 decimal - Test the decimal module against a large suite that
147 verifies compliance with standards.
148
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000149 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000150
Tim Peterseba28be2005-03-28 01:08:02 +0000151 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000152
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000153 urlfetch - It is okay to download files required on testing.
154
Guilherme Polo9de29af2009-01-28 20:40:48 +0000155 gui - Run tests that require a running GUI.
156
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000157To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000158example, to run all the tests except for the gui tests, give the
159option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000160"""
161
Brett Cannon45071902010-06-14 22:22:54 +0000162import builtins
Victor Stinner024e37a2011-03-31 01:31:06 +0200163import faulthandler
Guido van Rossum152494a1996-12-20 03:12:20 +0000164import getopt
Antoine Pitrou88909542009-06-29 13:54:42 +0000165import json
Christian Heimesb186d002008-03-18 15:15:01 +0000166import os
Skip Montanaroab1c7912000-06-30 16:39:27 +0000167import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000168import re
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000169import io
Christian Heimesb186d002008-03-18 15:15:01 +0000170import sys
171import time
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000172import traceback
Christian Heimesb186d002008-03-18 15:15:01 +0000173import warnings
Benjamin Petersone549ead2009-03-28 21:42:05 +0000174import unittest
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000175from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000176import tempfile
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000177import platform
178import sysconfig
Nick Coghlan7d819752010-12-05 06:45:03 +0000179import logging
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000180
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000181
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000182# Some times __path__ and __file__ are not absolute (e.g. while running from
183# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
184# imports might fail. This affects only the modules imported before os.chdir().
185# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
186# they are found in the CWD their __file__ and __path__ will be relative (this
187# happens before the chdir). All the modules imported after the chdir, are
188# not found in the CWD, and since the other paths in sys.path[1:] are absolute
189# (site.py absolutize them), the __file__ and __path__ will be absolute too.
190# Therefore it is necessary to absolutize manually the __file__ and __path__ of
191# the packages to prevent later imports to fail when the CWD is different.
192for module in sys.modules.values():
193 if hasattr(module, '__path__'):
194 module.__path__ = [os.path.abspath(path) for path in module.__path__]
195 if hasattr(module, '__file__'):
196 module.__file__ = os.path.abspath(module.__file__)
197
Guido van Rossumdc15c272002-08-12 21:55:51 +0000198
Guido van Rossumbb484652002-12-02 09:56:21 +0000199# MacOSX (a.k.a. Darwin) has a default stack size that is too small
200# for deeply recursive regular expressions. We see this as crashes in
201# the Python test suite when running test_re.py and test_sre.py. The
202# fix is to set the stack limit to 2048.
203# This approach may also be useful for other Unixy platforms that
204# suffer from small default stack limits.
205if sys.platform == 'darwin':
206 try:
207 import resource
208 except ImportError:
209 pass
210 else:
211 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
212 newsoft = min(hard, max(soft, 1024*2048))
213 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
214
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000215# Test result constants.
216PASSED = 1
217FAILED = 0
218ENV_CHANGED = -1
219SKIPPED = -2
220RESOURCE_DENIED = -3
221INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200222CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000223
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000224from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000225
Georg Brandl1158a332009-06-04 09:30:30 +0000226RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000227 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000228
Florent Xiclunaec882212010-08-09 16:56:43 +0000229TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000230
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000231def usage(msg):
232 print(msg, file=sys.stderr)
233 print("Use --help for usage", file=sys.stderr)
234 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000235
236
Antoine Pitrou88909542009-06-29 13:54:42 +0000237def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000238 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000239 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000240 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000241 random_seed=None, use_mp=None, verbose3=False, forever=False,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200242 header=False, timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000243 """Execute a test suite.
244
Thomas Wouters7e474022000-07-16 12:04:32 +0000245 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000246 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000247
248 tests -- a list of strings containing test names (optional)
249 testdir -- the directory in which to look for tests (optional)
250
251 Users other than the Python test suite will certainly want to
252 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000253 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000254
255 If the tests argument is omitted, the tests listed on the
256 command-line will be used. If that's empty, too, then all *.py
257 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000258
Antoine Pitrou88909542009-06-29 13:54:42 +0000259 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000260 single, randomize, findleaks, use_resources, trace, coverdir,
261 print_slow, and random_seed) allow programmers calling main()
262 directly to set the values that would normally be set by flags
263 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000264 """
Fred Drake004d5e62000-10-23 17:22:08 +0000265
Victor Stinner1802d3f2010-05-19 17:11:19 +0000266 replace_stdout()
267
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000268 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000269 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000270 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000271 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
272 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
273 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000274 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000275 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Victor Stinner0cc8d592011-03-31 18:10:13 +0200276 'start=', 'nowindows', 'header', 'testdir=', 'timeout='])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000277 except getopt.error as msg:
278 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000279
280 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000281 if random_seed is None:
282 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000283 if use_resources is None:
284 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000285 debug = False
286 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000287 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000288 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000289 print(__doc__)
290 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000291 elif o in ('-v', '--verbose'):
292 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000293 elif o in ('-w', '--verbose2'):
294 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000295 elif o in ('-d', '--debug'):
296 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000297 elif o in ('-W', '--verbose3'):
298 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000299 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000300 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000301 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000302 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000303 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000304 elif o in ('-S', '--start'):
305 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000306 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000307 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000308 elif o in ('-S', '--slow'):
309 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000310 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000311 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000312 elif o == '--randseed':
313 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000314 elif o in ('-f', '--fromfile'):
315 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000316 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000317 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000318 elif o in ('-L', '--runleaks'):
319 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000320 elif o in ('-t', '--threshold'):
321 import gc
322 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000323 elif o in ('-T', '--coverage'):
324 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000325 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400326 # CWD is replaced with a temporary dir before calling main(), so we
327 # need join it with the saved CWD so it goes where the user expects.
328 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000329 elif o in ('-N', '--nocoverdir'):
330 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000331 elif o in ('-R', '--huntrleaks'):
332 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000333 if len(huntrleaks) not in (2, 3):
334 print(a, huntrleaks)
335 usage('-R takes 2 or 3 colon-separated arguments')
336 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000337 huntrleaks[0] = 5
338 else:
339 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000340 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000341 huntrleaks[1] = 4
342 else:
343 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000344 if len(huntrleaks) == 2 or not huntrleaks[2]:
345 huntrleaks[2:] = ["reflog.txt"]
346 # Avoid false positives due to the character cache in
347 # stringobject.c filling slowly with random data
348 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000349 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000350 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000351 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000352 u = [x.lower() for x in a.split(',')]
353 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000354 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000355 use_resources[:] = RESOURCE_NAMES
356 continue
357 remove = False
358 if r[0] == '-':
359 remove = True
360 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000361 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000362 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000363 if remove:
364 if r in use_resources:
365 use_resources.remove(r)
366 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000367 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000368 elif o in ('-n', '--nowindows'):
369 import msvcrt
370 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
371 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
372 msvcrt.SEM_NOGPFAULTERRORBOX|
373 msvcrt.SEM_NOOPENFILEERRORBOX)
374 try:
375 msvcrt.CrtSetReportMode
376 except AttributeError:
377 # release build
378 pass
379 else:
380 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
381 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
382 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000383 elif o in ('-F', '--forever'):
384 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000385 elif o in ('-j', '--multiprocess'):
386 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100387 if use_mp <= 0:
388 try:
389 import multiprocessing
390 # Use all cores + extras for tests that like to sleep
391 use_mp = 2 + multiprocessing.cpu_count()
392 except (ImportError, NotImplementedError):
393 use_mp = 3
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000394 elif o == '--header':
395 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000396 elif o == '--slaveargs':
397 args, kwargs = json.loads(a)
398 try:
399 result = runtest(*args, **kwargs)
400 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000401 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000402 sys.stdout.flush()
403 print() # Force a newline (just in case)
404 print(json.dumps(result))
405 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400406 elif o == '--testdir':
407 # CWD is replaced with a temporary dir before calling main(), so we
408 # join it with the saved CWD so it ends up where the user expects.
409 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200410 elif o == '--timeout':
411 timeout = float(a)
R. David Murray35768ad2009-11-15 00:23:21 +0000412 else:
413 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000414 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000415 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000416 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000417 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000418 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000419 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000420 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000421 usage("-l and -j don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000422
Guido van Rossum152494a1996-12-20 03:12:20 +0000423 good = []
424 bad = []
425 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000426 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000427 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000428 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000429
Neil Schemenauerd569f232000-09-22 15:29:28 +0000430 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000431 try:
432 import gc
433 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000434 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000435 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000436 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000437 # Uncomment the line below to report garbage that is not
438 # freeable by reference counting alone. By default only
439 # garbage that is not collectable by the GC is reported.
440 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000441 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000442
Barry Warsawe11e3de1999-01-28 19:51:51 +0000443 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000444 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000445 try:
446 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000447 next_test = fp.read().strip()
448 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000449 fp.close()
450 except IOError:
451 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000452
453 if fromfile:
454 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000455 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000456 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000457 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000458 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000459 guts = line.split() # assuming no test has whitespace in its name
460 if guts and not guts[0].startswith('#'):
461 tests.extend(guts)
462 fp.close()
463
464 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000465 removepy(args)
466 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000467
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000468 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000469 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000470 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000471 for arg in args:
472 if arg in stdtests:
473 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000474 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000475 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000476
477 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000478 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000479 # Print basic platform information
480 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000481 print("== ", platform.platform(aliased=True),
482 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000483 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000484 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000485
R David Murrayb588f8d2011-03-24 14:42:58 -0400486 # if testdir is set, then we are not running the python tests suite, so
487 # don't add default tests to be executed or skipped (pass empty values)
488 if testdir:
489 alltests = findtests(testdir, list(), set())
490 else:
491 alltests = findtests(testdir, stdtests, nottests)
492
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000493 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000494 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000495 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000496 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000497 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000498 except IndexError:
499 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000500 # Remove all the tests that precede start if it's set.
501 if start:
502 try:
503 del tests[:tests.index(start)]
504 except ValueError:
505 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000506 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000507 random.seed(random_seed)
508 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000509 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000510 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000511 import trace, tempfile
512 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
513 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000514 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000515
Christian Heimesb186d002008-03-18 15:15:01 +0000516 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000517 support.verbose = verbose # Tell tests to be moderately quiet
518 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000519 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000520
521 def accumulate_result(test, result):
522 ok, test_time = result
523 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000524 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000525 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000526 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000527 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000528 elif ok == ENV_CHANGED:
529 bad.append(test)
530 environment_changed.append(test)
531 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000532 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000533 elif ok == RESOURCE_DENIED:
534 skipped.append(test)
535 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000536
R. David Murray7dc72cc2009-11-14 16:13:02 +0000537 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000538 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000539 while True:
540 for test in tests:
541 yield test
542 if bad:
543 return
544 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000545 test_count = ''
546 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000547 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000548 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000549 test_count = '/{}'.format(len(selected))
550 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000551
Antoine Pitrou88909542009-06-29 13:54:42 +0000552 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000553 try:
554 from threading import Thread
555 except ImportError:
556 print("Multiprocess option requires thread support")
557 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000558 from queue import Queue
559 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000560 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000561 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000562 def tests_and_args():
563 for test in tests:
564 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000565 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000566 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200567 debug=debug, rerun_failed=verbose3, timeout=timeout)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000568 )
569 yield (test, args_tuple)
570 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000571 opt_args = support.args_from_interpreter_flags()
572 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000573 def work():
574 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000575 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000576 while True:
577 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000578 test, args_tuple = next(pending)
579 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000580 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000581 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000582 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000583 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000584 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000585 universal_newlines=True,
586 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000587 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200588 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000589 # Strip last refcount output line if it exists, since it
590 # comes from the shutdown of the interpreter in the subcommand.
591 stderr = debug_output_pat.sub("", stderr)
592 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200593 if retcode != 0:
594 result = (CHILD_ERROR, "Exit code %s" % retcode)
595 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
596 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000597 if not result:
598 output.put((None, None, None, None))
599 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000600 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000601 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000602 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000603 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000604 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000605 workers = [Thread(target=work) for i in range(use_mp)]
606 for worker in workers:
607 worker.start()
608 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000609 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000610 try:
611 while finished < use_mp:
612 test, stdout, stderr, result = output.get()
613 if test is None:
614 finished += 1
615 continue
Georg Brandldee7b852010-08-02 18:59:52 +0000616 if not quiet:
Georg Brandle8e02e32010-08-03 07:56:50 +0000617 print("[{1:{0}}{2}] {3}".format(
Georg Brandldee7b852010-08-02 18:59:52 +0000618 test_count_width, test_index, test_count, test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000619 if stdout:
620 print(stdout)
621 if stderr:
622 print(stderr, file=sys.stderr)
623 if result[0] == INTERRUPTED:
624 assert result[1] == 'KeyboardInterrupt'
625 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200626 if result[0] == CHILD_ERROR:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200627 raise Exception("Child error: {}".format(result[1]))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000628 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000629 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000630 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000631 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000632 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000633 for worker in workers:
634 worker.join()
635 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000636 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000637 if not quiet:
Georg Brandle8e02e32010-08-03 07:56:50 +0000638 print("[{1:{0}}{2}] {3}".format(
Georg Brandldee7b852010-08-02 18:59:52 +0000639 test_count_width, test_index, test_count, test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000640 sys.stdout.flush()
641 if trace:
642 # If we're tracing code coverage, then we don't exit with status
643 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200644 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000645 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000646 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000647 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100648 result = runtest(test, verbose, quiet, huntrleaks, debug,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200649 rerun_failed=verbose3, timeout=timeout)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000650 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000651 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000652 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000653 break
654 except:
655 raise
656 if findleaks:
657 gc.collect()
658 if gc.garbage:
659 print("Warning: test created", len(gc.garbage), end=' ')
660 print("uncollectable object(s).")
661 # move the uncollectable objects somewhere so we don't see
662 # them again
663 found_garbage.extend(gc.garbage)
664 del gc.garbage[:]
665 # Unload the newly imported modules (best effort finalization)
666 for module in sys.modules.keys():
667 if module not in save_modules and module.startswith("test."):
668 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000669
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000670 if interrupted:
671 # print a newline after ^C
672 print()
673 print("Test suite interrupted by signal SIGINT.")
674 omitted = set(selected) - set(good) - set(bad) - set(skipped)
675 print(count(len(omitted), "test"), "omitted:")
676 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000677 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000678 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000679 print("All", end=' ')
680 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000681 if print_slow:
682 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000683 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000684 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000685 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000686 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000687 bad = sorted(set(bad) - set(environment_changed))
688 if bad:
689 print(count(len(bad), "test"), "failed:")
690 printlist(bad)
691 if environment_changed:
692 print("{} altered the execution environment:".format(
693 count(len(environment_changed), "test")))
694 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000695 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000696 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000697 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000698
Tim Petersb5b7b782001-08-12 01:20:39 +0000699 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000700 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000701 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000702 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000703 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000704 print(count(len(surprise), "skip"), \
705 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000706 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000707 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000708 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000709 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000710 print("Ask someone to teach regrtest.py about which tests are")
711 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000712
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000713 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000714 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000715 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000716 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000717 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000718 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000719 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200720 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000721 except KeyboardInterrupt:
722 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000723 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000724 break
725 except:
726 raise
727
Barry Warsawe11e3de1999-01-28 19:51:51 +0000728 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000729 if next_single_test:
730 with open(filename, 'w') as fp:
731 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000732 else:
733 os.unlink(filename)
734
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000735 if trace:
736 r = tracer.results()
737 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
738
Skip Montanaro0179a182004-06-06 15:53:18 +0000739 if runleaks:
740 os.system("leaks %d" % os.getpid())
741
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000742 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000743
Guido van Rossum152494a1996-12-20 03:12:20 +0000744
R David Murrayb588f8d2011-03-24 14:42:58 -0400745# small set of tests to determine if we have a basically functioning interpreter
746# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000747STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000748 'test_grammar',
749 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000750 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000751 'test_builtin',
752 'test_exceptions',
753 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000754 'test_unittest',
755 'test_doctest',
756 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000757]
Guido van Rossum152494a1996-12-20 03:12:20 +0000758
R David Murrayb588f8d2011-03-24 14:42:58 -0400759# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400760NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000761
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000762def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000763 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000764 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000765 names = os.listdir(testdir)
766 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000767 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000768 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400769 mod, ext = os.path.splitext(name)
770 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
771 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000772 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000773
Victor Stinnerf58087b2010-05-02 17:24:51 +0000774def replace_stdout():
775 """Set stdout encoder error handler to backslashreplace (as stderr error
776 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000777 if os.name == "nt":
778 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
779 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000780
781 import atexit
782
Victor Stinnerf58087b2010-05-02 17:24:51 +0000783 stdout = sys.stdout
784 sys.stdout = open(stdout.fileno(), 'w',
785 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000786 errors="backslashreplace",
787 closefd=False)
788
789 def restore_stdout():
790 sys.stdout.close()
791 sys.stdout = stdout
792 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000793
Antoine Pitrou88909542009-06-29 13:54:42 +0000794def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100795 huntrleaks=False, debug=False, use_resources=None,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200796 rerun_failed=False, timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000797 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000798
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000799 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000800 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000801 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000802 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000803 huntrleaks -- run multiple times to test for leaks; requires a debug
804 build; a triple corresponding to -R's three arguments
Antoine Pitrou293954d2011-03-23 23:01:49 +0100805 rerun_failed -- if true, re-run in verbose mode when failed
Victor Stinner0cc8d592011-03-31 18:10:13 +0200806 timeout -- dump the traceback and exit if a test takes more than
807 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000808
809 Returns one of the test result constants:
810 INTERRUPTED KeyboardInterrupt when run under -j
811 RESOURCE_DENIED test skipped because resource denied
812 SKIPPED test skipped for some other reason
813 ENV_CHANGED test failed because it changed the execution environment
814 FAILED test failed
815 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000816 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000817
Antoine Pitrou88909542009-06-29 13:54:42 +0000818 support.verbose = verbose # Tell tests to be moderately quiet
819 if use_resources is not None:
820 support.use_resources = use_resources
Victor Stinner7d648a02011-03-31 18:27:50 +0200821 use_timeout = (timeout is not None and timeout > 0)
822 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200823 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000824 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100825 result = runtest_inner(test, verbose, quiet, huntrleaks, debug)
826 if result[0] == FAILED and rerun_failed:
827 cleanup_test_droppings(test, verbose)
828 sys.stdout.flush()
829 sys.stderr.flush()
830 print("Re-running test {} in verbose mode".format(test))
Victor Stinner0cc8d592011-03-31 18:10:13 +0200831 runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100832 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000833 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200834 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200835 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000836 cleanup_test_droppings(test, verbose)
837
Nick Coghlan6ead5522009-10-18 13:19:33 +0000838# Unit tests are supposed to leave the execution environment unchanged
839# once they complete. But sometimes tests have bugs, especially when
840# tests fail, and the changes to environment go on to mess up other
841# tests. This can cause issues with buildbot stability, since tests
842# are run in random order and so problems may appear to come and go.
843# There are a few things we can save and restore to mitigate this, and
844# the following context manager handles this task.
845
846class saved_test_environment:
847 """Save bits of the test environment and restore them at block exit.
848
849 with saved_test_environment(testname, verbose, quiet):
850 #stuff
851
852 Unless quiet is True, a warning is printed to stderr if any of
853 the saved items was changed by the test. The attribute 'changed'
854 is initially False, but is set to True if a change is detected.
855
856 If verbose is more than 1, the before and after state of changed
857 items is also printed.
858 """
859
860 changed = False
861
862 def __init__(self, testname, verbose=0, quiet=False):
863 self.testname = testname
864 self.verbose = verbose
865 self.quiet = quiet
866
867 # To add things to save and restore, add a name XXX to the resources list
868 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
869 # return the value to be saved and compared against a second call to the
870 # get function when test execution completes. restore_XXX should accept
871 # the saved value and restore the resource using it. It will be called if
872 # and only if a change in the value is detected.
873 #
874 # Note: XXX will have any '.' replaced with '_' characters when determining
875 # the corresponding method names.
876
877 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000878 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000879 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200880 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Ezio Melotti0123e052011-03-20 15:09:26 +0200881 'sys.warnoptions')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000882
883 def get_sys_argv(self):
884 return id(sys.argv), sys.argv, sys.argv[:]
885 def restore_sys_argv(self, saved_argv):
886 sys.argv = saved_argv[1]
887 sys.argv[:] = saved_argv[2]
888
889 def get_cwd(self):
890 return os.getcwd()
891 def restore_cwd(self, saved_cwd):
892 os.chdir(saved_cwd)
893
894 def get_sys_stdout(self):
895 return sys.stdout
896 def restore_sys_stdout(self, saved_stdout):
897 sys.stdout = saved_stdout
898
899 def get_sys_stderr(self):
900 return sys.stderr
901 def restore_sys_stderr(self, saved_stderr):
902 sys.stderr = saved_stderr
903
904 def get_sys_stdin(self):
905 return sys.stdin
906 def restore_sys_stdin(self, saved_stdin):
907 sys.stdin = saved_stdin
908
909 def get_os_environ(self):
910 return id(os.environ), os.environ, dict(os.environ)
911 def restore_os_environ(self, saved_environ):
912 os.environ = saved_environ[1]
913 os.environ.clear()
914 os.environ.update(saved_environ[2])
915
916 def get_sys_path(self):
917 return id(sys.path), sys.path, sys.path[:]
918 def restore_sys_path(self, saved_path):
919 sys.path = saved_path[1]
920 sys.path[:] = saved_path[2]
921
Brett Cannon055470a2010-02-19 15:57:10 +0000922 def get_sys_path_hooks(self):
923 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
924 def restore_sys_path_hooks(self, saved_hooks):
925 sys.path_hooks = saved_hooks[1]
926 sys.path_hooks[:] = saved_hooks[2]
927
Brett Cannon31f59292011-02-21 19:29:56 +0000928 def get_sys_gettrace(self):
929 return sys.gettrace()
930 def restore_sys_gettrace(self, trace_fxn):
931 sys.settrace(trace_fxn)
932
Brett Cannon055470a2010-02-19 15:57:10 +0000933 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000934 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000935 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000936 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000937
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000938 def get_warnings_filters(self):
939 return id(warnings.filters), warnings.filters, warnings.filters[:]
940 def restore_warnings_filters(self, saved_filters):
941 warnings.filters = saved_filters[1]
942 warnings.filters[:] = saved_filters[2]
943
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000944 def get_asyncore_socket_map(self):
945 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000946 # XXX Making a copy keeps objects alive until __exit__ gets called.
947 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000948 def restore_asyncore_socket_map(self, saved_map):
949 asyncore = sys.modules.get('asyncore')
950 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000951 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000952 asyncore.socket_map.update(saved_map)
953
Nick Coghlan7d819752010-12-05 06:45:03 +0000954 def get_logging__handlers(self):
955 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000956 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +0000957 def restore_logging__handlers(self, saved_handlers):
958 # Can't easily revert the logging state
959 pass
960
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000961 def get_logging__handlerList(self):
962 # _handlerList is a list of weakrefs to handlers
963 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
964 def restore_logging__handlerList(self, saved_handlerList):
965 # Can't easily revert the logging state
966 pass
967
Ezio Melotti0123e052011-03-20 15:09:26 +0200968 def get_sys_warnoptions(self):
969 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
970 def restore_sys_warnoptions(self, saved_options):
971 sys.warnoptions = saved_options[1]
972 sys.warnoptions[:] = saved_options[2]
973
Nick Coghlan6ead5522009-10-18 13:19:33 +0000974 def resource_info(self):
975 for name in self.resources:
976 method_suffix = name.replace('.', '_')
977 get_name = 'get_' + method_suffix
978 restore_name = 'restore_' + method_suffix
979 yield name, getattr(self, get_name), getattr(self, restore_name)
980
981 def __enter__(self):
982 self.saved_values = dict((name, get()) for name, get, restore
983 in self.resource_info())
984 return self
985
986 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +0000987 saved_values = self.saved_values
988 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +0000989 for name, get, restore in self.resource_info():
990 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +0000991 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +0000992 # Check for changes to the resource's value
993 if current != original:
994 self.changed = True
995 restore(original)
996 if not self.quiet:
997 print("Warning -- {} was modified by {}".format(
998 name, self.testname),
999 file=sys.stderr)
1000 if self.verbose > 1:
1001 print(" Before: {}\n After: {} ".format(
1002 original, current),
1003 file=sys.stderr)
1004 return False
1005
1006
R. David Murray8e286c42010-12-27 20:09:32 +00001007def runtest_inner(test, verbose, quiet, huntrleaks=False, debug=False):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001008 support.unload(test)
Benjamin Petersonb48af542010-04-11 20:43:16 +00001009 if verbose:
1010 capture_stdout = None
1011 else:
1012 capture_stdout = io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001013
Antoine Pitrou88909542009-06-29 13:54:42 +00001014 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001015 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001016 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001017 if test.startswith('test.'):
1018 abstest = test
1019 else:
1020 # Always import it from the test package
1021 abstest = 'test.' + test
1022 with saved_test_environment(test, verbose, quiet) as environment:
1023 start_time = time.time()
1024 the_package = __import__(abstest, globals(), locals(), [])
1025 the_module = getattr(the_package, test)
1026 # Old tests run to completion simply as a side-effect of
1027 # being imported. For tests based on unittest or doctest,
1028 # explicitly invoke their test_main() function (if it exists).
1029 indirect_test = getattr(the_module, "test_main", None)
1030 if indirect_test is not None:
1031 indirect_test()
1032 if huntrleaks:
1033 refleak = dash_R(the_module, test, indirect_test,
1034 huntrleaks)
1035 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001036 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001037 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001038 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001039 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001040 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001041 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001042 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001043 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001044 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001045 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001046 except KeyboardInterrupt:
1047 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001048 except support.TestFailed as msg:
R. David Murray11cabcf2010-09-29 01:08:05 +00001049 print("test", test, "failed --", msg, file=sys.stderr)
1050 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001051 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001052 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001053 msg = traceback.format_exc()
1054 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001055 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001056 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001057 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001058 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001059 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001060 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001061 return ENV_CHANGED, test_time
1062 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001063
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001064def cleanup_test_droppings(testname, verbose):
1065 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001066 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001067 import gc
1068
1069 # First kill any dangling references to open files etc.
1070 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001071 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001072 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001073
1074 # Try to clean up junk commonly left behind. While tests shouldn't leave
1075 # any files or directories behind, when a test fails that can be tedious
1076 # for it to arrange. The consequences can be especially nasty on Windows,
1077 # since if a test leaves a file open, it cannot be deleted by name (while
1078 # there's nothing we can do about that here either, we can display the
1079 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001080 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001081 "db_home",
1082 ):
1083 if not os.path.exists(name):
1084 continue
1085
1086 if os.path.isdir(name):
1087 kind, nuker = "directory", shutil.rmtree
1088 elif os.path.isfile(name):
1089 kind, nuker = "file", os.unlink
1090 else:
1091 raise SystemError("os.path says %r exists but is neither "
1092 "directory nor file" % name)
1093
1094 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001095 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001096 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001097 # if we have chmod, fix possible permissions problems
1098 # that might prevent cleanup
1099 if (hasattr(os, 'chmod')):
1100 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001101 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001102 except Exception as msg:
1103 print(("%r left behind %s %r and it couldn't be "
1104 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001105
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001106def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001107 """Run a test multiple times, looking for reference leaks.
1108
1109 Returns:
1110 False if the test didn't leak references; True if we detected refleaks.
1111 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001112 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001113 import copyreg
1114 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001115
1116 if not hasattr(sys, 'gettotalrefcount'):
1117 raise Exception("Tracking reference leaks requires a debug build "
1118 "of Python")
1119
1120 # Save current values for dash_R_cleanup() to restore.
1121 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001122 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001123 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001124 try:
1125 import zipimport
1126 except ImportError:
1127 zdc = None # Run unmodified on platforms without zipimport support
1128 else:
1129 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001130 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001131 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001132 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001133 continue
1134 for obj in abc.__subclasses__() + [abc]:
1135 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001136
1137 if indirect_test:
1138 def run_the_test():
1139 indirect_test()
1140 else:
1141 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001142 del sys.modules[the_module.__name__]
1143 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001144
1145 deltas = []
1146 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001147 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001148 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001149 print("beginning", repcount, "repetitions", file=sys.stderr)
1150 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001151 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001152 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001153 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001154 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001155 run_the_test()
1156 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001157 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001158 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001159 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001160 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001161 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001162 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001163 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001164 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001165 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001166 sys.stderr.flush()
1167 with open(fname, "a") as refrep:
1168 print(msg, file=refrep)
1169 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001170 return True
1171 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001172
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001173def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001174 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001175 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001176 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001177 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001178 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001179 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001180
Christian Heimesdae2a892008-04-19 00:55:37 +00001181 # Clear the warnings registry, so they can be displayed again
1182 for mod in sys.modules.values():
1183 if hasattr(mod, '__warningregistry__'):
1184 del mod.__warningregistry__
1185
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001186 # Restore some original values.
1187 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001188 copyreg.dispatch_table.clear()
1189 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001190 sys.path_importer_cache.clear()
1191 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001192 try:
1193 import zipimport
1194 except ImportError:
1195 pass # Run unmodified on platforms without zipimport support
1196 else:
1197 zipimport._zip_directory_cache.clear()
1198 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001199
Christian Heimes26855632008-01-27 23:50:43 +00001200 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001201 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001202
Guido van Rossum3de862d2007-08-18 00:10:33 +00001203 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001204 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001205 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001206 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001207 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001208 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001209 obj._abc_cache.clear()
1210 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001211
Antoine Pitrou046467c2009-10-30 18:30:35 +00001212 # Flush standard output, so that buffered data is sent to the OS and
1213 # associated Python objects are reclaimed.
1214 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1215 if stream is not None:
1216 stream.flush()
1217
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001218 # Clear assorted module caches.
1219 _path_created.clear()
1220 re.purge()
1221 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001222 urllib.parse.clear_cache()
1223 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001224 linecache.clearcache()
1225 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001226 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001227 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001228 doctest.master = None
1229
1230 # Collect cyclic trash.
1231 gc.collect()
1232
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001233def warm_char_cache():
1234 s = bytes(range(256))
1235 for i in range(256):
1236 s[i:i+1]
1237
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001238def findtestdir(path=None):
1239 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001240
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001241def removepy(names):
1242 if not names:
1243 return
1244 for idx, name in enumerate(names):
1245 basename, ext = os.path.splitext(name)
1246 if ext == '.py':
1247 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001248
Guido van Rossum152494a1996-12-20 03:12:20 +00001249def count(n, word):
1250 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001251 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001252 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001253 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001254
Tim Petersa45da922001-08-12 03:45:50 +00001255def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001256 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001257
1258 Optional arg width (default 70) is the maximum line length.
1259 Optional arg indent (default 4) is the number of blanks with which to
1260 begin each line.
1261 """
1262
Tim Petersba78bc42002-07-04 19:45:06 +00001263 from textwrap import fill
1264 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001265 # Print the sorted list: 'x' may be a '--random' list or a set()
1266 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001267 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001268
Tim Petersde14a302002-04-01 05:04:46 +00001269# Map sys.platform to a string containing the basenames of tests
1270# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001271#
1272# Special cases:
1273# test_pep277
1274# The _ExpectedSkips constructor adds this to the set of expected
1275# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001276# test_timeout
1277# Controlled by test_timeout.skip_expected. Requires the network
1278# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001279#
1280# Tests that are expected to be skipped everywhere except on one platform
1281# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001282
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001283_expectations = {
1284 'win32':
1285 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001286 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001287 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001288 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001289 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001290 test_fcntl
1291 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001292 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001293 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001294 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001295 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001296 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001297 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001298 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001299 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001300 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001301 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001302 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001303 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001304 test_pty
1305 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001306 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001307 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001308 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001309 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001310 test_wait3
1311 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001312 """,
1313 'linux2':
1314 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001315 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001316 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001317 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001318 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001319 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001320 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001321 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001322 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001323 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001324 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001325 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001326 test_openpty
1327 test_pyexpat
1328 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001329 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001330 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001331 'openunix8':
1332 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001333 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001334 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001335 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001336 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001337 test_openpty
1338 test_pyexpat
1339 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001340 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001341 """,
1342 'sco_sv3':
1343 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001344 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001345 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001346 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001347 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001348 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001349 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001350 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001351 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001352 test_openpty
1353 test_pyexpat
1354 test_queue
1355 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001356 test_sundry
1357 test_thread
1358 test_threaded_import
1359 test_threadedtempfile
1360 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001361 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001362 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001363 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001364 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001365 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001366 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001367 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001368 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001369 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001370 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001371 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001372 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001373 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001374 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001375 'sunos5':
1376 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001377 test_curses
1378 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001379 test_epoll
1380 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001381 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001382 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001383 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001384 test_zipfile
1385 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001386 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001387 'hp-ux11':
1388 """
Skip Montanarob3230212002-03-15 02:54:03 +00001389 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001390 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001391 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001392 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001393 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001394 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001395 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001396 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001397 test_openpty
1398 test_pyexpat
1399 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001400 test_zipfile
1401 test_zlib
1402 """,
Jason Tishler25115942002-12-05 15:18:15 +00001403 'cygwin':
1404 """
Jason Tishler25115942002-12-05 15:18:15 +00001405 test_curses
1406 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001407 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001408 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001409 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001410 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001411 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001412 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001413 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001414 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001415 'os2emx':
1416 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001417 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001418 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001419 test_epoll
1420 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001421 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001422 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001423 test_openpty
1424 test_ossaudiodev
1425 test_pty
1426 test_resource
1427 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001428 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001429 'freebsd4':
1430 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001431 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001432 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001433 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001434 test_ossaudiodev
1435 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001436 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001437 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001438 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001439 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001440 test_ttk_guionly
1441 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001442 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001443 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001444 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001445 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001446 'aix5':
1447 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001448 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001449 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001450 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001451 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001452 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001453 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001454 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001455 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001456 test_ttk_guionly
1457 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001458 test_zipimport
1459 test_zlib
1460 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001461 'openbsd3':
1462 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001463 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001464 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001465 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001466 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001467 test_normalization
1468 test_ossaudiodev
1469 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001470 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001471 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001472 test_ttk_guionly
1473 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001474 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001475 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001476 'netbsd3':
1477 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001478 test_ctypes
1479 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001480 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001481 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001482 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001483 test_ossaudiodev
1484 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001485 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001486 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001487 test_ttk_guionly
1488 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001489 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001490 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001491}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001492_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001493_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001494_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001495_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001496
Tim Petersb5b7b782001-08-12 01:20:39 +00001497class _ExpectedSkips:
1498 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001499 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001500 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001501
Tim Peters7c7efe92002-08-23 17:55:54 +00001502 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001503 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001504 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001505 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001506
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001507 # These are broken tests, for now skipped on every platform.
1508 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001509 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001510
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001511 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001512 if not os.path.supports_unicode_filenames:
1513 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001514
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001515 # doctest, profile and cProfile tests fail when the codec for the
1516 # fs encoding isn't built in because PyUnicode_Decode() adds two
1517 # calls into Python.
1518 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1519 if sys.getfilesystemencoding().lower() not in encs:
1520 self.expected.add('test_profile')
1521 self.expected.add('test_cProfile')
1522 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001523
Neal Norwitz55b61d22003-02-28 19:57:03 +00001524 if test_timeout.skip_expected:
1525 self.expected.add('test_timeout')
1526
Tim Petersecd79eb2003-01-29 00:35:32 +00001527 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001528 # test_sqlite is only reliable on Windows where the library
1529 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001530 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001531 "test_winsound", "test_startfile",
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001532 "test_sqlite"}
1533 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001534
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001535 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001536 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001537
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001538 if support.python_is_optimized():
1539 self.expected.add("test_gdb")
1540
Tim Peters7c7efe92002-08-23 17:55:54 +00001541 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001542
1543 def isvalid(self):
1544 "Return true iff _ExpectedSkips knows about the current platform."
1545 return self.valid
1546
1547 def getexpected(self):
1548 """Return set of test names we expect to skip on current platform.
1549
1550 self.isvalid() must be true.
1551 """
1552
1553 assert self.isvalid()
1554 return self.expected
1555
Michael Foord3ab34cc2010-12-03 12:27:40 +00001556def _make_temp_dir_for_build(TEMPDIR):
1557 # When tests are run from the Python build directory, it is best practice
1558 # to keep the test files in a subfolder. It eases the cleanup of leftover
1559 # files using command "make distclean".
1560 if sysconfig.is_python_build():
1561 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1562 TEMPDIR = os.path.abspath(TEMPDIR)
1563 if not os.path.exists(TEMPDIR):
1564 os.mkdir(TEMPDIR)
1565
1566 # Define a writable temp dir that will be used as cwd while running
1567 # the tests. The name of the dir includes the pid to allow parallel
1568 # testing (see the -j option).
1569 TESTCWD = 'test_python_{}'.format(os.getpid())
1570
1571 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1572 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001573
Guido van Rossum152494a1996-12-20 03:12:20 +00001574if __name__ == '__main__':
Victor Stinner024e37a2011-03-31 01:31:06 +02001575 # Display the Python traceback on segfault and division by zero
1576 faulthandler.enable()
1577
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001578 # Remove regrtest.py's own directory from the module search path. Despite
1579 # the elimination of implicit relative imports, this is still needed to
1580 # ensure that submodules of the test package do not inappropriately appear
1581 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1582 # directly instead of using the -m switch
1583 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1584 i = len(sys.path)
1585 while i >= 0:
1586 i -= 1
1587 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1588 del sys.path[i]
1589
Florent Xiclunadc69e722010-09-13 16:35:02 +00001590 # findtestdir() gets the dirname out of __file__, so we have to make it
1591 # absolute before changing the working directory.
1592 # For example __file__ may be relative when running trace or profile.
1593 # See issue #9323.
1594 __file__ = os.path.abspath(__file__)
1595
1596 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001597 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001598
Michael Foord3ab34cc2010-12-03 12:27:40 +00001599 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001600
1601 # Run the tests in a context manager that temporary changes the CWD to a
1602 # temporary and writable directory. If it's not possible to create or
1603 # change the CWD, the original CWD will be used. The original CWD is
1604 # available from support.SAVEDCWD.
1605 with support.temp_cwd(TESTCWD, quiet=True):
1606 main()