blob: 10411522dffb02ea57b8acb567b145419a1ab260 [file] [log] [blame]
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -08001"""
2Tests of regrtest.py.
Victor Stinner6b415102015-09-28 23:16:17 +02003
4Note: test_regrtest cannot be run twice in parallel.
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -08005"""
6
Victor Stinner14b46622015-10-11 10:04:26 +02007import contextlib
Charles-François Natali833bf1f2013-09-08 12:27:33 +02008import faulthandler
Victor Stinner49671462015-10-11 10:03:28 +02009import io
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030010import os.path
Victor Stinner6b415102015-09-28 23:16:17 +020011import platform
12import re
13import subprocess
14import sys
Victor Stinnerda43ee42015-11-22 15:18:54 +010015import sysconfig
Victor Stinnerd6e25022016-03-30 02:33:52 +020016import tempfile
Victor Stinner6b415102015-09-28 23:16:17 +020017import textwrap
Antoine Pitroua6a4dc82017-09-07 18:56:24 +020018import threading
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080019import unittest
Victor Stinner3844fe52015-09-26 10:38:01 +020020from test import libregrtest
21from test import support
Victor Stinner6b415102015-09-28 23:16:17 +020022
23
24Py_DEBUG = hasattr(sys, 'getobjects')
25ROOT_DIR = os.path.join(os.path.dirname(__file__), '..', '..')
26ROOT_DIR = os.path.abspath(os.path.normpath(ROOT_DIR))
27
Victor Stinner3909e582015-10-11 10:37:25 +020028TEST_INTERRUPTED = textwrap.dedent("""
29 from signal import SIGINT
30 try:
31 from _testcapi import raise_signal
32 raise_signal(SIGINT)
33 except ImportError:
34 import os
35 os.kill(os.getpid(), SIGINT)
36 """)
37
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080038
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080039class ParseArgsTestCase(unittest.TestCase):
Victor Stinner6b415102015-09-28 23:16:17 +020040 """
41 Test regrtest's argument parsing, function _parse_args().
42 """
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080043
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030044 def checkError(self, args, msg):
45 with support.captured_stderr() as err, self.assertRaises(SystemExit):
Victor Stinner3844fe52015-09-26 10:38:01 +020046 libregrtest._parse_args(args)
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030047 self.assertIn(msg, err.getvalue())
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080048
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030049 def test_help(self):
50 for opt in '-h', '--help':
51 with self.subTest(opt=opt):
52 with support.captured_stdout() as out, \
53 self.assertRaises(SystemExit):
Victor Stinner3844fe52015-09-26 10:38:01 +020054 libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030055 self.assertIn('Run Python regression tests.', out.getvalue())
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080056
Charles-François Natali833bf1f2013-09-08 12:27:33 +020057 @unittest.skipUnless(hasattr(faulthandler, 'dump_traceback_later'),
58 "faulthandler.dump_traceback_later() required")
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030059 def test_timeout(self):
Victor Stinner3844fe52015-09-26 10:38:01 +020060 ns = libregrtest._parse_args(['--timeout', '4.2'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030061 self.assertEqual(ns.timeout, 4.2)
62 self.checkError(['--timeout'], 'expected one argument')
63 self.checkError(['--timeout', 'foo'], 'invalid float value')
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080064
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030065 def test_wait(self):
Victor Stinner3844fe52015-09-26 10:38:01 +020066 ns = libregrtest._parse_args(['--wait'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030067 self.assertTrue(ns.wait)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080068
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030069 def test_slaveargs(self):
Victor Stinner3844fe52015-09-26 10:38:01 +020070 ns = libregrtest._parse_args(['--slaveargs', '[[], {}]'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030071 self.assertEqual(ns.slaveargs, '[[], {}]')
72 self.checkError(['--slaveargs'], 'expected one argument')
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080073
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030074 def test_start(self):
75 for opt in '-S', '--start':
76 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +020077 ns = libregrtest._parse_args([opt, 'foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030078 self.assertEqual(ns.start, 'foo')
79 self.checkError([opt], 'expected one argument')
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080080
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030081 def test_verbose(self):
Victor Stinner3844fe52015-09-26 10:38:01 +020082 ns = libregrtest._parse_args(['-v'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030083 self.assertEqual(ns.verbose, 1)
Victor Stinner3844fe52015-09-26 10:38:01 +020084 ns = libregrtest._parse_args(['-vvv'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030085 self.assertEqual(ns.verbose, 3)
Victor Stinner3844fe52015-09-26 10:38:01 +020086 ns = libregrtest._parse_args(['--verbose'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030087 self.assertEqual(ns.verbose, 1)
Victor Stinner3844fe52015-09-26 10:38:01 +020088 ns = libregrtest._parse_args(['--verbose'] * 3)
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030089 self.assertEqual(ns.verbose, 3)
Victor Stinner3844fe52015-09-26 10:38:01 +020090 ns = libregrtest._parse_args([])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030091 self.assertEqual(ns.verbose, 0)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080092
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030093 def test_verbose2(self):
94 for opt in '-w', '--verbose2':
95 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +020096 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030097 self.assertTrue(ns.verbose2)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -080098
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +030099 def test_verbose3(self):
100 for opt in '-W', '--verbose3':
101 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200102 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300103 self.assertTrue(ns.verbose3)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800104
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300105 def test_quiet(self):
106 for opt in '-q', '--quiet':
107 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200108 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300109 self.assertTrue(ns.quiet)
110 self.assertEqual(ns.verbose, 0)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800111
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300112 def test_slow(self):
Victor Stinner6c446192016-08-17 11:25:43 +0200113 for opt in '-o', '--slowest':
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300114 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200115 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300116 self.assertTrue(ns.print_slow)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800117
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300118 def test_header(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200119 ns = libregrtest._parse_args(['--header'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300120 self.assertTrue(ns.header)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800121
Victor Stinner3d005682017-05-04 15:21:12 +0200122 ns = libregrtest._parse_args(['--verbose'])
123 self.assertTrue(ns.header)
124
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800125 def test_randomize(self):
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300126 for opt in '-r', '--randomize':
127 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200128 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300129 self.assertTrue(ns.randomize)
130
131 def test_randseed(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200132 ns = libregrtest._parse_args(['--randseed', '12345'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300133 self.assertEqual(ns.random_seed, 12345)
134 self.assertTrue(ns.randomize)
135 self.checkError(['--randseed'], 'expected one argument')
136 self.checkError(['--randseed', 'foo'], 'invalid int value')
137
138 def test_fromfile(self):
139 for opt in '-f', '--fromfile':
140 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200141 ns = libregrtest._parse_args([opt, 'foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300142 self.assertEqual(ns.fromfile, 'foo')
143 self.checkError([opt], 'expected one argument')
144 self.checkError([opt, 'foo', '-s'], "don't go together")
145
146 def test_exclude(self):
147 for opt in '-x', '--exclude':
148 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200149 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300150 self.assertTrue(ns.exclude)
151
152 def test_single(self):
153 for opt in '-s', '--single':
154 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200155 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300156 self.assertTrue(ns.single)
157 self.checkError([opt, '-f', 'foo'], "don't go together")
158
159 def test_match(self):
160 for opt in '-m', '--match':
161 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200162 ns = libregrtest._parse_args([opt, 'pattern'])
Victor Stinneref8320c2017-06-09 10:18:48 +0200163 self.assertEqual(ns.match_tests, ['pattern'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300164 self.checkError([opt], 'expected one argument')
165
Victor Stinneref8320c2017-06-09 10:18:48 +0200166 ns = libregrtest._parse_args(['-m', 'pattern1',
167 '-m', 'pattern2'])
168 self.assertEqual(ns.match_tests, ['pattern1', 'pattern2'])
169
170 self.addCleanup(support.unlink, support.TESTFN)
171 with open(support.TESTFN, "w") as fp:
172 print('matchfile1', file=fp)
173 print('matchfile2', file=fp)
174
175 filename = os.path.abspath(support.TESTFN)
176 ns = libregrtest._parse_args(['-m', 'match',
177 '--matchfile', filename])
178 self.assertEqual(ns.match_tests,
179 ['match', 'matchfile1', 'matchfile2'])
180
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300181 def test_failfast(self):
182 for opt in '-G', '--failfast':
183 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200184 ns = libregrtest._parse_args([opt, '-v'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300185 self.assertTrue(ns.failfast)
Victor Stinner3844fe52015-09-26 10:38:01 +0200186 ns = libregrtest._parse_args([opt, '-W'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300187 self.assertTrue(ns.failfast)
188 self.checkError([opt], '-G/--failfast needs either -v or -W')
189
190 def test_use(self):
191 for opt in '-u', '--use':
192 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200193 ns = libregrtest._parse_args([opt, 'gui,network'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300194 self.assertEqual(ns.use_resources, ['gui', 'network'])
Victor Stinner5b392bb2017-07-20 15:46:32 +0200195
Victor Stinner3844fe52015-09-26 10:38:01 +0200196 ns = libregrtest._parse_args([opt, 'gui,none,network'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300197 self.assertEqual(ns.use_resources, ['network'])
Victor Stinner5b392bb2017-07-20 15:46:32 +0200198
199 expected = list(libregrtest.ALL_RESOURCES)
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300200 expected.remove('gui')
Victor Stinner3844fe52015-09-26 10:38:01 +0200201 ns = libregrtest._parse_args([opt, 'all,-gui'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300202 self.assertEqual(ns.use_resources, expected)
203 self.checkError([opt], 'expected one argument')
204 self.checkError([opt, 'foo'], 'invalid resource')
205
Victor Stinner5b392bb2017-07-20 15:46:32 +0200206 # all + a resource not part of "all"
207 ns = libregrtest._parse_args([opt, 'all,tzdata'])
208 self.assertEqual(ns.use_resources,
209 list(libregrtest.ALL_RESOURCES) + ['tzdata'])
210
211 # test another resource which is not part of "all"
212 ns = libregrtest._parse_args([opt, 'extralargefile'])
213 self.assertEqual(ns.use_resources, ['extralargefile'])
214
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300215 def test_memlimit(self):
216 for opt in '-M', '--memlimit':
217 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200218 ns = libregrtest._parse_args([opt, '4G'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300219 self.assertEqual(ns.memlimit, '4G')
220 self.checkError([opt], 'expected one argument')
221
222 def test_testdir(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200223 ns = libregrtest._parse_args(['--testdir', 'foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300224 self.assertEqual(ns.testdir, os.path.join(support.SAVEDCWD, 'foo'))
225 self.checkError(['--testdir'], 'expected one argument')
226
Benjamin Peterson95536b82013-09-29 15:18:43 -0400227 def test_runleaks(self):
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300228 for opt in '-L', '--runleaks':
229 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200230 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300231 self.assertTrue(ns.runleaks)
232
Benjamin Peterson95536b82013-09-29 15:18:43 -0400233 def test_huntrleaks(self):
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300234 for opt in '-R', '--huntrleaks':
235 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200236 ns = libregrtest._parse_args([opt, ':'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300237 self.assertEqual(ns.huntrleaks, (5, 4, 'reflog.txt'))
Victor Stinner3844fe52015-09-26 10:38:01 +0200238 ns = libregrtest._parse_args([opt, '6:'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300239 self.assertEqual(ns.huntrleaks, (6, 4, 'reflog.txt'))
Victor Stinner3844fe52015-09-26 10:38:01 +0200240 ns = libregrtest._parse_args([opt, ':3'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300241 self.assertEqual(ns.huntrleaks, (5, 3, 'reflog.txt'))
Victor Stinner3844fe52015-09-26 10:38:01 +0200242 ns = libregrtest._parse_args([opt, '6:3:leaks.log'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300243 self.assertEqual(ns.huntrleaks, (6, 3, 'leaks.log'))
244 self.checkError([opt], 'expected one argument')
245 self.checkError([opt, '6'],
246 'needs 2 or 3 colon-separated arguments')
247 self.checkError([opt, 'foo:'], 'invalid huntrleaks value')
248 self.checkError([opt, '6:foo'], 'invalid huntrleaks value')
249
250 def test_multiprocess(self):
251 for opt in '-j', '--multiprocess':
252 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200253 ns = libregrtest._parse_args([opt, '2'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300254 self.assertEqual(ns.use_mp, 2)
255 self.checkError([opt], 'expected one argument')
256 self.checkError([opt, 'foo'], 'invalid int value')
257 self.checkError([opt, '2', '-T'], "don't go together")
258 self.checkError([opt, '2', '-l'], "don't go together")
Xiang Zhang270a21f2016-12-19 21:01:33 +0800259 self.checkError([opt, '0', '-T'], "don't go together")
260 self.checkError([opt, '0', '-l'], "don't go together")
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300261
Benjamin Peterson95536b82013-09-29 15:18:43 -0400262 def test_coverage(self):
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300263 for opt in '-T', '--coverage':
264 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200265 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300266 self.assertTrue(ns.trace)
267
268 def test_coverdir(self):
269 for opt in '-D', '--coverdir':
270 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200271 ns = libregrtest._parse_args([opt, 'foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300272 self.assertEqual(ns.coverdir,
273 os.path.join(support.SAVEDCWD, 'foo'))
274 self.checkError([opt], 'expected one argument')
275
276 def test_nocoverdir(self):
277 for opt in '-N', '--nocoverdir':
278 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200279 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300280 self.assertIsNone(ns.coverdir)
281
282 def test_threshold(self):
283 for opt in '-t', '--threshold':
284 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200285 ns = libregrtest._parse_args([opt, '1000'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300286 self.assertEqual(ns.threshold, 1000)
287 self.checkError([opt], 'expected one argument')
288 self.checkError([opt, 'foo'], 'invalid int value')
289
290 def test_nowindows(self):
291 for opt in '-n', '--nowindows':
292 with self.subTest(opt=opt):
Victor Stinner14b46622015-10-11 10:04:26 +0200293 with contextlib.redirect_stderr(io.StringIO()) as stderr:
294 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300295 self.assertTrue(ns.nowindows)
Victor Stinner14b46622015-10-11 10:04:26 +0200296 err = stderr.getvalue()
297 self.assertIn('the --nowindows (-n) option is deprecated', err)
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300298
299 def test_forever(self):
300 for opt in '-F', '--forever':
301 with self.subTest(opt=opt):
Victor Stinner3844fe52015-09-26 10:38:01 +0200302 ns = libregrtest._parse_args([opt])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300303 self.assertTrue(ns.forever)
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800304
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300305 def test_unrecognized_argument(self):
306 self.checkError(['--xxx'], 'usage:')
307
308 def test_long_option__partial(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200309 ns = libregrtest._parse_args(['--qui'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300310 self.assertTrue(ns.quiet)
311 self.assertEqual(ns.verbose, 0)
312
313 def test_two_options(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200314 ns = libregrtest._parse_args(['--quiet', '--exclude'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300315 self.assertTrue(ns.quiet)
316 self.assertEqual(ns.verbose, 0)
317 self.assertTrue(ns.exclude)
318
319 def test_option_with_empty_string_value(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200320 ns = libregrtest._parse_args(['--start', ''])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300321 self.assertEqual(ns.start, '')
322
323 def test_arg(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200324 ns = libregrtest._parse_args(['foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300325 self.assertEqual(ns.args, ['foo'])
326
327 def test_option_and_arg(self):
Victor Stinner3844fe52015-09-26 10:38:01 +0200328 ns = libregrtest._parse_args(['--quiet', 'foo'])
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300329 self.assertTrue(ns.quiet)
330 self.assertEqual(ns.verbose, 0)
331 self.assertEqual(ns.args, ['foo'])
332
Victor Stinner1f6b69b2016-10-17 18:11:03 +0200333 def test_arg_option_arg(self):
Victor Stinnera506a932016-10-17 18:13:46 +0200334 ns = libregrtest._parse_args(['test_unaryop', '-v', 'test_binop'])
Victor Stinner1f6b69b2016-10-17 18:11:03 +0200335 self.assertEqual(ns.verbose, 1)
336 self.assertEqual(ns.args, ['test_unaryop', 'test_binop'])
337
338 def test_unknown_option(self):
339 self.checkError(['--unknown-option'],
340 'unrecognized arguments: --unknown-option')
341
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800342
Victor Stinner6b415102015-09-28 23:16:17 +0200343class BaseTestCase(unittest.TestCase):
344 TEST_UNIQUE_ID = 1
345 TESTNAME_PREFIX = 'test_regrtest_'
Victor Stinner69649f22016-03-23 12:14:10 +0100346 TESTNAME_REGEX = r'test_[a-zA-Z0-9_]+'
Victor Stinner6b415102015-09-28 23:16:17 +0200347
348 def setUp(self):
Victor Stinnerda43ee42015-11-22 15:18:54 +0100349 self.testdir = os.path.realpath(os.path.dirname(__file__))
Victor Stinner6b415102015-09-28 23:16:17 +0200350
Victor Stinnerd6e25022016-03-30 02:33:52 +0200351 self.tmptestdir = tempfile.mkdtemp()
352 self.addCleanup(support.rmtree, self.tmptestdir)
Victor Stinner6b415102015-09-28 23:16:17 +0200353
354 def create_test(self, name=None, code=''):
355 if not name:
356 name = 'noop%s' % BaseTestCase.TEST_UNIQUE_ID
357 BaseTestCase.TEST_UNIQUE_ID += 1
358
359 # test_regrtest cannot be run twice in parallel because
360 # of setUp() and create_test()
Victor Stinnerd6e25022016-03-30 02:33:52 +0200361 name = self.TESTNAME_PREFIX + name
362 path = os.path.join(self.tmptestdir, name + '.py')
Victor Stinner6b415102015-09-28 23:16:17 +0200363
364 self.addCleanup(support.unlink, path)
365 # Use 'x' mode to ensure that we do not override existing tests
Victor Stinnerda43ee42015-11-22 15:18:54 +0100366 try:
367 with open(path, 'x', encoding='utf-8') as fp:
368 fp.write(code)
369 except PermissionError as exc:
370 if not sysconfig.is_python_build():
371 self.skipTest("cannot write %s: %s" % (path, exc))
372 raise
Victor Stinner6b415102015-09-28 23:16:17 +0200373 return name
374
375 def regex_search(self, regex, output):
376 match = re.search(regex, output, re.MULTILINE)
377 if not match:
378 self.fail("%r not found in %r" % (regex, output))
379 return match
380
381 def check_line(self, output, regex):
382 regex = re.compile(r'^' + regex, re.MULTILINE)
383 self.assertRegex(output, regex)
384
385 def parse_executed_tests(self, output):
Victor Stinner3d005682017-05-04 15:21:12 +0200386 regex = (r'^[0-9]+:[0-9]+:[0-9]+ (?:load avg: [0-9]+\.[0-9]{2} )?\[ *[0-9]+(?:/ *[0-9]+)*\] (%s)'
Victor Stinner69649f22016-03-23 12:14:10 +0100387 % self.TESTNAME_REGEX)
Victor Stinner9a142142015-09-30 13:51:17 +0200388 parser = re.finditer(regex, output, re.MULTILINE)
389 return list(match.group(1) for match in parser)
Victor Stinner6b415102015-09-28 23:16:17 +0200390
Victor Stinner9a142142015-09-30 13:51:17 +0200391 def check_executed_tests(self, output, tests, skipped=(), failed=(),
Victor Stinner63f54c62017-06-26 18:33:19 +0200392 env_changed=(), omitted=(),
Victor Stinnerc45fc762018-06-08 09:53:51 +0200393 rerun=(),
Victor Stinner63f54c62017-06-26 18:33:19 +0200394 randomize=False, interrupted=False,
395 fail_env_changed=False):
Victor Stinner6b415102015-09-28 23:16:17 +0200396 if isinstance(tests, str):
397 tests = [tests]
Victor Stinner9a142142015-09-30 13:51:17 +0200398 if isinstance(skipped, str):
399 skipped = [skipped]
400 if isinstance(failed, str):
401 failed = [failed]
Victor Stinner63f54c62017-06-26 18:33:19 +0200402 if isinstance(env_changed, str):
403 env_changed = [env_changed]
Victor Stinner3909e582015-10-11 10:37:25 +0200404 if isinstance(omitted, str):
405 omitted = [omitted]
Victor Stinnerc45fc762018-06-08 09:53:51 +0200406 if isinstance(rerun, str):
407 rerun = [rerun]
Victor Stinner6b415102015-09-28 23:16:17 +0200408
Victor Stinner9a142142015-09-30 13:51:17 +0200409 executed = self.parse_executed_tests(output)
410 if randomize:
411 self.assertEqual(set(executed), set(tests), output)
Victor Stinner6b415102015-09-28 23:16:17 +0200412 else:
Victor Stinner9a142142015-09-30 13:51:17 +0200413 self.assertEqual(executed, tests, output)
414
415 def plural(count):
416 return 's' if count != 1 else ''
417
418 def list_regex(line_format, tests):
419 count = len(tests)
420 names = ' '.join(sorted(tests))
421 regex = line_format % (count, plural(count))
422 regex = r'%s:\n %s$' % (regex, names)
423 return regex
424
425 if skipped:
426 regex = list_regex('%s test%s skipped', skipped)
427 self.check_line(output, regex)
428
429 if failed:
430 regex = list_regex('%s test%s failed', failed)
431 self.check_line(output, regex)
432
Victor Stinner63f54c62017-06-26 18:33:19 +0200433 if env_changed:
434 regex = list_regex('%s test%s altered the execution environment',
435 env_changed)
436 self.check_line(output, regex)
437
Victor Stinner3909e582015-10-11 10:37:25 +0200438 if omitted:
439 regex = list_regex('%s test%s omitted', omitted)
440 self.check_line(output, regex)
441
Victor Stinnerc45fc762018-06-08 09:53:51 +0200442 if rerun:
443 regex = list_regex('%s re-run test%s', rerun)
444 self.check_line(output, regex)
445 self.check_line(output, "Re-running failed tests in verbose mode")
446 for name in rerun:
447 regex = "Re-running test %r in verbose mode" % name
448 self.check_line(output, regex)
449
Victor Stinner63f54c62017-06-26 18:33:19 +0200450 good = (len(tests) - len(skipped) - len(failed)
451 - len(omitted) - len(env_changed))
Victor Stinner9a142142015-09-30 13:51:17 +0200452 if good:
453 regex = r'%s test%s OK\.$' % (good, plural(good))
454 if not skipped and not failed and good > 1:
455 regex = 'All %s' % regex
456 self.check_line(output, regex)
Victor Stinner6b415102015-09-28 23:16:17 +0200457
Victor Stinner63686032016-08-17 16:12:16 +0200458 if interrupted:
459 self.check_line(output, 'Test suite interrupted by signal SIGINT.')
460
Victor Stinnerc45fc762018-06-08 09:53:51 +0200461 result = []
Victor Stinner63f54c62017-06-26 18:33:19 +0200462 if failed:
Victor Stinnerc45fc762018-06-08 09:53:51 +0200463 result.append('FAILURE')
Victor Stinner63f54c62017-06-26 18:33:19 +0200464 elif fail_env_changed and env_changed:
Victor Stinnerc45fc762018-06-08 09:53:51 +0200465 result.append('ENV CHANGED')
466 if interrupted:
467 result.append('INTERRUPTED')
468 if not result:
469 result.append('SUCCESS')
470 result = ', '.join(result)
471 if rerun:
472 self.check_line(output, 'Tests result: %s' % result)
473 result = 'FAILURE then %s' % result
Victor Stinnerf2abf5c2016-08-19 17:54:25 +0200474 self.check_line(output, 'Tests result: %s' % result)
Victor Stinner63686032016-08-17 16:12:16 +0200475
Victor Stinner6b415102015-09-28 23:16:17 +0200476 def parse_random_seed(self, output):
477 match = self.regex_search(r'Using random seed ([0-9]+)', output)
478 randseed = int(match.group(1))
479 self.assertTrue(0 <= randseed <= 10000000, randseed)
480 return randseed
481
Victor Stinner076fc872015-10-03 00:20:56 +0200482 def run_command(self, args, input=None, exitcode=0, **kw):
Victor Stinner42771342015-09-29 14:17:09 +0200483 if not input:
484 input = ''
Victor Stinner076fc872015-10-03 00:20:56 +0200485 if 'stderr' not in kw:
486 kw['stderr'] = subprocess.PIPE
Victor Stinner9a142142015-09-30 13:51:17 +0200487 proc = subprocess.run(args,
488 universal_newlines=True,
489 input=input,
490 stdout=subprocess.PIPE,
Victor Stinner076fc872015-10-03 00:20:56 +0200491 **kw)
Victor Stinner9a142142015-09-30 13:51:17 +0200492 if proc.returncode != exitcode:
Victor Stinner076fc872015-10-03 00:20:56 +0200493 msg = ("Command %s failed with exit code %s\n"
494 "\n"
495 "stdout:\n"
496 "---\n"
497 "%s\n"
498 "---\n"
499 % (str(args), proc.returncode, proc.stdout))
500 if proc.stderr:
501 msg += ("\n"
502 "stderr:\n"
503 "---\n"
504 "%s"
505 "---\n"
506 % proc.stderr)
507 self.fail(msg)
Victor Stinner9a142142015-09-30 13:51:17 +0200508 return proc
Victor Stinner42771342015-09-29 14:17:09 +0200509
Victor Stinner42771342015-09-29 14:17:09 +0200510 def run_python(self, args, **kw):
511 args = [sys.executable, '-X', 'faulthandler', '-I', *args]
512 proc = self.run_command(args, **kw)
513 return proc.stdout
514
Victor Stinner6b415102015-09-28 23:16:17 +0200515
516class ProgramsTestCase(BaseTestCase):
517 """
518 Test various ways to run the Python test suite. Use options close
519 to options used on the buildbot.
520 """
521
522 NTEST = 4
523
524 def setUp(self):
525 super().setUp()
526
527 # Create NTEST tests doing nothing
528 self.tests = [self.create_test() for index in range(self.NTEST)]
529
530 self.python_args = ['-Wd', '-E', '-bb']
Victor Stinnerd6e25022016-03-30 02:33:52 +0200531 self.regrtest_args = ['-uall', '-rwW',
532 '--testdir=%s' % self.tmptestdir]
Victor Stinner49671462015-10-11 10:03:28 +0200533 if hasattr(faulthandler, 'dump_traceback_later'):
534 self.regrtest_args.extend(('--timeout', '3600', '-j4'))
Victor Stinner6b415102015-09-28 23:16:17 +0200535 if sys.platform == 'win32':
536 self.regrtest_args.append('-n')
537
538 def check_output(self, output):
539 self.parse_random_seed(output)
Victor Stinner9a142142015-09-30 13:51:17 +0200540 self.check_executed_tests(output, self.tests, randomize=True)
Victor Stinner6b415102015-09-28 23:16:17 +0200541
542 def run_tests(self, args):
Victor Stinner9a142142015-09-30 13:51:17 +0200543 output = self.run_python(args)
544 self.check_output(output)
Victor Stinner6b415102015-09-28 23:16:17 +0200545
546 def test_script_regrtest(self):
547 # Lib/test/regrtest.py
Victor Stinnerda43ee42015-11-22 15:18:54 +0100548 script = os.path.join(self.testdir, 'regrtest.py')
Victor Stinner6b415102015-09-28 23:16:17 +0200549
550 args = [*self.python_args, script, *self.regrtest_args, *self.tests]
551 self.run_tests(args)
552
553 def test_module_test(self):
554 # -m test
555 args = [*self.python_args, '-m', 'test',
556 *self.regrtest_args, *self.tests]
557 self.run_tests(args)
558
559 def test_module_regrtest(self):
560 # -m test.regrtest
561 args = [*self.python_args, '-m', 'test.regrtest',
562 *self.regrtest_args, *self.tests]
563 self.run_tests(args)
564
565 def test_module_autotest(self):
566 # -m test.autotest
567 args = [*self.python_args, '-m', 'test.autotest',
568 *self.regrtest_args, *self.tests]
569 self.run_tests(args)
570
571 def test_module_from_test_autotest(self):
572 # from test import autotest
573 code = 'from test import autotest'
574 args = [*self.python_args, '-c', code,
575 *self.regrtest_args, *self.tests]
576 self.run_tests(args)
577
578 def test_script_autotest(self):
579 # Lib/test/autotest.py
Victor Stinnerda43ee42015-11-22 15:18:54 +0100580 script = os.path.join(self.testdir, 'autotest.py')
Victor Stinner6b415102015-09-28 23:16:17 +0200581 args = [*self.python_args, script, *self.regrtest_args, *self.tests]
582 self.run_tests(args)
583
Victor Stinnerda43ee42015-11-22 15:18:54 +0100584 @unittest.skipUnless(sysconfig.is_python_build(),
585 'run_tests.py script is not installed')
Victor Stinner6b415102015-09-28 23:16:17 +0200586 def test_tools_script_run_tests(self):
587 # Tools/scripts/run_tests.py
588 script = os.path.join(ROOT_DIR, 'Tools', 'scripts', 'run_tests.py')
Victor Stinner8a34d412016-03-30 08:51:15 +0200589 args = [script, *self.regrtest_args, *self.tests]
Victor Stinnerd6e25022016-03-30 02:33:52 +0200590 self.run_tests(args)
Victor Stinner6b415102015-09-28 23:16:17 +0200591
Victor Stinnere143c1f2015-09-29 01:02:37 +0200592 def run_batch(self, *args):
Victor Stinner42771342015-09-29 14:17:09 +0200593 proc = self.run_command(args)
Victor Stinner6b415102015-09-28 23:16:17 +0200594 self.check_output(proc.stdout)
595
Victor Stinnerda43ee42015-11-22 15:18:54 +0100596 @unittest.skipUnless(sysconfig.is_python_build(),
597 'test.bat script is not installed')
Victor Stinner6b415102015-09-28 23:16:17 +0200598 @unittest.skipUnless(sys.platform == 'win32', 'Windows only')
599 def test_tools_buildbot_test(self):
600 # Tools\buildbot\test.bat
601 script = os.path.join(ROOT_DIR, 'Tools', 'buildbot', 'test.bat')
Victor Stinner732599f2016-03-30 08:38:05 +0200602 test_args = ['--testdir=%s' % self.tmptestdir]
Victor Stinnere143c1f2015-09-29 01:02:37 +0200603 if platform.architecture()[0] == '64bit':
604 test_args.append('-x64') # 64-bit build
605 if not Py_DEBUG:
606 test_args.append('+d') # Release build, use python.exe
607 self.run_batch(script, *test_args, *self.tests)
Victor Stinner6b415102015-09-28 23:16:17 +0200608
609 @unittest.skipUnless(sys.platform == 'win32', 'Windows only')
610 def test_pcbuild_rt(self):
611 # PCbuild\rt.bat
612 script = os.path.join(ROOT_DIR, r'PCbuild\rt.bat')
Andrés Delfinoc3b7a6d2018-04-30 13:30:53 -0300613 if not os.path.isfile(script):
614 self.skipTest(f'File "{script}" does not exist')
Victor Stinnere143c1f2015-09-29 01:02:37 +0200615 rt_args = ["-q"] # Quick, don't run tests twice
616 if platform.architecture()[0] == '64bit':
617 rt_args.append('-x64') # 64-bit build
618 if Py_DEBUG:
619 rt_args.append('-d') # Debug build, use python_d.exe
620 self.run_batch(script, *rt_args, *self.regrtest_args, *self.tests)
Victor Stinner6b415102015-09-28 23:16:17 +0200621
622
623class ArgsTestCase(BaseTestCase):
624 """
625 Test arguments of the Python test suite.
626 """
627
Victor Stinnerd6e25022016-03-30 02:33:52 +0200628 def run_tests(self, *testargs, **kw):
629 cmdargs = ['-m', 'test', '--testdir=%s' % self.tmptestdir, *testargs]
630 return self.run_python(cmdargs, **kw)
Victor Stinner9a142142015-09-30 13:51:17 +0200631
632 def test_failing_test(self):
633 # test a failing test
634 code = textwrap.dedent("""
635 import unittest
636
637 class FailingTest(unittest.TestCase):
638 def test_failing(self):
639 self.fail("bug")
640 """)
Victor Stinnerd6e25022016-03-30 02:33:52 +0200641 test_ok = self.create_test('ok')
642 test_failing = self.create_test('failing', code=code)
Victor Stinner9a142142015-09-30 13:51:17 +0200643 tests = [test_ok, test_failing]
644
Victor Stinner63f54c62017-06-26 18:33:19 +0200645 output = self.run_tests(*tests, exitcode=2)
Victor Stinner9a142142015-09-30 13:51:17 +0200646 self.check_executed_tests(output, tests, failed=test_failing)
Victor Stinner6b415102015-09-28 23:16:17 +0200647
648 def test_resources(self):
649 # test -u command line option
650 tests = {}
651 for resource in ('audio', 'network'):
652 code = 'from test import support\nsupport.requires(%r)' % resource
653 tests[resource] = self.create_test(resource, code)
654 test_names = sorted(tests.values())
655
656 # -u all: 2 resources enabled
657 output = self.run_tests('-u', 'all', *test_names)
658 self.check_executed_tests(output, test_names)
659
660 # -u audio: 1 resource enabled
661 output = self.run_tests('-uaudio', *test_names)
662 self.check_executed_tests(output, test_names,
663 skipped=tests['network'])
664
665 # no option: 0 resources enabled
666 output = self.run_tests(*test_names)
667 self.check_executed_tests(output, test_names,
668 skipped=test_names)
669
670 def test_random(self):
671 # test -r and --randseed command line option
672 code = textwrap.dedent("""
673 import random
674 print("TESTRANDOM: %s" % random.randint(1, 1000))
675 """)
676 test = self.create_test('random', code)
677
678 # first run to get the output with the random seed
679 output = self.run_tests('-r', test)
680 randseed = self.parse_random_seed(output)
681 match = self.regex_search(r'TESTRANDOM: ([0-9]+)', output)
682 test_random = int(match.group(1))
683
684 # try to reproduce with the random seed
685 output = self.run_tests('-r', '--randseed=%s' % randseed, test)
686 randseed2 = self.parse_random_seed(output)
687 self.assertEqual(randseed2, randseed)
688
689 match = self.regex_search(r'TESTRANDOM: ([0-9]+)', output)
690 test_random2 = int(match.group(1))
691 self.assertEqual(test_random2, test_random)
692
693 def test_fromfile(self):
694 # test --fromfile
695 tests = [self.create_test() for index in range(5)]
696
697 # Write the list of files using a format similar to regrtest output:
698 # [1/2] test_1
699 # [2/2] test_2
700 filename = support.TESTFN
701 self.addCleanup(support.unlink, filename)
Victor Stinner5de16e82016-03-24 09:43:00 +0100702
703 # test format '0:00:00 [2/7] test_opcodes -- test_grammar took 0 sec'
704 with open(filename, "w") as fp:
705 previous = None
706 for index, name in enumerate(tests, 1):
707 line = ("00:00:%02i [%s/%s] %s"
708 % (index, index, len(tests), name))
709 if previous:
710 line += " -- %s took 0 sec" % previous
711 print(line, file=fp)
712 previous = name
713
714 output = self.run_tests('--fromfile', filename)
715 self.check_executed_tests(output, tests)
716
717 # test format '[2/7] test_opcodes'
Victor Stinner6b415102015-09-28 23:16:17 +0200718 with open(filename, "w") as fp:
719 for index, name in enumerate(tests, 1):
720 print("[%s/%s] %s" % (index, len(tests), name), file=fp)
721
722 output = self.run_tests('--fromfile', filename)
723 self.check_executed_tests(output, tests)
724
Victor Stinner5de16e82016-03-24 09:43:00 +0100725 # test format 'test_opcodes'
726 with open(filename, "w") as fp:
727 for name in tests:
728 print(name, file=fp)
729
730 output = self.run_tests('--fromfile', filename)
731 self.check_executed_tests(output, tests)
732
Victor Stinnere8cfec52017-01-03 01:38:58 +0100733 # test format 'Lib/test/test_opcodes.py'
734 with open(filename, "w") as fp:
735 for name in tests:
736 print('Lib/test/%s.py' % name, file=fp)
737
738 output = self.run_tests('--fromfile', filename)
739 self.check_executed_tests(output, tests)
740
Victor Stinner3909e582015-10-11 10:37:25 +0200741 def test_interrupted(self):
742 code = TEST_INTERRUPTED
Victor Stinnerd6e25022016-03-30 02:33:52 +0200743 test = self.create_test('sigint', code=code)
Victor Stinner63f54c62017-06-26 18:33:19 +0200744 output = self.run_tests(test, exitcode=130)
Victor Stinner63686032016-08-17 16:12:16 +0200745 self.check_executed_tests(output, test, omitted=test,
746 interrupted=True)
Victor Stinner3909e582015-10-11 10:37:25 +0200747
Victor Stinner435eaf42016-08-17 12:22:52 +0200748 def test_slowest(self):
Victor Stinner6c446192016-08-17 11:25:43 +0200749 # test --slowest
Victor Stinner6b415102015-09-28 23:16:17 +0200750 tests = [self.create_test() for index in range(3)]
Victor Stinner6c446192016-08-17 11:25:43 +0200751 output = self.run_tests("--slowest", *tests)
Victor Stinner6b415102015-09-28 23:16:17 +0200752 self.check_executed_tests(output, tests)
753 regex = ('10 slowest tests:\n'
Victor Stinner435eaf42016-08-17 12:22:52 +0200754 '(?:- %s: .*\n){%s}'
Victor Stinner6b415102015-09-28 23:16:17 +0200755 % (self.TESTNAME_REGEX, len(tests)))
756 self.check_line(output, regex)
757
Victor Stinner3909e582015-10-11 10:37:25 +0200758 def test_slow_interrupted(self):
Victor Stinner6c446192016-08-17 11:25:43 +0200759 # Issue #25373: test --slowest with an interrupted test
Victor Stinner3909e582015-10-11 10:37:25 +0200760 code = TEST_INTERRUPTED
761 test = self.create_test("sigint", code=code)
762
Antoine Pitroua6a4dc82017-09-07 18:56:24 +0200763 for multiprocessing in (False, True):
Victor Stinner3909e582015-10-11 10:37:25 +0200764 if multiprocessing:
Victor Stinner6c446192016-08-17 11:25:43 +0200765 args = ("--slowest", "-j2", test)
Victor Stinner3909e582015-10-11 10:37:25 +0200766 else:
Victor Stinner6c446192016-08-17 11:25:43 +0200767 args = ("--slowest", test)
Victor Stinner63f54c62017-06-26 18:33:19 +0200768 output = self.run_tests(*args, exitcode=130)
Victor Stinner63686032016-08-17 16:12:16 +0200769 self.check_executed_tests(output, test,
770 omitted=test, interrupted=True)
771
Victor Stinner3909e582015-10-11 10:37:25 +0200772 regex = ('10 slowest tests:\n')
773 self.check_line(output, regex)
Victor Stinner3909e582015-10-11 10:37:25 +0200774
Victor Stinner6b415102015-09-28 23:16:17 +0200775 def test_coverage(self):
776 # test --coverage
Victor Stinnerd6e25022016-03-30 02:33:52 +0200777 test = self.create_test('coverage')
Victor Stinner6b415102015-09-28 23:16:17 +0200778 output = self.run_tests("--coverage", test)
Victor Stinner9a142142015-09-30 13:51:17 +0200779 self.check_executed_tests(output, [test])
R David Murray44b548d2016-09-08 13:59:53 -0400780 regex = (r'lines +cov% +module +\(path\)\n'
781 r'(?: *[0-9]+ *[0-9]{1,2}% *[^ ]+ +\([^)]+\)+)+')
Victor Stinner6b415102015-09-28 23:16:17 +0200782 self.check_line(output, regex)
783
Victor Stinner42771342015-09-29 14:17:09 +0200784 def test_wait(self):
785 # test --wait
Victor Stinnerd6e25022016-03-30 02:33:52 +0200786 test = self.create_test('wait')
Victor Stinner42771342015-09-29 14:17:09 +0200787 output = self.run_tests("--wait", test, input='key')
788 self.check_line(output, 'Press any key to continue')
789
Victor Stinner9a142142015-09-30 13:51:17 +0200790 def test_forever(self):
791 # test --forever
792 code = textwrap.dedent("""
Victor Stinnerd6e25022016-03-30 02:33:52 +0200793 import builtins
Victor Stinner9a142142015-09-30 13:51:17 +0200794 import unittest
795
796 class ForeverTester(unittest.TestCase):
Victor Stinner9a142142015-09-30 13:51:17 +0200797 def test_run(self):
Victor Stinnerd6e25022016-03-30 02:33:52 +0200798 # Store the state in the builtins module, because the test
799 # module is reload at each run
800 if 'RUN' in builtins.__dict__:
801 builtins.__dict__['RUN'] += 1
802 if builtins.__dict__['RUN'] >= 3:
803 self.fail("fail at the 3rd runs")
804 else:
805 builtins.__dict__['RUN'] = 1
Victor Stinner9a142142015-09-30 13:51:17 +0200806 """)
Victor Stinnerd6e25022016-03-30 02:33:52 +0200807 test = self.create_test('forever', code=code)
Victor Stinner63f54c62017-06-26 18:33:19 +0200808 output = self.run_tests('--forever', test, exitcode=2)
Victor Stinner9a142142015-09-30 13:51:17 +0200809 self.check_executed_tests(output, [test]*3, failed=test)
810
Victor Stinner48b5c422017-06-27 02:02:04 +0200811 def check_leak(self, code, what):
812 test = self.create_test('huntrleaks', code=code)
813
814 filename = 'reflog.txt'
815 self.addCleanup(support.unlink, filename)
816 output = self.run_tests('--huntrleaks', '3:3:', test,
817 exitcode=2,
818 stderr=subprocess.STDOUT)
819 self.check_executed_tests(output, [test], failed=test)
820
821 line = 'beginning 6 repetitions\n123456\n......\n'
822 self.check_line(output, re.escape(line))
823
824 line2 = '%s leaked [1, 1, 1] %s, sum=3\n' % (test, what)
825 self.assertIn(line2, output)
826
827 with open(filename) as fp:
828 reflog = fp.read()
829 self.assertIn(line2, reflog)
830
831 @unittest.skipUnless(Py_DEBUG, 'need a debug build')
832 def test_huntrleaks(self):
833 # test --huntrleaks
834 code = textwrap.dedent("""
835 import unittest
836
837 GLOBAL_LIST = []
838
839 class RefLeakTest(unittest.TestCase):
840 def test_leak(self):
841 GLOBAL_LIST.append(object())
842 """)
843 self.check_leak(code, 'references')
844
Victor Stinner46d75352015-10-03 02:21:35 +0200845 @unittest.skipUnless(Py_DEBUG, 'need a debug build')
Victor Stinner076fc872015-10-03 00:20:56 +0200846 def test_huntrleaks_fd_leak(self):
847 # test --huntrleaks for file descriptor leak
848 code = textwrap.dedent("""
849 import os
850 import unittest
851
852 class FDLeakTest(unittest.TestCase):
853 def test_leak(self):
854 fd = os.open(__file__, os.O_RDONLY)
Victor Stinner302bbbe2017-07-25 17:45:53 +0200855 # bug: never close the file descriptor
Victor Stinner076fc872015-10-03 00:20:56 +0200856 """)
Victor Stinner48b5c422017-06-27 02:02:04 +0200857 self.check_leak(code, 'file descriptors')
Victor Stinner076fc872015-10-03 00:20:56 +0200858
Victor Stinner5f9d3ac2015-10-03 00:21:12 +0200859 def test_list_tests(self):
860 # test --list-tests
861 tests = [self.create_test() for i in range(5)]
862 output = self.run_tests('--list-tests', *tests)
863 self.assertEqual(output.rstrip().splitlines(),
864 tests)
865
Louie Lub0c58d32017-06-16 19:15:28 +0800866 def test_list_cases(self):
867 # test --list-cases
868 code = textwrap.dedent("""
869 import unittest
870
871 class Tests(unittest.TestCase):
872 def test_method1(self):
873 pass
874 def test_method2(self):
875 pass
876 """)
877 testname = self.create_test(code=code)
Victor Stinnerace56d52017-06-26 14:18:51 +0200878
879 # Test --list-cases
Louie Lub0c58d32017-06-16 19:15:28 +0800880 all_methods = ['%s.Tests.test_method1' % testname,
881 '%s.Tests.test_method2' % testname]
882 output = self.run_tests('--list-cases', testname)
883 self.assertEqual(output.splitlines(), all_methods)
884
Victor Stinnerace56d52017-06-26 14:18:51 +0200885 # Test --list-cases with --match
886 all_methods = ['%s.Tests.test_method1' % testname]
887 output = self.run_tests('--list-cases',
888 '-m', 'test_method1',
889 testname)
890 self.assertEqual(output.splitlines(), all_methods)
891
Victor Stinner5bad70d2017-02-06 12:42:00 +0100892 def test_crashed(self):
893 # Any code which causes a crash
894 code = 'import faulthandler; faulthandler._sigsegv()'
895 crash_test = self.create_test(name="crash", code=code)
896 ok_test = self.create_test(name="ok")
897
898 tests = [crash_test, ok_test]
Victor Stinner63f54c62017-06-26 18:33:19 +0200899 output = self.run_tests("-j2", *tests, exitcode=2)
Victor Stinner5bad70d2017-02-06 12:42:00 +0100900 self.check_executed_tests(output, tests, failed=crash_test,
901 randomize=True)
902
Victor Stinneref8320c2017-06-09 10:18:48 +0200903 def parse_methods(self, output):
904 regex = re.compile("^(test[^ ]+).*ok$", flags=re.MULTILINE)
905 return [match.group(1) for match in regex.finditer(output)]
906
907 def test_matchfile(self):
Victor Stinneref8320c2017-06-09 10:18:48 +0200908 code = textwrap.dedent("""
909 import unittest
910
911 class Tests(unittest.TestCase):
912 def test_method1(self):
913 pass
914 def test_method2(self):
915 pass
916 def test_method3(self):
917 pass
918 def test_method4(self):
919 pass
920 """)
921 all_methods = ['test_method1', 'test_method2',
922 'test_method3', 'test_method4']
923 testname = self.create_test(code=code)
924
925 # by default, all methods should be run
926 output = self.run_tests("-v", testname)
927 methods = self.parse_methods(output)
928 self.assertEqual(methods, all_methods)
929
930 # only run a subset
931 filename = support.TESTFN
932 self.addCleanup(support.unlink, filename)
933
934 subset = [
935 # only match the method name
936 'test_method1',
937 # match the full identifier
938 '%s.Tests.test_method3' % testname]
939 with open(filename, "w") as fp:
940 for name in subset:
941 print(name, file=fp)
942
943 output = self.run_tests("-v", "--matchfile", filename, testname)
944 methods = self.parse_methods(output)
945 subset = ['test_method1', 'test_method3']
946 self.assertEqual(methods, subset)
947
Victor Stinner63f54c62017-06-26 18:33:19 +0200948 def test_env_changed(self):
949 code = textwrap.dedent("""
950 import unittest
951
952 class Tests(unittest.TestCase):
953 def test_env_changed(self):
954 open("env_changed", "w").close()
955 """)
956 testname = self.create_test(code=code)
957
958 # don't fail by default
959 output = self.run_tests(testname)
960 self.check_executed_tests(output, [testname], env_changed=testname)
961
962 # fail with --fail-env-changed
963 output = self.run_tests("--fail-env-changed", testname, exitcode=3)
964 self.check_executed_tests(output, [testname], env_changed=testname,
965 fail_env_changed=True)
966
Victor Stinnerc45fc762018-06-08 09:53:51 +0200967 def test_rerun_fail(self):
968 code = textwrap.dedent("""
969 import unittest
970
971 class Tests(unittest.TestCase):
972 def test_bug(self):
973 # test always fail
974 self.fail("bug")
975 """)
976 testname = self.create_test(code=code)
977
978 output = self.run_tests("-w", testname, exitcode=2)
979 self.check_executed_tests(output, [testname],
980 failed=testname, rerun=testname)
981
Victor Stinner6b415102015-09-28 23:16:17 +0200982
Chris Jerdonekd6c18dc2012-12-27 18:53:12 -0800983if __name__ == '__main__':
Serhiy Storchaka64f7c4e2013-08-29 12:26:23 +0300984 unittest.main()