blob: bbda525bea50a1b4f787bf2d6c16c09f83af203f [file] [log] [blame]
Benjamin Peterson90f5ba52010-03-11 22:53:45 +00001#! /usr/bin/env python3
Guido van Rossum152494a1996-12-20 03:12:20 +00002
R. David Murray0ba81e02010-04-26 17:02:32 +00003"""
4Usage:
Guido van Rossum152494a1996-12-20 03:12:20 +00005
Georg Brandlbe41a482011-01-05 21:47:47 +00006python -m test [options] [test_name1 [test_name2 ...]]
R. David Murray0ba81e02010-04-26 17:02:32 +00007python path/to/Lib/test/regrtest.py [options] [test_name1 [test_name2 ...]]
Guido van Rossum152494a1996-12-20 03:12:20 +00008
R. David Murray0ba81e02010-04-26 17:02:32 +00009
10If no arguments or options are provided, finds all files matching
11the pattern "test_*" in the Lib/test subdirectory and runs
12them in alphabetical order (but see -M and -u, below, for exceptions).
13
14For more rigorous testing, it is useful to use the following
15command line:
16
Georg Brandlbe41a482011-01-05 21:47:47 +000017python -E -Wd -m test [options] [test_name1 ...]
R. David Murray0ba81e02010-04-26 17:02:32 +000018
19
20Options:
Guido van Rossum152494a1996-12-20 03:12:20 +000021
Georg Brandl96f78972009-10-27 21:02:11 +000022-h/--help -- print this text and exit
Victor Stinner0cc8d592011-03-31 18:10:13 +020023--timeout TIMEOUT
24 -- dump the traceback and exit if a test takes more
Victor Stinner9e586c22011-06-06 12:04:36 +020025 than TIMEOUT seconds; disabled if TIMEOUT is negative
26 or equals to zero
Brian Curtinf6643452011-04-28 17:45:17 -050027--wait -- wait for user input, e.g., allow a debugger to be attached
Georg Brandl96f78972009-10-27 21:02:11 +000028
29Verbosity
30
31-v/--verbose -- run tests in verbose mode with output to stdout
32-w/--verbose2 -- re-run failed tests in verbose mode
Victor Stinnera7c33e52011-06-29 13:00:54 +020033-W/--verbose3 -- display test output on failure
Georg Brandl96f78972009-10-27 21:02:11 +000034-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000035-q/--quiet -- no output unless one or more tests fail
Georg Brandl96f78972009-10-27 21:02:11 +000036-S/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000037 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000038
39Selecting tests
40
R. David Murray0ba81e02010-04-26 17:02:32 +000041-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000042 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000043-f/--fromfile -- read names of tests to run from a file (see below)
44-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000045-s/--single -- single step through a set of tests (see below)
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
Brett Cannon45071902010-06-14 22:22:54 +0000168import builtins
Victor Stinner024e37a2011-03-31 01:31:06 +0200169import faulthandler
Guido van Rossum152494a1996-12-20 03:12:20 +0000170import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200171import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000172import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200173import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000174import os
Éric Araujo76558e12011-10-06 02:44:19 +0200175import packaging.database
Victor Stinner8313d6a2011-06-29 15:22:26 +0200176import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000177import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000178import re
Victor Stinnercb41cda2011-07-13 23:47:21 +0200179import signal
Christian Heimesb186d002008-03-18 15:15:01 +0000180import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000181import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200182import tempfile
183import time
184import traceback
185import unittest
186import warnings
187from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000188
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200189try:
190 import threading
191except ImportError:
192 threading = None
193try:
194 import multiprocessing.process
195except ImportError:
196 multiprocessing = None
197
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000198
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000199# Some times __path__ and __file__ are not absolute (e.g. while running from
200# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
201# imports might fail. This affects only the modules imported before os.chdir().
202# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
203# they are found in the CWD their __file__ and __path__ will be relative (this
204# happens before the chdir). All the modules imported after the chdir, are
205# not found in the CWD, and since the other paths in sys.path[1:] are absolute
206# (site.py absolutize them), the __file__ and __path__ will be absolute too.
207# Therefore it is necessary to absolutize manually the __file__ and __path__ of
208# the packages to prevent later imports to fail when the CWD is different.
209for module in sys.modules.values():
210 if hasattr(module, '__path__'):
211 module.__path__ = [os.path.abspath(path) for path in module.__path__]
212 if hasattr(module, '__file__'):
213 module.__file__ = os.path.abspath(module.__file__)
214
Guido van Rossumdc15c272002-08-12 21:55:51 +0000215
Guido van Rossumbb484652002-12-02 09:56:21 +0000216# MacOSX (a.k.a. Darwin) has a default stack size that is too small
217# for deeply recursive regular expressions. We see this as crashes in
218# the Python test suite when running test_re.py and test_sre.py. The
219# fix is to set the stack limit to 2048.
220# This approach may also be useful for other Unixy platforms that
221# suffer from small default stack limits.
222if sys.platform == 'darwin':
223 try:
224 import resource
225 except ImportError:
226 pass
227 else:
228 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
229 newsoft = min(hard, max(soft, 1024*2048))
230 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
231
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000232# Test result constants.
233PASSED = 1
234FAILED = 0
235ENV_CHANGED = -1
236SKIPPED = -2
237RESOURCE_DENIED = -3
238INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200239CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000240
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000241from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000242
Georg Brandl1158a332009-06-04 09:30:30 +0000243RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000244 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000245
Florent Xiclunaec882212010-08-09 16:56:43 +0000246TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000247
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000248def usage(msg):
249 print(msg, file=sys.stderr)
250 print("Use --help for usage", file=sys.stderr)
251 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000252
253
Antoine Pitrou88909542009-06-29 13:54:42 +0000254def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000255 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000256 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000257 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000258 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200259 header=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000260 """Execute a test suite.
261
Thomas Wouters7e474022000-07-16 12:04:32 +0000262 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000263 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000264
265 tests -- a list of strings containing test names (optional)
266 testdir -- the directory in which to look for tests (optional)
267
268 Users other than the Python test suite will certainly want to
269 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000270 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000271
272 If the tests argument is omitted, the tests listed on the
273 command-line will be used. If that's empty, too, then all *.py
274 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000275
Antoine Pitrou88909542009-06-29 13:54:42 +0000276 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000277 single, randomize, findleaks, use_resources, trace, coverdir,
278 print_slow, and random_seed) allow programmers calling main()
279 directly to set the values that would normally be set by flags
280 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000281 """
Victor Stinner3c18f252011-05-22 15:27:14 +0200282
Victor Stinnercb41cda2011-07-13 23:47:21 +0200283 # Display the Python traceback on fatal errors (e.g. segfault)
Victor Stinner3c18f252011-05-22 15:27:14 +0200284 faulthandler.enable(all_threads=True)
285
Victor Stinnercb41cda2011-07-13 23:47:21 +0200286 # Display the Python traceback on SIGALRM or SIGUSR1 signal
287 signals = []
288 if hasattr(signal, 'SIGALRM'):
289 signals.append(signal.SIGALRM)
290 if hasattr(signal, 'SIGUSR1'):
291 signals.append(signal.SIGUSR1)
292 for signum in signals:
293 faulthandler.register(signum, chain=True)
294
Victor Stinner1802d3f2010-05-19 17:11:19 +0000295 replace_stdout()
296
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000297 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000298 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200299 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:Gm:',
Georg Brandl96f78972009-10-27 21:02:11 +0000300 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
301 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
302 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000303 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000304 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitrou8dbd8572011-07-23 22:37:52 +0200305 'start=', 'nowindows', 'header', 'testdir=', 'timeout=', 'wait',
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200306 'failfast', 'match'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000307 except getopt.error as msg:
308 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000309
310 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000311 if random_seed is None:
312 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000313 if use_resources is None:
314 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000315 debug = False
316 start = None
Victor Stinner30196882011-06-03 12:53:26 +0200317 timeout = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000318 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000319 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000320 print(__doc__)
321 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000322 elif o in ('-v', '--verbose'):
323 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000324 elif o in ('-w', '--verbose2'):
325 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000326 elif o in ('-d', '--debug'):
327 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000328 elif o in ('-W', '--verbose3'):
329 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200330 elif o in ('-G', '--failfast'):
331 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000332 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000333 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000334 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000335 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000336 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000337 elif o in ('-S', '--start'):
338 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000339 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000340 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000341 elif o in ('-S', '--slow'):
342 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000343 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000344 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000345 elif o == '--randseed':
346 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000347 elif o in ('-f', '--fromfile'):
348 fromfile = a
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200349 elif o in ('-m', '--match'):
350 match_tests = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000351 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000352 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000353 elif o in ('-L', '--runleaks'):
354 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000355 elif o in ('-t', '--threshold'):
356 import gc
357 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000358 elif o in ('-T', '--coverage'):
359 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000360 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400361 # CWD is replaced with a temporary dir before calling main(), so we
362 # need join it with the saved CWD so it goes where the user expects.
363 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000364 elif o in ('-N', '--nocoverdir'):
365 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000366 elif o in ('-R', '--huntrleaks'):
367 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000368 if len(huntrleaks) not in (2, 3):
369 print(a, huntrleaks)
370 usage('-R takes 2 or 3 colon-separated arguments')
371 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000372 huntrleaks[0] = 5
373 else:
374 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000375 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000376 huntrleaks[1] = 4
377 else:
378 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000379 if len(huntrleaks) == 2 or not huntrleaks[2]:
380 huntrleaks[2:] = ["reflog.txt"]
381 # Avoid false positives due to the character cache in
382 # stringobject.c filling slowly with random data
383 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000384 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000385 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000386 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000387 u = [x.lower() for x in a.split(',')]
388 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000389 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000390 use_resources[:] = RESOURCE_NAMES
391 continue
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200392 if r == 'none':
393 del use_resources[:]
394 continue
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000395 remove = False
396 if r[0] == '-':
397 remove = True
398 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000399 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000400 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000401 if remove:
402 if r in use_resources:
403 use_resources.remove(r)
404 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000405 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000406 elif o in ('-n', '--nowindows'):
407 import msvcrt
408 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
409 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
410 msvcrt.SEM_NOGPFAULTERRORBOX|
411 msvcrt.SEM_NOOPENFILEERRORBOX)
412 try:
413 msvcrt.CrtSetReportMode
414 except AttributeError:
415 # release build
416 pass
417 else:
418 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
419 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
420 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000421 elif o in ('-F', '--forever'):
422 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000423 elif o in ('-j', '--multiprocess'):
424 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100425 if use_mp <= 0:
426 try:
427 import multiprocessing
428 # Use all cores + extras for tests that like to sleep
429 use_mp = 2 + multiprocessing.cpu_count()
430 except (ImportError, NotImplementedError):
431 use_mp = 3
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200432 if use_mp == 1:
433 use_mp = None
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000434 elif o == '--header':
435 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000436 elif o == '--slaveargs':
437 args, kwargs = json.loads(a)
438 try:
439 result = runtest(*args, **kwargs)
440 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000441 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000442 sys.stdout.flush()
443 print() # Force a newline (just in case)
444 print(json.dumps(result))
445 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400446 elif o == '--testdir':
447 # CWD is replaced with a temporary dir before calling main(), so we
448 # join it with the saved CWD so it ends up where the user expects.
449 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200450 elif o == '--timeout':
Victor Stinner88345682011-06-06 12:14:23 +0200451 if hasattr(faulthandler, 'dump_tracebacks_later'):
452 timeout = float(a)
453 if timeout <= 0:
454 timeout = None
455 else:
456 print("Warning: The timeout option requires "
457 "faulthandler.dump_tracebacks_later")
Victor Stinner9e586c22011-06-06 12:04:36 +0200458 timeout = None
Brian Curtinf6643452011-04-28 17:45:17 -0500459 elif o == '--wait':
460 input("Press any key to continue...")
R. David Murray35768ad2009-11-15 00:23:21 +0000461 else:
462 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000463 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000464 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000465 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000466 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000467 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000468 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000469 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000470 usage("-l and -j don't go together!")
Nadeem Vawdac1fba3e2011-08-27 15:22:05 +0200471 if use_mp and support.max_memuse:
472 usage("-M and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200473 if failfast and not (verbose or verbose3):
474 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000475
Guido van Rossum152494a1996-12-20 03:12:20 +0000476 good = []
477 bad = []
478 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000479 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000480 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000481 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000482
Neil Schemenauerd569f232000-09-22 15:29:28 +0000483 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000484 try:
485 import gc
486 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000487 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000488 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000489 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000490 # Uncomment the line below to report garbage that is not
491 # freeable by reference counting alone. By default only
492 # garbage that is not collectable by the GC is reported.
493 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000494 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000495
Barry Warsawe11e3de1999-01-28 19:51:51 +0000496 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000497 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000498 try:
499 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000500 next_test = fp.read().strip()
501 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000502 fp.close()
503 except IOError:
504 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000505
506 if fromfile:
507 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000508 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000509 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000510 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000511 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000512 guts = line.split() # assuming no test has whitespace in its name
513 if guts and not guts[0].startswith('#'):
514 tests.extend(guts)
515 fp.close()
516
517 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000518 removepy(args)
519 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000520
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000521 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000522 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000523 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000524 for arg in args:
525 if arg in stdtests:
526 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000527 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000528 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000529
530 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000531 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000532 # Print basic platform information
533 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000534 print("== ", platform.platform(aliased=True),
535 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000536 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000537 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000538
R David Murrayb588f8d2011-03-24 14:42:58 -0400539 # if testdir is set, then we are not running the python tests suite, so
540 # don't add default tests to be executed or skipped (pass empty values)
541 if testdir:
542 alltests = findtests(testdir, list(), set())
543 else:
544 alltests = findtests(testdir, stdtests, nottests)
545
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000546 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000547 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000548 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000549 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000550 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000551 except IndexError:
552 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000553 # Remove all the tests that precede start if it's set.
554 if start:
555 try:
556 del tests[:tests.index(start)]
557 except ValueError:
558 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000559 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000560 random.seed(random_seed)
561 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000562 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000563 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000564 import trace, tempfile
565 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
566 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000567 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000568
Christian Heimesb186d002008-03-18 15:15:01 +0000569 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000570 support.verbose = verbose # Tell tests to be moderately quiet
571 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000572 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000573
574 def accumulate_result(test, result):
575 ok, test_time = result
576 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000577 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000578 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000579 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000580 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000581 elif ok == ENV_CHANGED:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000582 environment_changed.append(test)
583 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000584 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000585 elif ok == RESOURCE_DENIED:
586 skipped.append(test)
587 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000588
R. David Murray7dc72cc2009-11-14 16:13:02 +0000589 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000590 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000591 while True:
592 for test in tests:
593 yield test
594 if bad:
595 return
596 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000597 test_count = ''
598 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000599 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000600 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000601 test_count = '/{}'.format(len(selected))
602 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000603
Antoine Pitrou88909542009-06-29 13:54:42 +0000604 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000605 try:
606 from threading import Thread
607 except ImportError:
608 print("Multiprocess option requires thread support")
609 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000610 from queue import Queue
611 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000612 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000613 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000614 def tests_and_args():
615 for test in tests:
616 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000617 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000618 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200619 debug=debug, output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200620 timeout=timeout, failfast=failfast,
621 match_tests=match_tests)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000622 )
623 yield (test, args_tuple)
624 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000625 opt_args = support.args_from_interpreter_flags()
626 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000627 def work():
628 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000629 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000630 while True:
631 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000632 test, args_tuple = next(pending)
633 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000634 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000635 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000636 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000637 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000638 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000639 universal_newlines=True,
640 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000641 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200642 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000643 # Strip last refcount output line if it exists, since it
644 # comes from the shutdown of the interpreter in the subcommand.
645 stderr = debug_output_pat.sub("", stderr)
646 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200647 if retcode != 0:
648 result = (CHILD_ERROR, "Exit code %s" % retcode)
649 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
650 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000651 if not result:
652 output.put((None, None, None, None))
653 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000654 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000655 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000656 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000657 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000658 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000659 workers = [Thread(target=work) for i in range(use_mp)]
660 for worker in workers:
661 worker.start()
662 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000663 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000664 try:
665 while finished < use_mp:
666 test, stdout, stderr, result = output.get()
667 if test is None:
668 finished += 1
669 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200670 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000671 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300672 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
673 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200674 test_count_width, test_index, test_count,
675 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000676 if stdout:
677 print(stdout)
678 if stderr:
679 print(stderr, file=sys.stderr)
680 if result[0] == INTERRUPTED:
681 assert result[1] == 'KeyboardInterrupt'
682 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200683 if result[0] == CHILD_ERROR:
Victor Stinner571e8fd2011-05-01 22:57:43 +0200684 raise Exception("Child error on {}: {}".format(test, result[1]))
Georg Brandldee7b852010-08-02 18:59:52 +0000685 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000686 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000687 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000688 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000689 for worker in workers:
690 worker.join()
691 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000692 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000693 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300694 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
695 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200696 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000697 sys.stdout.flush()
698 if trace:
699 # If we're tracing code coverage, then we don't exit with status
700 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200701 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000702 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000703 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000704 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100705 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200706 output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200707 timeout=timeout, failfast=failfast,
708 match_tests=match_tests)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000709 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000710 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000711 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000712 break
713 except:
714 raise
715 if findleaks:
716 gc.collect()
717 if gc.garbage:
718 print("Warning: test created", len(gc.garbage), end=' ')
719 print("uncollectable object(s).")
720 # move the uncollectable objects somewhere so we don't see
721 # them again
722 found_garbage.extend(gc.garbage)
723 del gc.garbage[:]
724 # Unload the newly imported modules (best effort finalization)
725 for module in sys.modules.keys():
726 if module not in save_modules and module.startswith("test."):
727 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000728
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000729 if interrupted:
730 # print a newline after ^C
731 print()
732 print("Test suite interrupted by signal SIGINT.")
733 omitted = set(selected) - set(good) - set(bad) - set(skipped)
734 print(count(len(omitted), "test"), "omitted:")
735 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000736 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000737 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000738 print("All", end=' ')
739 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000740 if print_slow:
741 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000742 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000743 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000744 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000745 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000746 bad = sorted(set(bad) - set(environment_changed))
747 if bad:
748 print(count(len(bad), "test"), "failed:")
749 printlist(bad)
750 if environment_changed:
751 print("{} altered the execution environment:".format(
752 count(len(environment_changed), "test")))
753 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000754 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000755 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000756 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000757
Tim Petersb5b7b782001-08-12 01:20:39 +0000758 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000759 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000760 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000761 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000762 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000763 print(count(len(surprise), "skip"), \
764 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000765 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000766 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000767 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000768 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000769 print("Ask someone to teach regrtest.py about which tests are")
770 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000771
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000772 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000773 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000774 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000775 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000776 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000777 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000778 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200779 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000780 except KeyboardInterrupt:
781 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000782 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000783 break
784 except:
785 raise
786
Barry Warsawe11e3de1999-01-28 19:51:51 +0000787 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000788 if next_single_test:
789 with open(filename, 'w') as fp:
790 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000791 else:
792 os.unlink(filename)
793
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000794 if trace:
795 r = tracer.results()
796 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
797
Skip Montanaro0179a182004-06-06 15:53:18 +0000798 if runleaks:
799 os.system("leaks %d" % os.getpid())
800
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000801 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000802
Guido van Rossum152494a1996-12-20 03:12:20 +0000803
R David Murrayb588f8d2011-03-24 14:42:58 -0400804# small set of tests to determine if we have a basically functioning interpreter
805# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000806STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000807 'test_grammar',
808 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000809 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000810 'test_builtin',
811 'test_exceptions',
812 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000813 'test_unittest',
814 'test_doctest',
815 'test_doctest2',
Eli Benderskyd18a0472011-07-27 20:21:45 +0300816 'test_support'
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000817]
Guido van Rossum152494a1996-12-20 03:12:20 +0000818
R David Murrayb588f8d2011-03-24 14:42:58 -0400819# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400820NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000821
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000822def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000823 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000824 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000825 names = os.listdir(testdir)
826 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000827 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000828 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400829 mod, ext = os.path.splitext(name)
830 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
831 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000832 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000833
Victor Stinnerf58087b2010-05-02 17:24:51 +0000834def replace_stdout():
835 """Set stdout encoder error handler to backslashreplace (as stderr error
836 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000837 import atexit
838
Victor Stinnerf58087b2010-05-02 17:24:51 +0000839 stdout = sys.stdout
840 sys.stdout = open(stdout.fileno(), 'w',
841 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000842 errors="backslashreplace",
Victor Stinnerbe621032011-05-25 02:01:55 +0200843 closefd=False,
844 newline='\n')
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000845
846 def restore_stdout():
847 sys.stdout.close()
848 sys.stdout = stdout
849 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000850
Antoine Pitrou88909542009-06-29 13:54:42 +0000851def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100852 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200853 output_on_failure=False, failfast=False, match_tests=None,
854 timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000855 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000856
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000857 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000858 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000859 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000860 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000861 huntrleaks -- run multiple times to test for leaks; requires a debug
862 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200863 output_on_failure -- if true, display test output on failure
Victor Stinner0cc8d592011-03-31 18:10:13 +0200864 timeout -- dump the traceback and exit if a test takes more than
865 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000866
867 Returns one of the test result constants:
868 INTERRUPTED KeyboardInterrupt when run under -j
869 RESOURCE_DENIED test skipped because resource denied
870 SKIPPED test skipped for some other reason
871 ENV_CHANGED test failed because it changed the execution environment
872 FAILED test failed
873 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000874 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000875
Antoine Pitrou88909542009-06-29 13:54:42 +0000876 if use_resources is not None:
877 support.use_resources = use_resources
Victor Stinner30196882011-06-03 12:53:26 +0200878 use_timeout = (timeout is not None)
Victor Stinner7d648a02011-03-31 18:27:50 +0200879 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200880 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000881 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200882 support.match_tests = match_tests
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200883 if failfast:
884 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200885 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200886 support.verbose = True
887
888 # Reuse the same instance to all calls to runtest(). Some
889 # tests keep a reference to sys.stdout or sys.stderr
890 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200891 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200892 stream = io.StringIO()
893 runtest.stringio = stream
894 else:
895 stream = runtest.stringio
896 stream.seek(0)
897 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200898
899 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200900 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200901 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200902 sys.stdout = stream
903 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200904 result = runtest_inner(test, verbose, quiet, huntrleaks,
905 debug, display_failure=False)
906 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200907 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200908 orig_stderr.write(output)
909 orig_stderr.flush()
910 finally:
911 sys.stdout = orig_stdout
912 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200913 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200914 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200915 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
916 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100917 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000918 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200919 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200920 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000921 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200922runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000923
Nick Coghlan6ead5522009-10-18 13:19:33 +0000924# Unit tests are supposed to leave the execution environment unchanged
925# once they complete. But sometimes tests have bugs, especially when
926# tests fail, and the changes to environment go on to mess up other
927# tests. This can cause issues with buildbot stability, since tests
928# are run in random order and so problems may appear to come and go.
929# There are a few things we can save and restore to mitigate this, and
930# the following context manager handles this task.
931
932class saved_test_environment:
933 """Save bits of the test environment and restore them at block exit.
934
935 with saved_test_environment(testname, verbose, quiet):
936 #stuff
937
938 Unless quiet is True, a warning is printed to stderr if any of
939 the saved items was changed by the test. The attribute 'changed'
940 is initially False, but is set to True if a change is detected.
941
942 If verbose is more than 1, the before and after state of changed
943 items is also printed.
944 """
945
946 changed = False
947
948 def __init__(self, testname, verbose=0, quiet=False):
949 self.testname = testname
950 self.verbose = verbose
951 self.quiet = quiet
952
953 # To add things to save and restore, add a name XXX to the resources list
954 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
955 # return the value to be saved and compared against a second call to the
956 # get function when test execution completes. restore_XXX should accept
957 # the saved value and restore the resource using it. It will be called if
958 # and only if a change in the value is detected.
959 #
960 # Note: XXX will have any '.' replaced with '_' characters when determining
961 # the corresponding method names.
962
963 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000964 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000965 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200966 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200967 'sys.warnoptions', 'threading._dangling',
Éric Araujo28df8de2011-09-19 05:10:45 +0200968 'multiprocessing.process._dangling',
969 'sysconfig._CONFIG_VARS', 'sysconfig._SCHEMES',
Éric Araujo76558e12011-10-06 02:44:19 +0200970 'packaging.database_caches',
Éric Araujo28df8de2011-09-19 05:10:45 +0200971 )
Nick Coghlan6ead5522009-10-18 13:19:33 +0000972
973 def get_sys_argv(self):
974 return id(sys.argv), sys.argv, sys.argv[:]
975 def restore_sys_argv(self, saved_argv):
976 sys.argv = saved_argv[1]
977 sys.argv[:] = saved_argv[2]
978
979 def get_cwd(self):
980 return os.getcwd()
981 def restore_cwd(self, saved_cwd):
982 os.chdir(saved_cwd)
983
984 def get_sys_stdout(self):
985 return sys.stdout
986 def restore_sys_stdout(self, saved_stdout):
987 sys.stdout = saved_stdout
988
989 def get_sys_stderr(self):
990 return sys.stderr
991 def restore_sys_stderr(self, saved_stderr):
992 sys.stderr = saved_stderr
993
994 def get_sys_stdin(self):
995 return sys.stdin
996 def restore_sys_stdin(self, saved_stdin):
997 sys.stdin = saved_stdin
998
999 def get_os_environ(self):
1000 return id(os.environ), os.environ, dict(os.environ)
1001 def restore_os_environ(self, saved_environ):
1002 os.environ = saved_environ[1]
1003 os.environ.clear()
1004 os.environ.update(saved_environ[2])
1005
1006 def get_sys_path(self):
1007 return id(sys.path), sys.path, sys.path[:]
1008 def restore_sys_path(self, saved_path):
1009 sys.path = saved_path[1]
1010 sys.path[:] = saved_path[2]
1011
Brett Cannon055470a2010-02-19 15:57:10 +00001012 def get_sys_path_hooks(self):
1013 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
1014 def restore_sys_path_hooks(self, saved_hooks):
1015 sys.path_hooks = saved_hooks[1]
1016 sys.path_hooks[:] = saved_hooks[2]
1017
Brett Cannon31f59292011-02-21 19:29:56 +00001018 def get_sys_gettrace(self):
1019 return sys.gettrace()
1020 def restore_sys_gettrace(self, trace_fxn):
1021 sys.settrace(trace_fxn)
1022
Brett Cannon055470a2010-02-19 15:57:10 +00001023 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +00001024 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +00001025 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +00001026 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +00001027
Brett Cannon29c0e4f2010-03-20 22:22:57 +00001028 def get_warnings_filters(self):
1029 return id(warnings.filters), warnings.filters, warnings.filters[:]
1030 def restore_warnings_filters(self, saved_filters):
1031 warnings.filters = saved_filters[1]
1032 warnings.filters[:] = saved_filters[2]
1033
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001034 def get_asyncore_socket_map(self):
1035 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +00001036 # XXX Making a copy keeps objects alive until __exit__ gets called.
1037 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001038 def restore_asyncore_socket_map(self, saved_map):
1039 asyncore = sys.modules.get('asyncore')
1040 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +00001041 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001042 asyncore.socket_map.update(saved_map)
1043
Nick Coghlan7d819752010-12-05 06:45:03 +00001044 def get_logging__handlers(self):
1045 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001046 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001047 def restore_logging__handlers(self, saved_handlers):
1048 # Can't easily revert the logging state
1049 pass
1050
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001051 def get_logging__handlerList(self):
1052 # _handlerList is a list of weakrefs to handlers
1053 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1054 def restore_logging__handlerList(self, saved_handlerList):
1055 # Can't easily revert the logging state
1056 pass
1057
Éric Araujo76558e12011-10-06 02:44:19 +02001058 def get_packaging_database_caches(self):
1059 # caching system used by the PEP 376 implementation
1060 # we have one boolean and four dictionaries, initially empty
1061 switch = packaging.database._cache_enabled
1062 saved = []
1063 for name in ('_cache_name', '_cache_name_egg',
1064 '_cache_path', '_cache_path_egg'):
1065 cache = getattr(packaging.database, name)
1066 saved.append((id(cache), cache, cache.copy()))
1067 return switch, saved
1068 def restore_packaging_database_caches(self, saved):
1069 switch, saved_caches = saved
1070 packaging.database._cache_enabled = switch
1071 for offset, name in enumerate(('_cache_name', '_cache_name_egg',
1072 '_cache_path', '_cache_path_egg')):
1073 _, cache, items = saved_caches[offset]
1074 # put back the same object in place
1075 setattr(packaging.database, name, cache)
1076 # now restore its items
1077 cache.clear()
1078 cache.update(items)
1079
Ezio Melotti0123e052011-03-20 15:09:26 +02001080 def get_sys_warnoptions(self):
1081 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1082 def restore_sys_warnoptions(self, saved_options):
1083 sys.warnoptions = saved_options[1]
1084 sys.warnoptions[:] = saved_options[2]
1085
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02001086 # Controlling dangling references to Thread objects can make it easier
1087 # to track reference leaks.
1088 def get_threading__dangling(self):
1089 if not threading:
1090 return None
1091 # This copies the weakrefs without making any strong reference
1092 return threading._dangling.copy()
1093 def restore_threading__dangling(self, saved):
1094 if not threading:
1095 return
1096 threading._dangling.clear()
1097 threading._dangling.update(saved)
1098
1099 # Same for Process objects
1100 def get_multiprocessing_process__dangling(self):
1101 if not multiprocessing:
1102 return None
1103 # This copies the weakrefs without making any strong reference
1104 return multiprocessing.process._dangling.copy()
1105 def restore_multiprocessing_process__dangling(self, saved):
1106 if not multiprocessing:
1107 return
1108 multiprocessing.process._dangling.clear()
1109 multiprocessing.process._dangling.update(saved)
1110
Éric Araujo28df8de2011-09-19 05:10:45 +02001111 def get_sysconfig__CONFIG_VARS(self):
1112 # make sure the dict is initialized
1113 sysconfig.get_config_var('prefix')
1114 return (id(sysconfig._CONFIG_VARS), sysconfig._CONFIG_VARS,
1115 dict(sysconfig._CONFIG_VARS))
1116 def restore_sysconfig__CONFIG_VARS(self, saved):
1117 sysconfig._CONFIG_VARS = saved[1]
1118 sysconfig._CONFIG_VARS.clear()
1119 sysconfig._CONFIG_VARS.update(saved[2])
1120
1121 def get_sysconfig__SCHEMES(self):
1122 # it's mildly evil to look at the internal attribute, but it's easier
1123 # than copying a RawConfigParser object
1124 return (id(sysconfig._SCHEMES), sysconfig._SCHEMES._sections,
1125 sysconfig._SCHEMES._sections.copy())
1126 def restore_sysconfig__SCHEMES(self, saved):
1127 sysconfig._SCHEMES._sections = saved[1]
1128 sysconfig._SCHEMES._sections.clear()
1129 sysconfig._SCHEMES._sections.update(saved[2])
1130
1131
Nick Coghlan6ead5522009-10-18 13:19:33 +00001132 def resource_info(self):
1133 for name in self.resources:
1134 method_suffix = name.replace('.', '_')
1135 get_name = 'get_' + method_suffix
1136 restore_name = 'restore_' + method_suffix
1137 yield name, getattr(self, get_name), getattr(self, restore_name)
1138
1139 def __enter__(self):
1140 self.saved_values = dict((name, get()) for name, get, restore
1141 in self.resource_info())
1142 return self
1143
1144 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001145 saved_values = self.saved_values
1146 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001147 for name, get, restore in self.resource_info():
1148 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001149 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001150 # Check for changes to the resource's value
1151 if current != original:
1152 self.changed = True
1153 restore(original)
1154 if not self.quiet:
1155 print("Warning -- {} was modified by {}".format(
1156 name, self.testname),
1157 file=sys.stderr)
1158 if self.verbose > 1:
1159 print(" Before: {}\n After: {} ".format(
1160 original, current),
1161 file=sys.stderr)
1162 return False
1163
1164
Victor Stinnera7c33e52011-06-29 13:00:54 +02001165def runtest_inner(test, verbose, quiet,
1166 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001167 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001168
Antoine Pitrou88909542009-06-29 13:54:42 +00001169 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001170 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001171 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001172 if test.startswith('test.'):
1173 abstest = test
1174 else:
1175 # Always import it from the test package
1176 abstest = 'test.' + test
1177 with saved_test_environment(test, verbose, quiet) as environment:
1178 start_time = time.time()
1179 the_package = __import__(abstest, globals(), locals(), [])
1180 the_module = getattr(the_package, test)
1181 # Old tests run to completion simply as a side-effect of
1182 # being imported. For tests based on unittest or doctest,
1183 # explicitly invoke their test_main() function (if it exists).
1184 indirect_test = getattr(the_module, "test_main", None)
1185 if indirect_test is not None:
1186 indirect_test()
1187 if huntrleaks:
1188 refleak = dash_R(the_module, test, indirect_test,
1189 huntrleaks)
1190 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001191 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001192 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001193 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001194 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001195 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001196 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001197 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001198 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001199 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001200 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001201 except KeyboardInterrupt:
1202 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001203 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001204 if display_failure:
1205 print("test", test, "failed --", msg, file=sys.stderr)
1206 else:
1207 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001208 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001209 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001210 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001211 msg = traceback.format_exc()
1212 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001213 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001214 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001215 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001216 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001217 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001218 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001219 return ENV_CHANGED, test_time
1220 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001221
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001222def cleanup_test_droppings(testname, verbose):
1223 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001224 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001225 import gc
1226
1227 # First kill any dangling references to open files etc.
1228 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001229 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001230 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001231
1232 # Try to clean up junk commonly left behind. While tests shouldn't leave
1233 # any files or directories behind, when a test fails that can be tedious
1234 # for it to arrange. The consequences can be especially nasty on Windows,
1235 # since if a test leaves a file open, it cannot be deleted by name (while
1236 # there's nothing we can do about that here either, we can display the
1237 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001238 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001239 "db_home",
1240 ):
1241 if not os.path.exists(name):
1242 continue
1243
1244 if os.path.isdir(name):
1245 kind, nuker = "directory", shutil.rmtree
1246 elif os.path.isfile(name):
1247 kind, nuker = "file", os.unlink
1248 else:
1249 raise SystemError("os.path says %r exists but is neither "
1250 "directory nor file" % name)
1251
1252 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001253 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001254 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001255 # if we have chmod, fix possible permissions problems
1256 # that might prevent cleanup
1257 if (hasattr(os, 'chmod')):
1258 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001259 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001260 except Exception as msg:
1261 print(("%r left behind %s %r and it couldn't be "
1262 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001263
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001264def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001265 """Run a test multiple times, looking for reference leaks.
1266
1267 Returns:
1268 False if the test didn't leak references; True if we detected refleaks.
1269 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001270 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001271 import copyreg
1272 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001273
1274 if not hasattr(sys, 'gettotalrefcount'):
1275 raise Exception("Tracking reference leaks requires a debug build "
1276 "of Python")
1277
1278 # Save current values for dash_R_cleanup() to restore.
1279 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001280 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001281 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001282 try:
1283 import zipimport
1284 except ImportError:
1285 zdc = None # Run unmodified on platforms without zipimport support
1286 else:
1287 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001288 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001289 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001290 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001291 continue
1292 for obj in abc.__subclasses__() + [abc]:
1293 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001294
1295 if indirect_test:
1296 def run_the_test():
1297 indirect_test()
1298 else:
1299 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001300 del sys.modules[the_module.__name__]
1301 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001302
1303 deltas = []
1304 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001305 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001306 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001307 print("beginning", repcount, "repetitions", file=sys.stderr)
1308 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001309 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001310 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001311 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001312 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001313 run_the_test()
1314 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001315 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001316 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001317 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001318 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001319 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001320 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001321 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001322 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001323 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001324 sys.stderr.flush()
1325 with open(fname, "a") as refrep:
1326 print(msg, file=refrep)
1327 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001328 return True
1329 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001330
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001331def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001332 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001333 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001334 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001335 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001336 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001337 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001338
Christian Heimesdae2a892008-04-19 00:55:37 +00001339 # Clear the warnings registry, so they can be displayed again
1340 for mod in sys.modules.values():
1341 if hasattr(mod, '__warningregistry__'):
1342 del mod.__warningregistry__
1343
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001344 # Restore some original values.
1345 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001346 copyreg.dispatch_table.clear()
1347 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001348 sys.path_importer_cache.clear()
1349 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001350 try:
1351 import zipimport
1352 except ImportError:
1353 pass # Run unmodified on platforms without zipimport support
1354 else:
1355 zipimport._zip_directory_cache.clear()
1356 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001357
Christian Heimes26855632008-01-27 23:50:43 +00001358 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001359 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001360
Guido van Rossum3de862d2007-08-18 00:10:33 +00001361 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001362 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001363 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001364 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001365 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001366 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001367 obj._abc_cache.clear()
1368 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001369
Antoine Pitrou046467c2009-10-30 18:30:35 +00001370 # Flush standard output, so that buffered data is sent to the OS and
1371 # associated Python objects are reclaimed.
1372 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1373 if stream is not None:
1374 stream.flush()
1375
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001376 # Clear assorted module caches.
1377 _path_created.clear()
1378 re.purge()
1379 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001380 urllib.parse.clear_cache()
1381 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001382 linecache.clearcache()
1383 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001384 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001385 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001386 doctest.master = None
1387
1388 # Collect cyclic trash.
1389 gc.collect()
1390
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001391def warm_char_cache():
1392 s = bytes(range(256))
1393 for i in range(256):
1394 s[i:i+1]
1395
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001396def findtestdir(path=None):
1397 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001398
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001399def removepy(names):
1400 if not names:
1401 return
1402 for idx, name in enumerate(names):
1403 basename, ext = os.path.splitext(name)
1404 if ext == '.py':
1405 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001406
Guido van Rossum152494a1996-12-20 03:12:20 +00001407def count(n, word):
1408 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001409 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001410 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001411 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001412
Tim Petersa45da922001-08-12 03:45:50 +00001413def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001414 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001415
1416 Optional arg width (default 70) is the maximum line length.
1417 Optional arg indent (default 4) is the number of blanks with which to
1418 begin each line.
1419 """
1420
Tim Petersba78bc42002-07-04 19:45:06 +00001421 from textwrap import fill
1422 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001423 # Print the sorted list: 'x' may be a '--random' list or a set()
1424 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001425 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001426
Tim Petersde14a302002-04-01 05:04:46 +00001427# Map sys.platform to a string containing the basenames of tests
1428# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001429#
1430# Special cases:
1431# test_pep277
1432# The _ExpectedSkips constructor adds this to the set of expected
1433# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001434# test_timeout
1435# Controlled by test_timeout.skip_expected. Requires the network
1436# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001437#
1438# Tests that are expected to be skipped everywhere except on one platform
1439# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001440
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001441_expectations = (
1442 ('win32',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001443 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001444 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001445 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001446 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001447 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001448 test_fcntl
1449 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001450 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001451 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001452 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001453 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001454 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001455 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001456 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001457 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001458 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001459 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001460 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001461 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001462 test_pty
1463 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001464 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001465 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001466 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001467 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001468 test_wait3
1469 test_wait4
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001470 """),
1471 ('linux',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001472 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001473 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001474 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001475 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001476 test_ossaudiodev
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001477 """),
1478 ('unixware',
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001479 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001480 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001481 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001482 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001483 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001484 test_openpty
1485 test_pyexpat
1486 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001487 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001488 """),
1489 ('openunix',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001490 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001491 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001492 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001493 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001494 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001495 test_openpty
1496 test_pyexpat
1497 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001498 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001499 """),
1500 ('sco_sv',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001501 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001502 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001503 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001504 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001505 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001506 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001507 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001508 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001509 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001510 test_openpty
1511 test_pyexpat
1512 test_queue
1513 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001514 test_sundry
1515 test_thread
1516 test_threaded_import
1517 test_threadedtempfile
1518 test_threading
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001519 """),
1520 ('darwin',
Jack Jansen398c2362001-12-02 21:41:36 +00001521 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001522 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001523 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001524 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001525 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001526 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001527 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001528 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001529 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001530 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001531 test_poll
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001532 """),
1533 ('sunos',
Guido van Rossum11c3f092002-07-17 15:08:24 +00001534 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001535 test_curses
1536 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001537 test_epoll
1538 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001539 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001540 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001541 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001542 test_zipfile
1543 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001544 """),
1545 ('hp-ux',
Skip Montanarob3230212002-03-15 02:54:03 +00001546 """
Skip Montanarob3230212002-03-15 02:54:03 +00001547 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001548 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001549 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001550 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001551 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001552 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001553 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001554 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001555 test_openpty
1556 test_pyexpat
1557 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001558 test_zipfile
1559 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001560 """),
1561 ('cygwin',
Jason Tishler25115942002-12-05 15:18:15 +00001562 """
Jason Tishler25115942002-12-05 15:18:15 +00001563 test_curses
1564 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001565 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001566 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001567 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001568 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001569 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001570 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001571 test_socketserver
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001572 """),
1573 ('os2emx',
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001574 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001575 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001576 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001577 test_epoll
1578 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001579 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001580 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001581 test_openpty
1582 test_ossaudiodev
1583 test_pty
1584 test_resource
1585 test_signal
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001586 """),
1587 ('freebsd',
Guido van Rossum944a6c32003-11-20 22:11:29 +00001588 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001589 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001590 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001591 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001592 test_ossaudiodev
1593 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001594 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001595 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001596 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001597 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001598 test_ttk_guionly
1599 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001600 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001601 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001602 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001603 """),
1604 ('aix',
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001605 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001606 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001607 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001608 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001609 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001610 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001611 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001612 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001613 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001614 test_ttk_guionly
1615 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001616 test_zipimport
1617 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001618 """),
1619 ('openbsd',
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001620 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001621 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001622 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001623 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001624 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001625 test_normalization
1626 test_ossaudiodev
1627 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001628 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001629 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001630 test_ttk_guionly
1631 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001632 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001633 """),
1634 ('netbsd',
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001635 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001636 test_ctypes
1637 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001638 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001639 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001640 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001641 test_ossaudiodev
1642 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001643 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001644 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001645 test_ttk_guionly
1646 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001647 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001648 """),
1649)
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001650
Tim Petersb5b7b782001-08-12 01:20:39 +00001651class _ExpectedSkips:
1652 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001653 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001654 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001655
Tim Peters7c7efe92002-08-23 17:55:54 +00001656 self.valid = False
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001657 expected = None
1658 for item in _expectations:
1659 if sys.platform.startswith(item[0]):
1660 expected = item[1]
1661 break
1662 if expected is not None:
1663 self.expected = set(expected.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001664
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001665 # These are broken tests, for now skipped on every platform.
1666 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001667 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001668
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001669 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001670 if not os.path.supports_unicode_filenames:
1671 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001672
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001673 # doctest, profile and cProfile tests fail when the codec for the
1674 # fs encoding isn't built in because PyUnicode_Decode() adds two
1675 # calls into Python.
1676 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1677 if sys.getfilesystemencoding().lower() not in encs:
1678 self.expected.add('test_profile')
1679 self.expected.add('test_cProfile')
1680 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001681
Neal Norwitz55b61d22003-02-28 19:57:03 +00001682 if test_timeout.skip_expected:
1683 self.expected.add('test_timeout')
1684
Tim Petersecd79eb2003-01-29 00:35:32 +00001685 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001686 # test_sqlite is only reliable on Windows where the library
1687 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001688 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001689 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001690 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001691 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001692
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001693 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001694 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001695
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001696 if support.python_is_optimized():
1697 self.expected.add("test_gdb")
1698
Tim Peters7c7efe92002-08-23 17:55:54 +00001699 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001700
1701 def isvalid(self):
1702 "Return true iff _ExpectedSkips knows about the current platform."
1703 return self.valid
1704
1705 def getexpected(self):
1706 """Return set of test names we expect to skip on current platform.
1707
1708 self.isvalid() must be true.
1709 """
1710
1711 assert self.isvalid()
1712 return self.expected
1713
Michael Foord3ab34cc2010-12-03 12:27:40 +00001714def _make_temp_dir_for_build(TEMPDIR):
1715 # When tests are run from the Python build directory, it is best practice
1716 # to keep the test files in a subfolder. It eases the cleanup of leftover
1717 # files using command "make distclean".
1718 if sysconfig.is_python_build():
1719 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1720 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001721 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001722 os.mkdir(TEMPDIR)
Florent Xicluna68f71a32011-10-28 16:06:23 +02001723 except FileExistsError:
1724 pass
Michael Foord3ab34cc2010-12-03 12:27:40 +00001725
1726 # Define a writable temp dir that will be used as cwd while running
1727 # the tests. The name of the dir includes the pid to allow parallel
1728 # testing (see the -j option).
1729 TESTCWD = 'test_python_{}'.format(os.getpid())
1730
1731 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1732 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001733
Guido van Rossum152494a1996-12-20 03:12:20 +00001734if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001735 # Remove regrtest.py's own directory from the module search path. Despite
1736 # the elimination of implicit relative imports, this is still needed to
1737 # ensure that submodules of the test package do not inappropriately appear
1738 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1739 # directly instead of using the -m switch
1740 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1741 i = len(sys.path)
1742 while i >= 0:
1743 i -= 1
1744 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1745 del sys.path[i]
1746
Florent Xiclunadc69e722010-09-13 16:35:02 +00001747 # findtestdir() gets the dirname out of __file__, so we have to make it
1748 # absolute before changing the working directory.
1749 # For example __file__ may be relative when running trace or profile.
1750 # See issue #9323.
1751 __file__ = os.path.abspath(__file__)
1752
1753 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001754 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001755
Michael Foord3ab34cc2010-12-03 12:27:40 +00001756 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001757
1758 # Run the tests in a context manager that temporary changes the CWD to a
1759 # temporary and writable directory. If it's not possible to create or
1760 # change the CWD, the original CWD will be used. The original CWD is
1761 # available from support.SAVEDCWD.
1762 with support.temp_cwd(TESTCWD, quiet=True):
1763 main()