blob: 8ca7f6c07f8f7cfdb32d72840ae6fc08af06e7af [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)
Georg Brandl96f78972009-10-27 21:02:11 +000041-u/--use RES1,RES2,...
42 -- specify which special resource intensive tests to run
43-M/--memlimit LIMIT
44 -- run very large memory-consuming tests
45
46Special runs
47
48-l/--findleaks -- if GC is available detect tests that leak memory
49-L/--runleaks -- run the leaks(1) command just before exit
50-R/--huntrleaks RUNCOUNTS
51 -- search for reference leaks (needs debug build, v. slow)
52-j/--multiprocess PROCESSES
53 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000054-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000055-D/--coverdir DIRECTORY
56 -- Directory where coverage files are put
57-N/--nocoverdir -- Put coverage files alongside modules
58-t/--threshold THRESHOLD
59 -- call gc.set_threshold(THRESHOLD)
60-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000061-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000062
R. David Murray0ba81e02010-04-26 17:02:32 +000063
64Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000065
Collin Winterfd12f492009-03-29 04:05:05 +000066-r randomizes test execution order. You can use --randseed=int to provide a
67int seed value for the randomizer; this is useful for reproducing troublesome
68test orders.
69
R. David Murray0ba81e02010-04-26 17:02:32 +000070-s On the first invocation of regrtest using -s, the first test file found
71or the first test file given on the command line is run, and the name of
72the next test is recorded in a file named pynexttest. If run from the
73Python build directory, pynexttest is located in the 'build' subdirectory,
74otherwise it is located in tempfile.gettempdir(). On subsequent runs,
75the test in pynexttest is run, and the next test is written to pynexttest.
76When the last test has been run, pynexttest is deleted. In this way it
77is possible to single step through the test files. This is useful when
78doing memory analysis on the Python interpreter, which process tends to
79consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000080
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000081-S is used to continue running tests after an aborted run. It will
82maintain the order a standard run (ie, this assumes -r is not used).
83This is useful after the tests have prematurely stopped for some external
84reason and you want to start running from where you left off rather
85than starting from the beginning.
86
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000087-f reads the names of tests from the file given as f's argument, one
88or more test names per line. Whitespace is ignored. Blank lines and
89lines beginning with '#' are ignored. This is especially useful for
90whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000091
Skip Montanaro0179a182004-06-06 15:53:18 +000092-L causes the leaks(1) command to be run just before exit if it exists.
93leaks(1) is available on Mac OS X and presumably on some other
94FreeBSD-derived systems.
95
Michael W. Hudson61147f62004-08-03 11:33:28 +000096-R runs each test several times and examines sys.gettotalrefcount() to
97see if the test appears to be leaking references. The argument should
98be of the form stab:run:fname where 'stab' is the number of times the
99test is run to let gettotalrefcount settle down, 'run' is the number
100of times further it is run and 'fname' is the name of the file the
101reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000102"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000103
Thomas Wouters477c8d52006-05-27 19:21:47 +0000104-M runs tests that require an exorbitant amount of memory. These tests
105typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001062 billion objects, which only works on 64-bit systems. There are also some
107tests that try to exhaust the address space of the process, which only makes
108sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000109which is a string in the form of '2.5Gb', determines howmuch memory the
110tests will limit themselves to (but they may go slightly over.) The number
111shouldn't be more memory than the machine has (including swap memory). You
112should also keep in mind that swap memory is generally much, much slower
113than RAM, and setting memlimit to all available RAM or higher will heavily
114tax the machine. On the other hand, it is no use running these tests with a
115limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
116to use more than memlimit memory will be skipped. The big-memory tests
117generally run very, very long.
118
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000119-u is used to specify which special resource intensive tests to run,
120such as those requiring large file support or network connectivity.
121The argument is a comma-separated list of words indicating the
122resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000123
Fred Drake3a15dac2002-04-11 16:39:16 +0000124 all - Enable all special resources.
125
Guido van Rossum315aa362003-03-11 14:46:48 +0000126 audio - Tests that use the audio device. (There are known
127 cases of broken audio drivers that can crash Python or
128 even the Linux kernel.)
129
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000130 curses - Tests that use curses and will modify the terminal's
131 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000132
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000133 largefile - It is okay to run some test that may create huge
134 files. These tests can take a long time and may
135 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000136
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000137 network - It is okay to run tests that use external network
138 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000139
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000140 decimal - Test the decimal module against a large suite that
141 verifies compliance with standards.
142
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000143 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000144
Tim Peterseba28be2005-03-28 01:08:02 +0000145 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000146
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000147 urlfetch - It is okay to download files required on testing.
148
Guilherme Polo9de29af2009-01-28 20:40:48 +0000149 gui - Run tests that require a running GUI.
150
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000151To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000152example, to run all the tests except for the gui tests, give the
153option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000154"""
155
Brett Cannon45071902010-06-14 22:22:54 +0000156import builtins
Victor Stinner8313d6a2011-06-29 15:22:26 +0200157import errno
Guido van Rossum152494a1996-12-20 03:12:20 +0000158import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200159import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000160import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200161import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000162import os
Victor Stinner8313d6a2011-06-29 15:22:26 +0200163import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000164import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000165import re
Christian Heimesb186d002008-03-18 15:15:01 +0000166import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000167import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200168import tempfile
169import time
170import traceback
171import unittest
172import warnings
173from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000174
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200175try:
176 import threading
177except ImportError:
178 threading = None
179try:
180 import multiprocessing.process
181except ImportError:
182 multiprocessing = None
183
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000184
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000185# Some times __path__ and __file__ are not absolute (e.g. while running from
186# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
187# imports might fail. This affects only the modules imported before os.chdir().
188# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
189# they are found in the CWD their __file__ and __path__ will be relative (this
190# happens before the chdir). All the modules imported after the chdir, are
191# not found in the CWD, and since the other paths in sys.path[1:] are absolute
192# (site.py absolutize them), the __file__ and __path__ will be absolute too.
193# Therefore it is necessary to absolutize manually the __file__ and __path__ of
194# the packages to prevent later imports to fail when the CWD is different.
195for module in sys.modules.values():
196 if hasattr(module, '__path__'):
197 module.__path__ = [os.path.abspath(path) for path in module.__path__]
198 if hasattr(module, '__file__'):
199 module.__file__ = os.path.abspath(module.__file__)
200
Guido van Rossumdc15c272002-08-12 21:55:51 +0000201
Guido van Rossumbb484652002-12-02 09:56:21 +0000202# MacOSX (a.k.a. Darwin) has a default stack size that is too small
203# for deeply recursive regular expressions. We see this as crashes in
204# the Python test suite when running test_re.py and test_sre.py. The
205# fix is to set the stack limit to 2048.
206# This approach may also be useful for other Unixy platforms that
207# suffer from small default stack limits.
208if sys.platform == 'darwin':
209 try:
210 import resource
211 except ImportError:
212 pass
213 else:
214 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
215 newsoft = min(hard, max(soft, 1024*2048))
216 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
217
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000218# Test result constants.
219PASSED = 1
220FAILED = 0
221ENV_CHANGED = -1
222SKIPPED = -2
223RESOURCE_DENIED = -3
224INTERRUPTED = -4
225
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000226from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000227
Georg Brandl1158a332009-06-04 09:30:30 +0000228RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000229 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000230
Florent Xiclunaec882212010-08-09 16:56:43 +0000231TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000232
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000233def usage(msg):
234 print(msg, file=sys.stderr)
235 print("Use --help for usage", file=sys.stderr)
236 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000237
238
Antoine Pitrou88909542009-06-29 13:54:42 +0000239def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000240 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000241 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000242 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000243 random_seed=None, use_mp=None, verbose3=False, forever=False,
244 header=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000245 """Execute a test suite.
246
Thomas Wouters7e474022000-07-16 12:04:32 +0000247 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000248 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000249
250 tests -- a list of strings containing test names (optional)
251 testdir -- the directory in which to look for tests (optional)
252
253 Users other than the Python test suite will certainly want to
254 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000255 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000256
257 If the tests argument is omitted, the tests listed on the
258 command-line will be used. If that's empty, too, then all *.py
259 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000260
Antoine Pitrou88909542009-06-29 13:54:42 +0000261 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000262 single, randomize, findleaks, use_resources, trace, coverdir,
263 print_slow, and random_seed) allow programmers calling main()
264 directly to set the values that would normally be set by flags
265 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000266 """
Fred Drake004d5e62000-10-23 17:22:08 +0000267
Victor Stinner1802d3f2010-05-19 17:11:19 +0000268 replace_stdout()
269
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000270 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000271 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000272 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:',
Georg Brandl96f78972009-10-27 21:02:11 +0000273 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
274 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
275 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000276 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000277 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
278 'start=', 'nowindows', 'header'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000279 except getopt.error as msg:
280 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000281
282 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000283 if random_seed is None:
284 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000285 if use_resources is None:
286 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000287 debug = False
288 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000289 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000290 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000291 print(__doc__)
292 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000293 elif o in ('-v', '--verbose'):
294 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000295 elif o in ('-w', '--verbose2'):
296 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000297 elif o in ('-d', '--debug'):
298 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000299 elif o in ('-W', '--verbose3'):
300 verbose3 = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000301 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000302 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000303 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000304 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000305 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000306 elif o in ('-S', '--start'):
307 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000308 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000309 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000310 elif o in ('-S', '--slow'):
311 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000312 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000313 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000314 elif o == '--randseed':
315 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000316 elif o in ('-f', '--fromfile'):
317 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000318 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000319 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000320 elif o in ('-L', '--runleaks'):
321 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000322 elif o in ('-t', '--threshold'):
323 import gc
324 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000325 elif o in ('-T', '--coverage'):
326 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000327 elif o in ('-D', '--coverdir'):
328 coverdir = os.path.join(os.getcwd(), a)
329 elif o in ('-N', '--nocoverdir'):
330 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000331 elif o in ('-R', '--huntrleaks'):
332 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000333 if len(huntrleaks) not in (2, 3):
334 print(a, huntrleaks)
335 usage('-R takes 2 or 3 colon-separated arguments')
336 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000337 huntrleaks[0] = 5
338 else:
339 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000340 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000341 huntrleaks[1] = 4
342 else:
343 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000344 if len(huntrleaks) == 2 or not huntrleaks[2]:
345 huntrleaks[2:] = ["reflog.txt"]
346 # Avoid false positives due to the character cache in
347 # stringobject.c filling slowly with random data
348 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000349 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000350 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000351 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000352 u = [x.lower() for x in a.split(',')]
353 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000354 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000355 use_resources[:] = RESOURCE_NAMES
356 continue
357 remove = False
358 if r[0] == '-':
359 remove = True
360 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000361 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000362 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000363 if remove:
364 if r in use_resources:
365 use_resources.remove(r)
366 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000367 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000368 elif o in ('-n', '--nowindows'):
369 import msvcrt
370 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
371 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
372 msvcrt.SEM_NOGPFAULTERRORBOX|
373 msvcrt.SEM_NOOPENFILEERRORBOX)
374 try:
375 msvcrt.CrtSetReportMode
376 except AttributeError:
377 # release build
378 pass
379 else:
380 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
381 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
382 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000383 elif o in ('-F', '--forever'):
384 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000385 elif o in ('-j', '--multiprocess'):
386 use_mp = int(a)
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000387 elif o == '--header':
388 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000389 elif o == '--slaveargs':
390 args, kwargs = json.loads(a)
391 try:
392 result = runtest(*args, **kwargs)
393 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000394 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000395 sys.stdout.flush()
396 print() # Force a newline (just in case)
397 print(json.dumps(result))
398 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000399 else:
400 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000401 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000402 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000403 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000404 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000405 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000406 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000407 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000408 usage("-l and -j don't go together!")
Barry Warsaw08fca522001-08-20 22:33:46 +0000409
Guido van Rossum152494a1996-12-20 03:12:20 +0000410 good = []
411 bad = []
412 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000413 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000414 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000415 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000416
Neil Schemenauerd569f232000-09-22 15:29:28 +0000417 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000418 try:
419 import gc
420 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000421 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000422 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000423 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000424 # Uncomment the line below to report garbage that is not
425 # freeable by reference counting alone. By default only
426 # garbage that is not collectable by the GC is reported.
427 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000428 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000429
Barry Warsawe11e3de1999-01-28 19:51:51 +0000430 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000431 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000432 try:
433 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000434 next_test = fp.read().strip()
435 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000436 fp.close()
437 except IOError:
438 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000439
440 if fromfile:
441 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000442 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000443 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000444 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000445 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000446 guts = line.split() # assuming no test has whitespace in its name
447 if guts and not guts[0].startswith('#'):
448 tests.extend(guts)
449 fp.close()
450
451 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000452 removepy(args)
453 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000454
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000455 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000456 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000457 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000458 for arg in args:
459 if arg in stdtests:
460 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000461 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000462 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000463
464 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000465 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000466 # Print basic platform information
467 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000468 print("== ", platform.platform(aliased=True),
469 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000470 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000471 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000472
R. David Murrayef1992b2009-12-16 15:19:27 +0000473 alltests = findtests(testdir, stdtests, nottests)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000474 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000475 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000476 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000477 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000478 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000479 except IndexError:
480 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000481 # Remove all the tests that precede start if it's set.
482 if start:
483 try:
484 del tests[:tests.index(start)]
485 except ValueError:
486 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000487 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000488 random.seed(random_seed)
489 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000490 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000491 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000492 import trace, tempfile
493 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
494 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000495 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000496
Christian Heimesb186d002008-03-18 15:15:01 +0000497 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000498 support.verbose = verbose # Tell tests to be moderately quiet
499 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000500 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000501
502 def accumulate_result(test, result):
503 ok, test_time = result
504 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000505 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000506 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000507 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000508 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000509 elif ok == ENV_CHANGED:
510 bad.append(test)
511 environment_changed.append(test)
512 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000513 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000514 elif ok == RESOURCE_DENIED:
515 skipped.append(test)
516 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000517
R. David Murray7dc72cc2009-11-14 16:13:02 +0000518 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000519 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000520 while True:
521 for test in tests:
522 yield test
523 if bad:
524 return
525 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000526 test_count = ''
527 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000528 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000529 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000530 test_count = '/{}'.format(len(selected))
531 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000532
Antoine Pitrou88909542009-06-29 13:54:42 +0000533 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000534 try:
535 from threading import Thread
536 except ImportError:
537 print("Multiprocess option requires thread support")
538 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000539 from queue import Queue
540 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000541 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000542 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000543 def tests_and_args():
544 for test in tests:
545 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000546 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000547 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Victor Stinner8313d6a2011-06-29 15:22:26 +0200548 debug=debug, output_on_failure=verbose3)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000549 )
550 yield (test, args_tuple)
551 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000552 opt_args = support.args_from_interpreter_flags()
553 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000554 def work():
555 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000556 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000557 while True:
558 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000559 test, args_tuple = next(pending)
560 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000561 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000562 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000563 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000564 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000565 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000566 universal_newlines=True,
567 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000568 stdout, stderr = popen.communicate()
569 # Strip last refcount output line if it exists, since it
570 # comes from the shutdown of the interpreter in the subcommand.
571 stderr = debug_output_pat.sub("", stderr)
572 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000573 if not result:
574 output.put((None, None, None, None))
575 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000576 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000577 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000578 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000579 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000580 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000581 workers = [Thread(target=work) for i in range(use_mp)]
582 for worker in workers:
583 worker.start()
584 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000585 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000586 try:
587 while finished < use_mp:
588 test, stdout, stderr, result = output.get()
589 if test is None:
590 finished += 1
591 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200592 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000593 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300594 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
595 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200596 test_count_width, test_index, test_count,
597 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000598 if stdout:
599 print(stdout)
600 if stderr:
601 print(stderr, file=sys.stderr)
602 if result[0] == INTERRUPTED:
603 assert result[1] == 'KeyboardInterrupt'
604 raise KeyboardInterrupt # What else?
Georg Brandldee7b852010-08-02 18:59:52 +0000605 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000606 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000607 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000608 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000609 for worker in workers:
610 worker.join()
611 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000612 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000613 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300614 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
615 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200616 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000617 sys.stdout.flush()
618 if trace:
619 # If we're tracing code coverage, then we don't exit with status
620 # if on a false return value from main.
R. David Murray8e286c42010-12-27 20:09:32 +0000621 tracer.runctx('runtest(test, verbose, quiet)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000622 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000623 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000624 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100625 result = runtest(test, verbose, quiet, huntrleaks, debug,
Victor Stinner8313d6a2011-06-29 15:22:26 +0200626 output_on_failure=verbose3)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000627 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000628 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000629 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000630 break
631 except:
632 raise
633 if findleaks:
634 gc.collect()
635 if gc.garbage:
636 print("Warning: test created", len(gc.garbage), end=' ')
637 print("uncollectable object(s).")
638 # move the uncollectable objects somewhere so we don't see
639 # them again
640 found_garbage.extend(gc.garbage)
641 del gc.garbage[:]
642 # Unload the newly imported modules (best effort finalization)
643 for module in sys.modules.keys():
644 if module not in save_modules and module.startswith("test."):
645 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000646
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000647 if interrupted:
648 # print a newline after ^C
649 print()
650 print("Test suite interrupted by signal SIGINT.")
651 omitted = set(selected) - set(good) - set(bad) - set(skipped)
652 print(count(len(omitted), "test"), "omitted:")
653 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000654 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000655 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000656 print("All", end=' ')
657 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000658 if print_slow:
659 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000660 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000661 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000662 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000663 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000664 bad = sorted(set(bad) - set(environment_changed))
665 if bad:
666 print(count(len(bad), "test"), "failed:")
667 printlist(bad)
668 if environment_changed:
669 print("{} altered the execution environment:".format(
670 count(len(environment_changed), "test")))
671 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000672 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000673 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000674 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000675
Tim Petersb5b7b782001-08-12 01:20:39 +0000676 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000677 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000678 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000679 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000680 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000681 print(count(len(surprise), "skip"), \
682 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000683 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000684 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000685 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000686 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000687 print("Ask someone to teach regrtest.py about which tests are")
688 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000689
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000690 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000691 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000692 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000693 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000694 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000695 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000696 verbose = True
R. David Murray8e286c42010-12-27 20:09:32 +0000697 ok = runtest(test, True, quiet, huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000698 except KeyboardInterrupt:
699 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000700 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000701 break
702 except:
703 raise
704
Barry Warsawe11e3de1999-01-28 19:51:51 +0000705 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000706 if next_single_test:
707 with open(filename, 'w') as fp:
708 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000709 else:
710 os.unlink(filename)
711
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000712 if trace:
713 r = tracer.results()
714 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
715
Skip Montanaro0179a182004-06-06 15:53:18 +0000716 if runleaks:
717 os.system("leaks %d" % os.getpid())
718
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000719 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000720
Guido van Rossum152494a1996-12-20 03:12:20 +0000721
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000722STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000723 'test_grammar',
724 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000725 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000726 'test_builtin',
727 'test_exceptions',
728 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000729 'test_unittest',
730 'test_doctest',
731 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000732]
Guido van Rossum152494a1996-12-20 03:12:20 +0000733
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000734NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000735 'test_future1',
736 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000737}
Guido van Rossum152494a1996-12-20 03:12:20 +0000738
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000739def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000740 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000741 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000742 names = os.listdir(testdir)
743 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000744 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000745 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000746 modname, ext = os.path.splitext(name)
747 if modname[:5] == "test_" and ext == ".py" and modname not in others:
748 tests.append(modname)
749 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000750
Victor Stinnerf58087b2010-05-02 17:24:51 +0000751def replace_stdout():
752 """Set stdout encoder error handler to backslashreplace (as stderr error
753 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000754 if os.name == "nt":
755 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
756 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000757
758 import atexit
759
Victor Stinnerf58087b2010-05-02 17:24:51 +0000760 stdout = sys.stdout
761 sys.stdout = open(stdout.fileno(), 'w',
762 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000763 errors="backslashreplace",
764 closefd=False)
765
766 def restore_stdout():
767 sys.stdout.close()
768 sys.stdout = stdout
769 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000770
Antoine Pitrou88909542009-06-29 13:54:42 +0000771def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100772 huntrleaks=False, debug=False, use_resources=None,
Victor Stinner8313d6a2011-06-29 15:22:26 +0200773 output_on_failure=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000774 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000775
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000776 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000777 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000778 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000779 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000780 huntrleaks -- run multiple times to test for leaks; requires a debug
781 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200782 output_on_failure -- if true, display test output on failure
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000783
784 Returns one of the test result constants:
785 INTERRUPTED KeyboardInterrupt when run under -j
786 RESOURCE_DENIED test skipped because resource denied
787 SKIPPED test skipped for some other reason
788 ENV_CHANGED test failed because it changed the execution environment
789 FAILED test failed
790 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000791 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000792
Antoine Pitrou88909542009-06-29 13:54:42 +0000793 if use_resources is not None:
794 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000795 try:
Victor Stinner8313d6a2011-06-29 15:22:26 +0200796 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200797 support.verbose = True
798
799 # Reuse the same instance to all calls to runtest(). Some
800 # tests keep a reference to sys.stdout or sys.stderr
801 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200802 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200803 stream = io.StringIO()
804 runtest.stringio = stream
805 else:
806 stream = runtest.stringio
807 stream.seek(0)
808 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200809
810 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200811 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200812 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200813 sys.stdout = stream
814 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200815 result = runtest_inner(test, verbose, quiet, huntrleaks,
816 debug, display_failure=False)
817 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200818 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200819 orig_stderr.write(output)
820 orig_stderr.flush()
821 finally:
822 sys.stdout = orig_stdout
823 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200824 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200825 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200826 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
827 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100828 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000829 finally:
830 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200831runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000832
Nick Coghlan6ead5522009-10-18 13:19:33 +0000833# Unit tests are supposed to leave the execution environment unchanged
834# once they complete. But sometimes tests have bugs, especially when
835# tests fail, and the changes to environment go on to mess up other
836# tests. This can cause issues with buildbot stability, since tests
837# are run in random order and so problems may appear to come and go.
838# There are a few things we can save and restore to mitigate this, and
839# the following context manager handles this task.
840
841class saved_test_environment:
842 """Save bits of the test environment and restore them at block exit.
843
844 with saved_test_environment(testname, verbose, quiet):
845 #stuff
846
847 Unless quiet is True, a warning is printed to stderr if any of
848 the saved items was changed by the test. The attribute 'changed'
849 is initially False, but is set to True if a change is detected.
850
851 If verbose is more than 1, the before and after state of changed
852 items is also printed.
853 """
854
855 changed = False
856
857 def __init__(self, testname, verbose=0, quiet=False):
858 self.testname = testname
859 self.verbose = verbose
860 self.quiet = quiet
861
862 # To add things to save and restore, add a name XXX to the resources list
863 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
864 # return the value to be saved and compared against a second call to the
865 # get function when test execution completes. restore_XXX should accept
866 # the saved value and restore the resource using it. It will be called if
867 # and only if a change in the value is detected.
868 #
869 # Note: XXX will have any '.' replaced with '_' characters when determining
870 # the corresponding method names.
871
872 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000873 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000874 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti0123e052011-03-20 15:09:26 +0200875 'logging._handlers', 'logging._handlerList',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200876 'sys.warnoptions', 'threading._dangling',
877 'multiprocessing.process._dangling')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000878
879 def get_sys_argv(self):
880 return id(sys.argv), sys.argv, sys.argv[:]
881 def restore_sys_argv(self, saved_argv):
882 sys.argv = saved_argv[1]
883 sys.argv[:] = saved_argv[2]
884
885 def get_cwd(self):
886 return os.getcwd()
887 def restore_cwd(self, saved_cwd):
888 os.chdir(saved_cwd)
889
890 def get_sys_stdout(self):
891 return sys.stdout
892 def restore_sys_stdout(self, saved_stdout):
893 sys.stdout = saved_stdout
894
895 def get_sys_stderr(self):
896 return sys.stderr
897 def restore_sys_stderr(self, saved_stderr):
898 sys.stderr = saved_stderr
899
900 def get_sys_stdin(self):
901 return sys.stdin
902 def restore_sys_stdin(self, saved_stdin):
903 sys.stdin = saved_stdin
904
905 def get_os_environ(self):
906 return id(os.environ), os.environ, dict(os.environ)
907 def restore_os_environ(self, saved_environ):
908 os.environ = saved_environ[1]
909 os.environ.clear()
910 os.environ.update(saved_environ[2])
911
912 def get_sys_path(self):
913 return id(sys.path), sys.path, sys.path[:]
914 def restore_sys_path(self, saved_path):
915 sys.path = saved_path[1]
916 sys.path[:] = saved_path[2]
917
Brett Cannon055470a2010-02-19 15:57:10 +0000918 def get_sys_path_hooks(self):
919 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
920 def restore_sys_path_hooks(self, saved_hooks):
921 sys.path_hooks = saved_hooks[1]
922 sys.path_hooks[:] = saved_hooks[2]
923
924 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000925 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000926 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000927 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000928
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000929 def get_warnings_filters(self):
930 return id(warnings.filters), warnings.filters, warnings.filters[:]
931 def restore_warnings_filters(self, saved_filters):
932 warnings.filters = saved_filters[1]
933 warnings.filters[:] = saved_filters[2]
934
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000935 def get_asyncore_socket_map(self):
936 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000937 # XXX Making a copy keeps objects alive until __exit__ gets called.
938 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000939 def restore_asyncore_socket_map(self, saved_map):
940 asyncore = sys.modules.get('asyncore')
941 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000942 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000943 asyncore.socket_map.update(saved_map)
944
Nick Coghlan7d819752010-12-05 06:45:03 +0000945 def get_logging__handlers(self):
946 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000947 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +0000948 def restore_logging__handlers(self, saved_handlers):
949 # Can't easily revert the logging state
950 pass
951
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000952 def get_logging__handlerList(self):
953 # _handlerList is a list of weakrefs to handlers
954 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
955 def restore_logging__handlerList(self, saved_handlerList):
956 # Can't easily revert the logging state
957 pass
958
Ezio Melotti0123e052011-03-20 15:09:26 +0200959 def get_sys_warnoptions(self):
960 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
961 def restore_sys_warnoptions(self, saved_options):
962 sys.warnoptions = saved_options[1]
963 sys.warnoptions[:] = saved_options[2]
964
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200965 # Controlling dangling references to Thread objects can make it easier
966 # to track reference leaks.
967 def get_threading__dangling(self):
968 if not threading:
969 return None
970 # This copies the weakrefs without making any strong reference
971 return threading._dangling.copy()
972 def restore_threading__dangling(self, saved):
973 if not threading:
974 return
975 threading._dangling.clear()
976 threading._dangling.update(saved)
977
978 # Same for Process objects
979 def get_multiprocessing_process__dangling(self):
980 if not multiprocessing:
981 return None
982 # This copies the weakrefs without making any strong reference
983 return multiprocessing.process._dangling.copy()
984 def restore_multiprocessing_process__dangling(self, saved):
985 if not multiprocessing:
986 return
987 multiprocessing.process._dangling.clear()
988 multiprocessing.process._dangling.update(saved)
989
Nick Coghlan6ead5522009-10-18 13:19:33 +0000990 def resource_info(self):
991 for name in self.resources:
992 method_suffix = name.replace('.', '_')
993 get_name = 'get_' + method_suffix
994 restore_name = 'restore_' + method_suffix
995 yield name, getattr(self, get_name), getattr(self, restore_name)
996
997 def __enter__(self):
998 self.saved_values = dict((name, get()) for name, get, restore
999 in self.resource_info())
1000 return self
1001
1002 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001003 saved_values = self.saved_values
1004 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001005 for name, get, restore in self.resource_info():
1006 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001007 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001008 # Check for changes to the resource's value
1009 if current != original:
1010 self.changed = True
1011 restore(original)
1012 if not self.quiet:
1013 print("Warning -- {} was modified by {}".format(
1014 name, self.testname),
1015 file=sys.stderr)
1016 if self.verbose > 1:
1017 print(" Before: {}\n After: {} ".format(
1018 original, current),
1019 file=sys.stderr)
1020 return False
1021
1022
Victor Stinnera7c33e52011-06-29 13:00:54 +02001023def runtest_inner(test, verbose, quiet,
1024 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001025 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001026
Antoine Pitrou88909542009-06-29 13:54:42 +00001027 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001028 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001029 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001030 if test.startswith('test.'):
1031 abstest = test
1032 else:
1033 # Always import it from the test package
1034 abstest = 'test.' + test
1035 with saved_test_environment(test, verbose, quiet) as environment:
1036 start_time = time.time()
1037 the_package = __import__(abstest, globals(), locals(), [])
1038 the_module = getattr(the_package, test)
1039 # Old tests run to completion simply as a side-effect of
1040 # being imported. For tests based on unittest or doctest,
1041 # explicitly invoke their test_main() function (if it exists).
1042 indirect_test = getattr(the_module, "test_main", None)
1043 if indirect_test is not None:
1044 indirect_test()
1045 if huntrleaks:
1046 refleak = dash_R(the_module, test, indirect_test,
1047 huntrleaks)
1048 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001049 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001050 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001051 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001052 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001053 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001054 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001055 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001056 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001057 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001058 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001059 except KeyboardInterrupt:
1060 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001061 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001062 if display_failure:
1063 print("test", test, "failed --", msg, file=sys.stderr)
1064 else:
1065 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001066 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001067 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001068 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001069 msg = traceback.format_exc()
1070 print("test", test, "crashed --", msg, file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001071 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001072 return FAILED, test_time
Guido van Rossum152494a1996-12-20 03:12:20 +00001073 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001074 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001075 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001076 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001077 return ENV_CHANGED, test_time
1078 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001079
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001080def cleanup_test_droppings(testname, verbose):
1081 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001082 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001083 import gc
1084
1085 # First kill any dangling references to open files etc.
1086 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001087 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001088 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001089
1090 # Try to clean up junk commonly left behind. While tests shouldn't leave
1091 # any files or directories behind, when a test fails that can be tedious
1092 # for it to arrange. The consequences can be especially nasty on Windows,
1093 # since if a test leaves a file open, it cannot be deleted by name (while
1094 # there's nothing we can do about that here either, we can display the
1095 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001096 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001097 "db_home",
1098 ):
1099 if not os.path.exists(name):
1100 continue
1101
1102 if os.path.isdir(name):
1103 kind, nuker = "directory", shutil.rmtree
1104 elif os.path.isfile(name):
1105 kind, nuker = "file", os.unlink
1106 else:
1107 raise SystemError("os.path says %r exists but is neither "
1108 "directory nor file" % name)
1109
1110 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001111 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001112 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001113 # if we have chmod, fix possible permissions problems
1114 # that might prevent cleanup
1115 if (hasattr(os, 'chmod')):
1116 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001117 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001118 except Exception as msg:
1119 print(("%r left behind %s %r and it couldn't be "
1120 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001121
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001122def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001123 """Run a test multiple times, looking for reference leaks.
1124
1125 Returns:
1126 False if the test didn't leak references; True if we detected refleaks.
1127 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001128 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001129 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001130
1131 if not hasattr(sys, 'gettotalrefcount'):
1132 raise Exception("Tracking reference leaks requires a debug build "
1133 "of Python")
1134
1135 # Save current values for dash_R_cleanup() to restore.
1136 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001137 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001138 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001139 try:
1140 import zipimport
1141 except ImportError:
1142 zdc = None # Run unmodified on platforms without zipimport support
1143 else:
1144 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001145 abcs = {}
1146 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001147 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001148 continue
1149 for obj in abc.__subclasses__() + [abc]:
1150 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001151
1152 if indirect_test:
1153 def run_the_test():
1154 indirect_test()
1155 else:
1156 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001157 del sys.modules[the_module.__name__]
1158 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001159
1160 deltas = []
1161 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001162 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001163 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001164 print("beginning", repcount, "repetitions", file=sys.stderr)
1165 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001166 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001167 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001168 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001169 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001170 run_the_test()
1171 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001172 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001173 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001174 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001175 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001176 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001177 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001178 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001179 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001180 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001181 sys.stderr.flush()
1182 with open(fname, "a") as refrep:
1183 print(msg, file=refrep)
1184 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001185 return True
1186 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001187
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001188def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001189 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001190 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001191 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001192 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001193 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001194 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001195
Christian Heimesdae2a892008-04-19 00:55:37 +00001196 # Clear the warnings registry, so they can be displayed again
1197 for mod in sys.modules.values():
1198 if hasattr(mod, '__warningregistry__'):
1199 del mod.__warningregistry__
1200
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001201 # Restore some original values.
1202 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001203 copyreg.dispatch_table.clear()
1204 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001205 sys.path_importer_cache.clear()
1206 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001207 try:
1208 import zipimport
1209 except ImportError:
1210 pass # Run unmodified on platforms without zipimport support
1211 else:
1212 zipimport._zip_directory_cache.clear()
1213 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001214
Christian Heimes26855632008-01-27 23:50:43 +00001215 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001216 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001217
Guido van Rossum3de862d2007-08-18 00:10:33 +00001218 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001219 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001220 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001221 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001222 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001223 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001224 obj._abc_cache.clear()
1225 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001226
Antoine Pitrou046467c2009-10-30 18:30:35 +00001227 # Flush standard output, so that buffered data is sent to the OS and
1228 # associated Python objects are reclaimed.
1229 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1230 if stream is not None:
1231 stream.flush()
1232
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001233 # Clear assorted module caches.
1234 _path_created.clear()
1235 re.purge()
1236 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001237 urllib.parse.clear_cache()
1238 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001239 linecache.clearcache()
1240 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001241 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001242 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001243 doctest.master = None
1244
1245 # Collect cyclic trash.
1246 gc.collect()
1247
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001248def warm_char_cache():
1249 s = bytes(range(256))
1250 for i in range(256):
1251 s[i:i+1]
1252
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001253def findtestdir(path=None):
1254 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001255
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001256def removepy(names):
1257 if not names:
1258 return
1259 for idx, name in enumerate(names):
1260 basename, ext = os.path.splitext(name)
1261 if ext == '.py':
1262 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001263
Guido van Rossum152494a1996-12-20 03:12:20 +00001264def count(n, word):
1265 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001266 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001267 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001268 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001269
Tim Petersa45da922001-08-12 03:45:50 +00001270def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001271 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001272
1273 Optional arg width (default 70) is the maximum line length.
1274 Optional arg indent (default 4) is the number of blanks with which to
1275 begin each line.
1276 """
1277
Tim Petersba78bc42002-07-04 19:45:06 +00001278 from textwrap import fill
1279 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001280 # Print the sorted list: 'x' may be a '--random' list or a set()
1281 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001282 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001283
Tim Petersde14a302002-04-01 05:04:46 +00001284# Map sys.platform to a string containing the basenames of tests
1285# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001286#
1287# Special cases:
1288# test_pep277
1289# The _ExpectedSkips constructor adds this to the set of expected
1290# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001291# test_timeout
1292# Controlled by test_timeout.skip_expected. Requires the network
1293# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001294#
1295# Tests that are expected to be skipped everywhere except on one platform
1296# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001297
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001298_expectations = {
1299 'win32':
1300 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001301 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001302 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001303 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001304 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001305 test_fcntl
1306 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001307 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001308 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001309 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001310 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001311 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001312 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001313 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001314 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001315 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001316 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001317 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001318 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001319 test_pty
1320 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001321 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001322 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001323 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001324 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001325 test_wait3
1326 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001327 """,
1328 'linux2':
1329 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001330 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001331 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001332 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001333 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001334 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001335 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001336 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001337 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001338 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001339 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001340 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001341 test_openpty
1342 test_pyexpat
1343 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001344 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001345 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001346 'openunix8':
1347 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001348 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001349 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001350 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001351 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001352 test_openpty
1353 test_pyexpat
1354 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001355 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001356 """,
1357 'sco_sv3':
1358 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001359 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001360 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001361 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001362 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001363 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001364 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001365 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001366 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001367 test_openpty
1368 test_pyexpat
1369 test_queue
1370 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001371 test_sundry
1372 test_thread
1373 test_threaded_import
1374 test_threadedtempfile
1375 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001376 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001377 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001378 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001379 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001380 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001381 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001382 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001383 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001384 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001385 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001386 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001387 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001388 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001389 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001390 'sunos5':
1391 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001392 test_curses
1393 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001394 test_epoll
1395 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001396 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001397 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001398 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001399 test_zipfile
1400 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001401 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001402 'hp-ux11':
1403 """
Skip Montanarob3230212002-03-15 02:54:03 +00001404 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001405 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001406 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001407 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001408 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001409 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001410 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001411 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001412 test_openpty
1413 test_pyexpat
1414 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001415 test_zipfile
1416 test_zlib
1417 """,
Jason Tishler25115942002-12-05 15:18:15 +00001418 'cygwin':
1419 """
Jason Tishler25115942002-12-05 15:18:15 +00001420 test_curses
1421 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001422 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001423 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001424 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001425 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001426 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001427 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001428 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001429 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001430 'os2emx':
1431 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001432 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001433 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001434 test_epoll
1435 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001436 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001437 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001438 test_openpty
1439 test_ossaudiodev
1440 test_pty
1441 test_resource
1442 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001443 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001444 'freebsd4':
1445 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001446 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001447 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001448 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001449 test_ossaudiodev
1450 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001451 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001452 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001453 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001454 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001455 test_ttk_guionly
1456 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001457 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001458 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001459 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001460 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001461 'aix5':
1462 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001463 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001464 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001465 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001466 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001467 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001468 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001469 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001470 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001471 test_ttk_guionly
1472 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001473 test_zipimport
1474 test_zlib
1475 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001476 'openbsd3':
1477 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001478 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001479 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001480 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001481 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001482 test_normalization
1483 test_ossaudiodev
1484 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001485 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001486 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001487 test_ttk_guionly
1488 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001489 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001490 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001491 'netbsd3':
1492 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001493 test_ctypes
1494 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001495 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001496 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001497 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001498 test_ossaudiodev
1499 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001500 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001501 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001502 test_ttk_guionly
1503 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001504 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001505 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001506}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001507_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001508_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001509_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001510_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001511
Tim Petersb5b7b782001-08-12 01:20:39 +00001512class _ExpectedSkips:
1513 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001514 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001515 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001516
Tim Peters7c7efe92002-08-23 17:55:54 +00001517 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001518 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001519 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001520 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001521
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001522 # These are broken tests, for now skipped on every platform.
1523 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001524 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001525
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001526 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001527 if not os.path.supports_unicode_filenames:
1528 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001529
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001530 # doctest, profile and cProfile tests fail when the codec for the
1531 # fs encoding isn't built in because PyUnicode_Decode() adds two
1532 # calls into Python.
1533 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1534 if sys.getfilesystemencoding().lower() not in encs:
1535 self.expected.add('test_profile')
1536 self.expected.add('test_cProfile')
1537 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001538
Neal Norwitz55b61d22003-02-28 19:57:03 +00001539 if test_timeout.skip_expected:
1540 self.expected.add('test_timeout')
1541
Tim Petersecd79eb2003-01-29 00:35:32 +00001542 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001543 # test_sqlite is only reliable on Windows where the library
1544 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001545 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001546 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001547 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001548 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001549
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001550 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001551 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001552
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001553 if support.python_is_optimized():
1554 self.expected.add("test_gdb")
1555
Tim Peters7c7efe92002-08-23 17:55:54 +00001556 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001557
1558 def isvalid(self):
1559 "Return true iff _ExpectedSkips knows about the current platform."
1560 return self.valid
1561
1562 def getexpected(self):
1563 """Return set of test names we expect to skip on current platform.
1564
1565 self.isvalid() must be true.
1566 """
1567
1568 assert self.isvalid()
1569 return self.expected
1570
Michael Foord3ab34cc2010-12-03 12:27:40 +00001571def _make_temp_dir_for_build(TEMPDIR):
1572 # When tests are run from the Python build directory, it is best practice
1573 # to keep the test files in a subfolder. It eases the cleanup of leftover
1574 # files using command "make distclean".
1575 if sysconfig.is_python_build():
1576 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1577 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001578 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001579 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001580 except OSError as e:
1581 if e.errno != errno.EEXIST:
1582 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001583
1584 # Define a writable temp dir that will be used as cwd while running
1585 # the tests. The name of the dir includes the pid to allow parallel
1586 # testing (see the -j option).
1587 TESTCWD = 'test_python_{}'.format(os.getpid())
1588
1589 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1590 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001591
Guido van Rossum152494a1996-12-20 03:12:20 +00001592if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001593 # Remove regrtest.py's own directory from the module search path. Despite
1594 # the elimination of implicit relative imports, this is still needed to
1595 # ensure that submodules of the test package do not inappropriately appear
1596 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1597 # directly instead of using the -m switch
1598 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1599 i = len(sys.path)
1600 while i >= 0:
1601 i -= 1
1602 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1603 del sys.path[i]
1604
Florent Xiclunadc69e722010-09-13 16:35:02 +00001605 # findtestdir() gets the dirname out of __file__, so we have to make it
1606 # absolute before changing the working directory.
1607 # For example __file__ may be relative when running trace or profile.
1608 # See issue #9323.
1609 __file__ = os.path.abspath(__file__)
1610
1611 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001612 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001613
Michael Foord3ab34cc2010-12-03 12:27:40 +00001614 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001615
1616 # Run the tests in a context manager that temporary changes the CWD to a
1617 # temporary and writable directory. If it's not possible to create or
1618 # change the CWD, the original CWD will be used. The original CWD is
1619 # available from support.SAVEDCWD.
1620 with support.temp_cwd(TESTCWD, quiet=True):
1621 main()