blob: a76492a0f9694a4cb03035bd87c5328686c380d9 [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 Stinnerd91a5ca2011-04-01 18:16:36 +020025 than TIMEOUT seconds (default: 30 minutes); disable
Victor Stinner305bff12011-04-01 15:59:59 +020026 the timeout if TIMEOUT is zero
Georg Brandl96f78972009-10-27 21:02:11 +000027
28Verbosity
29
30-v/--verbose -- run tests in verbose mode with output to stdout
31-w/--verbose2 -- re-run failed tests in verbose mode
32-W/--verbose3 -- re-run failed tests in verbose mode immediately
33-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000034-q/--quiet -- no output unless one or more tests fail
Georg Brandl96f78972009-10-27 21:02:11 +000035-S/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000036 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000037
38Selecting tests
39
R. David Murray0ba81e02010-04-26 17:02:32 +000040-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000041 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000042-f/--fromfile -- read names of tests to run from a file (see below)
43-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000044-s/--single -- single step through a set of tests (see below)
Georg Brandl96f78972009-10-27 21:02:11 +000045-u/--use RES1,RES2,...
46 -- specify which special resource intensive tests to run
47-M/--memlimit LIMIT
48 -- run very large memory-consuming tests
R David Murrayb588f8d2011-03-24 14:42:58 -040049 --testdir DIR
50 -- execute test files in the specified directory (instead
51 of the Python stdlib test suite)
Georg Brandl96f78972009-10-27 21:02:11 +000052
53Special runs
54
55-l/--findleaks -- if GC is available detect tests that leak memory
56-L/--runleaks -- run the leaks(1) command just before exit
57-R/--huntrleaks RUNCOUNTS
58 -- search for reference leaks (needs debug build, v. slow)
59-j/--multiprocess PROCESSES
60 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000061-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000062-D/--coverdir DIRECTORY
63 -- Directory where coverage files are put
64-N/--nocoverdir -- Put coverage files alongside modules
65-t/--threshold THRESHOLD
66 -- call gc.set_threshold(THRESHOLD)
67-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000068-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000069
R. David Murray0ba81e02010-04-26 17:02:32 +000070
71Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000072
Collin Winterfd12f492009-03-29 04:05:05 +000073-r randomizes test execution order. You can use --randseed=int to provide a
74int seed value for the randomizer; this is useful for reproducing troublesome
75test orders.
76
R. David Murray0ba81e02010-04-26 17:02:32 +000077-s On the first invocation of regrtest using -s, the first test file found
78or the first test file given on the command line is run, and the name of
79the next test is recorded in a file named pynexttest. If run from the
80Python build directory, pynexttest is located in the 'build' subdirectory,
81otherwise it is located in tempfile.gettempdir(). On subsequent runs,
82the test in pynexttest is run, and the next test is written to pynexttest.
83When the last test has been run, pynexttest is deleted. In this way it
84is possible to single step through the test files. This is useful when
85doing memory analysis on the Python interpreter, which process tends to
86consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000087
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000088-S is used to continue running tests after an aborted run. It will
89maintain the order a standard run (ie, this assumes -r is not used).
90This is useful after the tests have prematurely stopped for some external
91reason and you want to start running from where you left off rather
92than starting from the beginning.
93
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000094-f reads the names of tests from the file given as f's argument, one
95or more test names per line. Whitespace is ignored. Blank lines and
96lines beginning with '#' are ignored. This is especially useful for
97whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000098
Skip Montanaro0179a182004-06-06 15:53:18 +000099-L causes the leaks(1) command to be run just before exit if it exists.
100leaks(1) is available on Mac OS X and presumably on some other
101FreeBSD-derived systems.
102
Michael W. Hudson61147f62004-08-03 11:33:28 +0000103-R runs each test several times and examines sys.gettotalrefcount() to
104see if the test appears to be leaking references. The argument should
105be of the form stab:run:fname where 'stab' is the number of times the
106test is run to let gettotalrefcount settle down, 'run' is the number
107of times further it is run and 'fname' is the name of the file the
108reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000109"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000110
Thomas Wouters477c8d52006-05-27 19:21:47 +0000111-M runs tests that require an exorbitant amount of memory. These tests
112typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001132 billion objects, which only works on 64-bit systems. There are also some
114tests that try to exhaust the address space of the process, which only makes
115sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000116which is a string in the form of '2.5Gb', determines howmuch memory the
117tests will limit themselves to (but they may go slightly over.) The number
118shouldn't be more memory than the machine has (including swap memory). You
119should also keep in mind that swap memory is generally much, much slower
120than RAM, and setting memlimit to all available RAM or higher will heavily
121tax the machine. On the other hand, it is no use running these tests with a
122limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
123to use more than memlimit memory will be skipped. The big-memory tests
124generally run very, very long.
125
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000126-u is used to specify which special resource intensive tests to run,
127such as those requiring large file support or network connectivity.
128The argument is a comma-separated list of words indicating the
129resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000130
Fred Drake3a15dac2002-04-11 16:39:16 +0000131 all - Enable all special resources.
132
Guido van Rossum315aa362003-03-11 14:46:48 +0000133 audio - Tests that use the audio device. (There are known
134 cases of broken audio drivers that can crash Python or
135 even the Linux kernel.)
136
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000137 curses - Tests that use curses and will modify the terminal's
138 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000139
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000140 largefile - It is okay to run some test that may create huge
141 files. These tests can take a long time and may
142 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000143
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000144 network - It is okay to run tests that use external network
145 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000146
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000147 decimal - Test the decimal module against a large suite that
148 verifies compliance with standards.
149
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000150 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000151
Tim Peterseba28be2005-03-28 01:08:02 +0000152 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000153
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000154 urlfetch - It is okay to download files required on testing.
155
Guilherme Polo9de29af2009-01-28 20:40:48 +0000156 gui - Run tests that require a running GUI.
157
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000158To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000159example, to run all the tests except for the gui tests, give the
160option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000161"""
162
Brett Cannon45071902010-06-14 22:22:54 +0000163import builtins
Victor Stinner024e37a2011-03-31 01:31:06 +0200164import faulthandler
Guido van Rossum152494a1996-12-20 03:12:20 +0000165import getopt
Antoine Pitrou88909542009-06-29 13:54:42 +0000166import json
Christian Heimesb186d002008-03-18 15:15:01 +0000167import os
Skip Montanaroab1c7912000-06-30 16:39:27 +0000168import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000169import re
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000170import io
Christian Heimesb186d002008-03-18 15:15:01 +0000171import sys
172import time
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000173import traceback
Christian Heimesb186d002008-03-18 15:15:01 +0000174import warnings
Benjamin Petersone549ead2009-03-28 21:42:05 +0000175import unittest
Christian Heimesbe5b30b2008-03-03 19:18:51 +0000176from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000177import tempfile
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000178import platform
179import sysconfig
Nick Coghlan7d819752010-12-05 06:45:03 +0000180import logging
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000181
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000182
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000183# Some times __path__ and __file__ are not absolute (e.g. while running from
184# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
185# imports might fail. This affects only the modules imported before os.chdir().
186# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
187# they are found in the CWD their __file__ and __path__ will be relative (this
188# happens before the chdir). All the modules imported after the chdir, are
189# not found in the CWD, and since the other paths in sys.path[1:] are absolute
190# (site.py absolutize them), the __file__ and __path__ will be absolute too.
191# Therefore it is necessary to absolutize manually the __file__ and __path__ of
192# the packages to prevent later imports to fail when the CWD is different.
193for module in sys.modules.values():
194 if hasattr(module, '__path__'):
195 module.__path__ = [os.path.abspath(path) for path in module.__path__]
196 if hasattr(module, '__file__'):
197 module.__file__ = os.path.abspath(module.__file__)
198
Guido van Rossumdc15c272002-08-12 21:55:51 +0000199
Guido van Rossumbb484652002-12-02 09:56:21 +0000200# MacOSX (a.k.a. Darwin) has a default stack size that is too small
201# for deeply recursive regular expressions. We see this as crashes in
202# the Python test suite when running test_re.py and test_sre.py. The
203# fix is to set the stack limit to 2048.
204# This approach may also be useful for other Unixy platforms that
205# suffer from small default stack limits.
206if sys.platform == 'darwin':
207 try:
208 import resource
209 except ImportError:
210 pass
211 else:
212 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
213 newsoft = min(hard, max(soft, 1024*2048))
214 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
215
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000216# Test result constants.
217PASSED = 1
218FAILED = 0
219ENV_CHANGED = -1
220SKIPPED = -2
221RESOURCE_DENIED = -3
222INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200223CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000224
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000225from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000226
Georg Brandl1158a332009-06-04 09:30:30 +0000227RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000228 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000229
Florent Xiclunaec882212010-08-09 16:56:43 +0000230TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000231
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000232def usage(msg):
233 print(msg, file=sys.stderr)
234 print("Use --help for usage", file=sys.stderr)
235 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000236
237
Antoine Pitrou88909542009-06-29 13:54:42 +0000238def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000239 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000240 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000241 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000242 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub1eb6602011-04-07 23:22:28 +0200243 header=False, timeout=60*60):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000244 """Execute a test suite.
245
Thomas Wouters7e474022000-07-16 12:04:32 +0000246 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000247 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000248
249 tests -- a list of strings containing test names (optional)
250 testdir -- the directory in which to look for tests (optional)
251
252 Users other than the Python test suite will certainly want to
253 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000254 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000255
256 If the tests argument is omitted, the tests listed on the
257 command-line will be used. If that's empty, too, then all *.py
258 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000259
Antoine Pitrou88909542009-06-29 13:54:42 +0000260 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000261 single, randomize, findleaks, use_resources, trace, coverdir,
262 print_slow, and random_seed) allow programmers calling main()
263 directly to set the values that would normally be set by flags
264 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000265 """
Fred Drake004d5e62000-10-23 17:22:08 +0000266
Victor Stinner1802d3f2010-05-19 17:11:19 +0000267 replace_stdout()
268
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000269 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000270 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000271 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000272 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
273 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
274 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000275 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000276 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Victor Stinner0cc8d592011-03-31 18:10:13 +0200277 'start=', 'nowindows', 'header', 'testdir=', 'timeout='])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000278 except getopt.error as msg:
279 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000280
281 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000282 if random_seed is None:
283 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000284 if use_resources is None:
285 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000286 debug = False
287 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000288 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000289 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000290 print(__doc__)
291 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000292 elif o in ('-v', '--verbose'):
293 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000294 elif o in ('-w', '--verbose2'):
295 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000296 elif o in ('-d', '--debug'):
297 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000298 elif o in ('-W', '--verbose3'):
299 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000300 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000301 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000302 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000303 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000304 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000305 elif o in ('-S', '--start'):
306 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000307 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000308 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000309 elif o in ('-S', '--slow'):
310 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000311 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000312 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000313 elif o == '--randseed':
314 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000315 elif o in ('-f', '--fromfile'):
316 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000317 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000318 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000319 elif o in ('-L', '--runleaks'):
320 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000321 elif o in ('-t', '--threshold'):
322 import gc
323 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000324 elif o in ('-T', '--coverage'):
325 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000326 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400327 # CWD is replaced with a temporary dir before calling main(), so we
328 # need join it with the saved CWD so it goes where the user expects.
329 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000330 elif o in ('-N', '--nocoverdir'):
331 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000332 elif o in ('-R', '--huntrleaks'):
333 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000334 if len(huntrleaks) not in (2, 3):
335 print(a, huntrleaks)
336 usage('-R takes 2 or 3 colon-separated arguments')
337 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000338 huntrleaks[0] = 5
339 else:
340 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000341 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000342 huntrleaks[1] = 4
343 else:
344 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000345 if len(huntrleaks) == 2 or not huntrleaks[2]:
346 huntrleaks[2:] = ["reflog.txt"]
347 # Avoid false positives due to the character cache in
348 # stringobject.c filling slowly with random data
349 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000350 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000351 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000352 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000353 u = [x.lower() for x in a.split(',')]
354 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000355 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000356 use_resources[:] = RESOURCE_NAMES
357 continue
358 remove = False
359 if r[0] == '-':
360 remove = True
361 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000362 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000363 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000364 if remove:
365 if r in use_resources:
366 use_resources.remove(r)
367 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000368 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000369 elif o in ('-n', '--nowindows'):
370 import msvcrt
371 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
372 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
373 msvcrt.SEM_NOGPFAULTERRORBOX|
374 msvcrt.SEM_NOOPENFILEERRORBOX)
375 try:
376 msvcrt.CrtSetReportMode
377 except AttributeError:
378 # release build
379 pass
380 else:
381 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
382 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
383 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000384 elif o in ('-F', '--forever'):
385 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000386 elif o in ('-j', '--multiprocess'):
387 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100388 if use_mp <= 0:
389 try:
390 import multiprocessing
391 # Use all cores + extras for tests that like to sleep
392 use_mp = 2 + multiprocessing.cpu_count()
393 except (ImportError, NotImplementedError):
394 use_mp = 3
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000395 elif o == '--header':
396 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000397 elif o == '--slaveargs':
398 args, kwargs = json.loads(a)
399 try:
400 result = runtest(*args, **kwargs)
401 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000402 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000403 sys.stdout.flush()
404 print() # Force a newline (just in case)
405 print(json.dumps(result))
406 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400407 elif o == '--testdir':
408 # CWD is replaced with a temporary dir before calling main(), so we
409 # join it with the saved CWD so it ends up where the user expects.
410 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200411 elif o == '--timeout':
412 timeout = float(a)
R. David Murray35768ad2009-11-15 00:23:21 +0000413 else:
414 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000415 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000416 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000417 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000418 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000419 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000420 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000421 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000422 usage("-l and -j don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000423
Guido van Rossum152494a1996-12-20 03:12:20 +0000424 good = []
425 bad = []
426 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000427 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000428 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000429 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000430
Neil Schemenauerd569f232000-09-22 15:29:28 +0000431 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000432 try:
433 import gc
434 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000435 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000436 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000437 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000438 # Uncomment the line below to report garbage that is not
439 # freeable by reference counting alone. By default only
440 # garbage that is not collectable by the GC is reported.
441 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000442 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000443
Barry Warsawe11e3de1999-01-28 19:51:51 +0000444 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000445 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000446 try:
447 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000448 next_test = fp.read().strip()
449 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000450 fp.close()
451 except IOError:
452 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000453
454 if fromfile:
455 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000456 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000457 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000458 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000459 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000460 guts = line.split() # assuming no test has whitespace in its name
461 if guts and not guts[0].startswith('#'):
462 tests.extend(guts)
463 fp.close()
464
465 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000466 removepy(args)
467 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000468
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000469 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000470 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000471 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000472 for arg in args:
473 if arg in stdtests:
474 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000475 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000476 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000477
478 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000479 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000480 # Print basic platform information
481 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000482 print("== ", platform.platform(aliased=True),
483 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000484 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000485 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000486
R David Murrayb588f8d2011-03-24 14:42:58 -0400487 # if testdir is set, then we are not running the python tests suite, so
488 # don't add default tests to be executed or skipped (pass empty values)
489 if testdir:
490 alltests = findtests(testdir, list(), set())
491 else:
492 alltests = findtests(testdir, stdtests, nottests)
493
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000494 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000495 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000496 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000497 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000498 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000499 except IndexError:
500 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000501 # Remove all the tests that precede start if it's set.
502 if start:
503 try:
504 del tests[:tests.index(start)]
505 except ValueError:
506 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000507 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000508 random.seed(random_seed)
509 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000510 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000511 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000512 import trace, tempfile
513 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
514 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000515 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000516
Christian Heimesb186d002008-03-18 15:15:01 +0000517 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000518 support.verbose = verbose # Tell tests to be moderately quiet
519 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000520 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000521
522 def accumulate_result(test, result):
523 ok, test_time = result
524 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000525 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000526 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000527 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000528 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000529 elif ok == ENV_CHANGED:
530 bad.append(test)
531 environment_changed.append(test)
532 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000533 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000534 elif ok == RESOURCE_DENIED:
535 skipped.append(test)
536 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000537
R. David Murray7dc72cc2009-11-14 16:13:02 +0000538 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000539 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000540 while True:
541 for test in tests:
542 yield test
543 if bad:
544 return
545 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000546 test_count = ''
547 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000548 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000549 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000550 test_count = '/{}'.format(len(selected))
551 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000552
Antoine Pitrou88909542009-06-29 13:54:42 +0000553 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000554 try:
555 from threading import Thread
556 except ImportError:
557 print("Multiprocess option requires thread support")
558 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000559 from queue import Queue
560 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000561 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000562 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000563 def tests_and_args():
564 for test in tests:
565 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000566 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000567 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200568 debug=debug, rerun_failed=verbose3, timeout=timeout)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000569 )
570 yield (test, args_tuple)
571 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000572 opt_args = support.args_from_interpreter_flags()
573 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000574 def work():
575 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000576 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000577 while True:
578 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000579 test, args_tuple = next(pending)
580 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000581 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000582 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000583 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000584 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000585 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000586 universal_newlines=True,
587 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000588 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200589 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000590 # Strip last refcount output line if it exists, since it
591 # comes from the shutdown of the interpreter in the subcommand.
592 stderr = debug_output_pat.sub("", stderr)
593 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200594 if retcode != 0:
595 result = (CHILD_ERROR, "Exit code %s" % retcode)
596 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
597 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000598 if not result:
599 output.put((None, None, None, None))
600 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000601 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000602 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000603 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000604 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000605 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000606 workers = [Thread(target=work) for i in range(use_mp)]
607 for worker in workers:
608 worker.start()
609 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000610 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000611 try:
612 while finished < use_mp:
613 test, stdout, stderr, result = output.get()
614 if test is None:
615 finished += 1
616 continue
Georg Brandldee7b852010-08-02 18:59:52 +0000617 if not quiet:
Georg Brandle8e02e32010-08-03 07:56:50 +0000618 print("[{1:{0}}{2}] {3}".format(
Georg Brandldee7b852010-08-02 18:59:52 +0000619 test_count_width, test_index, test_count, test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000620 if stdout:
621 print(stdout)
622 if stderr:
623 print(stderr, file=sys.stderr)
624 if result[0] == INTERRUPTED:
625 assert result[1] == 'KeyboardInterrupt'
626 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200627 if result[0] == CHILD_ERROR:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200628 raise Exception("Child error: {}".format(result[1]))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000629 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000630 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000631 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000632 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000633 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000634 for worker in workers:
635 worker.join()
636 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000637 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000638 if not quiet:
Georg Brandle8e02e32010-08-03 07:56:50 +0000639 print("[{1:{0}}{2}] {3}".format(
Georg Brandldee7b852010-08-02 18:59:52 +0000640 test_count_width, test_index, test_count, test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000641 sys.stdout.flush()
642 if trace:
643 # If we're tracing code coverage, then we don't exit with status
644 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200645 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000646 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000647 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000648 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100649 result = runtest(test, verbose, quiet, huntrleaks, debug,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200650 rerun_failed=verbose3, timeout=timeout)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000651 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000652 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000653 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000654 break
655 except:
656 raise
657 if findleaks:
658 gc.collect()
659 if gc.garbage:
660 print("Warning: test created", len(gc.garbage), end=' ')
661 print("uncollectable object(s).")
662 # move the uncollectable objects somewhere so we don't see
663 # them again
664 found_garbage.extend(gc.garbage)
665 del gc.garbage[:]
666 # Unload the newly imported modules (best effort finalization)
667 for module in sys.modules.keys():
668 if module not in save_modules and module.startswith("test."):
669 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000670
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000671 if interrupted:
672 # print a newline after ^C
673 print()
674 print("Test suite interrupted by signal SIGINT.")
675 omitted = set(selected) - set(good) - set(bad) - set(skipped)
676 print(count(len(omitted), "test"), "omitted:")
677 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000678 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000679 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000680 print("All", end=' ')
681 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000682 if print_slow:
683 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000684 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000685 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000686 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000687 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000688 bad = sorted(set(bad) - set(environment_changed))
689 if bad:
690 print(count(len(bad), "test"), "failed:")
691 printlist(bad)
692 if environment_changed:
693 print("{} altered the execution environment:".format(
694 count(len(environment_changed), "test")))
695 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000696 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000697 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000698 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000699
Tim Petersb5b7b782001-08-12 01:20:39 +0000700 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000701 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000702 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000703 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000704 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000705 print(count(len(surprise), "skip"), \
706 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000707 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000708 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000709 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000710 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000711 print("Ask someone to teach regrtest.py about which tests are")
712 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000713
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000714 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000715 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000716 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000717 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000718 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000719 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000720 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200721 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000722 except KeyboardInterrupt:
723 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000724 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000725 break
726 except:
727 raise
728
Barry Warsawe11e3de1999-01-28 19:51:51 +0000729 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000730 if next_single_test:
731 with open(filename, 'w') as fp:
732 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000733 else:
734 os.unlink(filename)
735
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000736 if trace:
737 r = tracer.results()
738 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
739
Skip Montanaro0179a182004-06-06 15:53:18 +0000740 if runleaks:
741 os.system("leaks %d" % os.getpid())
742
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000743 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000744
Guido van Rossum152494a1996-12-20 03:12:20 +0000745
R David Murrayb588f8d2011-03-24 14:42:58 -0400746# small set of tests to determine if we have a basically functioning interpreter
747# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000748STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000749 'test_grammar',
750 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000751 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000752 'test_builtin',
753 'test_exceptions',
754 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000755 'test_unittest',
756 'test_doctest',
757 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000758]
Guido van Rossum152494a1996-12-20 03:12:20 +0000759
R David Murrayb588f8d2011-03-24 14:42:58 -0400760# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400761NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000762
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000763def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000764 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000765 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000766 names = os.listdir(testdir)
767 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000768 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000769 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400770 mod, ext = os.path.splitext(name)
771 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
772 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000773 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000774
Victor Stinnerf58087b2010-05-02 17:24:51 +0000775def replace_stdout():
776 """Set stdout encoder error handler to backslashreplace (as stderr error
777 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000778 if os.name == "nt":
779 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
780 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000781
782 import atexit
783
Victor Stinnerf58087b2010-05-02 17:24:51 +0000784 stdout = sys.stdout
785 sys.stdout = open(stdout.fileno(), 'w',
786 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000787 errors="backslashreplace",
788 closefd=False)
789
790 def restore_stdout():
791 sys.stdout.close()
792 sys.stdout = stdout
793 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000794
Antoine Pitrou88909542009-06-29 13:54:42 +0000795def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100796 huntrleaks=False, debug=False, use_resources=None,
Victor Stinner0cc8d592011-03-31 18:10:13 +0200797 rerun_failed=False, timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000798 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000799
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000800 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000801 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000802 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000803 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000804 huntrleaks -- run multiple times to test for leaks; requires a debug
805 build; a triple corresponding to -R's three arguments
Antoine Pitrou293954d2011-03-23 23:01:49 +0100806 rerun_failed -- if true, re-run in verbose mode when failed
Victor Stinner0cc8d592011-03-31 18:10:13 +0200807 timeout -- dump the traceback and exit if a test takes more than
808 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000809
810 Returns one of the test result constants:
811 INTERRUPTED KeyboardInterrupt when run under -j
812 RESOURCE_DENIED test skipped because resource denied
813 SKIPPED test skipped for some other reason
814 ENV_CHANGED test failed because it changed the execution environment
815 FAILED test failed
816 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000817 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000818
Antoine Pitrou88909542009-06-29 13:54:42 +0000819 support.verbose = verbose # Tell tests to be moderately quiet
820 if use_resources is not None:
821 support.use_resources = use_resources
Victor Stinner7d648a02011-03-31 18:27:50 +0200822 use_timeout = (timeout is not None and timeout > 0)
823 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200824 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000825 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100826 result = runtest_inner(test, verbose, quiet, huntrleaks, debug)
827 if result[0] == FAILED and rerun_failed:
828 cleanup_test_droppings(test, verbose)
829 sys.stdout.flush()
830 sys.stderr.flush()
831 print("Re-running test {} in verbose mode".format(test))
Victor Stinner0cc8d592011-03-31 18:10:13 +0200832 runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100833 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000834 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200835 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200836 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000837 cleanup_test_droppings(test, verbose)
838
Nick Coghlan6ead5522009-10-18 13:19:33 +0000839# Unit tests are supposed to leave the execution environment unchanged
840# once they complete. But sometimes tests have bugs, especially when
841# tests fail, and the changes to environment go on to mess up other
842# tests. This can cause issues with buildbot stability, since tests
843# are run in random order and so problems may appear to come and go.
844# There are a few things we can save and restore to mitigate this, and
845# the following context manager handles this task.
846
847class saved_test_environment:
848 """Save bits of the test environment and restore them at block exit.
849
850 with saved_test_environment(testname, verbose, quiet):
851 #stuff
852
853 Unless quiet is True, a warning is printed to stderr if any of
854 the saved items was changed by the test. The attribute 'changed'
855 is initially False, but is set to True if a change is detected.
856
857 If verbose is more than 1, the before and after state of changed
858 items is also printed.
859 """
860
861 changed = False
862
863 def __init__(self, testname, verbose=0, quiet=False):
864 self.testname = testname
865 self.verbose = verbose
866 self.quiet = quiet
867
868 # To add things to save and restore, add a name XXX to the resources list
869 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
870 # return the value to be saved and compared against a second call to the
871 # get function when test execution completes. restore_XXX should accept
872 # the saved value and restore the resource using it. It will be called if
873 # and only if a change in the value is detected.
874 #
875 # Note: XXX will have any '.' replaced with '_' characters when determining
876 # the corresponding method names.
877
878 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000879 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000880 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200881 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Ezio Melotti0123e052011-03-20 15:09:26 +0200882 'sys.warnoptions')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000883
884 def get_sys_argv(self):
885 return id(sys.argv), sys.argv, sys.argv[:]
886 def restore_sys_argv(self, saved_argv):
887 sys.argv = saved_argv[1]
888 sys.argv[:] = saved_argv[2]
889
890 def get_cwd(self):
891 return os.getcwd()
892 def restore_cwd(self, saved_cwd):
893 os.chdir(saved_cwd)
894
895 def get_sys_stdout(self):
896 return sys.stdout
897 def restore_sys_stdout(self, saved_stdout):
898 sys.stdout = saved_stdout
899
900 def get_sys_stderr(self):
901 return sys.stderr
902 def restore_sys_stderr(self, saved_stderr):
903 sys.stderr = saved_stderr
904
905 def get_sys_stdin(self):
906 return sys.stdin
907 def restore_sys_stdin(self, saved_stdin):
908 sys.stdin = saved_stdin
909
910 def get_os_environ(self):
911 return id(os.environ), os.environ, dict(os.environ)
912 def restore_os_environ(self, saved_environ):
913 os.environ = saved_environ[1]
914 os.environ.clear()
915 os.environ.update(saved_environ[2])
916
917 def get_sys_path(self):
918 return id(sys.path), sys.path, sys.path[:]
919 def restore_sys_path(self, saved_path):
920 sys.path = saved_path[1]
921 sys.path[:] = saved_path[2]
922
Brett Cannon055470a2010-02-19 15:57:10 +0000923 def get_sys_path_hooks(self):
924 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
925 def restore_sys_path_hooks(self, saved_hooks):
926 sys.path_hooks = saved_hooks[1]
927 sys.path_hooks[:] = saved_hooks[2]
928
Brett Cannon31f59292011-02-21 19:29:56 +0000929 def get_sys_gettrace(self):
930 return sys.gettrace()
931 def restore_sys_gettrace(self, trace_fxn):
932 sys.settrace(trace_fxn)
933
Brett Cannon055470a2010-02-19 15:57:10 +0000934 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000935 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000936 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000937 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000938
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000939 def get_warnings_filters(self):
940 return id(warnings.filters), warnings.filters, warnings.filters[:]
941 def restore_warnings_filters(self, saved_filters):
942 warnings.filters = saved_filters[1]
943 warnings.filters[:] = saved_filters[2]
944
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000945 def get_asyncore_socket_map(self):
946 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000947 # XXX Making a copy keeps objects alive until __exit__ gets called.
948 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000949 def restore_asyncore_socket_map(self, saved_map):
950 asyncore = sys.modules.get('asyncore')
951 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000952 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000953 asyncore.socket_map.update(saved_map)
954
Nick Coghlan7d819752010-12-05 06:45:03 +0000955 def get_logging__handlers(self):
956 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000957 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +0000958 def restore_logging__handlers(self, saved_handlers):
959 # Can't easily revert the logging state
960 pass
961
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000962 def get_logging__handlerList(self):
963 # _handlerList is a list of weakrefs to handlers
964 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
965 def restore_logging__handlerList(self, saved_handlerList):
966 # Can't easily revert the logging state
967 pass
968
Ezio Melotti0123e052011-03-20 15:09:26 +0200969 def get_sys_warnoptions(self):
970 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
971 def restore_sys_warnoptions(self, saved_options):
972 sys.warnoptions = saved_options[1]
973 sys.warnoptions[:] = saved_options[2]
974
Nick Coghlan6ead5522009-10-18 13:19:33 +0000975 def resource_info(self):
976 for name in self.resources:
977 method_suffix = name.replace('.', '_')
978 get_name = 'get_' + method_suffix
979 restore_name = 'restore_' + method_suffix
980 yield name, getattr(self, get_name), getattr(self, restore_name)
981
982 def __enter__(self):
983 self.saved_values = dict((name, get()) for name, get, restore
984 in self.resource_info())
985 return self
986
987 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +0000988 saved_values = self.saved_values
989 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +0000990 for name, get, restore in self.resource_info():
991 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +0000992 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +0000993 # Check for changes to the resource's value
994 if current != original:
995 self.changed = True
996 restore(original)
997 if not self.quiet:
998 print("Warning -- {} was modified by {}".format(
999 name, self.testname),
1000 file=sys.stderr)
1001 if self.verbose > 1:
1002 print(" Before: {}\n After: {} ".format(
1003 original, current),
1004 file=sys.stderr)
1005 return False
1006
1007
R. David Murray8e286c42010-12-27 20:09:32 +00001008def runtest_inner(test, verbose, quiet, huntrleaks=False, debug=False):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001009 support.unload(test)
Benjamin Petersonb48af542010-04-11 20:43:16 +00001010 if verbose:
1011 capture_stdout = None
1012 else:
1013 capture_stdout = io.StringIO()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001014
Antoine Pitrou88909542009-06-29 13:54:42 +00001015 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001016 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001017 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001018 if test.startswith('test.'):
1019 abstest = test
1020 else:
1021 # Always import it from the test package
1022 abstest = 'test.' + test
1023 with saved_test_environment(test, verbose, quiet) as environment:
1024 start_time = time.time()
1025 the_package = __import__(abstest, globals(), locals(), [])
1026 the_module = getattr(the_package, test)
1027 # Old tests run to completion simply as a side-effect of
1028 # being imported. For tests based on unittest or doctest,
1029 # explicitly invoke their test_main() function (if it exists).
1030 indirect_test = getattr(the_module, "test_main", None)
1031 if indirect_test is not None:
1032 indirect_test()
1033 if huntrleaks:
1034 refleak = dash_R(the_module, test, indirect_test,
1035 huntrleaks)
1036 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001037 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001038 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001039 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001040 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001041 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001042 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001043 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001044 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001045 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001046 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001047 except KeyboardInterrupt:
1048 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001049 except support.TestFailed as msg:
R. David Murray11cabcf2010-09-29 01:08:05 +00001050 print("test", test, "failed --", msg, file=sys.stderr)
1051 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001052 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001053 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001054 msg = traceback.format_exc()
1055 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001056 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001057 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001058 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001059 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001060 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001061 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001062 return ENV_CHANGED, test_time
1063 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001064
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001065def cleanup_test_droppings(testname, verbose):
1066 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001067 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001068 import gc
1069
1070 # First kill any dangling references to open files etc.
1071 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001072 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001073 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001074
1075 # Try to clean up junk commonly left behind. While tests shouldn't leave
1076 # any files or directories behind, when a test fails that can be tedious
1077 # for it to arrange. The consequences can be especially nasty on Windows,
1078 # since if a test leaves a file open, it cannot be deleted by name (while
1079 # there's nothing we can do about that here either, we can display the
1080 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001081 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001082 "db_home",
1083 ):
1084 if not os.path.exists(name):
1085 continue
1086
1087 if os.path.isdir(name):
1088 kind, nuker = "directory", shutil.rmtree
1089 elif os.path.isfile(name):
1090 kind, nuker = "file", os.unlink
1091 else:
1092 raise SystemError("os.path says %r exists but is neither "
1093 "directory nor file" % name)
1094
1095 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001096 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001097 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001098 # if we have chmod, fix possible permissions problems
1099 # that might prevent cleanup
1100 if (hasattr(os, 'chmod')):
1101 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001102 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001103 except Exception as msg:
1104 print(("%r left behind %s %r and it couldn't be "
1105 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001106
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001107def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001108 """Run a test multiple times, looking for reference leaks.
1109
1110 Returns:
1111 False if the test didn't leak references; True if we detected refleaks.
1112 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001113 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001114 import copyreg
1115 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001116
1117 if not hasattr(sys, 'gettotalrefcount'):
1118 raise Exception("Tracking reference leaks requires a debug build "
1119 "of Python")
1120
1121 # Save current values for dash_R_cleanup() to restore.
1122 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001123 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001124 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001125 try:
1126 import zipimport
1127 except ImportError:
1128 zdc = None # Run unmodified on platforms without zipimport support
1129 else:
1130 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001131 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001132 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001133 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001134 continue
1135 for obj in abc.__subclasses__() + [abc]:
1136 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001137
1138 if indirect_test:
1139 def run_the_test():
1140 indirect_test()
1141 else:
1142 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001143 del sys.modules[the_module.__name__]
1144 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001145
1146 deltas = []
1147 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001148 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001149 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001150 print("beginning", repcount, "repetitions", file=sys.stderr)
1151 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001152 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001153 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001154 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001155 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001156 run_the_test()
1157 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001158 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001159 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001160 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001161 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001162 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001163 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001164 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001165 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001166 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001167 sys.stderr.flush()
1168 with open(fname, "a") as refrep:
1169 print(msg, file=refrep)
1170 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001171 return True
1172 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001173
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001174def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001175 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001176 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001177 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001178 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001179 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001180 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001181
Christian Heimesdae2a892008-04-19 00:55:37 +00001182 # Clear the warnings registry, so they can be displayed again
1183 for mod in sys.modules.values():
1184 if hasattr(mod, '__warningregistry__'):
1185 del mod.__warningregistry__
1186
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001187 # Restore some original values.
1188 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001189 copyreg.dispatch_table.clear()
1190 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001191 sys.path_importer_cache.clear()
1192 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001193 try:
1194 import zipimport
1195 except ImportError:
1196 pass # Run unmodified on platforms without zipimport support
1197 else:
1198 zipimport._zip_directory_cache.clear()
1199 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001200
Christian Heimes26855632008-01-27 23:50:43 +00001201 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001202 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001203
Guido van Rossum3de862d2007-08-18 00:10:33 +00001204 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001205 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001206 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001207 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001208 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001209 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001210 obj._abc_cache.clear()
1211 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001212
Antoine Pitrou046467c2009-10-30 18:30:35 +00001213 # Flush standard output, so that buffered data is sent to the OS and
1214 # associated Python objects are reclaimed.
1215 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1216 if stream is not None:
1217 stream.flush()
1218
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001219 # Clear assorted module caches.
1220 _path_created.clear()
1221 re.purge()
1222 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001223 urllib.parse.clear_cache()
1224 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001225 linecache.clearcache()
1226 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001227 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001228 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001229 doctest.master = None
1230
1231 # Collect cyclic trash.
1232 gc.collect()
1233
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001234def warm_char_cache():
1235 s = bytes(range(256))
1236 for i in range(256):
1237 s[i:i+1]
1238
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001239def findtestdir(path=None):
1240 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001241
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001242def removepy(names):
1243 if not names:
1244 return
1245 for idx, name in enumerate(names):
1246 basename, ext = os.path.splitext(name)
1247 if ext == '.py':
1248 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001249
Guido van Rossum152494a1996-12-20 03:12:20 +00001250def count(n, word):
1251 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001252 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001253 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001254 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001255
Tim Petersa45da922001-08-12 03:45:50 +00001256def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001257 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001258
1259 Optional arg width (default 70) is the maximum line length.
1260 Optional arg indent (default 4) is the number of blanks with which to
1261 begin each line.
1262 """
1263
Tim Petersba78bc42002-07-04 19:45:06 +00001264 from textwrap import fill
1265 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001266 # Print the sorted list: 'x' may be a '--random' list or a set()
1267 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001268 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001269
Tim Petersde14a302002-04-01 05:04:46 +00001270# Map sys.platform to a string containing the basenames of tests
1271# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001272#
1273# Special cases:
1274# test_pep277
1275# The _ExpectedSkips constructor adds this to the set of expected
1276# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001277# test_timeout
1278# Controlled by test_timeout.skip_expected. Requires the network
1279# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001280#
1281# Tests that are expected to be skipped everywhere except on one platform
1282# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001283
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001284_expectations = {
1285 'win32':
1286 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001287 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001288 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001289 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001290 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001291 test_fcntl
1292 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001293 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001294 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001295 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001296 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001297 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001298 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001299 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001300 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001301 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001302 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001303 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001304 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001305 test_pty
1306 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001307 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001308 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001309 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001310 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001311 test_wait3
1312 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001313 """,
1314 'linux2':
1315 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001316 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001317 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001318 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001319 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001320 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001321 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001322 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001323 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001324 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001325 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001326 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001327 test_openpty
1328 test_pyexpat
1329 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001330 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001331 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001332 'openunix8':
1333 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001334 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001335 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001336 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001337 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001338 test_openpty
1339 test_pyexpat
1340 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001341 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001342 """,
1343 'sco_sv3':
1344 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001345 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001346 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001347 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001348 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001349 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001350 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001351 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001352 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001353 test_openpty
1354 test_pyexpat
1355 test_queue
1356 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001357 test_sundry
1358 test_thread
1359 test_threaded_import
1360 test_threadedtempfile
1361 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001362 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001363 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001364 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001365 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001366 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001367 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001368 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001369 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001370 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001371 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001372 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001373 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001374 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001375 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001376 'sunos5':
1377 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001378 test_curses
1379 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001380 test_epoll
1381 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001382 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001383 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001384 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001385 test_zipfile
1386 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001387 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001388 'hp-ux11':
1389 """
Skip Montanarob3230212002-03-15 02:54:03 +00001390 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001391 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001392 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001393 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001394 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001395 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001396 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001397 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001398 test_openpty
1399 test_pyexpat
1400 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001401 test_zipfile
1402 test_zlib
1403 """,
Jason Tishler25115942002-12-05 15:18:15 +00001404 'cygwin':
1405 """
Jason Tishler25115942002-12-05 15:18:15 +00001406 test_curses
1407 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001408 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001409 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001410 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001411 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001412 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001413 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001414 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001415 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001416 'os2emx':
1417 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001418 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001419 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001420 test_epoll
1421 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001422 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001423 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001424 test_openpty
1425 test_ossaudiodev
1426 test_pty
1427 test_resource
1428 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001429 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001430 'freebsd4':
1431 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001432 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001433 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001434 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001435 test_ossaudiodev
1436 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001437 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001438 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001439 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001440 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001441 test_ttk_guionly
1442 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001443 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001444 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001445 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001446 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001447 'aix5':
1448 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001449 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001450 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001451 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001452 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001453 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001454 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001455 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001456 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001457 test_ttk_guionly
1458 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001459 test_zipimport
1460 test_zlib
1461 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001462 'openbsd3':
1463 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001464 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001465 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001466 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001467 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001468 test_normalization
1469 test_ossaudiodev
1470 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001471 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001472 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001473 test_ttk_guionly
1474 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001475 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001476 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001477 'netbsd3':
1478 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001479 test_ctypes
1480 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001481 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001482 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001483 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001484 test_ossaudiodev
1485 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001486 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001487 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001488 test_ttk_guionly
1489 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001490 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001491 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001492}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001493_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001494_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001495_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001496_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001497
Tim Petersb5b7b782001-08-12 01:20:39 +00001498class _ExpectedSkips:
1499 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001500 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001501 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001502
Tim Peters7c7efe92002-08-23 17:55:54 +00001503 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001504 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001505 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001506 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001507
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001508 # These are broken tests, for now skipped on every platform.
1509 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001510 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001511
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001512 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001513 if not os.path.supports_unicode_filenames:
1514 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001515
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001516 # doctest, profile and cProfile tests fail when the codec for the
1517 # fs encoding isn't built in because PyUnicode_Decode() adds two
1518 # calls into Python.
1519 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1520 if sys.getfilesystemencoding().lower() not in encs:
1521 self.expected.add('test_profile')
1522 self.expected.add('test_cProfile')
1523 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001524
Neal Norwitz55b61d22003-02-28 19:57:03 +00001525 if test_timeout.skip_expected:
1526 self.expected.add('test_timeout')
1527
Tim Petersecd79eb2003-01-29 00:35:32 +00001528 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001529 # test_sqlite is only reliable on Windows where the library
1530 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001531 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001532 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001533 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001534 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001535
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001536 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001537 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001538
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001539 if support.python_is_optimized():
1540 self.expected.add("test_gdb")
1541
Tim Peters7c7efe92002-08-23 17:55:54 +00001542 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001543
1544 def isvalid(self):
1545 "Return true iff _ExpectedSkips knows about the current platform."
1546 return self.valid
1547
1548 def getexpected(self):
1549 """Return set of test names we expect to skip on current platform.
1550
1551 self.isvalid() must be true.
1552 """
1553
1554 assert self.isvalid()
1555 return self.expected
1556
Michael Foord3ab34cc2010-12-03 12:27:40 +00001557def _make_temp_dir_for_build(TEMPDIR):
1558 # When tests are run from the Python build directory, it is best practice
1559 # to keep the test files in a subfolder. It eases the cleanup of leftover
1560 # files using command "make distclean".
1561 if sysconfig.is_python_build():
1562 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1563 TEMPDIR = os.path.abspath(TEMPDIR)
1564 if not os.path.exists(TEMPDIR):
1565 os.mkdir(TEMPDIR)
1566
1567 # Define a writable temp dir that will be used as cwd while running
1568 # the tests. The name of the dir includes the pid to allow parallel
1569 # testing (see the -j option).
1570 TESTCWD = 'test_python_{}'.format(os.getpid())
1571
1572 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1573 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001574
Guido van Rossum152494a1996-12-20 03:12:20 +00001575if __name__ == '__main__':
Victor Stinner024e37a2011-03-31 01:31:06 +02001576 # Display the Python traceback on segfault and division by zero
1577 faulthandler.enable()
1578
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001579 # Remove regrtest.py's own directory from the module search path. Despite
1580 # the elimination of implicit relative imports, this is still needed to
1581 # ensure that submodules of the test package do not inappropriately appear
1582 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1583 # directly instead of using the -m switch
1584 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1585 i = len(sys.path)
1586 while i >= 0:
1587 i -= 1
1588 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1589 del sys.path[i]
1590
Florent Xiclunadc69e722010-09-13 16:35:02 +00001591 # findtestdir() gets the dirname out of __file__, so we have to make it
1592 # absolute before changing the working directory.
1593 # For example __file__ may be relative when running trace or profile.
1594 # See issue #9323.
1595 __file__ = os.path.abspath(__file__)
1596
1597 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001598 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001599
Michael Foord3ab34cc2010-12-03 12:27:40 +00001600 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001601
1602 # Run the tests in a context manager that temporary changes the CWD to a
1603 # temporary and writable directory. If it's not possible to create or
1604 # change the CWD, the original CWD will be used. The original CWD is
1605 # available from support.SAVEDCWD.
1606 with support.temp_cwd(TESTCWD, quiet=True):
1607 main()