blob: 10b515b5ab2d0bc53f40f7e583acef8de24e3806 [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
Georg Brandl96f78972009-10-27 21:02:11 +000031-S/--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
Christian Heimesb186d002008-03-18 15:15:01 +0000168import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000169import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200170import tempfile
171import time
172import traceback
173import unittest
174import warnings
175from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000176
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200177try:
178 import threading
179except ImportError:
180 threading = None
181try:
182 import multiprocessing.process
183except ImportError:
184 multiprocessing = None
185
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000186
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000187# Some times __path__ and __file__ are not absolute (e.g. while running from
188# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
189# imports might fail. This affects only the modules imported before os.chdir().
190# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
191# they are found in the CWD their __file__ and __path__ will be relative (this
192# happens before the chdir). All the modules imported after the chdir, are
193# not found in the CWD, and since the other paths in sys.path[1:] are absolute
194# (site.py absolutize them), the __file__ and __path__ will be absolute too.
195# Therefore it is necessary to absolutize manually the __file__ and __path__ of
196# the packages to prevent later imports to fail when the CWD is different.
197for module in sys.modules.values():
198 if hasattr(module, '__path__'):
199 module.__path__ = [os.path.abspath(path) for path in module.__path__]
200 if hasattr(module, '__file__'):
201 module.__file__ = os.path.abspath(module.__file__)
202
Guido van Rossumdc15c272002-08-12 21:55:51 +0000203
Guido van Rossumbb484652002-12-02 09:56:21 +0000204# MacOSX (a.k.a. Darwin) has a default stack size that is too small
205# for deeply recursive regular expressions. We see this as crashes in
206# the Python test suite when running test_re.py and test_sre.py. The
207# fix is to set the stack limit to 2048.
208# This approach may also be useful for other Unixy platforms that
209# suffer from small default stack limits.
210if sys.platform == 'darwin':
211 try:
212 import resource
213 except ImportError:
214 pass
215 else:
216 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
217 newsoft = min(hard, max(soft, 1024*2048))
218 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
219
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000220# Test result constants.
221PASSED = 1
222FAILED = 0
223ENV_CHANGED = -1
224SKIPPED = -2
225RESOURCE_DENIED = -3
226INTERRUPTED = -4
227
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000228from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000229
Georg Brandl1158a332009-06-04 09:30:30 +0000230RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000231 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000232
Florent Xiclunaec882212010-08-09 16:56:43 +0000233TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000234
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000235def usage(msg):
236 print(msg, file=sys.stderr)
237 print("Use --help for usage", file=sys.stderr)
238 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000239
240
Antoine Pitrou88909542009-06-29 13:54:42 +0000241def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000242 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000243 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000244 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000245 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200246 header=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000247 """Execute a test suite.
248
Thomas Wouters7e474022000-07-16 12:04:32 +0000249 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000250 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000251
252 tests -- a list of strings containing test names (optional)
253 testdir -- the directory in which to look for tests (optional)
254
255 Users other than the Python test suite will certainly want to
256 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000257 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000258
259 If the tests argument is omitted, the tests listed on the
260 command-line will be used. If that's empty, too, then all *.py
261 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000262
Antoine Pitrou88909542009-06-29 13:54:42 +0000263 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000264 single, randomize, findleaks, use_resources, trace, coverdir,
265 print_slow, and random_seed) allow programmers calling main()
266 directly to set the values that would normally be set by flags
267 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000268 """
Fred Drake004d5e62000-10-23 17:22:08 +0000269
Victor Stinner1802d3f2010-05-19 17:11:19 +0000270 replace_stdout()
271
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000272 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000273 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200274 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:Gm:',
Georg Brandl96f78972009-10-27 21:02:11 +0000275 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
276 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
277 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000278 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000279 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200280 'start=', 'nowindows', 'header', 'failfast', 'match'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000281 except getopt.error as msg:
282 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000283
284 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000285 if random_seed is None:
286 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000287 if use_resources is None:
288 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000289 debug = False
290 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000291 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000292 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000293 print(__doc__)
294 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000295 elif o in ('-v', '--verbose'):
296 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000297 elif o in ('-w', '--verbose2'):
298 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000299 elif o in ('-d', '--debug'):
300 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000301 elif o in ('-W', '--verbose3'):
302 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200303 elif o in ('-G', '--failfast'):
304 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000305 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000306 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000307 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000308 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000309 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000310 elif o in ('-S', '--start'):
311 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000312 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000313 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000314 elif o in ('-S', '--slow'):
315 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000316 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000317 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000318 elif o == '--randseed':
319 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000320 elif o in ('-f', '--fromfile'):
321 fromfile = a
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200322 elif o in ('-m', '--match'):
323 match_tests = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000324 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000325 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000326 elif o in ('-L', '--runleaks'):
327 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000328 elif o in ('-t', '--threshold'):
329 import gc
330 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000331 elif o in ('-T', '--coverage'):
332 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000333 elif o in ('-D', '--coverdir'):
334 coverdir = os.path.join(os.getcwd(), a)
335 elif o in ('-N', '--nocoverdir'):
336 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000337 elif o in ('-R', '--huntrleaks'):
338 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000339 if len(huntrleaks) not in (2, 3):
340 print(a, huntrleaks)
341 usage('-R takes 2 or 3 colon-separated arguments')
342 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000343 huntrleaks[0] = 5
344 else:
345 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000346 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000347 huntrleaks[1] = 4
348 else:
349 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000350 if len(huntrleaks) == 2 or not huntrleaks[2]:
351 huntrleaks[2:] = ["reflog.txt"]
352 # Avoid false positives due to the character cache in
353 # stringobject.c filling slowly with random data
354 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000355 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000356 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000357 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000358 u = [x.lower() for x in a.split(',')]
359 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000360 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000361 use_resources[:] = RESOURCE_NAMES
362 continue
363 remove = False
364 if r[0] == '-':
365 remove = True
366 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000367 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000368 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000369 if remove:
370 if r in use_resources:
371 use_resources.remove(r)
372 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000373 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000374 elif o in ('-n', '--nowindows'):
375 import msvcrt
376 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
377 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
378 msvcrt.SEM_NOGPFAULTERRORBOX|
379 msvcrt.SEM_NOOPENFILEERRORBOX)
380 try:
381 msvcrt.CrtSetReportMode
382 except AttributeError:
383 # release build
384 pass
385 else:
386 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
387 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
388 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000389 elif o in ('-F', '--forever'):
390 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000391 elif o in ('-j', '--multiprocess'):
392 use_mp = int(a)
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000393 elif o == '--header':
394 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000395 elif o == '--slaveargs':
396 args, kwargs = json.loads(a)
397 try:
398 result = runtest(*args, **kwargs)
399 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000400 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000401 sys.stdout.flush()
402 print() # Force a newline (just in case)
403 print(json.dumps(result))
404 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000405 else:
406 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000407 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000408 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000409 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000410 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000411 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000412 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000413 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000414 usage("-l and -j don't go together!")
Nadeem Vawdac1fba3e2011-08-27 15:22:05 +0200415 if use_mp and support.max_memuse:
416 usage("-M and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200417 if failfast and not (verbose or verbose3):
418 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000419
Guido van Rossum152494a1996-12-20 03:12:20 +0000420 good = []
421 bad = []
422 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000423 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000424 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000425 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000426
Neil Schemenauerd569f232000-09-22 15:29:28 +0000427 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000428 try:
429 import gc
430 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000431 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000432 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000433 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000434 # Uncomment the line below to report garbage that is not
435 # freeable by reference counting alone. By default only
436 # garbage that is not collectable by the GC is reported.
437 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000438 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000439
Barry Warsawe11e3de1999-01-28 19:51:51 +0000440 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000441 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000442 try:
443 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000444 next_test = fp.read().strip()
445 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000446 fp.close()
447 except IOError:
448 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000449
450 if fromfile:
451 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000452 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000453 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000454 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000455 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000456 guts = line.split() # assuming no test has whitespace in its name
457 if guts and not guts[0].startswith('#'):
458 tests.extend(guts)
459 fp.close()
460
461 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000462 removepy(args)
463 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000464
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000465 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000466 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000467 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000468 for arg in args:
469 if arg in stdtests:
470 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000471 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000472 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000473
474 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000475 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000476 # Print basic platform information
477 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000478 print("== ", platform.platform(aliased=True),
479 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000480 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000481 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000482
R. David Murrayef1992b2009-12-16 15:19:27 +0000483 alltests = findtests(testdir, stdtests, nottests)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000484 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000485 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000486 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000487 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000488 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000489 except IndexError:
490 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000491 # Remove all the tests that precede start if it's set.
492 if start:
493 try:
494 del tests[:tests.index(start)]
495 except ValueError:
496 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000497 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000498 random.seed(random_seed)
499 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000500 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000501 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000502 import trace, tempfile
503 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
504 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000505 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000506
Christian Heimesb186d002008-03-18 15:15:01 +0000507 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000508 support.verbose = verbose # Tell tests to be moderately quiet
509 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000510 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000511
512 def accumulate_result(test, result):
513 ok, test_time = result
514 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000515 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000516 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000517 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000518 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000519 elif ok == ENV_CHANGED:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000520 environment_changed.append(test)
521 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000522 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000523 elif ok == RESOURCE_DENIED:
524 skipped.append(test)
525 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000526
R. David Murray7dc72cc2009-11-14 16:13:02 +0000527 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000528 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000529 while True:
530 for test in tests:
531 yield test
532 if bad:
533 return
534 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000535 test_count = ''
536 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000537 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000538 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000539 test_count = '/{}'.format(len(selected))
540 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000541
Antoine Pitrou88909542009-06-29 13:54:42 +0000542 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000543 try:
544 from threading import Thread
545 except ImportError:
546 print("Multiprocess option requires thread support")
547 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000548 from queue import Queue
549 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000550 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000551 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000552 def tests_and_args():
553 for test in tests:
554 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000555 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000556 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200557 debug=debug, output_on_failure=verbose3,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200558 failfast=failfast, match_tests=match_tests)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000559 )
560 yield (test, args_tuple)
561 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000562 opt_args = support.args_from_interpreter_flags()
563 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000564 def work():
565 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000566 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000567 while True:
568 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000569 test, args_tuple = next(pending)
570 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000571 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000572 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000573 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000574 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000575 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000576 universal_newlines=True,
577 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000578 stdout, stderr = popen.communicate()
579 # Strip last refcount output line if it exists, since it
580 # comes from the shutdown of the interpreter in the subcommand.
581 stderr = debug_output_pat.sub("", stderr)
582 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000583 if not result:
584 output.put((None, None, None, None))
585 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000586 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000587 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000588 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000589 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000590 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000591 workers = [Thread(target=work) for i in range(use_mp)]
592 for worker in workers:
593 worker.start()
594 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000595 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000596 try:
597 while finished < use_mp:
598 test, stdout, stderr, result = output.get()
599 if test is None:
600 finished += 1
601 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200602 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000603 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300604 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
605 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200606 test_count_width, test_index, test_count,
607 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000608 if stdout:
609 print(stdout)
610 if stderr:
611 print(stderr, file=sys.stderr)
612 if result[0] == INTERRUPTED:
613 assert result[1] == 'KeyboardInterrupt'
614 raise KeyboardInterrupt # What else?
Georg Brandldee7b852010-08-02 18:59:52 +0000615 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000616 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000617 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000618 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000619 for worker in workers:
620 worker.join()
621 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000622 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000623 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300624 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
625 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200626 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000627 sys.stdout.flush()
628 if trace:
629 # If we're tracing code coverage, then we don't exit with status
630 # if on a false return value from main.
R. David Murray8e286c42010-12-27 20:09:32 +0000631 tracer.runctx('runtest(test, verbose, quiet)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000632 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000633 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000634 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100635 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200636 output_on_failure=verbose3,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200637 failfast=failfast, match_tests=match_tests)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000638 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000639 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000640 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000641 break
642 except:
643 raise
644 if findleaks:
645 gc.collect()
646 if gc.garbage:
647 print("Warning: test created", len(gc.garbage), end=' ')
648 print("uncollectable object(s).")
649 # move the uncollectable objects somewhere so we don't see
650 # them again
651 found_garbage.extend(gc.garbage)
652 del gc.garbage[:]
653 # Unload the newly imported modules (best effort finalization)
654 for module in sys.modules.keys():
655 if module not in save_modules and module.startswith("test."):
656 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000657
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000658 if interrupted:
659 # print a newline after ^C
660 print()
661 print("Test suite interrupted by signal SIGINT.")
662 omitted = set(selected) - set(good) - set(bad) - set(skipped)
663 print(count(len(omitted), "test"), "omitted:")
664 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000665 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000666 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000667 print("All", end=' ')
668 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000669 if print_slow:
670 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000671 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000672 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000673 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000674 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000675 bad = sorted(set(bad) - set(environment_changed))
676 if bad:
677 print(count(len(bad), "test"), "failed:")
678 printlist(bad)
679 if environment_changed:
680 print("{} altered the execution environment:".format(
681 count(len(environment_changed), "test")))
682 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000683 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000684 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000685 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000686
Tim Petersb5b7b782001-08-12 01:20:39 +0000687 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000688 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000689 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000690 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000691 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000692 print(count(len(surprise), "skip"), \
693 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000694 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000695 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000696 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000697 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000698 print("Ask someone to teach regrtest.py about which tests are")
699 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000700
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000701 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000702 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000703 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000704 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000705 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000706 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000707 verbose = True
R. David Murray8e286c42010-12-27 20:09:32 +0000708 ok = runtest(test, True, quiet, huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000709 except KeyboardInterrupt:
710 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000711 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000712 break
713 except:
714 raise
715
Barry Warsawe11e3de1999-01-28 19:51:51 +0000716 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000717 if next_single_test:
718 with open(filename, 'w') as fp:
719 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000720 else:
721 os.unlink(filename)
722
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000723 if trace:
724 r = tracer.results()
725 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
726
Skip Montanaro0179a182004-06-06 15:53:18 +0000727 if runleaks:
728 os.system("leaks %d" % os.getpid())
729
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000730 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000731
Guido van Rossum152494a1996-12-20 03:12:20 +0000732
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000733STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000734 'test_grammar',
735 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000736 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000737 'test_builtin',
738 'test_exceptions',
739 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000740 'test_unittest',
741 'test_doctest',
742 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000743]
Guido van Rossum152494a1996-12-20 03:12:20 +0000744
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000745NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000746 'test_future1',
747 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000748}
Guido van Rossum152494a1996-12-20 03:12:20 +0000749
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000750def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000751 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000752 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000753 names = os.listdir(testdir)
754 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000755 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000756 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000757 modname, ext = os.path.splitext(name)
758 if modname[:5] == "test_" and ext == ".py" and modname not in others:
759 tests.append(modname)
760 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000761
Victor Stinnerf58087b2010-05-02 17:24:51 +0000762def replace_stdout():
763 """Set stdout encoder error handler to backslashreplace (as stderr error
764 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000765 if os.name == "nt":
766 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
767 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000768
769 import atexit
770
Victor Stinnerf58087b2010-05-02 17:24:51 +0000771 stdout = sys.stdout
772 sys.stdout = open(stdout.fileno(), 'w',
773 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000774 errors="backslashreplace",
775 closefd=False)
776
777 def restore_stdout():
778 sys.stdout.close()
779 sys.stdout = stdout
780 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000781
Antoine Pitrou88909542009-06-29 13:54:42 +0000782def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100783 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200784 output_on_failure=False, failfast=False, match_tests=None):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000785 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000786
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000787 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000788 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000789 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000790 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000791 huntrleaks -- run multiple times to test for leaks; requires a debug
792 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200793 output_on_failure -- if true, display test output on failure
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000794
795 Returns one of the test result constants:
796 INTERRUPTED KeyboardInterrupt when run under -j
797 RESOURCE_DENIED test skipped because resource denied
798 SKIPPED test skipped for some other reason
799 ENV_CHANGED test failed because it changed the execution environment
800 FAILED test failed
801 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000802 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000803
Antoine Pitrou88909542009-06-29 13:54:42 +0000804 if use_resources is not None:
805 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000806 try:
Antoine Pitroub9c73e82011-07-29 23:53:38 +0200807 support.match_tests = match_tests
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200808 if failfast:
809 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200810 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200811 support.verbose = True
812
813 # Reuse the same instance to all calls to runtest(). Some
814 # tests keep a reference to sys.stdout or sys.stderr
815 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200816 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200817 stream = io.StringIO()
818 runtest.stringio = stream
819 else:
820 stream = runtest.stringio
821 stream.seek(0)
822 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200823
824 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200825 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200826 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200827 sys.stdout = stream
828 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200829 result = runtest_inner(test, verbose, quiet, huntrleaks,
830 debug, display_failure=False)
831 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200832 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200833 orig_stderr.write(output)
834 orig_stderr.flush()
835 finally:
836 sys.stdout = orig_stdout
837 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200838 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200839 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200840 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
841 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100842 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000843 finally:
844 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200845runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000846
Nick Coghlan6ead5522009-10-18 13:19:33 +0000847# Unit tests are supposed to leave the execution environment unchanged
848# once they complete. But sometimes tests have bugs, especially when
849# tests fail, and the changes to environment go on to mess up other
850# tests. This can cause issues with buildbot stability, since tests
851# are run in random order and so problems may appear to come and go.
852# There are a few things we can save and restore to mitigate this, and
853# the following context manager handles this task.
854
855class saved_test_environment:
856 """Save bits of the test environment and restore them at block exit.
857
858 with saved_test_environment(testname, verbose, quiet):
859 #stuff
860
861 Unless quiet is True, a warning is printed to stderr if any of
862 the saved items was changed by the test. The attribute 'changed'
863 is initially False, but is set to True if a change is detected.
864
865 If verbose is more than 1, the before and after state of changed
866 items is also printed.
867 """
868
869 changed = False
870
871 def __init__(self, testname, verbose=0, quiet=False):
872 self.testname = testname
873 self.verbose = verbose
874 self.quiet = quiet
875
876 # To add things to save and restore, add a name XXX to the resources list
877 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
878 # return the value to be saved and compared against a second call to the
879 # get function when test execution completes. restore_XXX should accept
880 # the saved value and restore the resource using it. It will be called if
881 # and only if a change in the value is detected.
882 #
883 # Note: XXX will have any '.' replaced with '_' characters when determining
884 # the corresponding method names.
885
886 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000887 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000888 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti0123e052011-03-20 15:09:26 +0200889 'logging._handlers', 'logging._handlerList',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200890 'sys.warnoptions', 'threading._dangling',
891 'multiprocessing.process._dangling')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000892
893 def get_sys_argv(self):
894 return id(sys.argv), sys.argv, sys.argv[:]
895 def restore_sys_argv(self, saved_argv):
896 sys.argv = saved_argv[1]
897 sys.argv[:] = saved_argv[2]
898
899 def get_cwd(self):
900 return os.getcwd()
901 def restore_cwd(self, saved_cwd):
902 os.chdir(saved_cwd)
903
904 def get_sys_stdout(self):
905 return sys.stdout
906 def restore_sys_stdout(self, saved_stdout):
907 sys.stdout = saved_stdout
908
909 def get_sys_stderr(self):
910 return sys.stderr
911 def restore_sys_stderr(self, saved_stderr):
912 sys.stderr = saved_stderr
913
914 def get_sys_stdin(self):
915 return sys.stdin
916 def restore_sys_stdin(self, saved_stdin):
917 sys.stdin = saved_stdin
918
919 def get_os_environ(self):
920 return id(os.environ), os.environ, dict(os.environ)
921 def restore_os_environ(self, saved_environ):
922 os.environ = saved_environ[1]
923 os.environ.clear()
924 os.environ.update(saved_environ[2])
925
926 def get_sys_path(self):
927 return id(sys.path), sys.path, sys.path[:]
928 def restore_sys_path(self, saved_path):
929 sys.path = saved_path[1]
930 sys.path[:] = saved_path[2]
931
Brett Cannon055470a2010-02-19 15:57:10 +0000932 def get_sys_path_hooks(self):
933 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
934 def restore_sys_path_hooks(self, saved_hooks):
935 sys.path_hooks = saved_hooks[1]
936 sys.path_hooks[:] = saved_hooks[2]
937
938 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000939 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000940 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000941 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000942
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000943 def get_warnings_filters(self):
944 return id(warnings.filters), warnings.filters, warnings.filters[:]
945 def restore_warnings_filters(self, saved_filters):
946 warnings.filters = saved_filters[1]
947 warnings.filters[:] = saved_filters[2]
948
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000949 def get_asyncore_socket_map(self):
950 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000951 # XXX Making a copy keeps objects alive until __exit__ gets called.
952 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000953 def restore_asyncore_socket_map(self, saved_map):
954 asyncore = sys.modules.get('asyncore')
955 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000956 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000957 asyncore.socket_map.update(saved_map)
958
Nick Coghlan7d819752010-12-05 06:45:03 +0000959 def get_logging__handlers(self):
960 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000961 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +0000962 def restore_logging__handlers(self, saved_handlers):
963 # Can't easily revert the logging state
964 pass
965
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000966 def get_logging__handlerList(self):
967 # _handlerList is a list of weakrefs to handlers
968 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
969 def restore_logging__handlerList(self, saved_handlerList):
970 # Can't easily revert the logging state
971 pass
972
Ezio Melotti0123e052011-03-20 15:09:26 +0200973 def get_sys_warnoptions(self):
974 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
975 def restore_sys_warnoptions(self, saved_options):
976 sys.warnoptions = saved_options[1]
977 sys.warnoptions[:] = saved_options[2]
978
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200979 # Controlling dangling references to Thread objects can make it easier
980 # to track reference leaks.
981 def get_threading__dangling(self):
982 if not threading:
983 return None
984 # This copies the weakrefs without making any strong reference
985 return threading._dangling.copy()
986 def restore_threading__dangling(self, saved):
987 if not threading:
988 return
989 threading._dangling.clear()
990 threading._dangling.update(saved)
991
992 # Same for Process objects
993 def get_multiprocessing_process__dangling(self):
994 if not multiprocessing:
995 return None
996 # This copies the weakrefs without making any strong reference
997 return multiprocessing.process._dangling.copy()
998 def restore_multiprocessing_process__dangling(self, saved):
999 if not multiprocessing:
1000 return
1001 multiprocessing.process._dangling.clear()
1002 multiprocessing.process._dangling.update(saved)
1003
Nick Coghlan6ead5522009-10-18 13:19:33 +00001004 def resource_info(self):
1005 for name in self.resources:
1006 method_suffix = name.replace('.', '_')
1007 get_name = 'get_' + method_suffix
1008 restore_name = 'restore_' + method_suffix
1009 yield name, getattr(self, get_name), getattr(self, restore_name)
1010
1011 def __enter__(self):
1012 self.saved_values = dict((name, get()) for name, get, restore
1013 in self.resource_info())
1014 return self
1015
1016 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001017 saved_values = self.saved_values
1018 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001019 for name, get, restore in self.resource_info():
1020 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001021 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001022 # Check for changes to the resource's value
1023 if current != original:
1024 self.changed = True
1025 restore(original)
1026 if not self.quiet:
1027 print("Warning -- {} was modified by {}".format(
1028 name, self.testname),
1029 file=sys.stderr)
1030 if self.verbose > 1:
1031 print(" Before: {}\n After: {} ".format(
1032 original, current),
1033 file=sys.stderr)
1034 return False
1035
1036
Victor Stinnera7c33e52011-06-29 13:00:54 +02001037def runtest_inner(test, verbose, quiet,
1038 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001039 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001040
Antoine Pitrou88909542009-06-29 13:54:42 +00001041 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001042 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001043 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001044 if test.startswith('test.'):
1045 abstest = test
1046 else:
1047 # Always import it from the test package
1048 abstest = 'test.' + test
1049 with saved_test_environment(test, verbose, quiet) as environment:
1050 start_time = time.time()
1051 the_package = __import__(abstest, globals(), locals(), [])
1052 the_module = getattr(the_package, test)
1053 # Old tests run to completion simply as a side-effect of
1054 # being imported. For tests based on unittest or doctest,
1055 # explicitly invoke their test_main() function (if it exists).
1056 indirect_test = getattr(the_module, "test_main", None)
1057 if indirect_test is not None:
1058 indirect_test()
1059 if huntrleaks:
1060 refleak = dash_R(the_module, test, indirect_test,
1061 huntrleaks)
1062 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001063 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001064 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001065 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001066 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001067 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001068 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001069 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001070 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001071 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001072 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001073 except KeyboardInterrupt:
1074 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001075 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001076 if display_failure:
1077 print("test", test, "failed --", msg, file=sys.stderr)
1078 else:
1079 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001080 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001081 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001082 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001083 msg = traceback.format_exc()
1084 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001085 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001086 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001087 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001088 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001089 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001090 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001091 return ENV_CHANGED, test_time
1092 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001093
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001094def cleanup_test_droppings(testname, verbose):
1095 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001096 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001097 import gc
1098
1099 # First kill any dangling references to open files etc.
1100 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001101 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001102 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001103
1104 # Try to clean up junk commonly left behind. While tests shouldn't leave
1105 # any files or directories behind, when a test fails that can be tedious
1106 # for it to arrange. The consequences can be especially nasty on Windows,
1107 # since if a test leaves a file open, it cannot be deleted by name (while
1108 # there's nothing we can do about that here either, we can display the
1109 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001110 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001111 "db_home",
1112 ):
1113 if not os.path.exists(name):
1114 continue
1115
1116 if os.path.isdir(name):
1117 kind, nuker = "directory", shutil.rmtree
1118 elif os.path.isfile(name):
1119 kind, nuker = "file", os.unlink
1120 else:
1121 raise SystemError("os.path says %r exists but is neither "
1122 "directory nor file" % name)
1123
1124 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001125 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001127 # if we have chmod, fix possible permissions problems
1128 # that might prevent cleanup
1129 if (hasattr(os, 'chmod')):
1130 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001131 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001132 except Exception as msg:
1133 print(("%r left behind %s %r and it couldn't be "
1134 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001135
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001136def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001137 """Run a test multiple times, looking for reference leaks.
1138
1139 Returns:
1140 False if the test didn't leak references; True if we detected refleaks.
1141 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001142 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001143 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001144
1145 if not hasattr(sys, 'gettotalrefcount'):
1146 raise Exception("Tracking reference leaks requires a debug build "
1147 "of Python")
1148
1149 # Save current values for dash_R_cleanup() to restore.
1150 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001151 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001152 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001153 try:
1154 import zipimport
1155 except ImportError:
1156 zdc = None # Run unmodified on platforms without zipimport support
1157 else:
1158 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001159 abcs = {}
1160 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001161 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001162 continue
1163 for obj in abc.__subclasses__() + [abc]:
1164 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001165
1166 if indirect_test:
1167 def run_the_test():
1168 indirect_test()
1169 else:
1170 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001171 del sys.modules[the_module.__name__]
1172 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001173
1174 deltas = []
1175 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001176 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001177 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001178 print("beginning", repcount, "repetitions", file=sys.stderr)
1179 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001180 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001181 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001182 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001183 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001184 run_the_test()
1185 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001186 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001187 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001188 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001189 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001190 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001191 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001192 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001193 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001194 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001195 sys.stderr.flush()
1196 with open(fname, "a") as refrep:
1197 print(msg, file=refrep)
1198 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001199 return True
1200 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001201
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001202def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001203 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001204 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001205 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001206 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001207 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001208 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001209
Christian Heimesdae2a892008-04-19 00:55:37 +00001210 # Clear the warnings registry, so they can be displayed again
1211 for mod in sys.modules.values():
1212 if hasattr(mod, '__warningregistry__'):
1213 del mod.__warningregistry__
1214
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001215 # Restore some original values.
1216 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001217 copyreg.dispatch_table.clear()
1218 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001219 sys.path_importer_cache.clear()
1220 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001221 try:
1222 import zipimport
1223 except ImportError:
1224 pass # Run unmodified on platforms without zipimport support
1225 else:
1226 zipimport._zip_directory_cache.clear()
1227 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001228
Christian Heimes26855632008-01-27 23:50:43 +00001229 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001230 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001231
Guido van Rossum3de862d2007-08-18 00:10:33 +00001232 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001233 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001234 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001235 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001236 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001237 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001238 obj._abc_cache.clear()
1239 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001240
Antoine Pitrou046467c2009-10-30 18:30:35 +00001241 # Flush standard output, so that buffered data is sent to the OS and
1242 # associated Python objects are reclaimed.
1243 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1244 if stream is not None:
1245 stream.flush()
1246
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001247 # Clear assorted module caches.
1248 _path_created.clear()
1249 re.purge()
1250 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001251 urllib.parse.clear_cache()
1252 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001253 linecache.clearcache()
1254 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001255 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001256 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001257 doctest.master = None
1258
1259 # Collect cyclic trash.
1260 gc.collect()
1261
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001262def warm_char_cache():
1263 s = bytes(range(256))
1264 for i in range(256):
1265 s[i:i+1]
1266
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001267def findtestdir(path=None):
1268 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001269
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001270def removepy(names):
1271 if not names:
1272 return
1273 for idx, name in enumerate(names):
1274 basename, ext = os.path.splitext(name)
1275 if ext == '.py':
1276 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001277
Guido van Rossum152494a1996-12-20 03:12:20 +00001278def count(n, word):
1279 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001280 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001281 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001282 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001283
Tim Petersa45da922001-08-12 03:45:50 +00001284def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001285 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001286
1287 Optional arg width (default 70) is the maximum line length.
1288 Optional arg indent (default 4) is the number of blanks with which to
1289 begin each line.
1290 """
1291
Tim Petersba78bc42002-07-04 19:45:06 +00001292 from textwrap import fill
1293 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001294 # Print the sorted list: 'x' may be a '--random' list or a set()
1295 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001296 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001297
Tim Petersde14a302002-04-01 05:04:46 +00001298# Map sys.platform to a string containing the basenames of tests
1299# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001300#
1301# Special cases:
1302# test_pep277
1303# The _ExpectedSkips constructor adds this to the set of expected
1304# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001305# test_timeout
1306# Controlled by test_timeout.skip_expected. Requires the network
1307# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001308#
1309# Tests that are expected to be skipped everywhere except on one platform
1310# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001311
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001312_expectations = {
1313 'win32':
1314 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001315 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001316 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001317 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001318 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001319 test_fcntl
1320 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001321 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001322 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001323 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001324 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001325 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001326 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001327 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001328 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001329 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001330 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001331 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001332 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001333 test_pty
1334 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001335 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001336 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001337 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001338 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001339 test_wait3
1340 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001341 """,
1342 'linux2':
1343 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001344 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001345 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001346 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001347 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001348 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001349 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001350 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001351 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001352 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001353 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001354 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001355 test_openpty
1356 test_pyexpat
1357 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001358 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001359 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001360 'openunix8':
1361 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001362 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001363 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001364 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001365 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001366 test_openpty
1367 test_pyexpat
1368 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001369 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001370 """,
1371 'sco_sv3':
1372 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001373 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001374 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001375 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001376 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001377 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001378 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001379 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001380 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001381 test_openpty
1382 test_pyexpat
1383 test_queue
1384 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001385 test_sundry
1386 test_thread
1387 test_threaded_import
1388 test_threadedtempfile
1389 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001390 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001391 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001392 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001393 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001394 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001395 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001396 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001397 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001398 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001399 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001400 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001401 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001402 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001403 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001404 'sunos5':
1405 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001406 test_curses
1407 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001408 test_epoll
1409 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001410 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001411 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001412 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001413 test_zipfile
1414 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001415 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001416 'hp-ux11':
1417 """
Skip Montanarob3230212002-03-15 02:54:03 +00001418 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001419 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001420 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001421 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001422 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001423 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001424 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001425 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001426 test_openpty
1427 test_pyexpat
1428 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001429 test_zipfile
1430 test_zlib
1431 """,
Jason Tishler25115942002-12-05 15:18:15 +00001432 'cygwin':
1433 """
Jason Tishler25115942002-12-05 15:18:15 +00001434 test_curses
1435 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001436 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001437 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001438 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001439 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001440 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001441 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001442 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001443 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001444 'os2emx':
1445 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001446 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001447 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001448 test_epoll
1449 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001450 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001451 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001452 test_openpty
1453 test_ossaudiodev
1454 test_pty
1455 test_resource
1456 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001457 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001458 'freebsd4':
1459 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001460 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001461 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001462 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001463 test_ossaudiodev
1464 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001465 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001466 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001467 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001468 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001469 test_ttk_guionly
1470 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001471 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001472 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001473 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001474 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001475 'aix5':
1476 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001477 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001478 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001479 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001480 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001481 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001482 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001483 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001484 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001485 test_ttk_guionly
1486 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001487 test_zipimport
1488 test_zlib
1489 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001490 'openbsd3':
1491 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001492 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001493 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001494 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001495 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001496 test_normalization
1497 test_ossaudiodev
1498 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001499 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001500 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001501 test_ttk_guionly
1502 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001503 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001504 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001505 'netbsd3':
1506 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001507 test_ctypes
1508 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001509 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001510 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001511 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001512 test_ossaudiodev
1513 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001514 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001515 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001516 test_ttk_guionly
1517 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001518 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001519 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001520}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001521_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001522_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001523_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001524_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001525
Tim Petersb5b7b782001-08-12 01:20:39 +00001526class _ExpectedSkips:
1527 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001528 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001529 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001530
Tim Peters7c7efe92002-08-23 17:55:54 +00001531 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001532 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001533 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001534 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001535
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001536 # These are broken tests, for now skipped on every platform.
1537 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001538 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001539
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001540 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001541 if not os.path.supports_unicode_filenames:
1542 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001543
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001544 # doctest, profile and cProfile tests fail when the codec for the
1545 # fs encoding isn't built in because PyUnicode_Decode() adds two
1546 # calls into Python.
1547 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1548 if sys.getfilesystemencoding().lower() not in encs:
1549 self.expected.add('test_profile')
1550 self.expected.add('test_cProfile')
1551 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001552
Neal Norwitz55b61d22003-02-28 19:57:03 +00001553 if test_timeout.skip_expected:
1554 self.expected.add('test_timeout')
1555
Tim Petersecd79eb2003-01-29 00:35:32 +00001556 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001557 # test_sqlite is only reliable on Windows where the library
1558 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001559 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001560 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001561 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001562 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001563
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001564 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001565 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001566
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001567 if support.python_is_optimized():
1568 self.expected.add("test_gdb")
1569
Tim Peters7c7efe92002-08-23 17:55:54 +00001570 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001571
1572 def isvalid(self):
1573 "Return true iff _ExpectedSkips knows about the current platform."
1574 return self.valid
1575
1576 def getexpected(self):
1577 """Return set of test names we expect to skip on current platform.
1578
1579 self.isvalid() must be true.
1580 """
1581
1582 assert self.isvalid()
1583 return self.expected
1584
Michael Foord3ab34cc2010-12-03 12:27:40 +00001585def _make_temp_dir_for_build(TEMPDIR):
1586 # When tests are run from the Python build directory, it is best practice
1587 # to keep the test files in a subfolder. It eases the cleanup of leftover
1588 # files using command "make distclean".
1589 if sysconfig.is_python_build():
1590 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1591 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001592 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001593 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001594 except OSError as e:
1595 if e.errno != errno.EEXIST:
1596 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001597
1598 # Define a writable temp dir that will be used as cwd while running
1599 # the tests. The name of the dir includes the pid to allow parallel
1600 # testing (see the -j option).
1601 TESTCWD = 'test_python_{}'.format(os.getpid())
1602
1603 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1604 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001605
Guido van Rossum152494a1996-12-20 03:12:20 +00001606if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001607 # Remove regrtest.py's own directory from the module search path. Despite
1608 # the elimination of implicit relative imports, this is still needed to
1609 # ensure that submodules of the test package do not inappropriately appear
1610 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1611 # directly instead of using the -m switch
1612 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1613 i = len(sys.path)
1614 while i >= 0:
1615 i -= 1
1616 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1617 del sys.path[i]
1618
Florent Xiclunadc69e722010-09-13 16:35:02 +00001619 # findtestdir() gets the dirname out of __file__, so we have to make it
1620 # absolute before changing the working directory.
1621 # For example __file__ may be relative when running trace or profile.
1622 # See issue #9323.
1623 __file__ = os.path.abspath(__file__)
1624
1625 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001626 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001627
Michael Foord3ab34cc2010-12-03 12:27:40 +00001628 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001629
1630 # Run the tests in a context manager that temporary changes the CWD to a
1631 # temporary and writable directory. If it's not possible to create or
1632 # change the CWD, the original CWD will be used. The original CWD is
1633 # available from support.SAVEDCWD.
1634 with support.temp_cwd(TESTCWD, quiet=True):
1635 main()