blob: af114168a998d38391d5c55ce5f01353f6dfd4f2 [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 Araujo4e377f22011-11-06 07:01:18 +0100175import packaging.command
Éric Araujo76558e12011-10-06 02:44:19 +0200176import packaging.database
Victor Stinner8313d6a2011-06-29 15:22:26 +0200177import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000178import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000179import re
Éric Araujoff913062011-11-29 16:45:07 +0100180import shutil
Victor Stinnercb41cda2011-07-13 23:47:21 +0200181import signal
Christian Heimesb186d002008-03-18 15:15:01 +0000182import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000183import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200184import tempfile
185import time
186import traceback
187import unittest
188import warnings
189from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000190
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200191try:
192 import threading
193except ImportError:
194 threading = None
195try:
196 import multiprocessing.process
197except ImportError:
198 multiprocessing = None
199
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000200
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000201# Some times __path__ and __file__ are not absolute (e.g. while running from
202# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
203# imports might fail. This affects only the modules imported before os.chdir().
204# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
205# they are found in the CWD their __file__ and __path__ will be relative (this
206# happens before the chdir). All the modules imported after the chdir, are
207# not found in the CWD, and since the other paths in sys.path[1:] are absolute
208# (site.py absolutize them), the __file__ and __path__ will be absolute too.
209# Therefore it is necessary to absolutize manually the __file__ and __path__ of
210# the packages to prevent later imports to fail when the CWD is different.
211for module in sys.modules.values():
212 if hasattr(module, '__path__'):
213 module.__path__ = [os.path.abspath(path) for path in module.__path__]
214 if hasattr(module, '__file__'):
215 module.__file__ = os.path.abspath(module.__file__)
216
Guido van Rossumdc15c272002-08-12 21:55:51 +0000217
Guido van Rossumbb484652002-12-02 09:56:21 +0000218# MacOSX (a.k.a. Darwin) has a default stack size that is too small
219# for deeply recursive regular expressions. We see this as crashes in
220# the Python test suite when running test_re.py and test_sre.py. The
221# fix is to set the stack limit to 2048.
222# This approach may also be useful for other Unixy platforms that
223# suffer from small default stack limits.
224if sys.platform == 'darwin':
225 try:
226 import resource
227 except ImportError:
228 pass
229 else:
230 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
231 newsoft = min(hard, max(soft, 1024*2048))
232 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
233
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000234# Test result constants.
235PASSED = 1
236FAILED = 0
237ENV_CHANGED = -1
238SKIPPED = -2
239RESOURCE_DENIED = -3
240INTERRUPTED = -4
Victor Stinner4b739882011-03-31 18:02:36 +0200241CHILD_ERROR = -5 # error in a child process
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000242
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000243from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000244
Georg Brandl1158a332009-06-04 09:30:30 +0000245RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000246 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000247
Florent Xiclunaec882212010-08-09 16:56:43 +0000248TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000249
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000250def usage(msg):
251 print(msg, file=sys.stderr)
252 print("Use --help for usage", file=sys.stderr)
253 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000254
255
Antoine Pitrou88909542009-06-29 13:54:42 +0000256def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000257 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000258 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000259 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000260 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200261 header=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000262 """Execute a test suite.
263
Thomas Wouters7e474022000-07-16 12:04:32 +0000264 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000265 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000266
267 tests -- a list of strings containing test names (optional)
268 testdir -- the directory in which to look for tests (optional)
269
270 Users other than the Python test suite will certainly want to
271 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000272 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000273
274 If the tests argument is omitted, the tests listed on the
275 command-line will be used. If that's empty, too, then all *.py
276 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000277
Antoine Pitrou88909542009-06-29 13:54:42 +0000278 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000279 single, randomize, findleaks, use_resources, trace, coverdir,
280 print_slow, and random_seed) allow programmers calling main()
281 directly to set the values that would normally be set by flags
282 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000283 """
Victor Stinner3c18f252011-05-22 15:27:14 +0200284
Victor Stinnercb41cda2011-07-13 23:47:21 +0200285 # Display the Python traceback on fatal errors (e.g. segfault)
Victor Stinner3c18f252011-05-22 15:27:14 +0200286 faulthandler.enable(all_threads=True)
287
Victor Stinnercb41cda2011-07-13 23:47:21 +0200288 # Display the Python traceback on SIGALRM or SIGUSR1 signal
289 signals = []
290 if hasattr(signal, 'SIGALRM'):
291 signals.append(signal.SIGALRM)
292 if hasattr(signal, 'SIGUSR1'):
293 signals.append(signal.SIGUSR1)
294 for signum in signals:
295 faulthandler.register(signum, chain=True)
296
Victor Stinner1802d3f2010-05-19 17:11:19 +0000297 replace_stdout()
298
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000299 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000300 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200301 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:Gm:',
Georg Brandl96f78972009-10-27 21:02:11 +0000302 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
303 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
304 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000305 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000306 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitrou8dbd8572011-07-23 22:37:52 +0200307 'start=', 'nowindows', 'header', 'testdir=', 'timeout=', 'wait',
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200308 'failfast', 'match'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000309 except getopt.error as msg:
310 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000311
312 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000313 if random_seed is None:
314 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000315 if use_resources is None:
316 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000317 debug = False
318 start = None
Victor Stinner30196882011-06-03 12:53:26 +0200319 timeout = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000320 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000321 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000322 print(__doc__)
323 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000324 elif o in ('-v', '--verbose'):
325 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000326 elif o in ('-w', '--verbose2'):
327 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000328 elif o in ('-d', '--debug'):
329 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000330 elif o in ('-W', '--verbose3'):
331 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200332 elif o in ('-G', '--failfast'):
333 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000334 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000335 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000336 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000337 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000338 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000339 elif o in ('-S', '--start'):
340 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000341 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000342 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000343 elif o in ('-S', '--slow'):
344 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000345 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000346 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000347 elif o == '--randseed':
348 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000349 elif o in ('-f', '--fromfile'):
350 fromfile = a
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200351 elif o in ('-m', '--match'):
352 match_tests = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000353 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000354 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000355 elif o in ('-L', '--runleaks'):
356 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000357 elif o in ('-t', '--threshold'):
358 import gc
359 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000360 elif o in ('-T', '--coverage'):
361 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000362 elif o in ('-D', '--coverdir'):
R David Murray03504fc2011-03-24 14:35:30 -0400363 # CWD is replaced with a temporary dir before calling main(), so we
364 # need join it with the saved CWD so it goes where the user expects.
365 coverdir = os.path.join(support.SAVEDCWD, a)
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000366 elif o in ('-N', '--nocoverdir'):
367 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000368 elif o in ('-R', '--huntrleaks'):
369 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000370 if len(huntrleaks) not in (2, 3):
371 print(a, huntrleaks)
372 usage('-R takes 2 or 3 colon-separated arguments')
373 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000374 huntrleaks[0] = 5
375 else:
376 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000377 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000378 huntrleaks[1] = 4
379 else:
380 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000381 if len(huntrleaks) == 2 or not huntrleaks[2]:
382 huntrleaks[2:] = ["reflog.txt"]
383 # Avoid false positives due to the character cache in
384 # stringobject.c filling slowly with random data
385 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000386 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000387 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000388 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000389 u = [x.lower() for x in a.split(',')]
390 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000391 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000392 use_resources[:] = RESOURCE_NAMES
393 continue
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200394 if r == 'none':
395 del use_resources[:]
396 continue
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000397 remove = False
398 if r[0] == '-':
399 remove = True
400 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000401 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000402 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000403 if remove:
404 if r in use_resources:
405 use_resources.remove(r)
406 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000407 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000408 elif o in ('-n', '--nowindows'):
409 import msvcrt
410 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
411 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
412 msvcrt.SEM_NOGPFAULTERRORBOX|
413 msvcrt.SEM_NOOPENFILEERRORBOX)
414 try:
415 msvcrt.CrtSetReportMode
416 except AttributeError:
417 # release build
418 pass
419 else:
420 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
421 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
422 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000423 elif o in ('-F', '--forever'):
424 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000425 elif o in ('-j', '--multiprocess'):
426 use_mp = int(a)
Antoine Pitrou6ab79d92011-03-23 20:17:45 +0100427 if use_mp <= 0:
428 try:
429 import multiprocessing
430 # Use all cores + extras for tests that like to sleep
431 use_mp = 2 + multiprocessing.cpu_count()
432 except (ImportError, NotImplementedError):
433 use_mp = 3
Nadeem Vawda3c01d162011-08-01 23:48:26 +0200434 if use_mp == 1:
435 use_mp = None
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000436 elif o == '--header':
437 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000438 elif o == '--slaveargs':
439 args, kwargs = json.loads(a)
440 try:
441 result = runtest(*args, **kwargs)
442 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000443 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000444 sys.stdout.flush()
445 print() # Force a newline (just in case)
446 print(json.dumps(result))
447 sys.exit(0)
R David Murrayb588f8d2011-03-24 14:42:58 -0400448 elif o == '--testdir':
449 # CWD is replaced with a temporary dir before calling main(), so we
450 # join it with the saved CWD so it ends up where the user expects.
451 testdir = os.path.join(support.SAVEDCWD, a)
Victor Stinner0cc8d592011-03-31 18:10:13 +0200452 elif o == '--timeout':
Victor Stinner88345682011-06-06 12:14:23 +0200453 if hasattr(faulthandler, 'dump_tracebacks_later'):
454 timeout = float(a)
455 if timeout <= 0:
456 timeout = None
457 else:
458 print("Warning: The timeout option requires "
459 "faulthandler.dump_tracebacks_later")
Victor Stinner9e586c22011-06-06 12:04:36 +0200460 timeout = None
Brian Curtinf6643452011-04-28 17:45:17 -0500461 elif o == '--wait':
462 input("Press any key to continue...")
R. David Murray35768ad2009-11-15 00:23:21 +0000463 else:
464 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000465 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000466 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000467 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000468 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000469 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000470 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000471 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000472 usage("-l and -j don't go together!")
Nadeem Vawdac1fba3e2011-08-27 15:22:05 +0200473 if use_mp and support.max_memuse:
474 usage("-M and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200475 if failfast and not (verbose or verbose3):
476 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000477
Guido van Rossum152494a1996-12-20 03:12:20 +0000478 good = []
479 bad = []
480 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000481 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000482 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000483 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000484
Neil Schemenauerd569f232000-09-22 15:29:28 +0000485 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000486 try:
487 import gc
488 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000489 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000490 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000491 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000492 # Uncomment the line below to report garbage that is not
493 # freeable by reference counting alone. By default only
494 # garbage that is not collectable by the GC is reported.
495 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000496 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000497
Barry Warsawe11e3de1999-01-28 19:51:51 +0000498 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000499 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000500 try:
501 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000502 next_test = fp.read().strip()
503 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000504 fp.close()
505 except IOError:
506 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000507
508 if fromfile:
509 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000510 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000511 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000512 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000513 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000514 guts = line.split() # assuming no test has whitespace in its name
515 if guts and not guts[0].startswith('#'):
516 tests.extend(guts)
517 fp.close()
518
519 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000520 removepy(args)
521 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000522
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000523 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000524 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000525 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000526 for arg in args:
527 if arg in stdtests:
528 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000529 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000530 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000531
532 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000533 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000534 # Print basic platform information
535 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000536 print("== ", platform.platform(aliased=True),
537 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000538 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000539 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000540
R David Murrayb588f8d2011-03-24 14:42:58 -0400541 # if testdir is set, then we are not running the python tests suite, so
542 # don't add default tests to be executed or skipped (pass empty values)
543 if testdir:
544 alltests = findtests(testdir, list(), set())
545 else:
546 alltests = findtests(testdir, stdtests, nottests)
547
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000548 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000549 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000550 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000551 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000552 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000553 except IndexError:
554 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000555 # Remove all the tests that precede start if it's set.
556 if start:
557 try:
558 del tests[:tests.index(start)]
559 except ValueError:
560 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000561 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000562 random.seed(random_seed)
563 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000564 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000565 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000566 import trace, tempfile
567 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
568 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000569 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000570
Christian Heimesb186d002008-03-18 15:15:01 +0000571 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000572 support.verbose = verbose # Tell tests to be moderately quiet
573 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000574 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000575
576 def accumulate_result(test, result):
577 ok, test_time = result
578 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000579 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000580 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000581 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000582 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000583 elif ok == ENV_CHANGED:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000584 environment_changed.append(test)
585 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000586 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000587 elif ok == RESOURCE_DENIED:
588 skipped.append(test)
589 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000590
R. David Murray7dc72cc2009-11-14 16:13:02 +0000591 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000592 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000593 while True:
594 for test in tests:
595 yield test
596 if bad:
597 return
598 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000599 test_count = ''
600 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000601 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000602 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000603 test_count = '/{}'.format(len(selected))
604 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000605
Antoine Pitrou88909542009-06-29 13:54:42 +0000606 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000607 try:
608 from threading import Thread
609 except ImportError:
610 print("Multiprocess option requires thread support")
611 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000612 from queue import Queue
613 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000614 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000615 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000616 def tests_and_args():
617 for test in tests:
618 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000619 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000620 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinnerd71bd9a2011-06-29 15:25:40 +0200621 debug=debug, output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200622 timeout=timeout, failfast=failfast,
623 match_tests=match_tests)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000624 )
625 yield (test, args_tuple)
626 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000627 opt_args = support.args_from_interpreter_flags()
628 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000629 def work():
630 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000631 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000632 while True:
633 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000634 test, args_tuple = next(pending)
635 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000636 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000637 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000638 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000639 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000640 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000641 universal_newlines=True,
642 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000643 stdout, stderr = popen.communicate()
Victor Stinner4b739882011-03-31 18:02:36 +0200644 retcode = popen.wait()
R. David Murray27144602009-10-19 15:26:16 +0000645 # Strip last refcount output line if it exists, since it
646 # comes from the shutdown of the interpreter in the subcommand.
647 stderr = debug_output_pat.sub("", stderr)
648 stdout, _, result = stdout.strip().rpartition("\n")
Victor Stinner4b739882011-03-31 18:02:36 +0200649 if retcode != 0:
650 result = (CHILD_ERROR, "Exit code %s" % retcode)
651 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
652 return
R. David Murray7dc72cc2009-11-14 16:13:02 +0000653 if not result:
654 output.put((None, None, None, None))
655 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000656 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000657 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000658 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000659 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000660 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000661 workers = [Thread(target=work) for i in range(use_mp)]
662 for worker in workers:
663 worker.start()
664 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000665 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000666 try:
667 while finished < use_mp:
668 test, stdout, stderr, result = output.get()
669 if test is None:
670 finished += 1
671 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200672 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000673 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300674 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
675 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200676 test_count_width, test_index, test_count,
677 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000678 if stdout:
679 print(stdout)
680 if stderr:
681 print(stderr, file=sys.stderr)
682 if result[0] == INTERRUPTED:
683 assert result[1] == 'KeyboardInterrupt'
684 raise KeyboardInterrupt # What else?
Victor Stinner4b739882011-03-31 18:02:36 +0200685 if result[0] == CHILD_ERROR:
Victor Stinner571e8fd2011-05-01 22:57:43 +0200686 raise Exception("Child error on {}: {}".format(test, result[1]))
Georg Brandldee7b852010-08-02 18:59:52 +0000687 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000688 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000689 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000690 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000691 for worker in workers:
692 worker.join()
693 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000694 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000695 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300696 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
697 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200698 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000699 sys.stdout.flush()
700 if trace:
701 # If we're tracing code coverage, then we don't exit with status
702 # if on a false return value from main.
Victor Stinner0cc8d592011-03-31 18:10:13 +0200703 tracer.runctx('runtest(test, verbose, quiet, timeout=timeout)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000704 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000705 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000706 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100707 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200708 output_on_failure=verbose3,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200709 timeout=timeout, failfast=failfast,
710 match_tests=match_tests)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000711 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000712 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000713 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000714 break
715 except:
716 raise
717 if findleaks:
718 gc.collect()
719 if gc.garbage:
720 print("Warning: test created", len(gc.garbage), end=' ')
721 print("uncollectable object(s).")
722 # move the uncollectable objects somewhere so we don't see
723 # them again
724 found_garbage.extend(gc.garbage)
725 del gc.garbage[:]
726 # Unload the newly imported modules (best effort finalization)
727 for module in sys.modules.keys():
728 if module not in save_modules and module.startswith("test."):
729 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000730
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000731 if interrupted:
732 # print a newline after ^C
733 print()
734 print("Test suite interrupted by signal SIGINT.")
735 omitted = set(selected) - set(good) - set(bad) - set(skipped)
736 print(count(len(omitted), "test"), "omitted:")
737 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000738 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000739 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000740 print("All", end=' ')
741 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000742 if print_slow:
743 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000744 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000745 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000746 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000747 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000748 bad = sorted(set(bad) - set(environment_changed))
749 if bad:
750 print(count(len(bad), "test"), "failed:")
751 printlist(bad)
752 if environment_changed:
753 print("{} altered the execution environment:".format(
754 count(len(environment_changed), "test")))
755 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000756 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000757 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000758 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000759
Tim Petersb5b7b782001-08-12 01:20:39 +0000760 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000761 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000762 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000763 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000764 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000765 print(count(len(surprise), "skip"), \
766 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000767 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000768 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000769 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000770 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000771 print("Ask someone to teach regrtest.py about which tests are")
772 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000773
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000774 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000775 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000776 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000777 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000778 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000779 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000780 verbose = True
Victor Stinner0cc8d592011-03-31 18:10:13 +0200781 ok = runtest(test, True, quiet, huntrleaks, debug, timeout=timeout)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000782 except KeyboardInterrupt:
783 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000784 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000785 break
786 except:
787 raise
788
Barry Warsawe11e3de1999-01-28 19:51:51 +0000789 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000790 if next_single_test:
791 with open(filename, 'w') as fp:
792 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000793 else:
794 os.unlink(filename)
795
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000796 if trace:
797 r = tracer.results()
798 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
799
Skip Montanaro0179a182004-06-06 15:53:18 +0000800 if runleaks:
801 os.system("leaks %d" % os.getpid())
802
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000803 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000804
Guido van Rossum152494a1996-12-20 03:12:20 +0000805
R David Murrayb588f8d2011-03-24 14:42:58 -0400806# small set of tests to determine if we have a basically functioning interpreter
807# (i.e. if any of these fail, then anything else is likely to follow)
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000808STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000809 'test_grammar',
810 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000811 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000812 'test_builtin',
813 'test_exceptions',
814 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000815 'test_unittest',
816 'test_doctest',
817 'test_doctest2',
Eli Benderskyd18a0472011-07-27 20:21:45 +0300818 'test_support'
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000819]
Guido van Rossum152494a1996-12-20 03:12:20 +0000820
R David Murrayb588f8d2011-03-24 14:42:58 -0400821# set of tests that we don't want to be executed when using regrtest
R David Murray57648302011-03-24 14:57:05 -0400822NOTTESTS = set()
Guido van Rossum152494a1996-12-20 03:12:20 +0000823
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000824def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000825 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000826 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000827 names = os.listdir(testdir)
828 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000829 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000830 for name in names:
R David Murray661720e2011-03-21 15:14:34 -0400831 mod, ext = os.path.splitext(name)
832 if mod[:5] == "test_" and ext in (".py", "") and mod not in others:
833 tests.append(mod)
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000834 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000835
Victor Stinnerf58087b2010-05-02 17:24:51 +0000836def replace_stdout():
837 """Set stdout encoder error handler to backslashreplace (as stderr error
838 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000839 import atexit
840
Victor Stinnerf58087b2010-05-02 17:24:51 +0000841 stdout = sys.stdout
842 sys.stdout = open(stdout.fileno(), 'w',
843 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000844 errors="backslashreplace",
Victor Stinnerbe621032011-05-25 02:01:55 +0200845 closefd=False,
846 newline='\n')
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000847
848 def restore_stdout():
849 sys.stdout.close()
850 sys.stdout = stdout
851 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000852
Antoine Pitrou88909542009-06-29 13:54:42 +0000853def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100854 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitrouf83e4ac2011-07-29 23:57:10 +0200855 output_on_failure=False, failfast=False, match_tests=None,
856 timeout=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000857 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000858
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000859 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000860 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000861 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000862 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000863 huntrleaks -- run multiple times to test for leaks; requires a debug
864 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200865 output_on_failure -- if true, display test output on failure
Victor Stinner0cc8d592011-03-31 18:10:13 +0200866 timeout -- dump the traceback and exit if a test takes more than
867 timeout seconds
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000868
869 Returns one of the test result constants:
870 INTERRUPTED KeyboardInterrupt when run under -j
871 RESOURCE_DENIED test skipped because resource denied
872 SKIPPED test skipped for some other reason
873 ENV_CHANGED test failed because it changed the execution environment
874 FAILED test failed
875 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000876 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000877
Antoine Pitrou88909542009-06-29 13:54:42 +0000878 if use_resources is not None:
879 support.use_resources = use_resources
Victor Stinner30196882011-06-03 12:53:26 +0200880 use_timeout = (timeout is not None)
Victor Stinner7d648a02011-03-31 18:27:50 +0200881 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200882 faulthandler.dump_tracebacks_later(timeout, exit=True)
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000883 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200884 support.match_tests = match_tests
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200885 if failfast:
886 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200887 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200888 support.verbose = True
889
890 # Reuse the same instance to all calls to runtest(). Some
891 # tests keep a reference to sys.stdout or sys.stderr
892 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200893 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200894 stream = io.StringIO()
895 runtest.stringio = stream
896 else:
897 stream = runtest.stringio
898 stream.seek(0)
899 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200900
901 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200902 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200903 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200904 sys.stdout = stream
905 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200906 result = runtest_inner(test, verbose, quiet, huntrleaks,
907 debug, display_failure=False)
908 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200909 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200910 orig_stderr.write(output)
911 orig_stderr.flush()
912 finally:
913 sys.stdout = orig_stdout
914 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200915 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200916 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200917 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
918 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100919 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000920 finally:
Victor Stinner7d648a02011-03-31 18:27:50 +0200921 if use_timeout:
Victor Stinner0cc8d592011-03-31 18:10:13 +0200922 faulthandler.cancel_dump_tracebacks_later()
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000923 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200924runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000925
Nick Coghlan6ead5522009-10-18 13:19:33 +0000926# Unit tests are supposed to leave the execution environment unchanged
927# once they complete. But sometimes tests have bugs, especially when
928# tests fail, and the changes to environment go on to mess up other
929# tests. This can cause issues with buildbot stability, since tests
930# are run in random order and so problems may appear to come and go.
931# There are a few things we can save and restore to mitigate this, and
932# the following context manager handles this task.
933
934class saved_test_environment:
935 """Save bits of the test environment and restore them at block exit.
936
937 with saved_test_environment(testname, verbose, quiet):
938 #stuff
939
940 Unless quiet is True, a warning is printed to stderr if any of
941 the saved items was changed by the test. The attribute 'changed'
942 is initially False, but is set to True if a change is detected.
943
944 If verbose is more than 1, the before and after state of changed
945 items is also printed.
946 """
947
948 changed = False
949
950 def __init__(self, testname, verbose=0, quiet=False):
951 self.testname = testname
952 self.verbose = verbose
953 self.quiet = quiet
954
955 # To add things to save and restore, add a name XXX to the resources list
956 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
957 # return the value to be saved and compared against a second call to the
958 # get function when test execution completes. restore_XXX should accept
959 # the saved value and restore the resource using it. It will be called if
960 # and only if a change in the value is detected.
961 #
962 # Note: XXX will have any '.' replaced with '_' characters when determining
963 # the corresponding method names.
964
965 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000966 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000967 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti45763d02011-03-20 15:34:28 +0200968 'logging._handlers', 'logging._handlerList', 'sys.gettrace',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200969 'sys.warnoptions', 'threading._dangling',
Éric Araujo28df8de2011-09-19 05:10:45 +0200970 'multiprocessing.process._dangling',
971 'sysconfig._CONFIG_VARS', 'sysconfig._SCHEMES',
Éric Araujo4e377f22011-11-06 07:01:18 +0100972 'packaging.command._COMMANDS', 'packaging.database_caches',
Éric Araujo28df8de2011-09-19 05:10:45 +0200973 )
Nick Coghlan6ead5522009-10-18 13:19:33 +0000974
975 def get_sys_argv(self):
976 return id(sys.argv), sys.argv, sys.argv[:]
977 def restore_sys_argv(self, saved_argv):
978 sys.argv = saved_argv[1]
979 sys.argv[:] = saved_argv[2]
980
981 def get_cwd(self):
982 return os.getcwd()
983 def restore_cwd(self, saved_cwd):
984 os.chdir(saved_cwd)
985
986 def get_sys_stdout(self):
987 return sys.stdout
988 def restore_sys_stdout(self, saved_stdout):
989 sys.stdout = saved_stdout
990
991 def get_sys_stderr(self):
992 return sys.stderr
993 def restore_sys_stderr(self, saved_stderr):
994 sys.stderr = saved_stderr
995
996 def get_sys_stdin(self):
997 return sys.stdin
998 def restore_sys_stdin(self, saved_stdin):
999 sys.stdin = saved_stdin
1000
1001 def get_os_environ(self):
1002 return id(os.environ), os.environ, dict(os.environ)
1003 def restore_os_environ(self, saved_environ):
1004 os.environ = saved_environ[1]
1005 os.environ.clear()
1006 os.environ.update(saved_environ[2])
1007
1008 def get_sys_path(self):
1009 return id(sys.path), sys.path, sys.path[:]
1010 def restore_sys_path(self, saved_path):
1011 sys.path = saved_path[1]
1012 sys.path[:] = saved_path[2]
1013
Brett Cannon055470a2010-02-19 15:57:10 +00001014 def get_sys_path_hooks(self):
1015 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
1016 def restore_sys_path_hooks(self, saved_hooks):
1017 sys.path_hooks = saved_hooks[1]
1018 sys.path_hooks[:] = saved_hooks[2]
1019
Brett Cannon31f59292011-02-21 19:29:56 +00001020 def get_sys_gettrace(self):
1021 return sys.gettrace()
1022 def restore_sys_gettrace(self, trace_fxn):
1023 sys.settrace(trace_fxn)
1024
Brett Cannon055470a2010-02-19 15:57:10 +00001025 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +00001026 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +00001027 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +00001028 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +00001029
Brett Cannon29c0e4f2010-03-20 22:22:57 +00001030 def get_warnings_filters(self):
1031 return id(warnings.filters), warnings.filters, warnings.filters[:]
1032 def restore_warnings_filters(self, saved_filters):
1033 warnings.filters = saved_filters[1]
1034 warnings.filters[:] = saved_filters[2]
1035
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001036 def get_asyncore_socket_map(self):
1037 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +00001038 # XXX Making a copy keeps objects alive until __exit__ gets called.
1039 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001040 def restore_asyncore_socket_map(self, saved_map):
1041 asyncore = sys.modules.get('asyncore')
1042 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +00001043 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +00001044 asyncore.socket_map.update(saved_map)
1045
Éric Araujoff913062011-11-29 16:45:07 +01001046 def get_shutil_archive_formats(self):
1047 # we could call get_archives_formats() but that only returns the
1048 # registry keys; we want to check the values too (the functions that
1049 # are registered)
1050 return shutil._ARCHIVE_FORMATS, shutil._ARCHIVE_FORMATS.copy()
1051 def restore_shutil_archive_formats(self, saved):
1052 shutil._ARCHIVE_FORMATS = saved[0]
1053 shutil._ARCHIVE_FORMATS.clear()
1054 shutil._ARCHIVE_FORMATS.update(saved[1])
1055
1056 def get_shutil_unpack_formats(self):
1057 return shutil._UNPACK_FORMATS, shutil._UNPACK_FORMATS.copy()
1058 def restore_shutil_unpack_formats(self, saved):
1059 shutil._UNPACK_FORMATS = saved[0]
1060 shutil._UNPACK_FORMATS.clear()
1061 shutil._UNPACK_FORMATS.update(saved[1])
1062
Nick Coghlan7d819752010-12-05 06:45:03 +00001063 def get_logging__handlers(self):
1064 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001065 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001066 def restore_logging__handlers(self, saved_handlers):
1067 # Can't easily revert the logging state
1068 pass
1069
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001070 def get_logging__handlerList(self):
1071 # _handlerList is a list of weakrefs to handlers
1072 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1073 def restore_logging__handlerList(self, saved_handlerList):
1074 # Can't easily revert the logging state
1075 pass
1076
Éric Araujo4e377f22011-11-06 07:01:18 +01001077 def get_packaging_command__COMMANDS(self):
1078 # registry mapping command names to full dotted path or to the actual
1079 # class (resolved on demand); this check only looks at the names, not
1080 # the types of the values (IOW, if a value changes from a string
1081 # (dotted path) to a class it's okay but if a key (i.e. command class)
1082 # is added we complain)
1083 id_ = id(packaging.command._COMMANDS)
1084 keys = set(packaging.command._COMMANDS)
1085 return id_, keys
1086 def restore_packaging_command__COMMANDS(self, saved):
Éric Araujobfc97292011-11-14 18:18:15 +01001087 # if command._COMMANDS was bound to another dict object, we can't
Éric Araujo4e377f22011-11-06 07:01:18 +01001088 # restore the previous object and contents, because the get_ method
1089 # above does not return the dict object (to ignore changes in values)
1090 for key in packaging.command._COMMANDS.keys() - saved[1]:
1091 del packaging.command._COMMANDS[key]
1092
Éric Araujo76558e12011-10-06 02:44:19 +02001093 def get_packaging_database_caches(self):
1094 # caching system used by the PEP 376 implementation
1095 # we have one boolean and four dictionaries, initially empty
1096 switch = packaging.database._cache_enabled
1097 saved = []
1098 for name in ('_cache_name', '_cache_name_egg',
1099 '_cache_path', '_cache_path_egg'):
1100 cache = getattr(packaging.database, name)
1101 saved.append((id(cache), cache, cache.copy()))
1102 return switch, saved
1103 def restore_packaging_database_caches(self, saved):
1104 switch, saved_caches = saved
1105 packaging.database._cache_enabled = switch
1106 for offset, name in enumerate(('_cache_name', '_cache_name_egg',
1107 '_cache_path', '_cache_path_egg')):
1108 _, cache, items = saved_caches[offset]
1109 # put back the same object in place
1110 setattr(packaging.database, name, cache)
1111 # now restore its items
1112 cache.clear()
1113 cache.update(items)
1114
Ezio Melotti0123e052011-03-20 15:09:26 +02001115 def get_sys_warnoptions(self):
1116 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1117 def restore_sys_warnoptions(self, saved_options):
1118 sys.warnoptions = saved_options[1]
1119 sys.warnoptions[:] = saved_options[2]
1120
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02001121 # Controlling dangling references to Thread objects can make it easier
1122 # to track reference leaks.
1123 def get_threading__dangling(self):
1124 if not threading:
1125 return None
1126 # This copies the weakrefs without making any strong reference
1127 return threading._dangling.copy()
1128 def restore_threading__dangling(self, saved):
1129 if not threading:
1130 return
1131 threading._dangling.clear()
1132 threading._dangling.update(saved)
1133
1134 # Same for Process objects
1135 def get_multiprocessing_process__dangling(self):
1136 if not multiprocessing:
1137 return None
1138 # This copies the weakrefs without making any strong reference
1139 return multiprocessing.process._dangling.copy()
1140 def restore_multiprocessing_process__dangling(self, saved):
1141 if not multiprocessing:
1142 return
1143 multiprocessing.process._dangling.clear()
1144 multiprocessing.process._dangling.update(saved)
1145
Éric Araujo28df8de2011-09-19 05:10:45 +02001146 def get_sysconfig__CONFIG_VARS(self):
1147 # make sure the dict is initialized
1148 sysconfig.get_config_var('prefix')
1149 return (id(sysconfig._CONFIG_VARS), sysconfig._CONFIG_VARS,
1150 dict(sysconfig._CONFIG_VARS))
1151 def restore_sysconfig__CONFIG_VARS(self, saved):
1152 sysconfig._CONFIG_VARS = saved[1]
1153 sysconfig._CONFIG_VARS.clear()
1154 sysconfig._CONFIG_VARS.update(saved[2])
1155
1156 def get_sysconfig__SCHEMES(self):
1157 # it's mildly evil to look at the internal attribute, but it's easier
1158 # than copying a RawConfigParser object
1159 return (id(sysconfig._SCHEMES), sysconfig._SCHEMES._sections,
1160 sysconfig._SCHEMES._sections.copy())
1161 def restore_sysconfig__SCHEMES(self, saved):
1162 sysconfig._SCHEMES._sections = saved[1]
1163 sysconfig._SCHEMES._sections.clear()
1164 sysconfig._SCHEMES._sections.update(saved[2])
1165
1166
Nick Coghlan6ead5522009-10-18 13:19:33 +00001167 def resource_info(self):
1168 for name in self.resources:
1169 method_suffix = name.replace('.', '_')
1170 get_name = 'get_' + method_suffix
1171 restore_name = 'restore_' + method_suffix
1172 yield name, getattr(self, get_name), getattr(self, restore_name)
1173
1174 def __enter__(self):
1175 self.saved_values = dict((name, get()) for name, get, restore
1176 in self.resource_info())
1177 return self
1178
1179 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001180 saved_values = self.saved_values
1181 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001182 for name, get, restore in self.resource_info():
1183 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001184 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001185 # Check for changes to the resource's value
1186 if current != original:
1187 self.changed = True
1188 restore(original)
1189 if not self.quiet:
1190 print("Warning -- {} was modified by {}".format(
1191 name, self.testname),
1192 file=sys.stderr)
1193 if self.verbose > 1:
1194 print(" Before: {}\n After: {} ".format(
1195 original, current),
1196 file=sys.stderr)
1197 return False
1198
1199
Victor Stinnera7c33e52011-06-29 13:00:54 +02001200def runtest_inner(test, verbose, quiet,
1201 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001202 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001203
Antoine Pitrou88909542009-06-29 13:54:42 +00001204 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001205 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001206 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001207 if test.startswith('test.'):
1208 abstest = test
1209 else:
1210 # Always import it from the test package
1211 abstest = 'test.' + test
1212 with saved_test_environment(test, verbose, quiet) as environment:
1213 start_time = time.time()
1214 the_package = __import__(abstest, globals(), locals(), [])
1215 the_module = getattr(the_package, test)
1216 # Old tests run to completion simply as a side-effect of
1217 # being imported. For tests based on unittest or doctest,
1218 # explicitly invoke their test_main() function (if it exists).
1219 indirect_test = getattr(the_module, "test_main", None)
1220 if indirect_test is not None:
1221 indirect_test()
1222 if huntrleaks:
1223 refleak = dash_R(the_module, test, indirect_test,
1224 huntrleaks)
1225 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001226 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001227 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001228 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001229 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001230 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001231 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001232 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001233 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001234 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001235 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001236 except KeyboardInterrupt:
1237 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001238 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001239 if display_failure:
1240 print("test", test, "failed --", msg, file=sys.stderr)
1241 else:
1242 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001243 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001244 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001245 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001246 msg = traceback.format_exc()
1247 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001248 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001249 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001250 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001251 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001252 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001253 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001254 return ENV_CHANGED, test_time
1255 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001256
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001257def cleanup_test_droppings(testname, verbose):
1258 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001259 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001260 import gc
1261
1262 # First kill any dangling references to open files etc.
1263 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001264 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001265 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001266
1267 # Try to clean up junk commonly left behind. While tests shouldn't leave
1268 # any files or directories behind, when a test fails that can be tedious
1269 # for it to arrange. The consequences can be especially nasty on Windows,
1270 # since if a test leaves a file open, it cannot be deleted by name (while
1271 # there's nothing we can do about that here either, we can display the
1272 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001273 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001274 "db_home",
1275 ):
1276 if not os.path.exists(name):
1277 continue
1278
1279 if os.path.isdir(name):
1280 kind, nuker = "directory", shutil.rmtree
1281 elif os.path.isfile(name):
1282 kind, nuker = "file", os.unlink
1283 else:
1284 raise SystemError("os.path says %r exists but is neither "
1285 "directory nor file" % name)
1286
1287 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001288 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001289 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001290 # if we have chmod, fix possible permissions problems
1291 # that might prevent cleanup
1292 if (hasattr(os, 'chmod')):
1293 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001294 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001295 except Exception as msg:
1296 print(("%r left behind %s %r and it couldn't be "
1297 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001298
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001299def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001300 """Run a test multiple times, looking for reference leaks.
1301
1302 Returns:
1303 False if the test didn't leak references; True if we detected refleaks.
1304 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001305 # This code is hackish and inelegant, but it seems to do the job.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001306 import copyreg
1307 import collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001308
1309 if not hasattr(sys, 'gettotalrefcount'):
1310 raise Exception("Tracking reference leaks requires a debug build "
1311 "of Python")
1312
1313 # Save current values for dash_R_cleanup() to restore.
1314 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001315 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001316 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001317 try:
1318 import zipimport
1319 except ImportError:
1320 zdc = None # Run unmodified on platforms without zipimport support
1321 else:
1322 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001323 abcs = {}
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001324 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001325 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001326 continue
1327 for obj in abc.__subclasses__() + [abc]:
1328 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001329
1330 if indirect_test:
1331 def run_the_test():
1332 indirect_test()
1333 else:
1334 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001335 del sys.modules[the_module.__name__]
1336 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001337
1338 deltas = []
1339 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001340 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001341 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001342 print("beginning", repcount, "repetitions", file=sys.stderr)
1343 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001344 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001345 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001346 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001347 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001348 run_the_test()
1349 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001350 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001351 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001352 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001353 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001354 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001355 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001356 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001357 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001358 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001359 sys.stderr.flush()
1360 with open(fname, "a") as refrep:
1361 print(msg, file=refrep)
1362 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001363 return True
1364 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001365
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001366def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001367 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001368 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001369 import urllib.parse, urllib.request, mimetypes, doctest
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001370 import struct, filecmp, collections.abc
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001371 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001372 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001373
Christian Heimesdae2a892008-04-19 00:55:37 +00001374 # Clear the warnings registry, so they can be displayed again
1375 for mod in sys.modules.values():
1376 if hasattr(mod, '__warningregistry__'):
1377 del mod.__warningregistry__
1378
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001379 # Restore some original values.
1380 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001381 copyreg.dispatch_table.clear()
1382 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001383 sys.path_importer_cache.clear()
1384 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001385 try:
1386 import zipimport
1387 except ImportError:
1388 pass # Run unmodified on platforms without zipimport support
1389 else:
1390 zipimport._zip_directory_cache.clear()
1391 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001392
Christian Heimes26855632008-01-27 23:50:43 +00001393 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001394 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001395
Guido van Rossum3de862d2007-08-18 00:10:33 +00001396 # Clear ABC registries, restoring previously saved ABC registries.
Raymond Hettinger158c9c22011-02-22 00:41:50 +00001397 for abc in [getattr(collections.abc, a) for a in collections.abc.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001398 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001399 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001400 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001401 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001402 obj._abc_cache.clear()
1403 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001404
Antoine Pitrou046467c2009-10-30 18:30:35 +00001405 # Flush standard output, so that buffered data is sent to the OS and
1406 # associated Python objects are reclaimed.
1407 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1408 if stream is not None:
1409 stream.flush()
1410
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001411 # Clear assorted module caches.
1412 _path_created.clear()
1413 re.purge()
1414 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001415 urllib.parse.clear_cache()
1416 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001417 linecache.clearcache()
1418 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001419 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001420 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001421 doctest.master = None
Meador Inge11e38132011-11-25 22:33:32 -06001422 try:
1423 import ctypes
1424 except ImportError:
1425 # Don't worry about resetting the cache if ctypes is not supported
1426 pass
1427 else:
1428 ctypes._reset_cache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001429
1430 # Collect cyclic trash.
1431 gc.collect()
1432
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001433def warm_char_cache():
1434 s = bytes(range(256))
1435 for i in range(256):
1436 s[i:i+1]
1437
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001438def findtestdir(path=None):
1439 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001440
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001441def removepy(names):
1442 if not names:
1443 return
1444 for idx, name in enumerate(names):
1445 basename, ext = os.path.splitext(name)
1446 if ext == '.py':
1447 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001448
Guido van Rossum152494a1996-12-20 03:12:20 +00001449def count(n, word):
1450 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001451 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001452 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001453 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001454
Tim Petersa45da922001-08-12 03:45:50 +00001455def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001456 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001457
1458 Optional arg width (default 70) is the maximum line length.
1459 Optional arg indent (default 4) is the number of blanks with which to
1460 begin each line.
1461 """
1462
Tim Petersba78bc42002-07-04 19:45:06 +00001463 from textwrap import fill
1464 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001465 # Print the sorted list: 'x' may be a '--random' list or a set()
1466 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001467 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001468
Tim Petersde14a302002-04-01 05:04:46 +00001469# Map sys.platform to a string containing the basenames of tests
1470# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001471#
1472# Special cases:
1473# test_pep277
1474# The _ExpectedSkips constructor adds this to the set of expected
1475# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001476# test_timeout
1477# Controlled by test_timeout.skip_expected. Requires the network
1478# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001479#
1480# Tests that are expected to be skipped everywhere except on one platform
1481# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001482
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001483_expectations = (
1484 ('win32',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001485 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001486 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001487 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001488 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001489 test_dbm
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001490 test_devpoll
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001491 test_fcntl
1492 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001493 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001494 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001495 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001496 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001497 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001498 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001499 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001500 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001501 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001502 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001503 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001504 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001505 test_pty
1506 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001507 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001508 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001509 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001510 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001511 test_wait3
1512 test_wait4
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001513 """),
1514 ('linux',
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001515 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001516 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001517 test_devpoll
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001518 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001519 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001520 test_ossaudiodev
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001521 """),
1522 ('unixware',
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001523 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001524 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001525 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001526 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001527 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001528 test_openpty
1529 test_pyexpat
1530 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001531 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001532 """),
1533 ('openunix',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001534 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001535 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001536 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001537 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001538 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001539 test_openpty
1540 test_pyexpat
1541 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001542 test_sundry
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001543 """),
1544 ('sco_sv',
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001545 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001546 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001547 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001548 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001549 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001550 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001551 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001552 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001553 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001554 test_openpty
1555 test_pyexpat
1556 test_queue
1557 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001558 test_sundry
1559 test_thread
1560 test_threaded_import
1561 test_threadedtempfile
1562 test_threading
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001563 """),
1564 ('darwin',
Jack Jansen398c2362001-12-02 21:41:36 +00001565 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001566 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001567 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001568 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001569 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001570 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001571 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001572 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001573 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001574 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001575 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001576 test_poll
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001577 """),
1578 ('sunos',
Guido van Rossum11c3f092002-07-17 15:08:24 +00001579 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001580 test_curses
1581 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001582 test_epoll
1583 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001584 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001585 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001586 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001587 test_zipfile
1588 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001589 """),
1590 ('hp-ux',
Skip Montanarob3230212002-03-15 02:54:03 +00001591 """
Skip Montanarob3230212002-03-15 02:54:03 +00001592 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001593 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001594 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001595 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001596 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001597 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001598 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001599 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001600 test_openpty
1601 test_pyexpat
1602 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001603 test_zipfile
1604 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001605 """),
1606 ('cygwin',
Jason Tishler25115942002-12-05 15:18:15 +00001607 """
Jason Tishler25115942002-12-05 15:18:15 +00001608 test_curses
1609 test_dbm
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001610 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001611 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001612 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001613 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001614 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001615 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001616 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001617 test_socketserver
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001618 """),
1619 ('os2emx',
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001620 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001621 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001622 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001623 test_epoll
1624 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001625 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001626 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001627 test_openpty
1628 test_ossaudiodev
1629 test_pty
1630 test_resource
1631 test_signal
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001632 """),
1633 ('freebsd',
Guido van Rossum944a6c32003-11-20 22:11:29 +00001634 """
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001635 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001636 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001637 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001638 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001639 test_ossaudiodev
1640 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001641 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001642 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +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
Guido van Rossum944a6c32003-11-20 22:11:29 +00001647 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001648 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001649 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001650 """),
1651 ('aix',
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001652 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001653 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001654 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001655 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001656 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001657 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001658 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001659 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001660 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001661 test_ttk_guionly
1662 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001663 test_zipimport
1664 test_zlib
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001665 """),
1666 ('openbsd',
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001667 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001668 test_ctypes
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001669 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001670 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001671 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001672 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001673 test_normalization
1674 test_ossaudiodev
1675 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001676 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001677 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001678 test_ttk_guionly
1679 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001680 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001681 """),
1682 ('netbsd',
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001683 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001684 test_ctypes
1685 test_curses
Antoine Pitrou13f8b682011-11-15 15:25:59 +01001686 test_devpoll
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001687 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001688 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001689 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001690 test_ossaudiodev
1691 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001692 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001693 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001694 test_ttk_guionly
1695 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001696 test_multiprocessing
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001697 """),
1698)
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001699
Tim Petersb5b7b782001-08-12 01:20:39 +00001700class _ExpectedSkips:
1701 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001702 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001703 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001704
Tim Peters7c7efe92002-08-23 17:55:54 +00001705 self.valid = False
Victor Stinner9c3de4a2011-08-17 20:49:41 +02001706 expected = None
1707 for item in _expectations:
1708 if sys.platform.startswith(item[0]):
1709 expected = item[1]
1710 break
1711 if expected is not None:
1712 self.expected = set(expected.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001713
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001714 # These are broken tests, for now skipped on every platform.
1715 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001716 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001717
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001718 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001719 if not os.path.supports_unicode_filenames:
1720 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001721
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001722 # doctest, profile and cProfile tests fail when the codec for the
1723 # fs encoding isn't built in because PyUnicode_Decode() adds two
1724 # calls into Python.
1725 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1726 if sys.getfilesystemencoding().lower() not in encs:
1727 self.expected.add('test_profile')
1728 self.expected.add('test_cProfile')
1729 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001730
Neal Norwitz55b61d22003-02-28 19:57:03 +00001731 if test_timeout.skip_expected:
1732 self.expected.add('test_timeout')
1733
Tim Petersecd79eb2003-01-29 00:35:32 +00001734 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001735 # test_sqlite is only reliable on Windows where the library
1736 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001737 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001738 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001739 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001740 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001741
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001742 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001743 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001744
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001745 if support.python_is_optimized():
1746 self.expected.add("test_gdb")
1747
Tim Peters7c7efe92002-08-23 17:55:54 +00001748 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001749
1750 def isvalid(self):
1751 "Return true iff _ExpectedSkips knows about the current platform."
1752 return self.valid
1753
1754 def getexpected(self):
1755 """Return set of test names we expect to skip on current platform.
1756
1757 self.isvalid() must be true.
1758 """
1759
1760 assert self.isvalid()
1761 return self.expected
1762
Michael Foord3ab34cc2010-12-03 12:27:40 +00001763def _make_temp_dir_for_build(TEMPDIR):
1764 # When tests are run from the Python build directory, it is best practice
1765 # to keep the test files in a subfolder. It eases the cleanup of leftover
1766 # files using command "make distclean".
1767 if sysconfig.is_python_build():
1768 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1769 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001770 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001771 os.mkdir(TEMPDIR)
Florent Xicluna68f71a32011-10-28 16:06:23 +02001772 except FileExistsError:
1773 pass
Michael Foord3ab34cc2010-12-03 12:27:40 +00001774
1775 # Define a writable temp dir that will be used as cwd while running
1776 # the tests. The name of the dir includes the pid to allow parallel
1777 # testing (see the -j option).
1778 TESTCWD = 'test_python_{}'.format(os.getpid())
1779
1780 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1781 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001782
Guido van Rossum152494a1996-12-20 03:12:20 +00001783if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001784 # Remove regrtest.py's own directory from the module search path. Despite
1785 # the elimination of implicit relative imports, this is still needed to
1786 # ensure that submodules of the test package do not inappropriately appear
1787 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1788 # directly instead of using the -m switch
1789 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1790 i = len(sys.path)
1791 while i >= 0:
1792 i -= 1
1793 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1794 del sys.path[i]
1795
Florent Xiclunadc69e722010-09-13 16:35:02 +00001796 # findtestdir() gets the dirname out of __file__, so we have to make it
1797 # absolute before changing the working directory.
1798 # For example __file__ may be relative when running trace or profile.
1799 # See issue #9323.
1800 __file__ = os.path.abspath(__file__)
1801
1802 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001803 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001804
Michael Foord3ab34cc2010-12-03 12:27:40 +00001805 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001806
1807 # Run the tests in a context manager that temporary changes the CWD to a
1808 # temporary and writable directory. If it's not possible to create or
1809 # change the CWD, the original CWD will be used. The original CWD is
1810 # available from support.SAVEDCWD.
1811 with support.temp_cwd(TESTCWD, quiet=True):
1812 main()