blob: 6fad78fc0b0991511b7e90a925edd770c3639875 [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 Pitrou216a3bc2011-07-23 22:33:39 +020041-G/--failfast -- fail as soon as a test fails (only with -v or -W)
Georg Brandl96f78972009-10-27 21:02:11 +000042-u/--use RES1,RES2,...
43 -- specify which special resource intensive tests to run
44-M/--memlimit LIMIT
45 -- run very large memory-consuming tests
46
47Special runs
48
49-l/--findleaks -- if GC is available detect tests that leak memory
50-L/--runleaks -- run the leaks(1) command just before exit
51-R/--huntrleaks RUNCOUNTS
52 -- search for reference leaks (needs debug build, v. slow)
53-j/--multiprocess PROCESSES
54 -- run PROCESSES processes at once
R. David Murray0ba81e02010-04-26 17:02:32 +000055-T/--coverage -- turn on code coverage tracing using the trace module
Georg Brandl96f78972009-10-27 21:02:11 +000056-D/--coverdir DIRECTORY
57 -- Directory where coverage files are put
58-N/--nocoverdir -- Put coverage files alongside modules
59-t/--threshold THRESHOLD
60 -- call gc.set_threshold(THRESHOLD)
61-n/--nowindows -- suppress error message boxes on Windows
R. David Murray0ba81e02010-04-26 17:02:32 +000062-F/--forever -- run the specified tests in a loop, until an error happens
Guido van Rossum152494a1996-12-20 03:12:20 +000063
R. David Murray0ba81e02010-04-26 17:02:32 +000064
65Additional Option Details:
Guido van Rossumf58ed251997-03-07 21:04:33 +000066
Collin Winterfd12f492009-03-29 04:05:05 +000067-r randomizes test execution order. You can use --randseed=int to provide a
68int seed value for the randomizer; this is useful for reproducing troublesome
69test orders.
70
R. David Murray0ba81e02010-04-26 17:02:32 +000071-s On the first invocation of regrtest using -s, the first test file found
72or the first test file given on the command line is run, and the name of
73the next test is recorded in a file named pynexttest. If run from the
74Python build directory, pynexttest is located in the 'build' subdirectory,
75otherwise it is located in tempfile.gettempdir(). On subsequent runs,
76the test in pynexttest is run, and the next test is written to pynexttest.
77When the last test has been run, pynexttest is deleted. In this way it
78is possible to single step through the test files. This is useful when
79doing memory analysis on the Python interpreter, which process tends to
80consume too many resources to run the full regression test non-stop.
Barry Warsawe11e3de1999-01-28 19:51:51 +000081
Neal Norwitz94fa2ee2008-03-31 02:55:15 +000082-S is used to continue running tests after an aborted run. It will
83maintain the order a standard run (ie, this assumes -r is not used).
84This is useful after the tests have prematurely stopped for some external
85reason and you want to start running from where you left off rather
86than starting from the beginning.
87
Guido van Rossum9e9d4f82002-06-07 15:17:03 +000088-f reads the names of tests from the file given as f's argument, one
89or more test names per line. Whitespace is ignored. Blank lines and
90lines beginning with '#' are ignored. This is especially useful for
91whittling down failures involving interactions among tests.
Tim Petersc5000df2002-06-02 21:42:01 +000092
Skip Montanaro0179a182004-06-06 15:53:18 +000093-L causes the leaks(1) command to be run just before exit if it exists.
94leaks(1) is available on Mac OS X and presumably on some other
95FreeBSD-derived systems.
96
Michael W. Hudson61147f62004-08-03 11:33:28 +000097-R runs each test several times and examines sys.gettotalrefcount() to
98see if the test appears to be leaking references. The argument should
99be of the form stab:run:fname where 'stab' is the number of times the
100test is run to let gettotalrefcount settle down, 'run' is the number
101of times further it is run and 'fname' is the name of the file the
102reports are written to. These parameters all have defaults (5, 4 and
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000103"reflog.txt" respectively), and the minimal invocation is '-R :'.
Michael W. Hudson61147f62004-08-03 11:33:28 +0000104
Thomas Wouters477c8d52006-05-27 19:21:47 +0000105-M runs tests that require an exorbitant amount of memory. These tests
106typically try to ascertain containers keep working when containing more than
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001072 billion objects, which only works on 64-bit systems. There are also some
108tests that try to exhaust the address space of the process, which only makes
109sense on 32-bit systems with at least 2Gb of memory. The passed-in memlimit,
Thomas Wouters477c8d52006-05-27 19:21:47 +0000110which is a string in the form of '2.5Gb', determines howmuch memory the
111tests will limit themselves to (but they may go slightly over.) The number
112shouldn't be more memory than the machine has (including swap memory). You
113should also keep in mind that swap memory is generally much, much slower
114than RAM, and setting memlimit to all available RAM or higher will heavily
115tax the machine. On the other hand, it is no use running these tests with a
116limit of less than 2.5Gb, and many require more than 20Gb. Tests that expect
117to use more than memlimit memory will be skipped. The big-memory tests
118generally run very, very long.
119
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000120-u is used to specify which special resource intensive tests to run,
121such as those requiring large file support or network connectivity.
122The argument is a comma-separated list of words indicating the
123resources to test. Currently only the following are defined:
Barry Warsaw08fca522001-08-20 22:33:46 +0000124
Fred Drake3a15dac2002-04-11 16:39:16 +0000125 all - Enable all special resources.
126
Guido van Rossum315aa362003-03-11 14:46:48 +0000127 audio - Tests that use the audio device. (There are known
128 cases of broken audio drivers that can crash Python or
129 even the Linux kernel.)
130
Andrew M. Kuchling2158df02001-10-22 15:26:09 +0000131 curses - Tests that use curses and will modify the terminal's
132 state and output modes.
Tim Peters1633a2e2001-10-30 05:56:40 +0000133
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000134 largefile - It is okay to run some test that may create huge
135 files. These tests can take a long time and may
136 consume >2GB of disk space temporarily.
Barry Warsaw08fca522001-08-20 22:33:46 +0000137
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000138 network - It is okay to run tests that use external network
139 resource, e.g. testing SSL support for sockets.
Martin v. Löwis1c6b1a22002-11-19 17:47:07 +0000140
Raymond Hettinger7c85fa42004-07-01 11:01:35 +0000141 decimal - Test the decimal module against a large suite that
142 verifies compliance with standards.
143
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000144 cpu - Used for certain CPU-heavy tests.
Jeremy Hylton4336eda2004-08-07 19:25:33 +0000145
Tim Peterseba28be2005-03-28 01:08:02 +0000146 subprocess Run all tests for the subprocess module.
Peter Astrandf7f1bb72005-03-03 20:47:37 +0000147
Hye-Shik Changaaa2f1d2005-12-10 17:44:27 +0000148 urlfetch - It is okay to download files required on testing.
149
Guilherme Polo9de29af2009-01-28 20:40:48 +0000150 gui - Run tests that require a running GUI.
151
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000152To enable all resources except one, use '-uall,-<resource>'. For
Georg Brandl1158a332009-06-04 09:30:30 +0000153example, to run all the tests except for the gui tests, give the
154option '-uall,-gui'.
Guido van Rossum152494a1996-12-20 03:12:20 +0000155"""
156
Brett Cannon45071902010-06-14 22:22:54 +0000157import builtins
Victor Stinner8313d6a2011-06-29 15:22:26 +0200158import errno
Guido van Rossum152494a1996-12-20 03:12:20 +0000159import getopt
Victor Stinner8313d6a2011-06-29 15:22:26 +0200160import io
Antoine Pitrou88909542009-06-29 13:54:42 +0000161import json
Victor Stinner8313d6a2011-06-29 15:22:26 +0200162import logging
Christian Heimesb186d002008-03-18 15:15:01 +0000163import os
Victor Stinner8313d6a2011-06-29 15:22:26 +0200164import platform
Skip Montanaroab1c7912000-06-30 16:39:27 +0000165import random
Thomas Wouters9ada3d62006-04-21 09:47:09 +0000166import re
Christian Heimesb186d002008-03-18 15:15:01 +0000167import sys
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000168import sysconfig
Victor Stinner8313d6a2011-06-29 15:22:26 +0200169import tempfile
170import time
171import traceback
172import unittest
173import warnings
174from inspect import isabstract
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000175
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200176try:
177 import threading
178except ImportError:
179 threading = None
180try:
181 import multiprocessing.process
182except ImportError:
183 multiprocessing = None
184
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000185
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000186# Some times __path__ and __file__ are not absolute (e.g. while running from
187# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
188# imports might fail. This affects only the modules imported before os.chdir().
189# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
190# they are found in the CWD their __file__ and __path__ will be relative (this
191# happens before the chdir). All the modules imported after the chdir, are
192# not found in the CWD, and since the other paths in sys.path[1:] are absolute
193# (site.py absolutize them), the __file__ and __path__ will be absolute too.
194# Therefore it is necessary to absolutize manually the __file__ and __path__ of
195# the packages to prevent later imports to fail when the CWD is different.
196for module in sys.modules.values():
197 if hasattr(module, '__path__'):
198 module.__path__ = [os.path.abspath(path) for path in module.__path__]
199 if hasattr(module, '__file__'):
200 module.__file__ = os.path.abspath(module.__file__)
201
Guido van Rossumdc15c272002-08-12 21:55:51 +0000202
Guido van Rossumbb484652002-12-02 09:56:21 +0000203# MacOSX (a.k.a. Darwin) has a default stack size that is too small
204# for deeply recursive regular expressions. We see this as crashes in
205# the Python test suite when running test_re.py and test_sre.py. The
206# fix is to set the stack limit to 2048.
207# This approach may also be useful for other Unixy platforms that
208# suffer from small default stack limits.
209if sys.platform == 'darwin':
210 try:
211 import resource
212 except ImportError:
213 pass
214 else:
215 soft, hard = resource.getrlimit(resource.RLIMIT_STACK)
216 newsoft = min(hard, max(soft, 1024*2048))
217 resource.setrlimit(resource.RLIMIT_STACK, (newsoft, hard))
218
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000219# Test result constants.
220PASSED = 1
221FAILED = 0
222ENV_CHANGED = -1
223SKIPPED = -2
224RESOURCE_DENIED = -3
225INTERRUPTED = -4
226
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000227from test import support
Fred Drake3a15dac2002-04-11 16:39:16 +0000228
Georg Brandl1158a332009-06-04 09:30:30 +0000229RESOURCE_NAMES = ('audio', 'curses', 'largefile', 'network',
Antoine Pitrou5bc4fa72010-10-14 15:34:31 +0000230 'decimal', 'cpu', 'subprocess', 'urlfetch', 'gui')
Fred Drake3a15dac2002-04-11 16:39:16 +0000231
Florent Xiclunaec882212010-08-09 16:56:43 +0000232TEMPDIR = os.path.abspath(tempfile.gettempdir())
Fred Drake3a15dac2002-04-11 16:39:16 +0000233
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000234def usage(msg):
235 print(msg, file=sys.stderr)
236 print("Use --help for usage", file=sys.stderr)
237 sys.exit(2)
Barry Warsaw08fca522001-08-20 22:33:46 +0000238
239
Antoine Pitrou88909542009-06-29 13:54:42 +0000240def main(tests=None, testdir=None, verbose=0, quiet=False,
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000241 exclude=False, single=False, randomize=False, fromfile=None,
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000242 findleaks=False, use_resources=None, trace=False, coverdir='coverage',
Collin Winterfd12f492009-03-29 04:05:05 +0000243 runleaks=False, huntrleaks=False, verbose2=False, print_slow=False,
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000244 random_seed=None, use_mp=None, verbose3=False, forever=False,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200245 header=False, failfast=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000246 """Execute a test suite.
247
Thomas Wouters7e474022000-07-16 12:04:32 +0000248 This also parses command-line options and modifies its behavior
Fred Drake004d5e62000-10-23 17:22:08 +0000249 accordingly.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000250
251 tests -- a list of strings containing test names (optional)
252 testdir -- the directory in which to look for tests (optional)
253
254 Users other than the Python test suite will certainly want to
255 specify testdir; if it's omitted, the directory containing the
Fred Drake004d5e62000-10-23 17:22:08 +0000256 Python test suite is searched for.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000257
258 If the tests argument is omitted, the tests listed on the
259 command-line will be used. If that's empty, too, then all *.py
260 files beginning with test_ will be used.
Skip Montanaroab1c7912000-06-30 16:39:27 +0000261
Antoine Pitrou88909542009-06-29 13:54:42 +0000262 The other default arguments (verbose, quiet, exclude,
Collin Winterfd12f492009-03-29 04:05:05 +0000263 single, randomize, findleaks, use_resources, trace, coverdir,
264 print_slow, and random_seed) allow programmers calling main()
265 directly to set the values that would normally be set by flags
266 on the command line.
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000267 """
Fred Drake004d5e62000-10-23 17:22:08 +0000268
Victor Stinner1802d3f2010-05-19 17:11:19 +0000269 replace_stdout()
270
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000271 support.record_original_stdout(sys.stdout)
Guido van Rossum152494a1996-12-20 03:12:20 +0000272 try:
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200273 opts, args = getopt.getopt(sys.argv[1:], 'hvqxsSrf:lu:t:TD:NLR:FwWM:nj:G',
Georg Brandl96f78972009-10-27 21:02:11 +0000274 ['help', 'verbose', 'verbose2', 'verbose3', 'quiet',
275 'exclude', 'single', 'slow', 'random', 'fromfile', 'findleaks',
276 'use=', 'threshold=', 'trace', 'coverdir=', 'nocoverdir',
R. David Murray7dc72cc2009-11-14 16:13:02 +0000277 'runleaks', 'huntrleaks=', 'memlimit=', 'randseed=',
Brett Cannon63eef1e2011-01-06 22:32:41 +0000278 'multiprocess=', 'coverage', 'slaveargs=', 'forever', 'debug',
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200279 'start=', 'nowindows', 'header', 'failfast'])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000280 except getopt.error as msg:
281 usage(msg)
Barry Warsaw08fca522001-08-20 22:33:46 +0000282
283 # Defaults
Collin Winterfd12f492009-03-29 04:05:05 +0000284 if random_seed is None:
285 random_seed = random.randrange(10000000)
Barry Warsaw08fca522001-08-20 22:33:46 +0000286 if use_resources is None:
287 use_resources = []
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000288 debug = False
289 start = None
Guido van Rossum152494a1996-12-20 03:12:20 +0000290 for o, a in opts:
Barry Warsaw08fca522001-08-20 22:33:46 +0000291 if o in ('-h', '--help'):
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000292 print(__doc__)
293 return
Barry Warsaw08fca522001-08-20 22:33:46 +0000294 elif o in ('-v', '--verbose'):
295 verbose += 1
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000296 elif o in ('-w', '--verbose2'):
297 verbose2 = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000298 elif o in ('-d', '--debug'):
299 debug = True
Georg Brandl96f78972009-10-27 21:02:11 +0000300 elif o in ('-W', '--verbose3'):
301 verbose3 = True
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200302 elif o in ('-G', '--failfast'):
303 failfast = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000304 elif o in ('-q', '--quiet'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000305 quiet = True;
Barry Warsaw08fca522001-08-20 22:33:46 +0000306 verbose = 0
Barry Warsaw08fca522001-08-20 22:33:46 +0000307 elif o in ('-x', '--exclude'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000308 exclude = True
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000309 elif o in ('-S', '--start'):
310 start = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000311 elif o in ('-s', '--single'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000312 single = True
Christian Heimesb186d002008-03-18 15:15:01 +0000313 elif o in ('-S', '--slow'):
314 print_slow = True
Barry Warsaw08fca522001-08-20 22:33:46 +0000315 elif o in ('-r', '--randomize'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000316 randomize = True
Collin Winterfd12f492009-03-29 04:05:05 +0000317 elif o == '--randseed':
318 random_seed = int(a)
Tim Petersc5000df2002-06-02 21:42:01 +0000319 elif o in ('-f', '--fromfile'):
320 fromfile = a
Barry Warsaw08fca522001-08-20 22:33:46 +0000321 elif o in ('-l', '--findleaks'):
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000322 findleaks = True
Skip Montanaro0179a182004-06-06 15:53:18 +0000323 elif o in ('-L', '--runleaks'):
324 runleaks = True
Guido van Rossum9e9d4f82002-06-07 15:17:03 +0000325 elif o in ('-t', '--threshold'):
326 import gc
327 gc.set_threshold(int(a))
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000328 elif o in ('-T', '--coverage'):
329 trace = True
Walter Dörwaldaee4da62004-11-12 18:51:27 +0000330 elif o in ('-D', '--coverdir'):
331 coverdir = os.path.join(os.getcwd(), a)
332 elif o in ('-N', '--nocoverdir'):
333 coverdir = None
Michael W. Hudson61147f62004-08-03 11:33:28 +0000334 elif o in ('-R', '--huntrleaks'):
335 huntrleaks = a.split(':')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000336 if len(huntrleaks) not in (2, 3):
337 print(a, huntrleaks)
338 usage('-R takes 2 or 3 colon-separated arguments')
339 if not huntrleaks[0]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000340 huntrleaks[0] = 5
341 else:
342 huntrleaks[0] = int(huntrleaks[0])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000343 if not huntrleaks[1]:
Michael W. Hudson61147f62004-08-03 11:33:28 +0000344 huntrleaks[1] = 4
345 else:
346 huntrleaks[1] = int(huntrleaks[1])
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000347 if len(huntrleaks) == 2 or not huntrleaks[2]:
348 huntrleaks[2:] = ["reflog.txt"]
349 # Avoid false positives due to the character cache in
350 # stringobject.c filling slowly with random data
351 warm_char_cache()
Thomas Wouters477c8d52006-05-27 19:21:47 +0000352 elif o in ('-M', '--memlimit'):
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000353 support.set_memlimit(a)
Barry Warsaw08fca522001-08-20 22:33:46 +0000354 elif o in ('-u', '--use'):
Guido van Rossumfe3f6962001-09-06 16:09:41 +0000355 u = [x.lower() for x in a.split(',')]
356 for r in u:
Fred Drake3a15dac2002-04-11 16:39:16 +0000357 if r == 'all':
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000358 use_resources[:] = RESOURCE_NAMES
359 continue
360 remove = False
361 if r[0] == '-':
362 remove = True
363 r = r[1:]
Fred Drake3a15dac2002-04-11 16:39:16 +0000364 if r not in RESOURCE_NAMES:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000365 usage('Invalid -u/--use option: ' + a)
Fred Drake4dd0f7e2002-11-26 21:44:56 +0000366 if remove:
367 if r in use_resources:
368 use_resources.remove(r)
369 elif r not in use_resources:
Andrew MacIntyree41abab2002-04-30 12:11:04 +0000370 use_resources.append(r)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000371 elif o in ('-n', '--nowindows'):
372 import msvcrt
373 msvcrt.SetErrorMode(msvcrt.SEM_FAILCRITICALERRORS|
374 msvcrt.SEM_NOALIGNMENTFAULTEXCEPT|
375 msvcrt.SEM_NOGPFAULTERRORBOX|
376 msvcrt.SEM_NOOPENFILEERRORBOX)
377 try:
378 msvcrt.CrtSetReportMode
379 except AttributeError:
380 # release build
381 pass
382 else:
383 for m in [msvcrt.CRT_WARN, msvcrt.CRT_ERROR, msvcrt.CRT_ASSERT]:
384 msvcrt.CrtSetReportMode(m, msvcrt.CRTDBG_MODE_FILE)
385 msvcrt.CrtSetReportFile(m, msvcrt.CRTDBG_FILE_STDERR)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000386 elif o in ('-F', '--forever'):
387 forever = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000388 elif o in ('-j', '--multiprocess'):
389 use_mp = int(a)
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000390 elif o == '--header':
391 header = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000392 elif o == '--slaveargs':
393 args, kwargs = json.loads(a)
394 try:
395 result = runtest(*args, **kwargs)
396 except BaseException as e:
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000397 result = INTERRUPTED, e.__class__.__name__
Antoine Pitrou88909542009-06-29 13:54:42 +0000398 sys.stdout.flush()
399 print() # Force a newline (just in case)
400 print(json.dumps(result))
401 sys.exit(0)
R. David Murray35768ad2009-11-15 00:23:21 +0000402 else:
403 print(("No handler for option {}. Please report this as a bug "
Georg Brandldee7b852010-08-02 18:59:52 +0000404 "at http://bugs.python.org.").format(o), file=sys.stderr)
R. David Murray35768ad2009-11-15 00:23:21 +0000405 sys.exit(1)
Tim Petersc5000df2002-06-02 21:42:01 +0000406 if single and fromfile:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000407 usage("-s and -f don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000408 if use_mp and trace:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000409 usage("-T and -j don't go together!")
Antoine Pitrou88909542009-06-29 13:54:42 +0000410 if use_mp and findleaks:
Florent Xicluna64fb18e2010-03-06 14:43:34 +0000411 usage("-l and -j don't go together!")
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200412 if failfast and not (verbose or verbose3):
413 usage("-G/--failfast needs either -v or -W")
Barry Warsaw08fca522001-08-20 22:33:46 +0000414
Guido van Rossum152494a1996-12-20 03:12:20 +0000415 good = []
416 bad = []
417 skipped = []
Fred Drake9a0db072003-02-03 15:19:30 +0000418 resource_denieds = []
Nick Coghlan6ead5522009-10-18 13:19:33 +0000419 environment_changed = []
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000420 interrupted = False
Barry Warsawe11e3de1999-01-28 19:51:51 +0000421
Neil Schemenauerd569f232000-09-22 15:29:28 +0000422 if findleaks:
Barry Warsawa873b032000-08-03 15:50:37 +0000423 try:
424 import gc
425 except ImportError:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000426 print('No GC available, disabling findleaks.')
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000427 findleaks = False
Barry Warsawa873b032000-08-03 15:50:37 +0000428 else:
Neil Schemenauer8a00abc2000-10-13 01:32:42 +0000429 # Uncomment the line below to report garbage that is not
430 # freeable by reference counting alone. By default only
431 # garbage that is not collectable by the GC is reported.
432 #gc.set_debug(gc.DEBUG_SAVEALL)
Neil Schemenauerd569f232000-09-22 15:29:28 +0000433 found_garbage = []
Barry Warsawa873b032000-08-03 15:50:37 +0000434
Barry Warsawe11e3de1999-01-28 19:51:51 +0000435 if single:
Florent Xiclunaec882212010-08-09 16:56:43 +0000436 filename = os.path.join(TEMPDIR, 'pynexttest')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000437 try:
438 fp = open(filename, 'r')
R. David Murray7dc72cc2009-11-14 16:13:02 +0000439 next_test = fp.read().strip()
440 tests = [next_test]
Barry Warsawe11e3de1999-01-28 19:51:51 +0000441 fp.close()
442 except IOError:
443 pass
Tim Petersc5000df2002-06-02 21:42:01 +0000444
445 if fromfile:
446 tests = []
Ezio Melotti184bdfb2010-02-18 09:37:05 +0000447 fp = open(os.path.join(support.SAVEDCWD, fromfile))
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000448 count_pat = re.compile(r'\[\s*\d+/\s*\d+\]')
Tim Petersc5000df2002-06-02 21:42:01 +0000449 for line in fp:
Antoine Pitroud95c7b52010-11-07 20:50:51 +0000450 line = count_pat.sub('', line)
Tim Petersc5000df2002-06-02 21:42:01 +0000451 guts = line.split() # assuming no test has whitespace in its name
452 if guts and not guts[0].startswith('#'):
453 tests.extend(guts)
454 fp.close()
455
456 # Strip .py extensions.
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000457 removepy(args)
458 removepy(tests)
Tim Petersc5000df2002-06-02 21:42:01 +0000459
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000460 stdtests = STDTESTS[:]
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000461 nottests = NOTTESTS.copy()
Guido van Rossum152494a1996-12-20 03:12:20 +0000462 if exclude:
Guido van Rossum6c74fea1998-08-25 12:29:08 +0000463 for arg in args:
464 if arg in stdtests:
465 stdtests.remove(arg)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000466 nottests.add(arg)
Guido van Rossum41360a41998-03-26 19:42:58 +0000467 args = []
Florent Xicluna0e62a142010-03-06 17:34:48 +0000468
469 # For a partial run, we do not need to clutter the output.
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000470 if verbose or header or not (quiet or single or tests or args):
Florent Xicluna0e62a142010-03-06 17:34:48 +0000471 # Print basic platform information
472 print("==", platform.python_implementation(), *sys.version.split())
Florent Xiclunaec882212010-08-09 16:56:43 +0000473 print("== ", platform.platform(aliased=True),
474 "%s-endian" % sys.byteorder)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000475 print("== ", os.getcwd())
Antoine Pitrou3c4402f2011-01-03 20:38:52 +0000476 print("Testing with flags:", sys.flags)
Florent Xicluna0e62a142010-03-06 17:34:48 +0000477
R. David Murrayef1992b2009-12-16 15:19:27 +0000478 alltests = findtests(testdir, stdtests, nottests)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000479 selected = tests or args or alltests
Barry Warsawe11e3de1999-01-28 19:51:51 +0000480 if single:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000481 selected = selected[:1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000482 try:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000483 next_single_test = alltests[alltests.index(selected[0])+1]
R. David Murrayef1992b2009-12-16 15:19:27 +0000484 except IndexError:
485 next_single_test = None
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000486 # Remove all the tests that precede start if it's set.
487 if start:
488 try:
489 del tests[:tests.index(start)]
490 except ValueError:
491 print("Couldn't find starting test (%s), using all tests" % start)
Skip Montanaroab1c7912000-06-30 16:39:27 +0000492 if randomize:
Collin Winterfd12f492009-03-29 04:05:05 +0000493 random.seed(random_seed)
494 print("Using random seed", random_seed)
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000495 random.shuffle(selected)
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000496 if trace:
Georg Brandl33c28812009-04-01 23:07:29 +0000497 import trace, tempfile
498 tracer = trace.Trace(ignoredirs=[sys.prefix, sys.exec_prefix,
499 tempfile.gettempdir()],
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000500 trace=False, count=True)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000501
Christian Heimesb186d002008-03-18 15:15:01 +0000502 test_times = []
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000503 support.verbose = verbose # Tell tests to be moderately quiet
504 support.use_resources = use_resources
Guido van Rossum5796d262000-04-21 21:35:06 +0000505 save_modules = sys.modules.keys()
Antoine Pitrou88909542009-06-29 13:54:42 +0000506
507 def accumulate_result(test, result):
508 ok, test_time = result
509 test_times.append((test_time, test))
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000510 if ok == PASSED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000511 good.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000512 elif ok == FAILED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000513 bad.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000514 elif ok == ENV_CHANGED:
515 bad.append(test)
516 environment_changed.append(test)
517 elif ok == SKIPPED:
Antoine Pitrou88909542009-06-29 13:54:42 +0000518 skipped.append(test)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000519 elif ok == RESOURCE_DENIED:
520 skipped.append(test)
521 resource_denieds.append(test)
Antoine Pitrou88909542009-06-29 13:54:42 +0000522
R. David Murray7dc72cc2009-11-14 16:13:02 +0000523 if forever:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000524 def test_forever(tests=list(selected)):
R. David Murray7dc72cc2009-11-14 16:13:02 +0000525 while True:
526 for test in tests:
527 yield test
528 if bad:
529 return
530 tests = test_forever()
Georg Brandle8e02e32010-08-03 07:56:50 +0000531 test_count = ''
532 test_count_width = 3
R. David Murray7dc72cc2009-11-14 16:13:02 +0000533 else:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000534 tests = iter(selected)
Georg Brandle8e02e32010-08-03 07:56:50 +0000535 test_count = '/{}'.format(len(selected))
536 test_count_width = len(test_count) - 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000537
Antoine Pitrou88909542009-06-29 13:54:42 +0000538 if use_mp:
Victor Stinner45df8202010-04-28 22:31:17 +0000539 try:
540 from threading import Thread
541 except ImportError:
542 print("Multiprocess option requires thread support")
543 sys.exit(2)
Georg Brandl1b37e872010-03-14 10:45:50 +0000544 from queue import Queue
545 from subprocess import Popen, PIPE
Antoine Pitrou88909542009-06-29 13:54:42 +0000546 debug_output_pat = re.compile(r"\[\d+ refs\]$")
Antoine Pitrou88909542009-06-29 13:54:42 +0000547 output = Queue()
R. David Murray7dc72cc2009-11-14 16:13:02 +0000548 def tests_and_args():
549 for test in tests:
550 args_tuple = (
R. David Murray8e286c42010-12-27 20:09:32 +0000551 (test, verbose, quiet),
R. David Murray7dc72cc2009-11-14 16:13:02 +0000552 dict(huntrleaks=huntrleaks, use_resources=use_resources,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200553 debug=debug, output_on_failure=verbose3,
554 failfast=failfast)
R. David Murray7dc72cc2009-11-14 16:13:02 +0000555 )
556 yield (test, args_tuple)
557 pending = tests_and_args()
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000558 opt_args = support.args_from_interpreter_flags()
559 base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest']
Antoine Pitrou88909542009-06-29 13:54:42 +0000560 def work():
561 # A worker thread.
Neal Norwitz14ca3272006-02-28 18:05:43 +0000562 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000563 while True:
564 try:
R. David Murray7dc72cc2009-11-14 16:13:02 +0000565 test, args_tuple = next(pending)
566 except StopIteration:
R. David Murray27144602009-10-19 15:26:16 +0000567 output.put((None, None, None, None))
Antoine Pitrou88909542009-06-29 13:54:42 +0000568 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000569 # -E is needed by some tests, e.g. test_import
Antoine Pitrou1b03f2c2010-10-14 11:12:00 +0000570 popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)],
R. David Murray27144602009-10-19 15:26:16 +0000571 stdout=PIPE, stderr=PIPE,
Antoine Pitrou36e778e2010-08-18 20:44:14 +0000572 universal_newlines=True,
573 close_fds=(os.name != 'nt'))
R. David Murray27144602009-10-19 15:26:16 +0000574 stdout, stderr = popen.communicate()
575 # Strip last refcount output line if it exists, since it
576 # comes from the shutdown of the interpreter in the subcommand.
577 stderr = debug_output_pat.sub("", stderr)
578 stdout, _, result = stdout.strip().rpartition("\n")
R. David Murray7dc72cc2009-11-14 16:13:02 +0000579 if not result:
580 output.put((None, None, None, None))
581 return
Antoine Pitrou88909542009-06-29 13:54:42 +0000582 result = json.loads(result)
R. David Murray27144602009-10-19 15:26:16 +0000583 output.put((test, stdout.rstrip(), stderr.rstrip(), result))
Antoine Pitrou88909542009-06-29 13:54:42 +0000584 except BaseException:
R. David Murray27144602009-10-19 15:26:16 +0000585 output.put((None, None, None, None))
Neal Norwitz14ca3272006-02-28 18:05:43 +0000586 raise
Antoine Pitrou88909542009-06-29 13:54:42 +0000587 workers = [Thread(target=work) for i in range(use_mp)]
588 for worker in workers:
589 worker.start()
590 finished = 0
Georg Brandldee7b852010-08-02 18:59:52 +0000591 test_index = 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000592 try:
593 while finished < use_mp:
594 test, stdout, stderr, result = output.get()
595 if test is None:
596 finished += 1
597 continue
Victor Stinnera2a895c2011-05-23 23:14:05 +0200598 accumulate_result(test, result)
Georg Brandldee7b852010-08-02 18:59:52 +0000599 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300600 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
601 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200602 test_count_width, test_index, test_count,
603 len(bad), test))
R. David Murray7dc72cc2009-11-14 16:13:02 +0000604 if stdout:
605 print(stdout)
606 if stderr:
607 print(stderr, file=sys.stderr)
608 if result[0] == INTERRUPTED:
609 assert result[1] == 'KeyboardInterrupt'
610 raise KeyboardInterrupt # What else?
Georg Brandldee7b852010-08-02 18:59:52 +0000611 test_index += 1
R. David Murray7dc72cc2009-11-14 16:13:02 +0000612 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000613 interrupted = True
R. David Murray7dc72cc2009-11-14 16:13:02 +0000614 pending.close()
Antoine Pitrou88909542009-06-29 13:54:42 +0000615 for worker in workers:
616 worker.join()
617 else:
Georg Brandldee7b852010-08-02 18:59:52 +0000618 for test_index, test in enumerate(tests, 1):
Antoine Pitrou88909542009-06-29 13:54:42 +0000619 if not quiet:
Ezio Melotti84f75c62011-05-24 01:00:10 +0300620 fmt = "[{1:{0}}{2}/{3}] {4}" if bad else "[{1:{0}}{2}] {4}"
621 print(fmt.format(
Victor Stinnera2a895c2011-05-23 23:14:05 +0200622 test_count_width, test_index, test_count, len(bad), test))
Antoine Pitrou88909542009-06-29 13:54:42 +0000623 sys.stdout.flush()
624 if trace:
625 # If we're tracing code coverage, then we don't exit with status
626 # if on a false return value from main.
R. David Murray8e286c42010-12-27 20:09:32 +0000627 tracer.runctx('runtest(test, verbose, quiet)',
Antoine Pitrou88909542009-06-29 13:54:42 +0000628 globals=globals(), locals=vars())
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000629 else:
Antoine Pitrou88909542009-06-29 13:54:42 +0000630 try:
Antoine Pitrou293954d2011-03-23 23:01:49 +0100631 result = runtest(test, verbose, quiet, huntrleaks, debug,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200632 output_on_failure=verbose3,
633 failfast=failfast)
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000634 accumulate_result(test, result)
Antoine Pitrou88909542009-06-29 13:54:42 +0000635 except KeyboardInterrupt:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000636 interrupted = True
Antoine Pitrou88909542009-06-29 13:54:42 +0000637 break
638 except:
639 raise
640 if findleaks:
641 gc.collect()
642 if gc.garbage:
643 print("Warning: test created", len(gc.garbage), end=' ')
644 print("uncollectable object(s).")
645 # move the uncollectable objects somewhere so we don't see
646 # them again
647 found_garbage.extend(gc.garbage)
648 del gc.garbage[:]
649 # Unload the newly imported modules (best effort finalization)
650 for module in sys.modules.keys():
651 if module not in save_modules and module.startswith("test."):
652 support.unload(module)
Jeremy Hylton7a1ea0e2001-10-17 13:45:28 +0000653
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000654 if interrupted:
655 # print a newline after ^C
656 print()
657 print("Test suite interrupted by signal SIGINT.")
658 omitted = set(selected) - set(good) - set(bad) - set(skipped)
659 print(count(len(omitted), "test"), "omitted:")
660 printlist(omitted)
Guido van Rossum152494a1996-12-20 03:12:20 +0000661 if good and not quiet:
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000662 if not bad and not skipped and not interrupted and len(good) > 1:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000663 print("All", end=' ')
664 print(count(len(good), "test"), "OK.")
Christian Heimesb186d002008-03-18 15:15:01 +0000665 if print_slow:
666 test_times.sort(reverse=True)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000667 print("10 slowest tests:")
Christian Heimesb186d002008-03-18 15:15:01 +0000668 for time, test in test_times[:10]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000669 print("%s: %.1fs" % (test, time))
Guido van Rossum152494a1996-12-20 03:12:20 +0000670 if bad:
Nick Coghlan6ead5522009-10-18 13:19:33 +0000671 bad = sorted(set(bad) - set(environment_changed))
672 if bad:
673 print(count(len(bad), "test"), "failed:")
674 printlist(bad)
675 if environment_changed:
676 print("{} altered the execution environment:".format(
677 count(len(environment_changed), "test")))
678 printlist(environment_changed)
Guido van Rossum152494a1996-12-20 03:12:20 +0000679 if skipped and not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000680 print(count(len(skipped), "test"), "skipped:")
Tim Petersa45da922001-08-12 03:45:50 +0000681 printlist(skipped)
Barry Warsawe11e3de1999-01-28 19:51:51 +0000682
Tim Petersb5b7b782001-08-12 01:20:39 +0000683 e = _ExpectedSkips()
Tim Petersa2be2d62001-08-12 02:01:09 +0000684 plat = sys.platform
Tim Petersb5b7b782001-08-12 01:20:39 +0000685 if e.isvalid():
Raymond Hettingera690a992003-11-16 16:17:49 +0000686 surprise = set(skipped) - e.getexpected() - set(resource_denieds)
Tim Petersb5b7b782001-08-12 01:20:39 +0000687 if surprise:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000688 print(count(len(surprise), "skip"), \
689 "unexpected on", plat + ":")
Tim Petersa45da922001-08-12 03:45:50 +0000690 printlist(surprise)
Tim Petersb5b7b782001-08-12 01:20:39 +0000691 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000692 print("Those skips are all expected on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000693 else:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000694 print("Ask someone to teach regrtest.py about which tests are")
695 print("expected to get skipped on", plat + ".")
Tim Petersb5b7b782001-08-12 01:20:39 +0000696
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000697 if verbose2 and bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000698 print("Re-running failed tests in verbose mode")
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000699 for test in bad:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000700 print("Re-running test %r in verbose mode" % test)
Tim Peters922dd7d2006-03-10 23:37:10 +0000701 sys.stdout.flush()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000702 try:
Antoine Pitrou88909542009-06-29 13:54:42 +0000703 verbose = True
R. David Murray8e286c42010-12-27 20:09:32 +0000704 ok = runtest(test, True, quiet, huntrleaks, debug)
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000705 except KeyboardInterrupt:
706 # print a newline separate from the ^C
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000707 print()
Martin v. Löwis04824ce2006-03-10 21:26:16 +0000708 break
709 except:
710 raise
711
Barry Warsawe11e3de1999-01-28 19:51:51 +0000712 if single:
R. David Murrayef1992b2009-12-16 15:19:27 +0000713 if next_single_test:
714 with open(filename, 'w') as fp:
715 fp.write(next_single_test + '\n')
Barry Warsawe11e3de1999-01-28 19:51:51 +0000716 else:
717 os.unlink(filename)
718
Barry Warsaw3b6d0252004-02-07 22:43:03 +0000719 if trace:
720 r = tracer.results()
721 r.write_results(show_missing=True, summary=True, coverdir=coverdir)
722
Skip Montanaro0179a182004-06-06 15:53:18 +0000723 if runleaks:
724 os.system("leaks %d" % os.getpid())
725
Florent Xiclunad6995eb2010-03-30 19:43:09 +0000726 sys.exit(len(bad) > 0 or interrupted)
Barry Warsaw08fca522001-08-20 22:33:46 +0000727
Guido van Rossum152494a1996-12-20 03:12:20 +0000728
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000729STDTESTS = [
Guido van Rossum152494a1996-12-20 03:12:20 +0000730 'test_grammar',
731 'test_opcodes',
Guido van Rossumd8faa362007-04-27 19:54:29 +0000732 'test_dict',
Guido van Rossum152494a1996-12-20 03:12:20 +0000733 'test_builtin',
734 'test_exceptions',
735 'test_types',
Collin Winter7afaa882007-03-08 19:54:43 +0000736 'test_unittest',
737 'test_doctest',
738 'test_doctest2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000739]
Guido van Rossum152494a1996-12-20 03:12:20 +0000740
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000741NOTTESTS = {
Jeremy Hylton62e2c7e2001-02-28 17:48:06 +0000742 'test_future1',
743 'test_future2',
Neal Norwitz94fa2ee2008-03-31 02:55:15 +0000744}
Guido van Rossum152494a1996-12-20 03:12:20 +0000745
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000746def findtests(testdir=None, stdtests=STDTESTS, nottests=NOTTESTS):
Guido van Rossum152494a1996-12-20 03:12:20 +0000747 """Return a list of all applicable test modules."""
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000748 testdir = findtestdir(testdir)
Guido van Rossum152494a1996-12-20 03:12:20 +0000749 names = os.listdir(testdir)
750 tests = []
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000751 others = set(stdtests) | nottests
Guido van Rossum152494a1996-12-20 03:12:20 +0000752 for name in names:
Florent Xiclunada7bfd52010-03-06 11:43:55 +0000753 modname, ext = os.path.splitext(name)
754 if modname[:5] == "test_" and ext == ".py" and modname not in others:
755 tests.append(modname)
756 return stdtests + sorted(tests)
Guido van Rossum152494a1996-12-20 03:12:20 +0000757
Victor Stinnerf58087b2010-05-02 17:24:51 +0000758def replace_stdout():
759 """Set stdout encoder error handler to backslashreplace (as stderr error
760 handler) to avoid UnicodeEncodeError when printing a traceback"""
Victor Stinnera3032802010-05-07 00:57:12 +0000761 if os.name == "nt":
762 # Replace sys.stdout breaks the stdout newlines on Windows: issue #8533
763 return
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000764
765 import atexit
766
Victor Stinnerf58087b2010-05-02 17:24:51 +0000767 stdout = sys.stdout
768 sys.stdout = open(stdout.fileno(), 'w',
769 encoding=stdout.encoding,
Victor Stinner4b2b43d2011-01-05 03:54:26 +0000770 errors="backslashreplace",
771 closefd=False)
772
773 def restore_stdout():
774 sys.stdout.close()
775 sys.stdout = stdout
776 atexit.register(restore_stdout)
Victor Stinnerf58087b2010-05-02 17:24:51 +0000777
Antoine Pitrou88909542009-06-29 13:54:42 +0000778def runtest(test, verbose, quiet,
Antoine Pitrou293954d2011-03-23 23:01:49 +0100779 huntrleaks=False, debug=False, use_resources=None,
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200780 output_on_failure=False, failfast=False):
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000781 """Run a single test.
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000782
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000783 test -- the name of the test
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000784 verbose -- if true, print more messages
Trent Mickf29f47b2000-08-11 19:02:59 +0000785 quiet -- if true, don't print 'skipped' messages (probably redundant)
Christian Heimesb186d002008-03-18 15:15:01 +0000786 test_times -- a list of (time, test_name) pairs
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000787 huntrleaks -- run multiple times to test for leaks; requires a debug
788 build; a triple corresponding to -R's three arguments
Victor Stinner8313d6a2011-06-29 15:22:26 +0200789 output_on_failure -- if true, display test output on failure
Benjamin Petersona0dfa822009-11-13 02:25:08 +0000790
791 Returns one of the test result constants:
792 INTERRUPTED KeyboardInterrupt when run under -j
793 RESOURCE_DENIED test skipped because resource denied
794 SKIPPED test skipped for some other reason
795 ENV_CHANGED test failed because it changed the execution environment
796 FAILED test failed
797 PASSED test passed
Guido van Rossum6fd83b71998-08-01 17:04:08 +0000798 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +0000799
Antoine Pitrou88909542009-06-29 13:54:42 +0000800 if use_resources is not None:
801 support.use_resources = use_resources
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000802 try:
Antoine Pitrou216a3bc2011-07-23 22:33:39 +0200803 if failfast:
804 support.failfast = True
Victor Stinner8313d6a2011-06-29 15:22:26 +0200805 if output_on_failure:
Victor Stinnerea95de72011-06-29 15:34:48 +0200806 support.verbose = True
807
808 # Reuse the same instance to all calls to runtest(). Some
809 # tests keep a reference to sys.stdout or sys.stderr
810 # (eg. test_argparse).
Victor Stinner8313d6a2011-06-29 15:22:26 +0200811 if runtest.stringio is None:
Victor Stinnerfcc2a212011-06-29 20:01:29 +0200812 stream = io.StringIO()
813 runtest.stringio = stream
814 else:
815 stream = runtest.stringio
816 stream.seek(0)
817 stream.truncate()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200818
819 orig_stdout = sys.stdout
Victor Stinnera7c33e52011-06-29 13:00:54 +0200820 orig_stderr = sys.stderr
Victor Stinner8313d6a2011-06-29 15:22:26 +0200821 try:
Victor Stinnerea95de72011-06-29 15:34:48 +0200822 sys.stdout = stream
823 sys.stderr = stream
Victor Stinner8313d6a2011-06-29 15:22:26 +0200824 result = runtest_inner(test, verbose, quiet, huntrleaks,
825 debug, display_failure=False)
826 if result[0] == FAILED:
Victor Stinnerea95de72011-06-29 15:34:48 +0200827 output = stream.getvalue()
Victor Stinner8313d6a2011-06-29 15:22:26 +0200828 orig_stderr.write(output)
829 orig_stderr.flush()
830 finally:
831 sys.stdout = orig_stdout
832 sys.stderr = orig_stderr
Victor Stinnera7c33e52011-06-29 13:00:54 +0200833 else:
Victor Stinnerea95de72011-06-29 15:34:48 +0200834 support.verbose = verbose # Tell tests to be moderately quiet
Victor Stinnera7c33e52011-06-29 13:00:54 +0200835 result = runtest_inner(test, verbose, quiet, huntrleaks, debug,
836 display_failure=not verbose)
Antoine Pitrou293954d2011-03-23 23:01:49 +0100837 return result
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000838 finally:
839 cleanup_test_droppings(test, verbose)
Victor Stinner8313d6a2011-06-29 15:22:26 +0200840runtest.stringio = None
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000841
Nick Coghlan6ead5522009-10-18 13:19:33 +0000842# Unit tests are supposed to leave the execution environment unchanged
843# once they complete. But sometimes tests have bugs, especially when
844# tests fail, and the changes to environment go on to mess up other
845# tests. This can cause issues with buildbot stability, since tests
846# are run in random order and so problems may appear to come and go.
847# There are a few things we can save and restore to mitigate this, and
848# the following context manager handles this task.
849
850class saved_test_environment:
851 """Save bits of the test environment and restore them at block exit.
852
853 with saved_test_environment(testname, verbose, quiet):
854 #stuff
855
856 Unless quiet is True, a warning is printed to stderr if any of
857 the saved items was changed by the test. The attribute 'changed'
858 is initially False, but is set to True if a change is detected.
859
860 If verbose is more than 1, the before and after state of changed
861 items is also printed.
862 """
863
864 changed = False
865
866 def __init__(self, testname, verbose=0, quiet=False):
867 self.testname = testname
868 self.verbose = verbose
869 self.quiet = quiet
870
871 # To add things to save and restore, add a name XXX to the resources list
872 # and add corresponding get_XXX/restore_XXX functions. get_XXX should
873 # return the value to be saved and compared against a second call to the
874 # get function when test execution completes. restore_XXX should accept
875 # the saved value and restore the resource using it. It will be called if
876 # and only if a change in the value is detected.
877 #
878 # Note: XXX will have any '.' replaced with '_' characters when determining
879 # the corresponding method names.
880
881 resources = ('sys.argv', 'cwd', 'sys.stdin', 'sys.stdout', 'sys.stderr',
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000882 'os.environ', 'sys.path', 'sys.path_hooks', '__import__',
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000883 'warnings.filters', 'asyncore.socket_map',
Ezio Melotti0123e052011-03-20 15:09:26 +0200884 'logging._handlers', 'logging._handlerList',
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200885 'sys.warnoptions', 'threading._dangling',
886 'multiprocessing.process._dangling')
Nick Coghlan6ead5522009-10-18 13:19:33 +0000887
888 def get_sys_argv(self):
889 return id(sys.argv), sys.argv, sys.argv[:]
890 def restore_sys_argv(self, saved_argv):
891 sys.argv = saved_argv[1]
892 sys.argv[:] = saved_argv[2]
893
894 def get_cwd(self):
895 return os.getcwd()
896 def restore_cwd(self, saved_cwd):
897 os.chdir(saved_cwd)
898
899 def get_sys_stdout(self):
900 return sys.stdout
901 def restore_sys_stdout(self, saved_stdout):
902 sys.stdout = saved_stdout
903
904 def get_sys_stderr(self):
905 return sys.stderr
906 def restore_sys_stderr(self, saved_stderr):
907 sys.stderr = saved_stderr
908
909 def get_sys_stdin(self):
910 return sys.stdin
911 def restore_sys_stdin(self, saved_stdin):
912 sys.stdin = saved_stdin
913
914 def get_os_environ(self):
915 return id(os.environ), os.environ, dict(os.environ)
916 def restore_os_environ(self, saved_environ):
917 os.environ = saved_environ[1]
918 os.environ.clear()
919 os.environ.update(saved_environ[2])
920
921 def get_sys_path(self):
922 return id(sys.path), sys.path, sys.path[:]
923 def restore_sys_path(self, saved_path):
924 sys.path = saved_path[1]
925 sys.path[:] = saved_path[2]
926
Brett Cannon055470a2010-02-19 15:57:10 +0000927 def get_sys_path_hooks(self):
928 return id(sys.path_hooks), sys.path_hooks, sys.path_hooks[:]
929 def restore_sys_path_hooks(self, saved_hooks):
930 sys.path_hooks = saved_hooks[1]
931 sys.path_hooks[:] = saved_hooks[2]
932
933 def get___import__(self):
Brett Cannon45071902010-06-14 22:22:54 +0000934 return builtins.__import__
Brett Cannon055470a2010-02-19 15:57:10 +0000935 def restore___import__(self, import_):
Brett Cannon45071902010-06-14 22:22:54 +0000936 builtins.__import__ = import_
Brett Cannon055470a2010-02-19 15:57:10 +0000937
Brett Cannon29c0e4f2010-03-20 22:22:57 +0000938 def get_warnings_filters(self):
939 return id(warnings.filters), warnings.filters, warnings.filters[:]
940 def restore_warnings_filters(self, saved_filters):
941 warnings.filters = saved_filters[1]
942 warnings.filters[:] = saved_filters[2]
943
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000944 def get_asyncore_socket_map(self):
945 asyncore = sys.modules.get('asyncore')
Antoine Pitrouaa879652010-10-29 11:54:38 +0000946 # XXX Making a copy keeps objects alive until __exit__ gets called.
947 return asyncore and asyncore.socket_map.copy() or {}
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000948 def restore_asyncore_socket_map(self, saved_map):
949 asyncore = sys.modules.get('asyncore')
950 if asyncore is not None:
Antoine Pitrouaa879652010-10-29 11:54:38 +0000951 asyncore.close_all(ignore_all=True)
Antoine Pitroub14ac8c2010-08-16 00:28:05 +0000952 asyncore.socket_map.update(saved_map)
953
Nick Coghlan7d819752010-12-05 06:45:03 +0000954 def get_logging__handlers(self):
955 # _handlers is a WeakValueDictionary
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000956 return id(logging._handlers), logging._handlers, logging._handlers.copy()
Nick Coghlan7d819752010-12-05 06:45:03 +0000957 def restore_logging__handlers(self, saved_handlers):
958 # Can't easily revert the logging state
959 pass
960
Nick Coghlan7bd5dbe2010-12-05 07:17:25 +0000961 def get_logging__handlerList(self):
962 # _handlerList is a list of weakrefs to handlers
963 return id(logging._handlerList), logging._handlerList, logging._handlerList[:]
964 def restore_logging__handlerList(self, saved_handlerList):
965 # Can't easily revert the logging state
966 pass
967
Ezio Melotti0123e052011-03-20 15:09:26 +0200968 def get_sys_warnoptions(self):
969 return id(sys.warnoptions), sys.warnoptions, sys.warnoptions[:]
970 def restore_sys_warnoptions(self, saved_options):
971 sys.warnoptions = saved_options[1]
972 sys.warnoptions[:] = saved_options[2]
973
Antoine Pitrouc081c0c2011-07-15 22:12:24 +0200974 # Controlling dangling references to Thread objects can make it easier
975 # to track reference leaks.
976 def get_threading__dangling(self):
977 if not threading:
978 return None
979 # This copies the weakrefs without making any strong reference
980 return threading._dangling.copy()
981 def restore_threading__dangling(self, saved):
982 if not threading:
983 return
984 threading._dangling.clear()
985 threading._dangling.update(saved)
986
987 # Same for Process objects
988 def get_multiprocessing_process__dangling(self):
989 if not multiprocessing:
990 return None
991 # This copies the weakrefs without making any strong reference
992 return multiprocessing.process._dangling.copy()
993 def restore_multiprocessing_process__dangling(self, saved):
994 if not multiprocessing:
995 return
996 multiprocessing.process._dangling.clear()
997 multiprocessing.process._dangling.update(saved)
998
Nick Coghlan6ead5522009-10-18 13:19:33 +0000999 def resource_info(self):
1000 for name in self.resources:
1001 method_suffix = name.replace('.', '_')
1002 get_name = 'get_' + method_suffix
1003 restore_name = 'restore_' + method_suffix
1004 yield name, getattr(self, get_name), getattr(self, restore_name)
1005
1006 def __enter__(self):
1007 self.saved_values = dict((name, get()) for name, get, restore
1008 in self.resource_info())
1009 return self
1010
1011 def __exit__(self, exc_type, exc_val, exc_tb):
Antoine Pitrouaa879652010-10-29 11:54:38 +00001012 saved_values = self.saved_values
1013 del self.saved_values
Nick Coghlan6ead5522009-10-18 13:19:33 +00001014 for name, get, restore in self.resource_info():
1015 current = get()
Antoine Pitrouaa879652010-10-29 11:54:38 +00001016 original = saved_values.pop(name)
Nick Coghlan6ead5522009-10-18 13:19:33 +00001017 # Check for changes to the resource's value
1018 if current != original:
1019 self.changed = True
1020 restore(original)
1021 if not self.quiet:
1022 print("Warning -- {} was modified by {}".format(
1023 name, self.testname),
1024 file=sys.stderr)
1025 if self.verbose > 1:
1026 print(" Before: {}\n After: {} ".format(
1027 original, current),
1028 file=sys.stderr)
1029 return False
1030
1031
Victor Stinnera7c33e52011-06-29 13:00:54 +02001032def runtest_inner(test, verbose, quiet,
1033 huntrleaks=False, debug=False, display_failure=True):
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001034 support.unload(test)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001035
Antoine Pitrou88909542009-06-29 13:54:42 +00001036 test_time = 0.0
Collin Wintera5503d52009-05-15 01:20:21 +00001037 refleak = False # True if the test leaked references.
Guido van Rossum152494a1996-12-20 03:12:20 +00001038 try:
R. David Murray0bc11ae2009-10-18 22:18:17 +00001039 if test.startswith('test.'):
1040 abstest = test
1041 else:
1042 # Always import it from the test package
1043 abstest = 'test.' + test
1044 with saved_test_environment(test, verbose, quiet) as environment:
1045 start_time = time.time()
1046 the_package = __import__(abstest, globals(), locals(), [])
1047 the_module = getattr(the_package, test)
1048 # Old tests run to completion simply as a side-effect of
1049 # being imported. For tests based on unittest or doctest,
1050 # explicitly invoke their test_main() function (if it exists).
1051 indirect_test = getattr(the_module, "test_main", None)
1052 if indirect_test is not None:
1053 indirect_test()
1054 if huntrleaks:
1055 refleak = dash_R(the_module, test, indirect_test,
1056 huntrleaks)
1057 test_time = time.time() - start_time
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001058 except support.ResourceDenied as msg:
Fred Drake9a0db072003-02-03 15:19:30 +00001059 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001060 print(test, "skipped --", msg)
Fred Drake9a0db072003-02-03 15:19:30 +00001061 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001062 return RESOURCE_DENIED, test_time
R. David Murraya21e4ca2009-03-31 23:16:50 +00001063 except unittest.SkipTest as msg:
Trent Mickf29f47b2000-08-11 19:02:59 +00001064 if not quiet:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001065 print(test, "skipped --", msg)
Guido van Rossum3cda93e2002-09-13 21:28:03 +00001066 sys.stdout.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001067 return SKIPPED, test_time
Fred Drakefe5c22a2000-08-18 16:04:05 +00001068 except KeyboardInterrupt:
1069 raise
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001070 except support.TestFailed as msg:
Victor Stinnera7c33e52011-06-29 13:00:54 +02001071 if display_failure:
1072 print("test", test, "failed --", msg, file=sys.stderr)
1073 else:
1074 print("test", test, "failed", file=sys.stderr)
R. David Murray11cabcf2010-09-29 01:08:05 +00001075 sys.stderr.flush()
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001076 return FAILED, test_time
Guido van Rossum9e48b271997-07-16 01:56:13 +00001077 except:
Antoine Pitrou779a5b02011-03-21 19:55:16 +01001078 msg = traceback.format_exc()
1079 print("test", test, "crashed --", msg, 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 Rossum152494a1996-12-20 03:12:20 +00001082 else:
Collin Wintera5503d52009-05-15 01:20:21 +00001083 if refleak:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001084 return FAILED, test_time
Nick Coghlan6ead5522009-10-18 13:19:33 +00001085 if environment.changed:
Benjamin Petersona0dfa822009-11-13 02:25:08 +00001086 return ENV_CHANGED, test_time
1087 return PASSED, test_time
Guido van Rossum0fcca4e2001-09-21 20:31:52 +00001088
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001089def cleanup_test_droppings(testname, verbose):
1090 import shutil
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001091 import stat
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001092 import gc
1093
1094 # First kill any dangling references to open files etc.
1095 # This can also issue some ResourceWarnings which would otherwise get
Antoine Pitrou2b40efd2010-10-29 19:36:37 +00001096 # triggered during the following test run, and possibly produce failures.
Antoine Pitrouc14efc42010-10-29 19:34:45 +00001097 gc.collect()
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001098
1099 # Try to clean up junk commonly left behind. While tests shouldn't leave
1100 # any files or directories behind, when a test fails that can be tedious
1101 # for it to arrange. The consequences can be especially nasty on Windows,
1102 # since if a test leaves a file open, it cannot be deleted by name (while
1103 # there's nothing we can do about that here either, we can display the
1104 # name of the offending test, which is a real help).
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001105 for name in (support.TESTFN,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001106 "db_home",
1107 ):
1108 if not os.path.exists(name):
1109 continue
1110
1111 if os.path.isdir(name):
1112 kind, nuker = "directory", shutil.rmtree
1113 elif os.path.isfile(name):
1114 kind, nuker = "file", os.unlink
1115 else:
1116 raise SystemError("os.path says %r exists but is neither "
1117 "directory nor file" % name)
1118
1119 if verbose:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001120 print("%r left behind %s %r" % (testname, kind, name))
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001121 try:
Benjamin Peterson25c95f12009-05-08 20:42:26 +00001122 # if we have chmod, fix possible permissions problems
1123 # that might prevent cleanup
1124 if (hasattr(os, 'chmod')):
1125 os.chmod(name, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001126 nuker(name)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001127 except Exception as msg:
1128 print(("%r left behind %s %r and it couldn't be "
1129 "removed: %s" % (testname, kind, name, msg)), file=sys.stderr)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001130
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001131def dash_R(the_module, test, indirect_test, huntrleaks):
Collin Wintera5503d52009-05-15 01:20:21 +00001132 """Run a test multiple times, looking for reference leaks.
1133
1134 Returns:
1135 False if the test didn't leak references; True if we detected refleaks.
1136 """
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001137 # This code is hackish and inelegant, but it seems to do the job.
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001138 import copyreg, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001139
1140 if not hasattr(sys, 'gettotalrefcount'):
1141 raise Exception("Tracking reference leaks requires a debug build "
1142 "of Python")
1143
1144 # Save current values for dash_R_cleanup() to restore.
1145 fs = warnings.filters[:]
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001146 ps = copyreg.dispatch_table.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001147 pic = sys.path_importer_cache.copy()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001148 try:
1149 import zipimport
1150 except ImportError:
1151 zdc = None # Run unmodified on platforms without zipimport support
1152 else:
1153 zdc = zipimport._zip_directory_cache.copy()
Christian Heimes93852662007-12-01 12:22:32 +00001154 abcs = {}
1155 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001156 if not isabstract(abc):
Christian Heimes93852662007-12-01 12:22:32 +00001157 continue
1158 for obj in abc.__subclasses__() + [abc]:
1159 abcs[obj] = obj._abc_registry.copy()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001160
1161 if indirect_test:
1162 def run_the_test():
1163 indirect_test()
1164 else:
1165 def run_the_test():
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001166 del sys.modules[the_module.__name__]
1167 exec('import ' + the_module.__name__)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001168
1169 deltas = []
1170 nwarmup, ntracked, fname = huntrleaks
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001171 fname = os.path.join(support.SAVEDCWD, fname)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001172 repcount = nwarmup + ntracked
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001173 print("beginning", repcount, "repetitions", file=sys.stderr)
1174 print(("1234567890"*(repcount//10 + 1))[:repcount], file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001175 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001176 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001177 for i in range(repcount):
Collin Winter58bee592010-03-17 02:08:57 +00001178 rc_before = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001179 run_the_test()
1180 sys.stderr.write('.')
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001181 sys.stderr.flush()
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001182 dash_R_cleanup(fs, ps, pic, zdc, abcs)
Collin Winter58bee592010-03-17 02:08:57 +00001183 rc_after = sys.gettotalrefcount()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001184 if i >= nwarmup:
Collin Winter58bee592010-03-17 02:08:57 +00001185 deltas.append(rc_after - rc_before)
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001186 print(file=sys.stderr)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001187 if any(deltas):
Guido van Rossum360e4b82007-05-14 22:51:27 +00001188 msg = '%s leaked %s references, sum=%s' % (test, deltas, sum(deltas))
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001189 print(msg, file=sys.stderr)
Antoine Pitrou88909542009-06-29 13:54:42 +00001190 sys.stderr.flush()
1191 with open(fname, "a") as refrep:
1192 print(msg, file=refrep)
1193 refrep.flush()
Collin Wintera5503d52009-05-15 01:20:21 +00001194 return True
1195 return False
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001196
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001197def dash_R_cleanup(fs, ps, pic, zdc, abcs):
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001198 import gc, copyreg
Brett Cannonf4fd9932008-05-10 21:11:46 +00001199 import _strptime, linecache
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001200 import urllib.parse, urllib.request, mimetypes, doctest
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001201 import struct, filecmp, _abcoll
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001202 from distutils.dir_util import _path_created
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001203 from weakref import WeakSet
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001204
Christian Heimesdae2a892008-04-19 00:55:37 +00001205 # Clear the warnings registry, so they can be displayed again
1206 for mod in sys.modules.values():
1207 if hasattr(mod, '__warningregistry__'):
1208 del mod.__warningregistry__
1209
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001210 # Restore some original values.
1211 warnings.filters[:] = fs
Alexandre Vassalottif7fa63d2008-05-11 08:55:36 +00001212 copyreg.dispatch_table.clear()
1213 copyreg.dispatch_table.update(ps)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001214 sys.path_importer_cache.clear()
1215 sys.path_importer_cache.update(pic)
Nick Coghlan260bd3e2009-11-16 06:49:25 +00001216 try:
1217 import zipimport
1218 except ImportError:
1219 pass # Run unmodified on platforms without zipimport support
1220 else:
1221 zipimport._zip_directory_cache.clear()
1222 zipimport._zip_directory_cache.update(zdc)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001223
Christian Heimes26855632008-01-27 23:50:43 +00001224 # clear type cache
Christian Heimes15ebc882008-02-04 18:48:49 +00001225 sys._clear_type_cache()
Christian Heimes26855632008-01-27 23:50:43 +00001226
Guido van Rossum3de862d2007-08-18 00:10:33 +00001227 # Clear ABC registries, restoring previously saved ABC registries.
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001228 for abc in [getattr(_abcoll, a) for a in _abcoll.__all__]:
Christian Heimesbe5b30b2008-03-03 19:18:51 +00001229 if not isabstract(abc):
Christian Heimes941973a2007-11-30 21:53:03 +00001230 continue
Guido van Rossum7eaf8222007-06-18 17:58:50 +00001231 for obj in abc.__subclasses__() + [abc]:
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001232 obj._abc_registry = abcs.get(obj, WeakSet()).copy()
Guido van Rossumc1e315d2007-08-20 19:29:24 +00001233 obj._abc_cache.clear()
1234 obj._abc_negative_cache.clear()
Guido van Rossumcd16bf62007-06-13 18:07:49 +00001235
Antoine Pitrou046467c2009-10-30 18:30:35 +00001236 # Flush standard output, so that buffered data is sent to the OS and
1237 # associated Python objects are reclaimed.
1238 for stream in (sys.stdout, sys.stderr, sys.__stdout__, sys.__stderr__):
1239 if stream is not None:
1240 stream.flush()
1241
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001242 # Clear assorted module caches.
1243 _path_created.clear()
1244 re.purge()
1245 _strptime._regex_cache.clear()
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001246 urllib.parse.clear_cache()
1247 urllib.request.urlcleanup()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001248 linecache.clearcache()
1249 mimetypes._default_mime_types()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001250 filecmp._cache.clear()
Christian Heimesa34706f2008-01-04 03:06:10 +00001251 struct._clearcache()
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00001252 doctest.master = None
1253
1254 # Collect cyclic trash.
1255 gc.collect()
1256
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001257def warm_char_cache():
1258 s = bytes(range(256))
1259 for i in range(256):
1260 s[i:i+1]
1261
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001262def findtestdir(path=None):
1263 return path or os.path.dirname(__file__) or os.curdir
Guido van Rossum152494a1996-12-20 03:12:20 +00001264
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001265def removepy(names):
1266 if not names:
1267 return
1268 for idx, name in enumerate(names):
1269 basename, ext = os.path.splitext(name)
1270 if ext == '.py':
1271 names[idx] = basename
Tim Petersc5000df2002-06-02 21:42:01 +00001272
Guido van Rossum152494a1996-12-20 03:12:20 +00001273def count(n, word):
1274 if n == 1:
Guido van Rossum41360a41998-03-26 19:42:58 +00001275 return "%d %s" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001276 else:
Guido van Rossum41360a41998-03-26 19:42:58 +00001277 return "%d %ss" % (n, word)
Guido van Rossum152494a1996-12-20 03:12:20 +00001278
Tim Petersa45da922001-08-12 03:45:50 +00001279def printlist(x, width=70, indent=4):
Tim Peters7c7efe92002-08-23 17:55:54 +00001280 """Print the elements of iterable x to stdout.
Tim Petersa45da922001-08-12 03:45:50 +00001281
1282 Optional arg width (default 70) is the maximum line length.
1283 Optional arg indent (default 4) is the number of blanks with which to
1284 begin each line.
1285 """
1286
Tim Petersba78bc42002-07-04 19:45:06 +00001287 from textwrap import fill
1288 blanks = ' ' * indent
Florent Xiclunafd1b0932010-03-28 00:25:02 +00001289 # Print the sorted list: 'x' may be a '--random' list or a set()
1290 print(fill(' '.join(str(elt) for elt in sorted(x)), width,
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001291 initial_indent=blanks, subsequent_indent=blanks))
Tim Petersa45da922001-08-12 03:45:50 +00001292
Tim Petersde14a302002-04-01 05:04:46 +00001293# Map sys.platform to a string containing the basenames of tests
1294# expected to be skipped on that platform.
Tim Peters2a182db2002-10-09 01:07:11 +00001295#
1296# Special cases:
1297# test_pep277
1298# The _ExpectedSkips constructor adds this to the set of expected
1299# skips if not os.path.supports_unicode_filenames.
Neal Norwitz55b61d22003-02-28 19:57:03 +00001300# test_timeout
1301# Controlled by test_timeout.skip_expected. Requires the network
1302# resource and a socket module.
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001303#
1304# Tests that are expected to be skipped everywhere except on one platform
1305# are also handled separately.
Tim Petersde14a302002-04-01 05:04:46 +00001306
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001307_expectations = {
1308 'win32':
1309 """
Tim Petersc7c516a2003-09-20 22:06:13 +00001310 test__locale
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001311 test_crypt
Tim Petersd7030572001-10-22 22:06:08 +00001312 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001313 test_dbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001314 test_fcntl
1315 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001316 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001317 test_dbm_gnu
Kristján Valur Jónsson42a40c52009-04-01 11:28:47 +00001318 test_dbm_ndbm
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001319 test_grp
Tim Petersfd8e6e52003-03-04 00:26:38 +00001320 test_ioctl
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001321 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001322 test_kqueue
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001323 test_openpty
Tim Petersefc4b122002-12-10 18:47:56 +00001324 test_ossaudiodev
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001325 test_pipes
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001326 test_poll
Tim Peters003eb302003-02-17 21:48:48 +00001327 test_posix
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001328 test_pty
1329 test_pwd
Tim Peters1e33ffa2002-04-23 23:09:02 +00001330 test_resource
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001331 test_signal
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001332 test_syslog
Tim Peterscea2cc42004-08-04 02:32:03 +00001333 test_threadsignals
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001334 test_wait3
1335 test_wait4
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001336 """,
1337 'linux2':
1338 """
Guido van Rossumf66dacd2001-10-23 15:10:55 +00001339 test_curses
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001340 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001341 test_kqueue
Guido van Rossum4507ec72003-02-14 19:29:22 +00001342 test_ossaudiodev
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001343 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001344 'unixware7':
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001345 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001346 test_epoll
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001347 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001348 test_kqueue
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001349 test_minidom
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001350 test_openpty
1351 test_pyexpat
1352 test_sax
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001353 test_sundry
Martin v. Löwis0ace3262001-09-05 14:38:48 +00001354 """,
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001355 'openunix8':
1356 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001357 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001358 test_largefile
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001359 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001360 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001361 test_openpty
1362 test_pyexpat
1363 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001364 test_sundry
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001365 """,
1366 'sco_sv3':
1367 """
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001368 test_asynchat
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001369 test_fork1
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001370 test_epoll
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001371 test_gettext
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001372 test_largefile
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001373 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001374 test_kqueue
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001375 test_minidom
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001376 test_openpty
1377 test_pyexpat
1378 test_queue
1379 test_sax
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001380 test_sundry
1381 test_thread
1382 test_threaded_import
1383 test_threadedtempfile
1384 test_threading
Martin v. Löwis21ee4092002-09-30 16:19:48 +00001385 """,
Jack Jansen8a97f4a2001-12-05 23:27:32 +00001386 'darwin':
Jack Jansen398c2362001-12-02 21:41:36 +00001387 """
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001388 test__locale
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001389 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001390 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001391 test_dbm_gnu
Skip Montanaro97019ff2010-10-25 01:35:48 +00001392 test_gdb
Jack Jansen398c2362001-12-02 21:41:36 +00001393 test_largefile
Jack Jansenacda3392002-12-30 23:03:13 +00001394 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001395 test_minidom
Jack Jansenacda3392002-12-30 23:03:13 +00001396 test_ossaudiodev
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001397 test_poll
Jack Jansen398c2362001-12-02 21:41:36 +00001398 """,
Guido van Rossum11c3f092002-07-17 15:08:24 +00001399 'sunos5':
1400 """
Guido van Rossum11c3f092002-07-17 15:08:24 +00001401 test_curses
1402 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001403 test_epoll
1404 test_kqueue
Benjamin Peterson4d480532008-05-26 19:08:31 +00001405 test_dbm_gnu
Guido van Rossum11c3f092002-07-17 15:08:24 +00001406 test_gzip
Guido van Rossum11c3f092002-07-17 15:08:24 +00001407 test_openpty
Guido van Rossum11c3f092002-07-17 15:08:24 +00001408 test_zipfile
1409 test_zlib
Jeremy Hyltoned375e12002-07-17 15:56:55 +00001410 """,
Skip Montanarob3230212002-03-15 02:54:03 +00001411 'hp-ux11':
1412 """
Skip Montanarob3230212002-03-15 02:54:03 +00001413 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001414 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001415 test_dbm_gnu
Skip Montanarob3230212002-03-15 02:54:03 +00001416 test_gzip
Skip Montanarob3230212002-03-15 02:54:03 +00001417 test_largefile
Skip Montanarob3230212002-03-15 02:54:03 +00001418 test_locale
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001419 test_kqueue
Skip Montanarob3230212002-03-15 02:54:03 +00001420 test_minidom
Skip Montanarob3230212002-03-15 02:54:03 +00001421 test_openpty
1422 test_pyexpat
1423 test_sax
Skip Montanarob3230212002-03-15 02:54:03 +00001424 test_zipfile
1425 test_zlib
1426 """,
Jason Tishler25115942002-12-05 15:18:15 +00001427 'cygwin':
1428 """
Jason Tishler25115942002-12-05 15:18:15 +00001429 test_curses
1430 test_dbm
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001431 test_epoll
Jason Tishlerc23f39c2003-07-22 18:35:58 +00001432 test_ioctl
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001433 test_kqueue
Jason Tishler25115942002-12-05 15:18:15 +00001434 test_largefile
Jason Tishler25115942002-12-05 15:18:15 +00001435 test_locale
Jason Tishler5c4ded22003-02-05 16:46:01 +00001436 test_ossaudiodev
Jason Tishler25115942002-12-05 15:18:15 +00001437 test_socketserver
Jason Tishler25115942002-12-05 15:18:15 +00001438 """,
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001439 'os2emx':
1440 """
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001441 test_audioop
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001442 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001443 test_epoll
1444 test_kqueue
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001445 test_largefile
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001446 test_mmap
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001447 test_openpty
1448 test_ossaudiodev
1449 test_pty
1450 test_resource
1451 test_signal
Andrew MacIntyrefd07e7d2002-12-31 11:26:50 +00001452 """,
Guido van Rossum944a6c32003-11-20 22:11:29 +00001453 'freebsd4':
1454 """
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001455 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001456 test_dbm_gnu
Guido van Rossum944a6c32003-11-20 22:11:29 +00001457 test_locale
Guido van Rossum944a6c32003-11-20 22:11:29 +00001458 test_ossaudiodev
1459 test_pep277
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001460 test_pty
Guido van Rossum944a6c32003-11-20 22:11:29 +00001461 test_socketserver
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001462 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001463 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001464 test_ttk_guionly
1465 test_ttk_textonly
Guido van Rossum944a6c32003-11-20 22:11:29 +00001466 test_timeout
Guido van Rossum944a6c32003-11-20 22:11:29 +00001467 test_urllibnet
Benjamin Petersone5384b02008-10-04 22:00:42 +00001468 test_multiprocessing
Martin v. Löwis56f88112003-06-07 20:01:37 +00001469 """,
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001470 'aix5':
1471 """
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001472 test_bz2
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001473 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001474 test_dbm_gnu
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001475 test_gzip
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001476 test_kqueue
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001477 test_ossaudiodev
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001478 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001479 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001480 test_ttk_guionly
1481 test_ttk_textonly
Guido van Rossum8ee3e5a2005-09-14 18:09:42 +00001482 test_zipimport
1483 test_zlib
1484 """,
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001485 'openbsd3':
1486 """
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001487 test_ctypes
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001488 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001489 test_dbm_gnu
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001490 test_locale
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001491 test_normalization
1492 test_ossaudiodev
1493 test_pep277
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001494 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001495 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001496 test_ttk_guionly
1497 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001498 test_multiprocessing
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001499 """,
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001500 'netbsd3':
1501 """
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001502 test_ctypes
1503 test_curses
Christian Heimes4fbc72b2008-03-22 00:47:35 +00001504 test_epoll
Benjamin Peterson4d480532008-05-26 19:08:31 +00001505 test_dbm_gnu
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001506 test_locale
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001507 test_ossaudiodev
1508 test_pep277
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001509 test_tcl
Guilherme Poloa91790a2009-02-09 20:40:42 +00001510 test_tk
Guilherme Polo9de29af2009-01-28 20:40:48 +00001511 test_ttk_guionly
1512 test_ttk_textonly
Benjamin Petersone5384b02008-10-04 22:00:42 +00001513 test_multiprocessing
Thomas Wouters0e3f5912006-08-11 14:57:12 +00001514 """,
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001515}
Martin v. Löwis32d0c1b2004-07-26 12:09:13 +00001516_expectations['freebsd5'] = _expectations['freebsd4']
Hye-Shik Changf64700a2004-08-18 15:13:41 +00001517_expectations['freebsd6'] = _expectations['freebsd4']
Hye-Shik Chang4e422812005-07-17 02:36:59 +00001518_expectations['freebsd7'] = _expectations['freebsd4']
Guido van Rossum8ce8a782007-11-01 19:42:39 +00001519_expectations['freebsd8'] = _expectations['freebsd4']
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001520
Tim Petersb5b7b782001-08-12 01:20:39 +00001521class _ExpectedSkips:
1522 def __init__(self):
Tim Peters2a182db2002-10-09 01:07:11 +00001523 import os.path
Neal Norwitz55b61d22003-02-28 19:57:03 +00001524 from test import test_timeout
Tim Peters1b445d32002-11-24 18:53:11 +00001525
Tim Peters7c7efe92002-08-23 17:55:54 +00001526 self.valid = False
Tim Petersde14a302002-04-01 05:04:46 +00001527 if sys.platform in _expectations:
Guido van Rossumf73e30c2001-08-12 02:22:19 +00001528 s = _expectations[sys.platform]
Raymond Hettingera690a992003-11-16 16:17:49 +00001529 self.expected = set(s.split())
Tim Peters1b445d32002-11-24 18:53:11 +00001530
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001531 # These are broken tests, for now skipped on every platform.
1532 # XXX Fix these!
Benjamin Petersone9ea19e2008-08-19 23:02:38 +00001533 self.expected.add('test_nis')
Benjamin Peterson4fde0c42008-03-31 02:36:22 +00001534
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001535 # expected to be skipped on every platform, even Linux
Tim Peters2a182db2002-10-09 01:07:11 +00001536 if not os.path.supports_unicode_filenames:
1537 self.expected.add('test_pep277')
Tim Peters1b445d32002-11-24 18:53:11 +00001538
Neal Norwitz94fa2ee2008-03-31 02:55:15 +00001539 # doctest, profile and cProfile tests fail when the codec for the
1540 # fs encoding isn't built in because PyUnicode_Decode() adds two
1541 # calls into Python.
1542 encs = ("utf-8", "latin-1", "ascii", "mbcs", "utf-16", "utf-32")
1543 if sys.getfilesystemencoding().lower() not in encs:
1544 self.expected.add('test_profile')
1545 self.expected.add('test_cProfile')
1546 self.expected.add('test_doctest')
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001547
Neal Norwitz55b61d22003-02-28 19:57:03 +00001548 if test_timeout.skip_expected:
1549 self.expected.add('test_timeout')
1550
Tim Petersecd79eb2003-01-29 00:35:32 +00001551 if sys.platform != "win32":
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001552 # test_sqlite is only reliable on Windows where the library
1553 # is distributed with Python
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001554 WIN_ONLY = {"test_unicode_file", "test_winreg",
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001555 "test_winsound", "test_startfile",
Ross Lagerwall226580e2011-04-09 20:05:04 +02001556 "test_sqlite", "test_msilib"}
Benjamin Peterson54ad0be2010-10-29 21:33:10 +00001557 self.expected |= WIN_ONLY
Tim Petersf2715e02003-02-19 02:35:07 +00001558
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001559 if sys.platform != 'sunos5':
Guido van Rossumb5a755e2007-07-18 18:15:48 +00001560 self.expected.add('test_nis')
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001561
Benjamin Peterson65c66ab2010-10-29 21:31:35 +00001562 if support.python_is_optimized():
1563 self.expected.add("test_gdb")
1564
Tim Peters7c7efe92002-08-23 17:55:54 +00001565 self.valid = True
Tim Petersb5b7b782001-08-12 01:20:39 +00001566
1567 def isvalid(self):
1568 "Return true iff _ExpectedSkips knows about the current platform."
1569 return self.valid
1570
1571 def getexpected(self):
1572 """Return set of test names we expect to skip on current platform.
1573
1574 self.isvalid() must be true.
1575 """
1576
1577 assert self.isvalid()
1578 return self.expected
1579
Michael Foord3ab34cc2010-12-03 12:27:40 +00001580def _make_temp_dir_for_build(TEMPDIR):
1581 # When tests are run from the Python build directory, it is best practice
1582 # to keep the test files in a subfolder. It eases the cleanup of leftover
1583 # files using command "make distclean".
1584 if sysconfig.is_python_build():
1585 TEMPDIR = os.path.join(sysconfig.get_config_var('srcdir'), 'build')
1586 TEMPDIR = os.path.abspath(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001587 try:
Michael Foord3ab34cc2010-12-03 12:27:40 +00001588 os.mkdir(TEMPDIR)
Antoine Pitrouee429342011-04-16 18:53:59 +02001589 except OSError as e:
1590 if e.errno != errno.EEXIST:
1591 raise
Michael Foord3ab34cc2010-12-03 12:27:40 +00001592
1593 # Define a writable temp dir that will be used as cwd while running
1594 # the tests. The name of the dir includes the pid to allow parallel
1595 # testing (see the -j option).
1596 TESTCWD = 'test_python_{}'.format(os.getpid())
1597
1598 TESTCWD = os.path.join(TEMPDIR, TESTCWD)
1599 return TEMPDIR, TESTCWD
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001600
Guido van Rossum152494a1996-12-20 03:12:20 +00001601if __name__ == '__main__':
Nick Coghlan4c4c0f22010-12-03 07:44:33 +00001602 # Remove regrtest.py's own directory from the module search path. Despite
1603 # the elimination of implicit relative imports, this is still needed to
1604 # ensure that submodules of the test package do not inappropriately appear
1605 # as top-level modules even when people (or buildbots!) invoke regrtest.py
1606 # directly instead of using the -m switch
1607 mydir = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
1608 i = len(sys.path)
1609 while i >= 0:
1610 i -= 1
1611 if os.path.abspath(os.path.normpath(sys.path[i])) == mydir:
1612 del sys.path[i]
1613
Florent Xiclunadc69e722010-09-13 16:35:02 +00001614 # findtestdir() gets the dirname out of __file__, so we have to make it
1615 # absolute before changing the working directory.
1616 # For example __file__ may be relative when running trace or profile.
1617 # See issue #9323.
1618 __file__ = os.path.abspath(__file__)
1619
1620 # sanity check
Florent Xiclunada7bfd52010-03-06 11:43:55 +00001621 assert __file__ == os.path.abspath(sys.argv[0])
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001622
Michael Foord3ab34cc2010-12-03 12:27:40 +00001623 TEMPDIR, TESTCWD = _make_temp_dir_for_build(TEMPDIR)
Ezio Melotti184bdfb2010-02-18 09:37:05 +00001624
1625 # Run the tests in a context manager that temporary changes the CWD to a
1626 # temporary and writable directory. If it's not possible to create or
1627 # change the CWD, the original CWD will be used. The original CWD is
1628 # available from support.SAVEDCWD.
1629 with support.temp_cwd(TESTCWD, quiet=True):
1630 main()