blob: 29f2bf042ca44bbcff0900b0e7d3dc84e95e01c8 [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
23
24Verbosity
25
26-v/--verbose -- run tests in verbose mode with output to stdout
27-w/--verbose2 -- re-run failed tests in verbose mode
Victor Stinnera7c33e52011-06-29 13:00:54 +020028-W/--verbose3 -- display test output on failure
Georg Brandl96f78972009-10-27 21:02:11 +000029-d/--debug -- print traceback for failed tests
R. David Murray0ba81e02010-04-26 17:02:32 +000030-q/--quiet -- no output unless one or more tests fail
Senthil Kumaranbcfe56d2012-01-14 18:26:27 +080031-o/--slow -- print the slowest 10 tests
Antoine Pitrou3c4402f2011-01-03 20:38:52 +000032 --header -- print header with interpreter info
Georg Brandl96f78972009-10-27 21:02:11 +000033
34Selecting tests
35
R. David Murray0ba81e02010-04-26 17:02:32 +000036-r/--random -- randomize test execution order (see below)
Antoine Pitrou68530ac2011-01-03 20:40:07 +000037 --randseed -- pass a random seed to reproduce a previous random run
Georg Brandl96f78972009-10-27 21:02:11 +000038-f/--fromfile -- read names of tests to run from a file (see below)
39-x/--exclude -- arguments are tests to *exclude*
R. David Murray0ba81e02010-04-26 17:02:32 +000040-s/--single -- single step through a set of tests (see below)
Antoine Pitroub9c73e82011-07-29 23:53:38 +020041-m/--match PAT -- match test cases and methods with glob pattern PAT
Antoine Pitrou216a3bc2011-07-23 22:33:39 +020042-G/--failfast -- fail as soon as a test fails (only with -v or -W)
Georg Brandl96f78972009-10-27 21:02:11 +000043-u/--use RES1,RES2,...
44 -- specify which special resource intensive tests to run
45-M/--memlimit LIMIT
46 -- run very large memory-consuming tests
47
48Special runs
49
50-l/--findleaks -- if GC is available detect tests that leak memory
51-L/--runleaks -- run the leaks(1) command just before exit
52-R/--huntrleaks RUNCOUNTS
53 -- search for reference leaks (needs debug build, v. slow)
54-j/--multiprocess PROCESSES
55 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000056-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000057-D/--coverdir DIRECTORY
58 -- Directory where coverage files are put
59-N/--nocoverdir -- Put coverage files alongside modules
60-t/--threshold THRESHOLD
61 -- call gc.set_threshold(THRESHOLD)
62-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000063-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000064
R. David Murray0ba81e02010-04-26 17:02:32 +000065
66Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000067
Collin Winterfd12f492009-03-29 04:05:05 +000068-r randomizes test execution order. You can use --randseed=int to provide a
69int seed value for the randomizer; this is useful for reproducing troublesome
70test orders.
71
R. David Murray0ba81e02010-04-26 17:02:32 +000072-s On the first invocation of regrtest using -s, the first test file found
73or the first test file given on the command line is run, and the name of
74the next test is recorded in a file named pynexttest. If run from the
75Python build directory, pynexttest is located in the 'build' subdirectory,
76otherwise it is located in tempfile.gettempdir(). On subsequent runs,
77the test in pynexttest is run, and the next test is written to pynexttest.
78When the last test has been run, pynexttest is deleted. In this way it
79is possible to single step through the test files. This is useful when
80doing memory analysis on the Python interpreter, which process tends to
81consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000082
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000083-S is used to continue running tests after an aborted run. It will
84maintain the order a standard run (ie, this assumes -r is not used).
85This is useful after the tests have prematurely stopped for some external
86reason and you want to start running from where you left off rather
87than starting from the beginning.
88
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000089-f reads the names of tests from the file given as f's argument, one
90or more test names per line. Whitespace is ignored. Blank lines and
91lines beginning with '#' are ignored. This is especially useful for
92whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000093
Skip Montanaro0179a182004-06-06 15:53:18 +000094-L causes the leaks(1) command to be run just before exit if it exists.
95leaks(1) is available on Mac OS X and presumably on some other
96FreeBSD-derived systems.
97
Michael W. Hudson61147f62004-08-03 11:33:28 +000098-R runs each test several times and examines sys.gettotalrefcount() to
99see if the test appears to be leaking references. The argument should
100be of the form stab:run:fname where 'stab' is the number of times the
101test is run to let gettotalrefcount settle down, 'run' is the number
102of times further it is run and 'fname' is the name of the file the
103reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000104"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000105
Thomas Wouters477c8d52006-05-27 19:21:47 +0000106-M runs tests that require an exorbitant amount of memory. These tests
107typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001082 billion objects, which only works on 64-bit systems. There are also some
109tests that try to exhaust the address space of the process, which only makes
110sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000111which is a string in the form of '2.5Gb', determines howmuch memory the
112tests will limit themselves to (but they may go slightly over.) The number
113shouldn't be more memory than the machine has (including swap memory). You
114should also keep in mind that swap memory is generally much, much slower
115than RAM, and setting memlimit to all available RAM or higher will heavily
116tax the machine. On the other hand, it is no use running these tests with a
117limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
118to use more than memlimit memory will be skipped. The big-memory tests
119generally run very, very long.
120
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000121-u is used to specify which special resource intensive tests to run,
122such as those requiring large file support or network connectivity.
123The argument is a comma-separated list of words indicating the
124resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000125
Fred Drake3a15dac2002-04-11 16:39:16 +0000126 all - Enable all special resources.
127
Guido van Rossum315aa362003-03-11 14:46:48 +0000128 audio - Tests that use the audio device. (There are known
129 cases of broken audio drivers that can crash Python or
130 even the Linux kernel.)
131
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000132 curses - Tests that use curses and will modify the terminal's
133 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000134
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000135 largefile - It is okay to run some test that may create huge
136 files. These tests can take a long time and may
137 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000138
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000139 network - It is okay to run tests that use external network
140 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000141
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000142 decimal - Test the decimal module against a large suite that
143 verifies compliance with standards.
144
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000145 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000146
Tim Peterseba28be2005-03-28 01:08:02 +0000147 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000148
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000149 urlfetch - It is okay to download files required on testing.
150
Guilherme Polo9de29af2009-01-28 20:40:48 +0000151 gui - Run tests that require a running GUI.
152
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000153To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000154example, to run all the tests except for the gui tests, give the
155option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000156"""
157
Brett Cannon45071902010-06-14 22:22:54 +0000158import builtins
Victor Stinner8313d6a2011-06-29 15:22:26 +0200159import errno
Guido van Rossum152494a1996-12-20 03:12:20 +0000160import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200161import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000162import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200163import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000164import os
Victor Stinner8313d6a2011-06-29 15:22:26 +0200165import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000166import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000167import re
Éric Araujoff913062011-11-29 16:45:07 +0100168import shutil
Christian Heimesb186d002008-03-18 15:15:01 +0000169import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000170import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200171import tempfile
172import time
173import traceback
174import unittest
175import warnings
176from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000177
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200178try:
179 import threading
180except ImportError:
181 threading = None
182try:
183 import multiprocessing.process
184except ImportError:
185 multiprocessing = None
186
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000187
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000188# Some times __path__ and __file__ are not absolute (e.g. while running from
189# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
190# imports might fail. This affects only the modules imported before os.chdir().
191# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
192# they are found in the CWD their __file__ and __path__ will be relative (this
193# happens before the chdir). All the modules imported after the chdir, are
194# not found in the CWD, and since the other paths in sys.path[1:] are absolute
195# (site.py absolutize them), the __file__ and __path__ will be absolute too.
196# Therefore it is necessary to absolutize manually the __file__ and __path__ of
197# the packages to prevent later imports to fail when the CWD is different.
198for module in sys.modules.values():
199 if hasattr(module, '__path__'):
200 module.__path__ = [os.path.abspath(path) for path in module.__path__]
201 if hasattr(module, '__file__'):
202 module.__file__ = os.path.abspath(module.__file__)
203
Guido van Rossumdc15c272002-08-12 21:55:51 +0000204
Guido van Rossumbb484652002-12-02 09:56:21 +0000205# MacOSX (a.k.a. Darwin) has a default stack size that is too small
206# for deeply recursive regular expressions. We see this as crashes in
207# the Python test suite when running test_re.py and test_sre.py. The
208# fix is to set the stack limit to 2048.
209# This approach may also be useful for other Unixy platforms that
210# suffer from small default stack limits.
211if sys.platform == 'darwin':
212 try:
213 import resource
214 except ImportError:
215 pass
216 else:
217 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
218 newsoft = min(hard, max(soft, 1024*2048))
219 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
220
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000221# Test result constants.
222PASSED = 1
223FAILED = 0
224ENV_CHANGED = -1
225SKIPPED = -2
226RESOURCE_DENIED = -3
227INTERRUPTED = -4
228
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000229from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000230
Georg Brandl1158a332009-06-04 09:30:30 +0000231RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000232 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000233
Florent Xiclunaec882212010-08-09 16:56:43 +0000234TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000235
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000236def usage(msg):
237 print(msg, file=sys.stderr)
238 print("Use --help for usage", file=sys.stderr)
239 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000240
241
Antoine Pitrou88909542009-06-29 13:54:42 +0000242def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000243 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000244 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000245 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000246 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200247 header=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000248 """Execute a test suite.
249
Thomas Wouters7e474022000-07-16 12:04:32 +0000250 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000251 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000252
253 tests -- a list of strings containing test names (optional)
254 testdir -- the directory in which to look for tests (optional)
255
256 Users other than the Python test suite will certainly want to
257 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000258 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000259
260 If the tests argument is omitted, the tests listed on the
261 command-line will be used. If that's empty, too, then all *.py
262 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000263
Antoine Pitrou88909542009-06-29 13:54:42 +0000264 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000265 single, randomize, findleaks, use_resources, trace, coverdir,
266 print_slow, and random_seed) allow programmers calling main()
267 directly to set the values that would normally be set by flags
268 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000269 """
Fred Drake004d5e62000-10-23 17:22:08 +0000270
Victor Stinner1802d3f2010-05-19 17:11:19 +0000271 replace_stdout()
272
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000273 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000274 try:
Meador Inge3f55e592012-01-14 09:12:55 -0600275 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsoS:rf:lu:t:TD:NLR:FdwWM:nj:Gm:',
Georg Brandl96f78972009-10-27 21:02:11 +0000276 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
277 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
Meador Ingeb0ae0ac2012-01-14 11:50:33 -0600278 'use=', 'threshold=', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000279 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000280 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200281 'start=', 'nowindows', 'header', 'failfast', 'match'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000282 except getopt.error as msg:
283 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000284
285 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000286 if random_seed is None:
287 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000288 if use_resources is None:
289 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000290 debug = False
291 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000292 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000293 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000294 print(__doc__)
295 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000296 elif o in ('-v', '--verbose'):
297 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000298 elif o in ('-w', '--verbose2'):
299 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000300 elif o in ('-d', '--debug'):
301 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000302 elif o in ('-W', '--verbose3'):
303 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200304 elif o in ('-G', '--failfast'):
305 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000306 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000307 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000308 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000309 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000310 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000311 elif o in ('-S', '--start'):
312 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000313 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000314 single = True
Senthil Kumaranbcfe56d2012-01-14 18:26:27 +0800315 elif o in ('-o', '--slow'):
Christian Heimesb186d002008-03-18 15:15:01 +0000316 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000317 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000318 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000319 elif o == '--randseed':
320 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000321 elif o in ('-f', '--fromfile'):
322 fromfile = a
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200323 elif o in ('-m', '--match'):
324 match_tests = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000325 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000326 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000327 elif o in ('-L', '--runleaks'):
328 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000329 elif o in ('-t', '--threshold'):
330 import gc
331 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000332 elif o in ('-T', '--coverage'):
333 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000334 elif o in ('-D', '--coverdir'):
335 coverdir = os.path.join(os.getcwd(), a)
336 elif o in ('-N', '--nocoverdir'):
337 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000338 elif o in ('-R', '--huntrleaks'):
339 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000340 if len(huntrleaks) not in (2, 3):
341 print(a, huntrleaks)
342 usage('-R takes 2 or 3 colon-separated arguments')
343 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000344 huntrleaks[0] = 5
345 else:
346 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000347 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000348 huntrleaks[1] = 4
349 else:
350 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000351 if len(huntrleaks) == 2 or not huntrleaks[2]:
352 huntrleaks[2:] = ["reflog.txt"]
353 # Avoid false positives due to the character cache in
354 # stringobject.c filling slowly with random data
355 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000356 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000357 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000358 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000359 u = [x.lower() for x in a.split(',')]
360 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000361 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000362 use_resources[:] = RESOURCE_NAMES
363 continue
364 remove = False
365 if r[0] == '-':
366 remove = True
367 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000368 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000369 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000370 if remove:
371 if r in use_resources:
372 use_resources.remove(r)
373 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000374 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000375 elif o in ('-n', '--nowindows'):
376 import msvcrt
377 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
378 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
379 msvcrt.SEM_NOGPFAULTERRORBOX|
380 msvcrt.SEM_NOOPENFILEERRORBOX)
381 try:
382 msvcrt.CrtSetReportMode
383 except AttributeError:
384 # release build
385 pass
386 else:
387 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
388 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
389 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000390 elif o in ('-F', '--forever'):
391 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000392 elif o in ('-j', '--multiprocess'):
393 use_mp = int(a)
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000394 elif o == '--header':
395 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000396 elif o == '--slaveargs':
397 args, kwargs = json.loads(a)
398 try:
399 result = runtest(*args, **kwargs)
400 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000401 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000402 sys.stdout.flush()
403 print() # Force a newline (just in case)
404 print(json.dumps(result))
405 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000406 else:
407 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000408 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000409 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000410 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000411 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000412 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000413 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000414 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000415 usage("-l and -j don't go together!")
Nadeem Vawdac1fba3e2011-08-27 15:22:05 +0200416 if use_mp and support.max_memuse:
417 usage("-M and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200418 if failfast and not (verbose or verbose3):
419 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000420
Guido van Rossum152494a1996-12-20 03:12:20 +0000421 good = []
422 bad = []
423 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000424 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000425 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000426 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000427
Neil Schemenauerd569f232000-09-22 15:29:28 +0000428 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000429 try:
430 import gc
431 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000432 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000433 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000434 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000435 # Uncomment the line below to report garbage that is not
436 # freeable by reference counting alone. By default only
437 # garbage that is not collectable by the GC is reported.
438 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000439 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000440
Barry Warsawe11e3de1999-01-28 19:51:51 +0000441 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000442 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000443 try:
444 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000445 next_test = fp.read().strip()
446 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000447 fp.close()
448 except IOError:
449 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000450
451 if fromfile:
452 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000453 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000454 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000455 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000456 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000457 guts = line.split() # assuming no test has whitespace in its name
458 if guts and not guts[0].startswith('#'):
459 tests.extend(guts)
460 fp.close()
461
462 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000463 removepy(args)
464 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000465
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000466 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000467 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000468 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000469 for arg in args:
470 if arg in stdtests:
471 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000472 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000473 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000474
475 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000476 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000477 # Print basic platform information
478 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000479 print("== ", platform.platform(aliased=True),
480 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000481 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000482 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000483
R. David Murrayef1992b2009-12-16 15:19:27 +0000484 alltests = findtests(testdir, stdtests, nottests)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000485 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000486 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000487 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000488 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000489 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000490 except IndexError:
491 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000492 # Remove all the tests that precede start if it's set.
493 if start:
494 try:
495 del tests[:tests.index(start)]
496 except ValueError:
497 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000498 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000499 random.seed(random_seed)
500 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000501 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000502 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000503 import trace, tempfile
504 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
505 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000506 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000507
Christian Heimesb186d002008-03-18 15:15:01 +0000508 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000509 support.verbose = verbose # Tell tests to be moderately quiet
510 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000511 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000512
513 def accumulate_result(test, result):
514 ok, test_time = result
515 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000516 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000517 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000518 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000519 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000520 elif ok == ENV_CHANGED:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000521 environment_changed.append(test)
522 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000523 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000524 elif ok == RESOURCE_DENIED:
525 skipped.append(test)
526 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000527
R. David Murray7dc72cc2009-11-14 16:13:02 +0000528 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000529 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000530 while True:
531 for test in tests:
532 yield test
533 if bad:
534 return
535 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000536 test_count = ''
537 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000538 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000539 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000540 test_count = '/{}'.format(len(selected))
541 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000542
Antoine Pitrou88909542009-06-29 13:54:42 +0000543 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000544 try:
545 from threading import Thread
546 except ImportError:
547 print("Multiprocess option requires thread support")
548 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000549 from queue import Queue
550 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000551 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000552 output = Queue()
Antoine Pitrou09f2e6f2012-07-26 00:45:19 +0200553 pending = MultiprocessTests(tests)
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000554 opt_args = support.args_from_interpreter_flags()
555 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000556 def work():
557 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000558 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000559 while True:
560 try:
Antoine Pitrou09f2e6f2012-07-26 00:45:19 +0200561 test = next(pending)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000562 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000563 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000564 return
Antoine Pitrou09f2e6f2012-07-26 00:45:19 +0200565 args_tuple = (
566 (test, verbose, quiet),
567 dict(huntrleaks=huntrleaks, use_resources=use_resources,
568 debug=debug, output_on_failure=verbose3,
569 failfast=failfast, match_tests=match_tests)
570 )
Antoine Pitrou88909542009-06-29 13:54:42 +0000571 # -E is needed by some tests, e.g. test_import
Antoine Pitrou282083d2012-07-11 19:19:14 +0200572 # Running the child from the same working directory ensures
573 # that TEMPDIR for the child is the same when
574 # sysconfig.is_python_build() is true. See issue 15300.
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000575 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000576 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000577 universal_newlines=True,
Antoine Pitrou282083d2012-07-11 19:19:14 +0200578 close_fds=(os.name != 'nt'),
579 cwd=support.SAVEDCWD)
R. David Murray27144602009-10-19 15:26:16 +0000580 stdout, stderr = popen.communicate()
581 # Strip last refcount output line if it exists, since it
582 # comes from the shutdown of the interpreter in the subcommand.
583 stderr = debug_output_pat.sub("", stderr)
584 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000585 if not result:
586 output.put((None, None, None, None))
587 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000588 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000589 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000590 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000591 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000592 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000593 workers = [Thread(target=work) for i in range(use_mp)]
594 for worker in workers:
595 worker.start()
596 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000597 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000598 try:
599 while finished < use_mp:
600 test, stdout, stderr, result = output.get()
601 if test is None:
602 finished += 1
603 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200604 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000605 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300606 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
607 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200608 test_count_width, test_index, test_count,
609 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000610 if stdout:
611 print(stdout)
612 if stderr:
613 print(stderr, file=sys.stderr)
Antoine Pitrou82372582012-06-27 17:41:07 +0200614 sys.stdout.flush()
615 sys.stderr.flush()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000616 if result[0] == INTERRUPTED:
617 assert result[1] == 'KeyboardInterrupt'
618 raise KeyboardInterrupt # What else?
Georg Brandldee7b852010-08-02 18:59:52 +0000619 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000620 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000621 interrupted = True
Antoine Pitrou09f2e6f2012-07-26 00:45:19 +0200622 pending.interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000623 for worker in workers:
624 worker.join()
625 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000626 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000627 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300628 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
629 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200630 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000631 sys.stdout.flush()
632 if trace:
633 # If we're tracing code coverage, then we don't exit with status
634 # if on a false return value from main.
R. David Murray8e286c42010-12-27 20:09:32 +0000635 tracer.runctx('runtest(test, verbose, quiet)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000636 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000637 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000638 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100639 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200640 output_on_failure=verbose3,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200641 failfast=failfast, match_tests=match_tests)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000642 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000643 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000644 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000645 break
646 except:
647 raise
648 if findleaks:
649 gc.collect()
650 if gc.garbage:
651 print("Warning: test created", len(gc.garbage), end=' ')
652 print("uncollectable object(s).")
653 # move the uncollectable objects somewhere so we don't see
654 # them again
655 found_garbage.extend(gc.garbage)
656 del gc.garbage[:]
657 # Unload the newly imported modules (best effort finalization)
658 for module in sys.modules.keys():
659 if module not in save_modules and module.startswith("test."):
660 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000661
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000662 if interrupted:
663 # print a newline after ^C
664 print()
665 print("Test suite interrupted by signal SIGINT.")
666 omitted = set(selected) - set(good) - set(bad) - set(skipped)
667 print(count(len(omitted), "test"), "omitted:")
668 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000669 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000670 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000671 print("All", end=' ')
672 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000673 if print_slow:
674 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000675 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000676 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000677 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000678 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000679 bad = sorted(set(bad) - set(environment_changed))
680 if bad:
681 print(count(len(bad), "test"), "failed:")
682 printlist(bad)
Vinay Sajipf9596182012-03-02 01:01:13 +0000683 if environment_changed:
684 print("{} altered the execution environment:".format(
685 count(len(environment_changed), "test")))
686 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000687 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000688 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000689 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000690
Tim Petersb5b7b782001-08-12 01:20:39 +0000691 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000692 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000693 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000694 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000695 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000696 print(count(len(surprise), "skip"), \
697 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000698 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000699 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000700 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000701 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000702 print("Ask someone to teach regrtest.py about which tests are")
703 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000704
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000705 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000706 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000707 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000708 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000709 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000710 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000711 verbose = True
R. David Murray8e286c42010-12-27 20:09:32 +0000712 ok = runtest(test, True, quiet, huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000713 except KeyboardInterrupt:
714 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000715 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000716 break
717 except:
718 raise
719
Barry Warsawe11e3de1999-01-28 19:51:51 +0000720 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000721 if next_single_test:
722 with open(filename, 'w') as fp:
723 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000724 else:
725 os.unlink(filename)
726
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000727 if trace:
728 r = tracer.results()
729 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
730
Skip Montanaro0179a182004-06-06 15:53:18 +0000731 if runleaks:
732 os.system("leaks %d" % os.getpid())
733
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000734 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000735
Guido van Rossum152494a1996-12-20 03:12:20 +0000736
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000737STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000738 'test_grammar',
739 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000740 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000741 'test_builtin',
742 'test_exceptions',
743 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000744 'test_unittest',
745 'test_doctest',
746 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000747]
Guido van Rossum152494a1996-12-20 03:12:20 +0000748
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000749NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000750 'test_future1',
751 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000752}
Guido van Rossum152494a1996-12-20 03:12:20 +0000753
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000754def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000755 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000756 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000757 names = os.listdir(testdir)
758 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000759 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000760 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000761 modname, ext = os.path.splitext(name)
762 if modname[:5] == "test_" and ext == ".py" and modname not in others:
763 tests.append(modname)
764 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000765
Antoine Pitrou09f2e6f2012-07-26 00:45:19 +0200766# We do not use a generator so multiple threads can call next().
767class MultiprocessTests(object):
768
769 """A thread-safe iterator over tests for multiprocess mode."""
770
771 def __init__(self, tests):
772 self.interrupted = False
773 self.lock = threading.Lock()
774 self.tests = tests
775
776 def __iter__(self):
777 return self
778
779 def __next__(self):
780 with self.lock:
781 if self.interrupted:
782 raise StopIteration('tests interrupted')
783 return next(self.tests)
784
Victor Stinnerf58087b2010-05-02 17:24:51 +0000785def replace_stdout():
786 """Set stdout encoder error handler to backslashreplace (as stderr error
787 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000788 if os.name == "nt":
789 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
790 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000791
792 import atexit
793
Victor Stinnerf58087b2010-05-02 17:24:51 +0000794 stdout = sys.stdout
795 sys.stdout = open(stdout.fileno(), 'w',
796 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000797 errors="backslashreplace",
798 closefd=False)
799
800 def restore_stdout():
801 sys.stdout.close()
802 sys.stdout = stdout
803 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000804
Antoine Pitrou88909542009-06-29 13:54:42 +0000805def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100806 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200807 output_on_failure=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000808 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000809
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000810 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000811 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000812 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000813 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000814 huntrleaks -- run multiple times to test for leaks; requires a debug
815 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200816 output_on_failure -- if true, display test output on failure
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000817
818 Returns one of the test result constants:
819 INTERRUPTED KeyboardInterrupt when run under -j
820 RESOURCE_DENIED test skipped because resource denied
821 SKIPPED test skipped for some other reason
822 ENV_CHANGED test failed because it changed the execution environment
823 FAILED test failed
824 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000825 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000826
Antoine Pitrou88909542009-06-29 13:54:42 +0000827 if use_resources is not None:
828 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000829 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200830 support.match_tests = match_tests
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200831 if failfast:
832 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200833 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200834 support.verbose = True
835
836 # Reuse the same instance to all calls to runtest(). Some
837 # tests keep a reference to sys.stdout or sys.stderr
838 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200839 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200840 stream = io.StringIO()
841 runtest.stringio = stream
842 else:
843 stream = runtest.stringio
844 stream.seek(0)
845 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200846
847 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200848 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200849 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200850 sys.stdout = stream
851 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200852 result = runtest_inner(test, verbose, quiet, huntrleaks,
853 debug, display_failure=False)
854 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200855 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200856 orig_stderr.write(output)
857 orig_stderr.flush()
858 finally:
859 sys.stdout = orig_stdout
860 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200861 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200862 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200863 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
864 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100865 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000866 finally:
867 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200868runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000869
Nick Coghlan6ead5522009-10-18 13:19:33 +0000870# Unit tests are supposed to leave the execution environment unchanged
871# once they complete. But sometimes tests have bugs, especially when
872# tests fail, and the changes to environment go on to mess up other
873# tests. This can cause issues with buildbot stability, since tests
874# are run in random order and so problems may appear to come and go.
875# There are a few things we can save and restore to mitigate this, and
876# the following context manager handles this task.
877
878class saved_test_environment:
879 """Save bits of the test environment and restore them at block exit.
880
881 with saved_test_environment(testname, verbose, quiet):
882 #stuff
883
884 Unless quiet is True, a warning is printed to stderr if any of
885 the saved items was changed by the test. The attribute 'changed'
886 is initially False, but is set to True if a change is detected.
887
888 If verbose is more than 1, the before and after state of changed
889 items is also printed.
890 """
891
892 changed = False
893
894 def __init__(self, testname, verbose=0, quiet=False):
895 self.testname = testname
896 self.verbose = verbose
897 self.quiet = quiet
898
899 # To add things to save and restore, add a name XXX to the resources list
900 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
901 # return the value to be saved and compared against a second call to the
902 # get function when test execution completes. restore_XXX should accept
903 # the saved value and restore the resource using it. It will be called if
904 # and only if a change in the value is detected.
905 #
906 # Note: XXX will have any '.' replaced with '_' characters when determining
907 # the corresponding method names.
908
909 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000910 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000911 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti0123e052011-03-20 15:09:26 +0200912 'logging._handlers', 'logging._handlerList',
Éric Araujoff913062011-11-29 16:45:07 +0100913 'shutil.archive_formats', 'shutil.unpack_formats',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200914 'sys.warnoptions', 'threading._dangling',
Vinay Sajipf9596182012-03-02 01:01:13 +0000915 'multiprocessing.process._dangling',
916 'support.TESTFN',
917 )
Nick Coghlan6ead5522009-10-18 13:19:33 +0000918
919 def get_sys_argv(self):
920 return id(sys.argv), sys.argv, sys.argv[:]
921 def restore_sys_argv(self, saved_argv):
922 sys.argv = saved_argv[1]
923 sys.argv[:] = saved_argv[2]
924
925 def get_cwd(self):
926 return os.getcwd()
927 def restore_cwd(self, saved_cwd):
928 os.chdir(saved_cwd)
929
930 def get_sys_stdout(self):
931 return sys.stdout
932 def restore_sys_stdout(self, saved_stdout):
933 sys.stdout = saved_stdout
934
935 def get_sys_stderr(self):
936 return sys.stderr
937 def restore_sys_stderr(self, saved_stderr):
938 sys.stderr = saved_stderr
939
940 def get_sys_stdin(self):
941 return sys.stdin
942 def restore_sys_stdin(self, saved_stdin):
943 sys.stdin = saved_stdin
944
945 def get_os_environ(self):
946 return id(os.environ), os.environ, dict(os.environ)
947 def restore_os_environ(self, saved_environ):
948 os.environ = saved_environ[1]
949 os.environ.clear()
950 os.environ.update(saved_environ[2])
951
952 def get_sys_path(self):
953 return id(sys.path), sys.path, sys.path[:]
954 def restore_sys_path(self, saved_path):
955 sys.path = saved_path[1]
956 sys.path[:] = saved_path[2]
957
Brett Cannon055470a2010-02-19 15:57:10 +0000958 def get_sys_path_hooks(self):
959 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
960 def restore_sys_path_hooks(self, saved_hooks):
961 sys.path_hooks = saved_hooks[1]
962 sys.path_hooks[:] = saved_hooks[2]
963
964 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000965 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000966 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000967 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000968
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000969 def get_warnings_filters(self):
970 return id(warnings.filters), warnings.filters, warnings.filters[:]
971 def restore_warnings_filters(self, saved_filters):
972 warnings.filters = saved_filters[1]
973 warnings.filters[:] = saved_filters[2]
974
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000975 def get_asyncore_socket_map(self):
976 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000977 # XXX Making a copy keeps objects alive until __exit__ gets called.
978 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000979 def restore_asyncore_socket_map(self, saved_map):
980 asyncore = sys.modules.get('asyncore')
981 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000982 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000983 asyncore.socket_map.update(saved_map)
984
Éric Araujoff913062011-11-29 16:45:07 +0100985 def get_shutil_archive_formats(self):
986 # we could call get_archives_formats() but that only returns the
987 # registry keys; we want to check the values too (the functions that
988 # are registered)
989 return shutil._ARCHIVE_FORMATS, shutil._ARCHIVE_FORMATS.copy()
990 def restore_shutil_archive_formats(self, saved):
991 shutil._ARCHIVE_FORMATS = saved[0]
992 shutil._ARCHIVE_FORMATS.clear()
993 shutil._ARCHIVE_FORMATS.update(saved[1])
994
995 def get_shutil_unpack_formats(self):
996 return shutil._UNPACK_FORMATS, shutil._UNPACK_FORMATS.copy()
997 def restore_shutil_unpack_formats(self, saved):
998 shutil._UNPACK_FORMATS = saved[0]
999 shutil._UNPACK_FORMATS.clear()
1000 shutil._UNPACK_FORMATS.update(saved[1])
1001
Nick Coghlan7d819752010-12-05 06:45:03 +00001002 def get_logging__handlers(self):
1003 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001004 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +00001005 def restore_logging__handlers(self, saved_handlers):
1006 # Can't easily revert the logging state
1007 pass
1008
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +00001009 def get_logging__handlerList(self):
1010 # _handlerList is a list of weakrefs to handlers
1011 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
1012 def restore_logging__handlerList(self, saved_handlerList):
1013 # Can't easily revert the logging state
1014 pass
1015
Ezio Melotti0123e052011-03-20 15:09:26 +02001016 def get_sys_warnoptions(self):
1017 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
1018 def restore_sys_warnoptions(self, saved_options):
1019 sys.warnoptions = saved_options[1]
1020 sys.warnoptions[:] = saved_options[2]
1021
Antoine Pitrouc081c0c2011-07-15 22:12:24 +02001022 # Controlling dangling references to Thread objects can make it easier
1023 # to track reference leaks.
1024 def get_threading__dangling(self):
1025 if not threading:
1026 return None
1027 # This copies the weakrefs without making any strong reference
1028 return threading._dangling.copy()
1029 def restore_threading__dangling(self, saved):
1030 if not threading:
1031 return
1032 threading._dangling.clear()
1033 threading._dangling.update(saved)
1034
1035 # Same for Process objects
1036 def get_multiprocessing_process__dangling(self):
1037 if not multiprocessing:
1038 return None
1039 # This copies the weakrefs without making any strong reference
1040 return multiprocessing.process._dangling.copy()
1041 def restore_multiprocessing_process__dangling(self, saved):
1042 if not multiprocessing:
1043 return
1044 multiprocessing.process._dangling.clear()
1045 multiprocessing.process._dangling.update(saved)
1046
Vinay Sajipf9596182012-03-02 01:01:13 +00001047 def get_support_TESTFN(self):
1048 if os.path.isfile(support.TESTFN):
1049 result = 'f'
1050 elif os.path.isdir(support.TESTFN):
1051 result = 'd'
1052 else:
1053 result = None
1054 return result
1055 def restore_support_TESTFN(self, saved_value):
1056 if saved_value is None:
1057 if os.path.isfile(support.TESTFN):
1058 os.unlink(support.TESTFN)
1059 elif os.path.isdir(support.TESTFN):
1060 shutil.rmtree(support.TESTFN)
1061
Nick Coghlan6ead5522009-10-18 13:19:33 +00001062 def resource_info(self):
1063 for name in self.resources:
1064 method_suffix = name.replace('.', '_')
1065 get_name = 'get_' + method_suffix
1066 restore_name = 'restore_' + method_suffix
1067 yield name, getattr(self, get_name), getattr(self, restore_name)
1068
1069 def __enter__(self):
1070 self.saved_values = dict((name, get()) for name, get, restore
1071 in self.resource_info())
1072 return self
1073
1074 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001075 saved_values = self.saved_values
1076 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001077 for name, get, restore in self.resource_info():
1078 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001079 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001080 # Check for changes to the resource's value
1081 if current != original:
1082 self.changed = True
1083 restore(original)
1084 if not self.quiet:
1085 print("Warning -- {} was modified by {}".format(
1086 name, self.testname),
1087 file=sys.stderr)
1088 if self.verbose > 1:
1089 print(" Before: {}\n After: {} ".format(
1090 original, current),
1091 file=sys.stderr)
1092 return False
1093
1094
Victor Stinnera7c33e52011-06-29 13:00:54 +02001095def runtest_inner(test, verbose, quiet,
1096 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001097 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001098
Antoine Pitrou88909542009-06-29 13:54:42 +00001099 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001100 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001101 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001102 if test.startswith('test.'):
1103 abstest = test
1104 else:
1105 # Always import it from the test package
1106 abstest = 'test.' + test
1107 with saved_test_environment(test, verbose, quiet) as environment:
1108 start_time = time.time()
1109 the_package = __import__(abstest, globals(), locals(), [])
1110 the_module = getattr(the_package, test)
1111 # Old tests run to completion simply as a side-effect of
1112 # being imported. For tests based on unittest or doctest,
1113 # explicitly invoke their test_main() function (if it exists).
1114 indirect_test = getattr(the_module, "test_main", None)
1115 if indirect_test is not None:
1116 indirect_test()
1117 if huntrleaks:
1118 refleak = dash_R(the_module, test, indirect_test,
1119 huntrleaks)
1120 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001121 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001122 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001123 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001124 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001125 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001126 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001127 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001128 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001129 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001130 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001131 except KeyboardInterrupt:
1132 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001133 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001134 if display_failure:
1135 print("test", test, "failed --", msg, file=sys.stderr)
1136 else:
1137 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001138 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001139 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001140 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001141 msg = traceback.format_exc()
1142 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001143 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001144 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001145 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001146 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001147 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001148 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001149 return ENV_CHANGED, test_time
1150 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001151
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001152def cleanup_test_droppings(testname, verbose):
1153 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001154 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001155 import gc
1156
1157 # First kill any dangling references to open files etc.
1158 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001159 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001160 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001161
1162 # Try to clean up junk commonly left behind. While tests shouldn't leave
1163 # any files or directories behind, when a test fails that can be tedious
1164 # for it to arrange. The consequences can be especially nasty on Windows,
1165 # since if a test leaves a file open, it cannot be deleted by name (while
1166 # there's nothing we can do about that here either, we can display the
1167 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001168 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001169 "db_home",
1170 ):
1171 if not os.path.exists(name):
1172 continue
1173
1174 if os.path.isdir(name):
1175 kind, nuker = "directory", shutil.rmtree
1176 elif os.path.isfile(name):
1177 kind, nuker = "file", os.unlink
1178 else:
1179 raise SystemError("os.path says %r exists but is neither "
1180 "directory nor file" % name)
1181
1182 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001183 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001184 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001185 # if we have chmod, fix possible permissions problems
1186 # that might prevent cleanup
1187 if (hasattr(os, 'chmod')):
1188 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001189 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001190 except Exception as msg:
1191 print(("%r left behind %s %r and it couldn't be "
1192 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001193
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001194def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001195 """Run a test multiple times, looking for reference leaks.
1196
1197 Returns:
1198 False if the test didn't leak references; True if we detected refleaks.
1199 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001200 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001201 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001202
1203 if not hasattr(sys, 'gettotalrefcount'):
1204 raise Exception("Tracking reference leaks requires a debug build "
1205 "of Python")
1206
1207 # Save current values for dash_R_cleanup() to restore.
1208 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001209 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001210 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001211 try:
1212 import zipimport
1213 except ImportError:
1214 zdc = None # Run unmodified on platforms without zipimport support
1215 else:
1216 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001217 abcs = {}
1218 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001219 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001220 continue
1221 for obj in abc.__subclasses__() + [abc]:
1222 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001223
1224 if indirect_test:
1225 def run_the_test():
1226 indirect_test()
1227 else:
1228 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001229 del sys.modules[the_module.__name__]
1230 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001231
1232 deltas = []
1233 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001234 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001235 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001236 print("beginning", repcount, "repetitions", file=sys.stderr)
1237 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001238 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001239 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001240 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001241 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001242 run_the_test()
1243 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001244 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001245 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001246 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001247 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001248 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001249 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001250 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001251 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001252 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001253 sys.stderr.flush()
1254 with open(fname, "a") as refrep:
1255 print(msg, file=refrep)
1256 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001257 return True
1258 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001259
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001260def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001261 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001262 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001263 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001264 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001265 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001266 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001267
Christian Heimesdae2a892008-04-19 00:55:37 +00001268 # Clear the warnings registry, so they can be displayed again
1269 for mod in sys.modules.values():
1270 if hasattr(mod, '__warningregistry__'):
1271 del mod.__warningregistry__
1272
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001273 # Restore some original values.
1274 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001275 copyreg.dispatch_table.clear()
1276 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001277 sys.path_importer_cache.clear()
1278 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001279 try:
1280 import zipimport
1281 except ImportError:
1282 pass # Run unmodified on platforms without zipimport support
1283 else:
1284 zipimport._zip_directory_cache.clear()
1285 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001286
Christian Heimes26855632008-01-27 23:50:43 +00001287 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001288 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001289
Guido van Rossum3de862d2007-08-18 00:10:33 +00001290 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001291 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001292 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001293 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001294 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001295 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001296 obj._abc_cache.clear()
1297 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001298
Antoine Pitrou046467c2009-10-30 18:30:35 +00001299 # Flush standard output, so that buffered data is sent to the OS and
1300 # associated Python objects are reclaimed.
1301 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1302 if stream is not None:
1303 stream.flush()
1304
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001305 # Clear assorted module caches.
1306 _path_created.clear()
1307 re.purge()
1308 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001309 urllib.parse.clear_cache()
1310 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001311 linecache.clearcache()
1312 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001313 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001314 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001315 doctest.master = None
Meador Inge11e38132011-11-25 22:33:32 -06001316 try:
1317 import ctypes
1318 except ImportError:
1319 # Don't worry about resetting the cache if ctypes is not supported
1320 pass
1321 else:
1322 ctypes._reset_cache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001323
1324 # Collect cyclic trash.
1325 gc.collect()
1326
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001327def warm_char_cache():
1328 s = bytes(range(256))
1329 for i in range(256):
1330 s[i:i+1]
1331
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001332def findtestdir(path=None):
1333 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001334
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001335def removepy(names):
1336 if not names:
1337 return
1338 for idx, name in enumerate(names):
1339 basename, ext = os.path.splitext(name)
1340 if ext == '.py':
1341 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001342
Guido van Rossum152494a1996-12-20 03:12:20 +00001343def count(n, word):
1344 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001345 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001346 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001347 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001348
Tim Petersa45da922001-08-12 03:45:50 +00001349def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001350 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001351
1352 Optional arg width (default 70) is the maximum line length.
1353 Optional arg indent (default 4) is the number of blanks with which to
1354 begin each line.
1355 """
1356
Tim Petersba78bc42002-07-04 19:45:06 +00001357 from textwrap import fill
1358 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001359 # Print the sorted list: 'x' may be a '--random' list or a set()
1360 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001361 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001362
Tim Petersde14a302002-04-01 05:04:46 +00001363# Map sys.platform to a string containing the basenames of tests
1364# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001365#
1366# Special cases:
1367# test_pep277
1368# The _ExpectedSkips constructor adds this to the set of expected
1369# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001370# test_timeout
1371# Controlled by test_timeout.skip_expected. Requires the network
1372# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001373#
1374# Tests that are expected to be skipped everywhere except on one platform
1375# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001376
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001377_expectations = {
1378 'win32':
1379 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001380 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001381 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001382 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001383 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001384 test_fcntl
1385 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001386 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001387 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001388 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001389 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001390 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001391 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001392 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001393 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001394 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001395 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001396 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001397 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001398 test_pty
1399 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001400 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001401 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001402 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001403 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001404 test_wait3
1405 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001406 """,
1407 'linux2':
1408 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001409 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001410 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001411 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001412 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001413 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001414 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001415 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001416 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001417 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001418 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001419 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001420 test_openpty
1421 test_pyexpat
1422 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001423 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001424 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001425 'openunix8':
1426 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001427 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001428 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001429 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001430 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001431 test_openpty
1432 test_pyexpat
1433 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001434 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001435 """,
1436 'sco_sv3':
1437 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001438 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001439 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001440 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001441 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001442 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001443 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001444 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001445 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001446 test_openpty
1447 test_pyexpat
1448 test_queue
1449 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001450 test_sundry
1451 test_thread
1452 test_threaded_import
1453 test_threadedtempfile
1454 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001455 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001456 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001457 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001458 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001459 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001460 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001461 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001462 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001463 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001464 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001465 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001466 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001467 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001468 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001469 'sunos5':
1470 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001471 test_curses
1472 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001473 test_epoll
1474 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001475 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001476 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001477 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001478 test_zipfile
1479 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001480 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001481 'hp-ux11':
1482 """
Skip Montanarob3230212002-03-15 02:54:03 +00001483 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001484 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001485 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001486 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001487 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001488 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001489 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001490 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001491 test_openpty
1492 test_pyexpat
1493 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001494 test_zipfile
1495 test_zlib
1496 """,
Jason Tishler25115942002-12-05 15:18:15 +00001497 'cygwin':
1498 """
Jason Tishler25115942002-12-05 15:18:15 +00001499 test_curses
1500 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001501 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001502 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001503 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001504 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001505 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001506 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001507 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001508 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001509 'os2emx':
1510 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001511 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001512 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001513 test_epoll
1514 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001515 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001516 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001517 test_openpty
1518 test_ossaudiodev
1519 test_pty
1520 test_resource
1521 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001522 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001523 'freebsd4':
1524 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001525 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001526 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001527 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001528 test_ossaudiodev
1529 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001530 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001531 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001532 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001533 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001534 test_ttk_guionly
1535 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001536 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001537 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001538 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001539 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001540 'aix5':
1541 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001542 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001543 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001544 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001545 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001546 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001547 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001548 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001549 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001550 test_ttk_guionly
1551 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001552 test_zipimport
1553 test_zlib
1554 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001555 'openbsd3':
1556 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001557 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001558 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001559 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001560 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001561 test_normalization
1562 test_ossaudiodev
1563 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001564 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001565 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001566 test_ttk_guionly
1567 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001568 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001569 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001570 'netbsd3':
1571 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001572 test_ctypes
1573 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001574 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001575 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001576 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001577 test_ossaudiodev
1578 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001579 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001580 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001581 test_ttk_guionly
1582 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001583 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001584 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001585}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001586_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001587_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001588_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001589_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001590
Tim Petersb5b7b782001-08-12 01:20:39 +00001591class _ExpectedSkips:
1592 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001593 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001594 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001595
Tim Peters7c7efe92002-08-23 17:55:54 +00001596 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001597 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001598 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001599 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001600
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001601 # These are broken tests, for now skipped on every platform.
1602 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001603 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001604
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001605 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001606 if not os.path.supports_unicode_filenames:
1607 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001608
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001609 # doctest, profile and cProfile tests fail when the codec for the
1610 # fs encoding isn't built in because PyUnicode_Decode() adds two
1611 # calls into Python.
1612 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1613 if sys.getfilesystemencoding().lower() not in encs:
1614 self.expected.add('test_profile')
1615 self.expected.add('test_cProfile')
1616 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001617
Neal Norwitz55b61d22003-02-28 19:57:03 +00001618 if test_timeout.skip_expected:
1619 self.expected.add('test_timeout')
1620
Tim Petersecd79eb2003-01-29 00:35:32 +00001621 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001622 # test_sqlite is only reliable on Windows where the library
1623 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001624 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001625 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001626 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001627 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001628
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001629 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001630 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001631
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001632 if support.python_is_optimized():
1633 self.expected.add("test_gdb")
1634
Tim Peters7c7efe92002-08-23 17:55:54 +00001635 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001636
1637 def isvalid(self):
1638 "Return true iff _ExpectedSkips knows about the current platform."
1639 return self.valid
1640
1641 def getexpected(self):
1642 """Return set of test names we expect to skip on current platform.
1643
1644 self.isvalid() must be true.
1645 """
1646
1647 assert self.isvalid()
1648 return self.expected
1649
Michael Foord3ab34cc2010-12-03 12:27:40 +00001650def _make_temp_dir_for_build(TEMPDIR):
1651 # When tests are run from the Python build directory, it is best practice
1652 # to keep the test files in a subfolder. It eases the cleanup of leftover
1653 # files using command "make distclean".
1654 if sysconfig.is_python_build():
1655 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1656 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001657 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001658 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001659 except OSError as e:
1660 if e.errno != errno.EEXIST:
1661 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001662
1663 # Define a writable temp dir that will be used as cwd while running
1664 # the tests. The name of the dir includes the pid to allow parallel
1665 # testing (see the -j option).
1666 TESTCWD = 'test_python_{}'.format(os.getpid())
1667
1668 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1669 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001670
Guido van Rossum152494a1996-12-20 03:12:20 +00001671if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001672 # Remove regrtest.py's own directory from the module search path. Despite
1673 # the elimination of implicit relative imports, this is still needed to
1674 # ensure that submodules of the test package do not inappropriately appear
1675 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1676 # directly instead of using the -m switch
1677 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1678 i = len(sys.path)
1679 while i >= 0:
1680 i -= 1
1681 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1682 del sys.path[i]
1683
Florent Xiclunadc69e722010-09-13 16:35:02 +00001684 # findtestdir() gets the dirname out of __file__, so we have to make it
1685 # absolute before changing the working directory.
1686 # For example __file__ may be relative when running trace or profile.
1687 # See issue #9323.
1688 __file__ = os.path.abspath(__file__)
1689
1690 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001691 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001692
Michael Foord3ab34cc2010-12-03 12:27:40 +00001693 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001694
1695 # Run the tests in a context manager that temporary changes the CWD to a
1696 # temporary and writable directory. If it's not possible to create or
1697 # change the CWD, the original CWD will be used. The original CWD is
1698 # available from support.SAVEDCWD.
1699 with support.temp_cwd(TESTCWD, quiet=True):
1700 main()