blob: 3c8359aca1cd1efef8cf890591c62b90ed4ed1ba [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
Guido van Rossum152494a1996-12-20 03:12:20 +00002
R. David Murray0ba81e02010-04-26 17:02:32 +00003"""
4Usage:
Guido van Rossum152494a1996-12-20 03:12:20 +00005
Georg Brandlbe41a482011-01-05 21:47:47 +00006python -m test [options] [test_name1 [test_name2 ...]]
R. David Murray0ba81e02010-04-26 17:02:32 +00007python path/to/Lib/test/regrtest.py [options] [test_name1 [test_name2 ...]]
Guido van Rossum152494a1996-12-20 03:12:20 +00008
R. David Murray0ba81e02010-04-26 17:02:32 +00009
10If no arguments or options are provided, finds all files matching
11the pattern "test_*" in the Lib/test subdirectory and runs
12them in alphabetical order (but see -M and -u, below, for exceptions).
13
14For more rigorous testing, it is useful to use the following
15command line:
16
Georg Brandlbe41a482011-01-05 21:47:47 +000017python -E -Wd -m test [options] [test_name1 ...]
R. David Murray0ba81e02010-04-26 17:02:32 +000018
19
20Options:
Guido van Rossum152494a1996-12-20 03:12:20 +000021
Georg Brandl96f78972009-10-27 21:02:11 +000022-h/--help -- print this text and exit
Victor Stinner0cc8d592011-03-31 18:10:13 +020023--timeout TIMEOUT
24 -- dump the traceback and exit if a test takes more
Victor Stinner9e586c22011-06-06 12:04:36 +020025 than TIMEOUT seconds; disabled if TIMEOUT is negative
26 or equals to zero
Brian Curtinf6643452011-04-28 17:45:17 -050027--wait -- wait for user input, e.g., allow a debugger to be attached
Georg Brandl96f78972009-10-27 21:02:11 +000028
29Verbosity
30
31-v/--verbose -- run tests in verbose mode with output to stdout
32-w/--verbose2 -- re-run failed tests in verbose mode
Victor Stinnera7c33e52011-06-29 13:00:54 +020033-W/--verbose3 -- display test output on failure
Georg Brandl96f78972009-10-27 21:02:11 +000034-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000035-q/--quiet -- no output unless one or more tests fail
Senthil Kumaranbcfe56d2012-01-14 18:26:27 +080036-o/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000037 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000038
39Selecting tests
40
R. David Murray0ba81e02010-04-26 17:02:32 +000041-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000042 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000043-f/--fromfile -- read names of tests to run from a file (see below)
44-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000045-s/--single -- single step through a set of tests (see below)
Antoine Pitroub9c73e82011-07-29 23:53:38 +020046-m/--match PAT -- match test cases and methods with glob pattern PAT
Antoine Pitrou216a3bc2011-07-23 22:33:39 +020047-G/--failfast -- fail as soon as a test fails (only with -v or -W)
Georg Brandl96f78972009-10-27 21:02:11 +000048-u/--use RES1,RES2,...
49 -- specify which special resource intensive tests to run
50-M/--memlimit LIMIT
51 -- run very large memory-consuming tests
R David Murrayb588f8d2011-03-24 14:42:58 -040052 --testdir DIR
53 -- execute test files in the specified directory (instead
54 of the Python stdlib test suite)
Georg Brandl96f78972009-10-27 21:02:11 +000055
56Special runs
57
58-l/--findleaks -- if GC is available detect tests that leak memory
59-L/--runleaks -- run the leaks(1) command just before exit
60-R/--huntrleaks RUNCOUNTS
61 -- search for reference leaks (needs debug build, v. slow)
62-j/--multiprocess PROCESSES
63 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000064-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000065-D/--coverdir DIRECTORY
66 -- Directory where coverage files are put
67-N/--nocoverdir -- Put coverage files alongside modules
68-t/--threshold THRESHOLD
69 -- call gc.set_threshold(THRESHOLD)
70-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000071-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000072
R. David Murray0ba81e02010-04-26 17:02:32 +000073
74Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000075
Collin Winterfd12f492009-03-29 04:05:05 +000076-r randomizes test execution order. You can use --randseed=int to provide a
77int seed value for the randomizer; this is useful for reproducing troublesome
78test orders.
79
R. David Murray0ba81e02010-04-26 17:02:32 +000080-s On the first invocation of regrtest using -s, the first test file found
81or the first test file given on the command line is run, and the name of
82the next test is recorded in a file named pynexttest. If run from the
83Python build directory, pynexttest is located in the 'build' subdirectory,
84otherwise it is located in tempfile.gettempdir(). On subsequent runs,
85the test in pynexttest is run, and the next test is written to pynexttest.
86When the last test has been run, pynexttest is deleted. In this way it
87is possible to single step through the test files. This is useful when
88doing memory analysis on the Python interpreter, which process tends to
89consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000090
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000091-S is used to continue running tests after an aborted run. It will
92maintain the order a standard run (ie, this assumes -r is not used).
93This is useful after the tests have prematurely stopped for some external
94reason and you want to start running from where you left off rather
95than starting from the beginning.
96
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000097-f reads the names of tests from the file given as f's argument, one
98or more test names per line. Whitespace is ignored. Blank lines and
99lines beginning with '#' are ignored. This is especially useful for
100whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +0000101
Skip Montanaro0179a182004-06-06 15:53:18 +0000102-L causes the leaks(1) command to be run just before exit if it exists.
103leaks(1) is available on Mac OS X and presumably on some other
104FreeBSD-derived systems.
105
Michael W. Hudson61147f62004-08-03 11:33:28 +0000106-R runs each test several times and examines sys.gettotalrefcount() to
107see if the test appears to be leaking references. The argument should
108be of the form stab:run:fname where 'stab' is the number of times the
109test is run to let gettotalrefcount settle down, 'run' is the number
110of times further it is run and 'fname' is the name of the file the
111reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000112"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000113
Thomas Wouters477c8d52006-05-27 19:21:47 +0000114-M runs tests that require an exorbitant amount of memory. These tests
115typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001162 billion objects, which only works on 64-bit systems. There are also some
117tests that try to exhaust the address space of the process, which only makes
118sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000119which is a string in the form of '2.5Gb', determines howmuch memory the
120tests will limit themselves to (but they may go slightly over.) The number
121shouldn't be more memory than the machine has (including swap memory). You
122should also keep in mind that swap memory is generally much, much slower
123than RAM, and setting memlimit to all available RAM or higher will heavily
124tax the machine. On the other hand, it is no use running these tests with a
125limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
126to use more than memlimit memory will be skipped. The big-memory tests
127generally run very, very long.
128
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000129-u is used to specify which special resource intensive tests to run,
130such as those requiring large file support or network connectivity.
131The argument is a comma-separated list of words indicating the
132resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000133
Fred Drake3a15dac2002-04-11 16:39:16 +0000134 all - Enable all special resources.
135
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200136 none - Disable all special resources (this is the default).
137
Guido van Rossum315aa362003-03-11 14:46:48 +0000138 audio - Tests that use the audio device. (There are known
139 cases of broken audio drivers that can crash Python or
140 even the Linux kernel.)
141
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000142 curses - Tests that use curses and will modify the terminal's
143 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000144
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000145 largefile - It is okay to run some test that may create huge
146 files. These tests can take a long time and may
147 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000148
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000149 network - It is okay to run tests that use external network
150 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000151
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000152 decimal - Test the decimal module against a large suite that
153 verifies compliance with standards.
154
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000155 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000156
Tim Peterseba28be2005-03-28 01:08:02 +0000157 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000158
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000159 urlfetch - It is okay to download files required on testing.
160
Guilherme Polo9de29af2009-01-28 20:40:48 +0000161 gui - Run tests that require a running GUI.
162
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000163To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000164example, to run all the tests except for the gui tests, give the
165option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000166"""
167
Nick Coghlanbe7e49f2012-07-20 23:40:09 +1000168# We import importlib *ASAP* in order to test #15386
169import importlib
170
Brett Cannon45071902010-06-14 22:22:54 +0000171import builtins
Victor Stinner024e37a2011-03-31 01:31:06 +0200172import faulthandler
Guido van Rossum152494a1996-12-20 03:12:20 +0000173import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200174import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000175import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200176import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000177import os
Victor Stinner8313d6a2011-06-29 15:22:26 +0200178import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000179import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000180import re
Éric Araujoff913062011-11-29 16:45:07 +0100181import shutil
Victor Stinnercb41cda2011-07-13 23:47:21 +0200182import signal
Christian Heimesb186d002008-03-18 15:15:01 +0000183import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000184import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200185import tempfile
186import time
187import traceback
188import unittest
189import warnings
190from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000191
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200192try:
193 import threading
194except ImportError:
195 threading = None
196try:
197 import multiprocessing.process
198except ImportError:
199 multiprocessing = None
200
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000201
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000202# Some times __path__ and __file__ are not absolute (e.g. while running from
203# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
204# imports might fail. This affects only the modules imported before os.chdir().
205# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
206# they are found in the CWD their __file__ and __path__ will be relative (this
207# happens before the chdir). All the modules imported after the chdir, are
208# not found in the CWD, and since the other paths in sys.path[1:] are absolute
209# (site.py absolutize them), the __file__ and __path__ will be absolute too.
210# Therefore it is necessary to absolutize manually the __file__ and __path__ of
211# the packages to prevent later imports to fail when the CWD is different.
212for module in sys.modules.values():
213 if hasattr(module, '__path__'):
214 module.__path__ = [os.path.abspath(path) for path in module.__path__]
215 if hasattr(module, '__file__'):
216 module.__file__ = os.path.abspath(module.__file__)
217
Guido van Rossumdc15c272002-08-12 21:55:51 +0000218
Guido van Rossumbb484652002-12-02 09:56:21 +0000219# MacOSX (a.k.a. Darwin) has a default stack size that is too small
220# for deeply recursive regular expressions. We see this as crashes in
221# the Python test suite when running test_re.py and test_sre.py. The
222# fix is to set the stack limit to 2048.
223# This approach may also be useful for other Unixy platforms that
224# suffer from small default stack limits.
225if sys.platform == 'darwin':
226 try:
227 import resource
228 except ImportError:
229 pass
230 else:
231 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
232 newsoft = min(hard, max(soft, 1024*2048))
233 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
234
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000235# Test result constants.
236PASSED = 1
237FAILED = 0
238ENV_CHANGED = -1
239SKIPPED = -2
240RESOURCE_DENIED = -3
241INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200242CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000243
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000244from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000245
Georg Brandl1158a332009-06-04 09:30:30 +0000246RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000247 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000248
Florent Xiclunaec882212010-08-09 16:56:43 +0000249TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000250
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000251def usage(msg):
252 print(msg, file=sys.stderr)
253 print("Use --help for usage", file=sys.stderr)
254 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000255
256
Antoine Pitrou88909542009-06-29 13:54:42 +0000257def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000258 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000259 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000260 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000261 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200262 header=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000263 """Execute a test suite.
264
Thomas Wouters7e474022000-07-16 12:04:32 +0000265 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000266 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000267
268 tests -- a list of strings containing test names (optional)
269 testdir -- the directory in which to look for tests (optional)
270
271 Users other than the Python test suite will certainly want to
272 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000273 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000274
275 If the tests argument is omitted, the tests listed on the
276 command-line will be used. If that's empty, too, then all *.py
277 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000278
Antoine Pitrou88909542009-06-29 13:54:42 +0000279 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000280 single, randomize, findleaks, use_resources, trace, coverdir,
281 print_slow, and random_seed) allow programmers calling main()
282 directly to set the values that would normally be set by flags
283 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000284 """
Victor Stinner3c18f252011-05-22 15:27:14 +0200285
Victor Stinnercb41cda2011-07-13 23:47:21 +0200286 # Display the Python traceback on fatal errors (e.g. segfault)
Victor Stinner3c18f252011-05-22 15:27:14 +0200287 faulthandler.enable(all_threads=True)
288
Victor Stinnercb41cda2011-07-13 23:47:21 +0200289 # Display the Python traceback on SIGALRM or SIGUSR1 signal
290 signals = []
291 if hasattr(signal, 'SIGALRM'):
292 signals.append(signal.SIGALRM)
293 if hasattr(signal, 'SIGUSR1'):
294 signals.append(signal.SIGUSR1)
295 for signum in signals:
296 faulthandler.register(signum, chain=True)
297
Victor Stinner1802d3f2010-05-19 17:11:19 +0000298 replace_stdout()
299
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000300 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000301 try:
Meador Inge3f55e592012-01-14 09:12:55 -0600302 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsoS:rf:lu:t:TD:NLR:FdwWM:nj:Gm:',
Georg Brandl96f78972009-10-27 21:02:11 +0000303 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
304 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
Meador Ingeb0ae0ac2012-01-14 11:50:33 -0600305 'use=', 'threshold=', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000306 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000307 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitrou8dbd8572011-07-23 22:37:52 +0200308 'start=', 'nowindows', 'header', 'testdir=', 'timeout=', 'wait',
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200309 'failfast', 'match'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000310 except getopt.error as msg:
311 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000312
313 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000314 if random_seed is None:
315 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000316 if use_resources is None:
317 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000318 debug = False
319 start = None
Victor Stinner30196882011-06-03 12:53:26 +0200320 timeout = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000321 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000322 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000323 print(__doc__)
324 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000325 elif o in ('-v', '--verbose'):
326 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000327 elif o in ('-w', '--verbose2'):
328 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000329 elif o in ('-d', '--debug'):
330 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000331 elif o in ('-W', '--verbose3'):
332 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200333 elif o in ('-G', '--failfast'):
334 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000335 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000336 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000337 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000338 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000339 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000340 elif o in ('-S', '--start'):
341 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000342 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000343 single = True
Senthil Kumaranbcfe56d2012-01-14 18:26:27 +0800344 elif o in ('-o', '--slow'):
Christian Heimesb186d002008-03-18 15:15:01 +0000345 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000346 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000347 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000348 elif o == '--randseed':
349 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000350 elif o in ('-f', '--fromfile'):
351 fromfile = a
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200352 elif o in ('-m', '--match'):
353 match_tests = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000354 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000355 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000356 elif o in ('-L', '--runleaks'):
357 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000358 elif o in ('-t', '--threshold'):
359 import gc
360 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000361 elif o in ('-T', '--coverage'):
362 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000363 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400364 # CWD is replaced with a temporary dir before calling main(), so we
365 # need join it with the saved CWD so it goes where the user expects.
366 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000367 elif o in ('-N', '--nocoverdir'):
368 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000369 elif o in ('-R', '--huntrleaks'):
370 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000371 if len(huntrleaks) not in (2, 3):
372 print(a, huntrleaks)
373 usage('-R takes 2 or 3 colon-separated arguments')
374 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000375 huntrleaks[0] = 5
376 else:
377 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000378 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000379 huntrleaks[1] = 4
380 else:
381 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000382 if len(huntrleaks) == 2 or not huntrleaks[2]:
383 huntrleaks[2:] = ["reflog.txt"]
384 # Avoid false positives due to the character cache in
385 # stringobject.c filling slowly with random data
386 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000387 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000388 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000389 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000390 u = [x.lower() for x in a.split(',')]
391 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000392 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000393 use_resources[:] = RESOURCE_NAMES
394 continue
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200395 if r == 'none':
396 del use_resources[:]
397 continue
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000398 remove = False
399 if r[0] == '-':
400 remove = True
401 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000402 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000403 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000404 if remove:
405 if r in use_resources:
406 use_resources.remove(r)
407 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000408 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000409 elif o in ('-n', '--nowindows'):
410 import msvcrt
411 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
412 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
413 msvcrt.SEM_NOGPFAULTERRORBOX|
414 msvcrt.SEM_NOOPENFILEERRORBOX)
415 try:
416 msvcrt.CrtSetReportMode
417 except AttributeError:
418 # release build
419 pass
420 else:
421 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
422 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
423 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000424 elif o in ('-F', '--forever'):
425 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000426 elif o in ('-j', '--multiprocess'):
427 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100428 if use_mp <= 0:
429 try:
430 import multiprocessing
431 # Use all cores + extras for tests that like to sleep
432 use_mp = 2 + multiprocessing.cpu_count()
433 except (ImportError, NotImplementedError):
434 use_mp = 3
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200435 if use_mp == 1:
436 use_mp = None
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000437 elif o == '--header':
438 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000439 elif o == '--slaveargs':
440 args, kwargs = json.loads(a)
441 try:
442 result = runtest(*args, **kwargs)
443 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000444 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000445 sys.stdout.flush()
446 print() # Force a newline (just in case)
447 print(json.dumps(result))
448 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400449 elif o == '--testdir':
450 # CWD is replaced with a temporary dir before calling main(), so we
451 # join it with the saved CWD so it ends up where the user expects.
452 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200453 elif o == '--timeout':
Victor Stinner88345682011-06-06 12:14:23 +0200454 if hasattr(faulthandler, 'dump_tracebacks_later'):
455 timeout = float(a)
456 if timeout <= 0:
457 timeout = None
458 else:
459 print("Warning: The timeout option requires "
460 "faulthandler.dump_tracebacks_later")
Victor Stinner9e586c22011-06-06 12:04:36 +0200461 timeout = None
Brian Curtinf6643452011-04-28 17:45:17 -0500462 elif o == '--wait':
463 input("Press any key to continue...")
R. David Murray35768ad2009-11-15 00:23:21 +0000464 else:
465 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000466 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000467 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000468 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000469 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000470 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000471 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000472 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000473 usage("-l and -j don't go together!")
Nadeem Vawdac1fba3e2011-08-27 15:22:05 +0200474 if use_mp and support.max_memuse:
475 usage("-M and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200476 if failfast and not (verbose or verbose3):
477 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000478
Guido van Rossum152494a1996-12-20 03:12:20 +0000479 good = []
480 bad = []
481 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000482 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000483 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000484 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000485
Neil Schemenauerd569f232000-09-22 15:29:28 +0000486 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000487 try:
488 import gc
489 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000490 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000491 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000492 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000493 # Uncomment the line below to report garbage that is not
494 # freeable by reference counting alone. By default only
495 # garbage that is not collectable by the GC is reported.
496 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000497 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000498
Barry Warsawe11e3de1999-01-28 19:51:51 +0000499 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000500 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000501 try:
502 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000503 next_test = fp.read().strip()
504 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000505 fp.close()
506 except IOError:
507 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000508
509 if fromfile:
510 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000511 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000512 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000513 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000514 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000515 guts = line.split() # assuming no test has whitespace in its name
516 if guts and not guts[0].startswith('#'):
517 tests.extend(guts)
518 fp.close()
519
520 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000521 removepy(args)
522 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000523
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000524 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000525 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000526 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000527 for arg in args:
528 if arg in stdtests:
529 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000530 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000531 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000532
533 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000534 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000535 # Print basic platform information
536 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000537 print("== ", platform.platform(aliased=True),
538 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000539 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000540 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000541
R David Murrayb588f8d2011-03-24 14:42:58 -0400542 # if testdir is set, then we are not running the python tests suite, so
543 # don't add default tests to be executed or skipped (pass empty values)
544 if testdir:
545 alltests = findtests(testdir, list(), set())
546 else:
547 alltests = findtests(testdir, stdtests, nottests)
548
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000549 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000550 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000551 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000552 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000553 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000554 except IndexError:
555 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000556 # Remove all the tests that precede start if it's set.
557 if start:
558 try:
559 del tests[:tests.index(start)]
560 except ValueError:
561 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000562 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000563 random.seed(random_seed)
564 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000565 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000566 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000567 import trace, tempfile
Vinay Sajip7ded1f02012-05-26 03:45:29 +0100568 tracer = trace.Trace(ignoredirs=[sys.base_prefix, sys.base_exec_prefix,
Georg Brandl33c28812009-04-01 23:07:29 +0000569 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000570 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000571
Christian Heimesb186d002008-03-18 15:15:01 +0000572 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000573 support.verbose = verbose # Tell tests to be moderately quiet
574 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000575 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000576
577 def accumulate_result(test, result):
578 ok, test_time = result
579 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000580 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000581 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000582 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000583 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000584 elif ok == ENV_CHANGED:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000585 environment_changed.append(test)
586 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000587 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000588 elif ok == RESOURCE_DENIED:
589 skipped.append(test)
590 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000591
R. David Murray7dc72cc2009-11-14 16:13:02 +0000592 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000593 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000594 while True:
595 for test in tests:
596 yield test
597 if bad:
598 return
599 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000600 test_count = ''
601 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000602 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000603 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000604 test_count = '/{}'.format(len(selected))
605 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000606
Antoine Pitrou88909542009-06-29 13:54:42 +0000607 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000608 try:
609 from threading import Thread
610 except ImportError:
611 print("Multiprocess option requires thread support")
612 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000613 from queue import Queue
614 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000615 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000616 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000617 def tests_and_args():
618 for test in tests:
619 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000620 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000621 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200622 debug=debug, output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200623 timeout=timeout, failfast=failfast,
624 match_tests=match_tests)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000625 )
626 yield (test, args_tuple)
627 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000628 opt_args = support.args_from_interpreter_flags()
629 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000630 def work():
631 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000632 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000633 while True:
634 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000635 test, args_tuple = next(pending)
636 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000637 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000638 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000639 # -E is needed by some tests, e.g. test_import
Antoine Pitrou282083d2012-07-11 19:19:14 +0200640 # Running the child from the same working directory ensures
641 # that TEMPDIR for the child is the same when
642 # sysconfig.is_python_build() is true. See issue 15300.
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000643 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000644 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000645 universal_newlines=True,
Antoine Pitrou282083d2012-07-11 19:19:14 +0200646 close_fds=(os.name != 'nt'),
647 cwd=support.SAVEDCWD)
R. David Murray27144602009-10-19 15:26:16 +0000648 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200649 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000650 # Strip last refcount output line if it exists, since it
651 # comes from the shutdown of the interpreter in the subcommand.
652 stderr = debug_output_pat.sub("", stderr)
653 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200654 if retcode != 0:
655 result = (CHILD_ERROR, "Exit code %s" % retcode)
656 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
657 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000658 if not result:
659 output.put((None, None, None, None))
660 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000661 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000662 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000663 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000664 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000665 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000666 workers = [Thread(target=work) for i in range(use_mp)]
667 for worker in workers:
668 worker.start()
669 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000670 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000671 try:
672 while finished < use_mp:
673 test, stdout, stderr, result = output.get()
674 if test is None:
675 finished += 1
676 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200677 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000678 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300679 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
680 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200681 test_count_width, test_index, test_count,
682 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000683 if stdout:
684 print(stdout)
685 if stderr:
686 print(stderr, file=sys.stderr)
Antoine Pitrou82372582012-06-27 17:41:07 +0200687 sys.stdout.flush()
688 sys.stderr.flush()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000689 if result[0] == INTERRUPTED:
690 assert result[1] == 'KeyboardInterrupt'
691 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200692 if result[0] == CHILD_ERROR:
Victor Stinner571e8fd2011-05-01 22:57:43 +0200693 raise Exception("Child error on {}: {}".format(test, result[1]))
Georg Brandldee7b852010-08-02 18:59:52 +0000694 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000695 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000696 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000697 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000698 for worker in workers:
699 worker.join()
700 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000701 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000702 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300703 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
704 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200705 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000706 sys.stdout.flush()
707 if trace:
708 # If we're tracing code coverage, then we don't exit with status
709 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200710 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000711 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000712 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000713 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100714 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200715 output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200716 timeout=timeout, failfast=failfast,
717 match_tests=match_tests)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000718 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000719 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000720 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000721 break
722 except:
723 raise
724 if findleaks:
725 gc.collect()
726 if gc.garbage:
727 print("Warning: test created", len(gc.garbage), end=' ')
728 print("uncollectable object(s).")
729 # move the uncollectable objects somewhere so we don't see
730 # them again
731 found_garbage.extend(gc.garbage)
732 del gc.garbage[:]
733 # Unload the newly imported modules (best effort finalization)
734 for module in sys.modules.keys():
735 if module not in save_modules and module.startswith("test."):
736 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000737
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000738 if interrupted:
739 # print a newline after ^C
740 print()
741 print("Test suite interrupted by signal SIGINT.")
742 omitted = set(selected) - set(good) - set(bad) - set(skipped)
743 print(count(len(omitted), "test"), "omitted:")
744 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000745 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000746 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000747 print("All", end=' ')
748 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000749 if print_slow:
750 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000751 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000752 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000753 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000754 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000755 bad = sorted(set(bad) - set(environment_changed))
756 if bad:
757 print(count(len(bad), "test"), "failed:")
758 printlist(bad)
Vinay Sajipf9596182012-03-02 01:01:13 +0000759 if environment_changed:
760 print("{} altered the execution environment:".format(
761 count(len(environment_changed), "test")))
762 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000763 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000764 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000765 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000766
Tim Petersb5b7b782001-08-12 01:20:39 +0000767 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000768 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000769 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000770 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000771 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000772 print(count(len(surprise), "skip"), \
773 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000774 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000775 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000776 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000777 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000778 print("Ask someone to teach regrtest.py about which tests are")
779 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000780
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000781 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000782 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000783 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000784 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000785 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000786 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000787 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200788 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000789 except KeyboardInterrupt:
790 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000791 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000792 break
793 except:
794 raise
795
Barry Warsawe11e3de1999-01-28 19:51:51 +0000796 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000797 if next_single_test:
798 with open(filename, 'w') as fp:
799 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000800 else:
801 os.unlink(filename)
802
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000803 if trace:
804 r = tracer.results()
805 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
806
Skip Montanaro0179a182004-06-06 15:53:18 +0000807 if runleaks:
808 os.system("leaks %d" % os.getpid())
809
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000810 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000811
Guido van Rossum152494a1996-12-20 03:12:20 +0000812
R David Murrayb588f8d2011-03-24 14:42:58 -0400813# small set of tests to determine if we have a basically functioning interpreter
814# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000815STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000816 'test_grammar',
817 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000818 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000819 'test_builtin',
820 'test_exceptions',
821 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000822 'test_unittest',
823 'test_doctest',
824 'test_doctest2',
Eli Benderskyd18a0472011-07-27 20:21:45 +0300825 'test_support'
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000826]
Guido van Rossum152494a1996-12-20 03:12:20 +0000827
R David Murrayb588f8d2011-03-24 14:42:58 -0400828# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400829NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000830
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000831def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000832 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000833 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000834 names = os.listdir(testdir)
835 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000836 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000837 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400838 mod, ext = os.path.splitext(name)
839 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
840 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000841 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000842
Victor Stinnerf58087b2010-05-02 17:24:51 +0000843def replace_stdout():
844 """Set stdout encoder error handler to backslashreplace (as stderr error
845 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000846 import atexit
847
Victor Stinnerf58087b2010-05-02 17:24:51 +0000848 stdout = sys.stdout
849 sys.stdout = open(stdout.fileno(), 'w',
850 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000851 errors="backslashreplace",
Victor Stinnerbe621032011-05-25 02:01:55 +0200852 closefd=False,
853 newline='\n')
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000854
855 def restore_stdout():
856 sys.stdout.close()
857 sys.stdout = stdout
858 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000859
Antoine Pitrou88909542009-06-29 13:54:42 +0000860def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100861 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200862 output_on_failure=False, failfast=False, match_tests=None,
863 timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000864 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000865
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000866 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000867 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000868 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000869 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000870 huntrleaks -- run multiple times to test for leaks; requires a debug
871 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200872 output_on_failure -- if true, display test output on failure
Victor Stinner0cc8d592011-03-31 18:10:13 +0200873 timeout -- dump the traceback and exit if a test takes more than
874 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000875
876 Returns one of the test result constants:
877 INTERRUPTED KeyboardInterrupt when run under -j
878 RESOURCE_DENIED test skipped because resource denied
879 SKIPPED test skipped for some other reason
880 ENV_CHANGED test failed because it changed the execution environment
881 FAILED test failed
882 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000883 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000884
Antoine Pitrou88909542009-06-29 13:54:42 +0000885 if use_resources is not None:
886 support.use_resources = use_resources
Victor Stinner30196882011-06-03 12:53:26 +0200887 use_timeout = (timeout is not None)
Victor Stinner7d648a02011-03-31 18:27:50 +0200888 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200889 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000890 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200891 support.match_tests = match_tests
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200892 if failfast:
893 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200894 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200895 support.verbose = True
896
897 # Reuse the same instance to all calls to runtest(). Some
898 # tests keep a reference to sys.stdout or sys.stderr
899 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200900 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200901 stream = io.StringIO()
902 runtest.stringio = stream
903 else:
904 stream = runtest.stringio
905 stream.seek(0)
906 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200907
908 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200909 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200910 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200911 sys.stdout = stream
912 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200913 result = runtest_inner(test, verbose, quiet, huntrleaks,
914 debug, display_failure=False)
915 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200916 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200917 orig_stderr.write(output)
918 orig_stderr.flush()
919 finally:
920 sys.stdout = orig_stdout
921 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200922 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200923 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200924 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
925 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100926 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000927 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200928 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200929 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000930 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200931runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000932
Nick Coghlan6ead5522009-10-18 13:19:33 +0000933# Unit tests are supposed to leave the execution environment unchanged
934# once they complete. But sometimes tests have bugs, especially when
935# tests fail, and the changes to environment go on to mess up other
936# tests. This can cause issues with buildbot stability, since tests
937# are run in random order and so problems may appear to come and go.
938# There are a few things we can save and restore to mitigate this, and
939# the following context manager handles this task.
940
941class saved_test_environment:
942 """Save bits of the test environment and restore them at block exit.
943
944 with saved_test_environment(testname, verbose, quiet):
945 #stuff
946
947 Unless quiet is True, a warning is printed to stderr if any of
948 the saved items was changed by the test. The attribute 'changed'
949 is initially False, but is set to True if a change is detected.
950
951 If verbose is more than 1, the before and after state of changed
952 items is also printed.
953 """
954
955 changed = False
956
957 def __init__(self, testname, verbose=0, quiet=False):
958 self.testname = testname
959 self.verbose = verbose
960 self.quiet = quiet
961
962 # To add things to save and restore, add a name XXX to the resources list
963 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
964 # return the value to be saved and compared against a second call to the
965 # get function when test execution completes. restore_XXX should accept
966 # the saved value and restore the resource using it. It will be called if
967 # and only if a change in the value is detected.
968 #
969 # Note: XXX will have any '.' replaced with '_' characters when determining
970 # the corresponding method names.
971
972 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000973 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000974 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200975 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200976 'sys.warnoptions', 'threading._dangling',
Éric Araujo28df8de2011-09-19 05:10:45 +0200977 'multiprocessing.process._dangling',
Éric Araujoec177c12012-06-24 03:27:43 -0400978 'sysconfig._CONFIG_VARS', 'sysconfig._INSTALL_SCHEMES',
Vinay Sajipf9596182012-03-02 01:01:13 +0000979 'support.TESTFN',
Éric Araujo28df8de2011-09-19 05:10:45 +0200980 )
Nick Coghlan6ead5522009-10-18 13:19:33 +0000981
982 def get_sys_argv(self):
983 return id(sys.argv), sys.argv, sys.argv[:]
984 def restore_sys_argv(self, saved_argv):
985 sys.argv = saved_argv[1]
986 sys.argv[:] = saved_argv[2]
987
988 def get_cwd(self):
989 return os.getcwd()
990 def restore_cwd(self, saved_cwd):
991 os.chdir(saved_cwd)
992
993 def get_sys_stdout(self):
994 return sys.stdout
995 def restore_sys_stdout(self, saved_stdout):
996 sys.stdout = saved_stdout
997
998 def get_sys_stderr(self):
999 return sys.stderr
1000 def restore_sys_stderr(self, saved_stderr):
1001 sys.stderr = saved_stderr
1002
1003 def get_sys_stdin(self):
1004 return sys.stdin
1005 def restore_sys_stdin(self, saved_stdin):
1006 sys.stdin = saved_stdin
1007
1008 def get_os_environ(self):
1009 return id(os.environ), os.environ, dict(os.environ)
1010 def restore_os_environ(self, saved_environ):
1011 os.environ = saved_environ[1]
1012 os.environ.clear()
1013 os.environ.update(saved_environ[2])
1014
1015 def get_sys_path(self):
1016 return id(sys.path), sys.path, sys.path[:]
1017 def restore_sys_path(self, saved_path):
1018 sys.path = saved_path[1]
1019 sys.path[:] = saved_path[2]
1020
Brett Cannon055470a2010-02-19 15:57:10 +00001021 def get_sys_path_hooks(self):
1022 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
1023 def restore_sys_path_hooks(self, saved_hooks):
1024 sys.path_hooks = saved_hooks[1]
1025 sys.path_hooks[:] = saved_hooks[2]
1026
Brett Cannon31f59292011-02-21 19:29:56 +00001027 def get_sys_gettrace(self):
1028 return sys.gettrace()
1029 def restore_sys_gettrace(self, trace_fxn):
1030 sys.settrace(trace_fxn)
1031
Brett Cannon055470a2010-02-19 15:57:10 +00001032 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +00001033 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +00001034 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +00001035 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +00001036
Brett Cannon29c0e4f2010-03-20 22:22:57 +00001037 def get_warnings_filters(self):
1038 return id(warnings.filters), warnings.filters, warnings.filters[:]
1039 def restore_warnings_filters(self, saved_filters):
1040 warnings.filters = saved_filters[1]
1041 warnings.filters[:] = saved_filters[2]
1042
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001043 def get_asyncore_socket_map(self):
1044 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +00001045 # XXX Making a copy keeps objects alive until __exit__ gets called.
1046 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001047 def restore_asyncore_socket_map(self, saved_map):
1048 asyncore = sys.modules.get('asyncore')
1049 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +00001050 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001051 asyncore.socket_map.update(saved_map)
1052
Éric Araujoff913062011-11-29 16:45:07 +01001053 def get_shutil_archive_formats(self):
1054 # we could call get_archives_formats() but that only returns the
1055 # registry keys; we want to check the values too (the functions that
1056 # are registered)
1057 return shutil._ARCHIVE_FORMATS, shutil._ARCHIVE_FORMATS.copy()
1058 def restore_shutil_archive_formats(self, saved):
1059 shutil._ARCHIVE_FORMATS = saved[0]
1060 shutil._ARCHIVE_FORMATS.clear()
1061 shutil._ARCHIVE_FORMATS.update(saved[1])
1062
1063 def get_shutil_unpack_formats(self):
1064 return shutil._UNPACK_FORMATS, shutil._UNPACK_FORMATS.copy()
1065 def restore_shutil_unpack_formats(self, saved):
1066 shutil._UNPACK_FORMATS = saved[0]
1067 shutil._UNPACK_FORMATS.clear()
1068 shutil._UNPACK_FORMATS.update(saved[1])
1069
Nick Coghlan7d819752010-12-05 06:45:03 +00001070 def get_logging__handlers(self):
1071 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001072 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001073 def restore_logging__handlers(self, saved_handlers):
1074 # Can't easily revert the logging state
1075 pass
1076
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001077 def get_logging__handlerList(self):
1078 # _handlerList is a list of weakrefs to handlers
1079 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1080 def restore_logging__handlerList(self, saved_handlerList):
1081 # Can't easily revert the logging state
1082 pass
1083
Ezio Melotti0123e052011-03-20 15:09:26 +02001084 def get_sys_warnoptions(self):
1085 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1086 def restore_sys_warnoptions(self, saved_options):
1087 sys.warnoptions = saved_options[1]
1088 sys.warnoptions[:] = saved_options[2]
1089
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02001090 # Controlling dangling references to Thread objects can make it easier
1091 # to track reference leaks.
1092 def get_threading__dangling(self):
1093 if not threading:
1094 return None
1095 # This copies the weakrefs without making any strong reference
1096 return threading._dangling.copy()
1097 def restore_threading__dangling(self, saved):
1098 if not threading:
1099 return
1100 threading._dangling.clear()
1101 threading._dangling.update(saved)
1102
1103 # Same for Process objects
1104 def get_multiprocessing_process__dangling(self):
1105 if not multiprocessing:
1106 return None
1107 # This copies the weakrefs without making any strong reference
1108 return multiprocessing.process._dangling.copy()
1109 def restore_multiprocessing_process__dangling(self, saved):
1110 if not multiprocessing:
1111 return
1112 multiprocessing.process._dangling.clear()
1113 multiprocessing.process._dangling.update(saved)
1114
Éric Araujo28df8de2011-09-19 05:10:45 +02001115 def get_sysconfig__CONFIG_VARS(self):
1116 # make sure the dict is initialized
1117 sysconfig.get_config_var('prefix')
1118 return (id(sysconfig._CONFIG_VARS), sysconfig._CONFIG_VARS,
1119 dict(sysconfig._CONFIG_VARS))
1120 def restore_sysconfig__CONFIG_VARS(self, saved):
1121 sysconfig._CONFIG_VARS = saved[1]
1122 sysconfig._CONFIG_VARS.clear()
1123 sysconfig._CONFIG_VARS.update(saved[2])
1124
Éric Araujoec177c12012-06-24 03:27:43 -04001125 def get_sysconfig__INSTALL_SCHEMES(self):
1126 return (id(sysconfig._INSTALL_SCHEMES), sysconfig._INSTALL_SCHEMES,
1127 sysconfig._INSTALL_SCHEMES.copy())
1128 def restore_sysconfig__INSTALL_SCHEMES(self, saved):
1129 sysconfig._INSTALL_SCHEMES = saved[1]
1130 sysconfig._INSTALL_SCHEMES.clear()
1131 sysconfig._INSTALL_SCHEMES.update(saved[2])
Éric Araujo28df8de2011-09-19 05:10:45 +02001132
Vinay Sajipf9596182012-03-02 01:01:13 +00001133 def get_support_TESTFN(self):
1134 if os.path.isfile(support.TESTFN):
1135 result = 'f'
1136 elif os.path.isdir(support.TESTFN):
1137 result = 'd'
1138 else:
1139 result = None
1140 return result
1141 def restore_support_TESTFN(self, saved_value):
1142 if saved_value is None:
1143 if os.path.isfile(support.TESTFN):
1144 os.unlink(support.TESTFN)
1145 elif os.path.isdir(support.TESTFN):
1146 shutil.rmtree(support.TESTFN)
Éric Araujo28df8de2011-09-19 05:10:45 +02001147
Nick Coghlan6ead5522009-10-18 13:19:33 +00001148 def resource_info(self):
1149 for name in self.resources:
1150 method_suffix = name.replace('.', '_')
1151 get_name = 'get_' + method_suffix
1152 restore_name = 'restore_' + method_suffix
1153 yield name, getattr(self, get_name), getattr(self, restore_name)
1154
1155 def __enter__(self):
1156 self.saved_values = dict((name, get()) for name, get, restore
1157 in self.resource_info())
1158 return self
1159
1160 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001161 saved_values = self.saved_values
1162 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001163 for name, get, restore in self.resource_info():
1164 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001165 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001166 # Check for changes to the resource's value
1167 if current != original:
1168 self.changed = True
1169 restore(original)
1170 if not self.quiet:
1171 print("Warning -- {} was modified by {}".format(
1172 name, self.testname),
1173 file=sys.stderr)
1174 if self.verbose > 1:
1175 print(" Before: {}\n After: {} ".format(
1176 original, current),
1177 file=sys.stderr)
1178 return False
1179
1180
Victor Stinnera7c33e52011-06-29 13:00:54 +02001181def runtest_inner(test, verbose, quiet,
1182 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001183 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001184
Antoine Pitrou88909542009-06-29 13:54:42 +00001185 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001186 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001187 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001188 if test.startswith('test.'):
1189 abstest = test
1190 else:
1191 # Always import it from the test package
1192 abstest = 'test.' + test
1193 with saved_test_environment(test, verbose, quiet) as environment:
1194 start_time = time.time()
1195 the_package = __import__(abstest, globals(), locals(), [])
1196 the_module = getattr(the_package, test)
R David Murray78fc25c2012-04-09 08:55:42 -04001197 # If the test has a test_main, that will run the appropriate
1198 # tests. If not, use normal unittest test loading.
1199 test_runner = getattr(the_module, "test_main", None)
1200 if test_runner is None:
1201 tests = unittest.TestLoader().loadTestsFromModule(the_module)
1202 test_runner = lambda: support.run_unittest(tests)
1203 test_runner()
R. David Murray0bc11ae2009-10-18 22:18:17 +00001204 if huntrleaks:
R David Murray78fc25c2012-04-09 08:55:42 -04001205 refleak = dash_R(the_module, test, test_runner,
R. David Murray0bc11ae2009-10-18 22:18:17 +00001206 huntrleaks)
1207 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001208 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001209 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001210 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001211 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001212 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001213 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001214 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001215 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001216 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001217 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001218 except KeyboardInterrupt:
1219 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001220 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001221 if display_failure:
1222 print("test", test, "failed --", msg, file=sys.stderr)
1223 else:
1224 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001225 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001226 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001227 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001228 msg = traceback.format_exc()
1229 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001230 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001231 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001232 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001233 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001234 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001235 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001236 return ENV_CHANGED, test_time
1237 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001238
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001239def cleanup_test_droppings(testname, verbose):
1240 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001241 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001242 import gc
1243
1244 # First kill any dangling references to open files etc.
1245 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001246 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001247 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001248
1249 # Try to clean up junk commonly left behind. While tests shouldn't leave
1250 # any files or directories behind, when a test fails that can be tedious
1251 # for it to arrange. The consequences can be especially nasty on Windows,
1252 # since if a test leaves a file open, it cannot be deleted by name (while
1253 # there's nothing we can do about that here either, we can display the
1254 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001255 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001256 "db_home",
1257 ):
1258 if not os.path.exists(name):
1259 continue
1260
1261 if os.path.isdir(name):
1262 kind, nuker = "directory", shutil.rmtree
1263 elif os.path.isfile(name):
1264 kind, nuker = "file", os.unlink
1265 else:
1266 raise SystemError("os.path says %r exists but is neither "
1267 "directory nor file" % name)
1268
1269 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001270 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001271 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001272 # if we have chmod, fix possible permissions problems
1273 # that might prevent cleanup
1274 if (hasattr(os, 'chmod')):
1275 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001276 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001277 except Exception as msg:
1278 print(("%r left behind %s %r and it couldn't be "
1279 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001280
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001281def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001282 """Run a test multiple times, looking for reference leaks.
1283
1284 Returns:
1285 False if the test didn't leak references; True if we detected refleaks.
1286 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001287 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001288 import copyreg
1289 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001290
1291 if not hasattr(sys, 'gettotalrefcount'):
1292 raise Exception("Tracking reference leaks requires a debug build "
1293 "of Python")
1294
1295 # Save current values for dash_R_cleanup() to restore.
1296 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001297 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001298 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001299 try:
1300 import zipimport
1301 except ImportError:
1302 zdc = None # Run unmodified on platforms without zipimport support
1303 else:
1304 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001305 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001306 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001307 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001308 continue
1309 for obj in abc.__subclasses__() + [abc]:
1310 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001311
1312 if indirect_test:
1313 def run_the_test():
1314 indirect_test()
1315 else:
1316 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001317 del sys.modules[the_module.__name__]
1318 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001319
1320 deltas = []
1321 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001322 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001323 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001324 print("beginning", repcount, "repetitions", file=sys.stderr)
1325 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001326 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001327 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001328 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001329 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001330 run_the_test()
1331 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001332 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001333 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001334 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001335 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001336 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001337 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001338 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001339 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001340 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001341 sys.stderr.flush()
1342 with open(fname, "a") as refrep:
1343 print(msg, file=refrep)
1344 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001345 return True
1346 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001347
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001348def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001349 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001350 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001351 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001352 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001353 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001354 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001355
Christian Heimesdae2a892008-04-19 00:55:37 +00001356 # Clear the warnings registry, so they can be displayed again
1357 for mod in sys.modules.values():
1358 if hasattr(mod, '__warningregistry__'):
1359 del mod.__warningregistry__
1360
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001361 # Restore some original values.
1362 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001363 copyreg.dispatch_table.clear()
1364 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001365 sys.path_importer_cache.clear()
1366 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001367 try:
1368 import zipimport
1369 except ImportError:
1370 pass # Run unmodified on platforms without zipimport support
1371 else:
1372 zipimport._zip_directory_cache.clear()
1373 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001374
Christian Heimes26855632008-01-27 23:50:43 +00001375 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001376 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001377
Guido van Rossum3de862d2007-08-18 00:10:33 +00001378 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001379 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001380 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001381 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001382 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001383 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001384 obj._abc_cache.clear()
1385 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001386
Antoine Pitrou046467c2009-10-30 18:30:35 +00001387 # Flush standard output, so that buffered data is sent to the OS and
1388 # associated Python objects are reclaimed.
1389 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1390 if stream is not None:
1391 stream.flush()
1392
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001393 # Clear assorted module caches.
1394 _path_created.clear()
1395 re.purge()
1396 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001397 urllib.parse.clear_cache()
1398 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001399 linecache.clearcache()
1400 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001401 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001402 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001403 doctest.master = None
Meador Inge11e38132011-11-25 22:33:32 -06001404 try:
1405 import ctypes
1406 except ImportError:
1407 # Don't worry about resetting the cache if ctypes is not supported
1408 pass
1409 else:
1410 ctypes._reset_cache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001411
1412 # Collect cyclic trash.
1413 gc.collect()
1414
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001415def warm_char_cache():
1416 s = bytes(range(256))
1417 for i in range(256):
1418 s[i:i+1]
1419
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001420def findtestdir(path=None):
1421 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001422
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001423def removepy(names):
1424 if not names:
1425 return
1426 for idx, name in enumerate(names):
1427 basename, ext = os.path.splitext(name)
1428 if ext == '.py':
1429 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001430
Guido van Rossum152494a1996-12-20 03:12:20 +00001431def count(n, word):
1432 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001433 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001434 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001435 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001436
Tim Petersa45da922001-08-12 03:45:50 +00001437def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001438 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001439
1440 Optional arg width (default 70) is the maximum line length.
1441 Optional arg indent (default 4) is the number of blanks with which to
1442 begin each line.
1443 """
1444
Tim Petersba78bc42002-07-04 19:45:06 +00001445 from textwrap import fill
1446 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001447 # Print the sorted list: 'x' may be a '--random' list or a set()
1448 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001449 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001450
Tim Petersde14a302002-04-01 05:04:46 +00001451# Map sys.platform to a string containing the basenames of tests
1452# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001453#
1454# Special cases:
1455# test_pep277
1456# The _ExpectedSkips constructor adds this to the set of expected
1457# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001458# test_timeout
1459# Controlled by test_timeout.skip_expected. Requires the network
1460# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001461#
1462# Tests that are expected to be skipped everywhere except on one platform
1463# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001464
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001465_expectations = (
1466 ('win32',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001467 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001468 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001469 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001470 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001471 test_dbm
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001472 test_devpoll
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001473 test_fcntl
1474 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001475 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001476 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001477 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001478 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001479 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001480 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001481 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001482 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001483 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001484 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001485 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001486 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001487 test_pty
1488 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001489 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001490 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001491 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001492 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001493 test_wait3
1494 test_wait4
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001495 """),
1496 ('linux',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001497 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001498 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001499 test_devpoll
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001500 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001501 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001502 test_ossaudiodev
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001503 """),
1504 ('unixware',
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001505 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001506 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001507 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001508 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001509 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001510 test_openpty
1511 test_pyexpat
1512 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001513 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001514 """),
1515 ('openunix',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001516 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001517 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001518 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001519 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001520 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001521 test_openpty
1522 test_pyexpat
1523 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001524 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001525 """),
1526 ('sco_sv',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001527 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001528 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001529 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001530 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001531 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001532 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001533 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001534 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001535 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001536 test_openpty
1537 test_pyexpat
1538 test_queue
1539 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001540 test_sundry
1541 test_thread
1542 test_threaded_import
1543 test_threadedtempfile
1544 test_threading
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001545 """),
1546 ('darwin',
Jack Jansen398c2362001-12-02 21:41:36 +00001547 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001548 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001549 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001550 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001551 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001552 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001553 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001554 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001555 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001556 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001557 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001558 test_poll
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001559 """),
1560 ('sunos',
Guido van Rossum11c3f092002-07-17 15:08:24 +00001561 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001562 test_curses
1563 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001564 test_epoll
1565 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001566 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001567 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001568 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001569 test_zipfile
1570 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001571 """),
1572 ('hp-ux',
Skip Montanarob3230212002-03-15 02:54:03 +00001573 """
Skip Montanarob3230212002-03-15 02:54:03 +00001574 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001575 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001576 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001577 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001578 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001579 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001580 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001581 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001582 test_openpty
1583 test_pyexpat
1584 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001585 test_zipfile
1586 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001587 """),
1588 ('cygwin',
Jason Tishler25115942002-12-05 15:18:15 +00001589 """
Jason Tishler25115942002-12-05 15:18:15 +00001590 test_curses
1591 test_dbm
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001592 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001593 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001594 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001595 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001596 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001597 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001598 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001599 test_socketserver
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001600 """),
1601 ('os2emx',
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001602 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001603 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001604 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001605 test_epoll
1606 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001607 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001608 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001609 test_openpty
1610 test_ossaudiodev
1611 test_pty
1612 test_resource
1613 test_signal
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001614 """),
1615 ('freebsd',
Guido van Rossum944a6c32003-11-20 22:11:29 +00001616 """
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001617 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001618 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001619 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001620 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001621 test_ossaudiodev
1622 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001623 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001624 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001625 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001626 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001627 test_ttk_guionly
1628 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001629 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001630 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001631 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001632 """),
1633 ('aix',
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001634 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001635 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001636 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001637 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001638 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001639 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001640 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001641 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001642 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001643 test_ttk_guionly
1644 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001645 test_zipimport
1646 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001647 """),
1648 ('openbsd',
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001649 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001650 test_ctypes
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001651 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001652 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001653 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001654 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001655 test_normalization
1656 test_ossaudiodev
1657 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001658 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001659 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001660 test_ttk_guionly
1661 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001662 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001663 """),
1664 ('netbsd',
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001665 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001666 test_ctypes
1667 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001668 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001669 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001670 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001671 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001672 test_ossaudiodev
1673 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001674 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001675 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001676 test_ttk_guionly
1677 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001678 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001679 """),
1680)
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001681
Tim Petersb5b7b782001-08-12 01:20:39 +00001682class _ExpectedSkips:
1683 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001684 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001685 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001686
Tim Peters7c7efe92002-08-23 17:55:54 +00001687 self.valid = False
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001688 expected = None
1689 for item in _expectations:
1690 if sys.platform.startswith(item[0]):
1691 expected = item[1]
1692 break
1693 if expected is not None:
1694 self.expected = set(expected.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001695
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001696 # These are broken tests, for now skipped on every platform.
1697 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001698 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001699
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001700 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001701 if not os.path.supports_unicode_filenames:
1702 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001703
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001704 # doctest, profile and cProfile tests fail when the codec for the
1705 # fs encoding isn't built in because PyUnicode_Decode() adds two
1706 # calls into Python.
1707 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1708 if sys.getfilesystemencoding().lower() not in encs:
1709 self.expected.add('test_profile')
1710 self.expected.add('test_cProfile')
1711 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001712
Neal Norwitz55b61d22003-02-28 19:57:03 +00001713 if test_timeout.skip_expected:
1714 self.expected.add('test_timeout')
1715
Tim Petersecd79eb2003-01-29 00:35:32 +00001716 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001717 # test_sqlite is only reliable on Windows where the library
1718 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001719 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001720 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001721 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001722 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001723
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001724 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001725 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001726
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001727 if support.python_is_optimized():
1728 self.expected.add("test_gdb")
1729
Tim Peters7c7efe92002-08-23 17:55:54 +00001730 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001731
1732 def isvalid(self):
1733 "Return true iff _ExpectedSkips knows about the current platform."
1734 return self.valid
1735
1736 def getexpected(self):
1737 """Return set of test names we expect to skip on current platform.
1738
1739 self.isvalid() must be true.
1740 """
1741
1742 assert self.isvalid()
1743 return self.expected
1744
Michael Foord3ab34cc2010-12-03 12:27:40 +00001745def _make_temp_dir_for_build(TEMPDIR):
1746 # When tests are run from the Python build directory, it is best practice
1747 # to keep the test files in a subfolder. It eases the cleanup of leftover
1748 # files using command "make distclean".
1749 if sysconfig.is_python_build():
1750 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1751 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001752 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001753 os.mkdir(TEMPDIR)
Florent Xicluna68f71a32011-10-28 16:06:23 +02001754 except FileExistsError:
1755 pass
Michael Foord3ab34cc2010-12-03 12:27:40 +00001756
1757 # Define a writable temp dir that will be used as cwd while running
1758 # the tests. The name of the dir includes the pid to allow parallel
1759 # testing (see the -j option).
1760 TESTCWD = 'test_python_{}'.format(os.getpid())
1761
1762 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1763 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001764
Guido van Rossum152494a1996-12-20 03:12:20 +00001765if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001766 # Remove regrtest.py's own directory from the module search path. Despite
1767 # the elimination of implicit relative imports, this is still needed to
1768 # ensure that submodules of the test package do not inappropriately appear
1769 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1770 # directly instead of using the -m switch
1771 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1772 i = len(sys.path)
1773 while i >= 0:
1774 i -= 1
1775 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1776 del sys.path[i]
1777
Florent Xiclunadc69e722010-09-13 16:35:02 +00001778 # findtestdir() gets the dirname out of __file__, so we have to make it
1779 # absolute before changing the working directory.
1780 # For example __file__ may be relative when running trace or profile.
1781 # See issue #9323.
1782 __file__ = os.path.abspath(__file__)
1783
1784 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001785 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001786
Michael Foord3ab34cc2010-12-03 12:27:40 +00001787 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001788
1789 # Run the tests in a context manager that temporary changes the CWD to a
1790 # temporary and writable directory. If it's not possible to create or
1791 # change the CWD, the original CWD will be used. The original CWD is
1792 # available from support.SAVEDCWD.
1793 with support.temp_cwd(TESTCWD, quiet=True):
1794 main()