##// END OF EJS Templates
run-tests: start to report test results against TestResult...
Gregory Szorc -
r21428:3df2ecf8 default
parent child Browse files
Show More
@@ -1,1517 +1,1537 b''
1 1 #!/usr/bin/env python
2 2 #
3 3 # run-tests.py - Run a set of tests on Mercurial
4 4 #
5 5 # Copyright 2006 Matt Mackall <mpm@selenic.com>
6 6 #
7 7 # This software may be used and distributed according to the terms of the
8 8 # GNU General Public License version 2 or any later version.
9 9
10 10 # Modifying this script is tricky because it has many modes:
11 11 # - serial (default) vs parallel (-jN, N > 1)
12 12 # - no coverage (default) vs coverage (-c, -C, -s)
13 13 # - temp install (default) vs specific hg script (--with-hg, --local)
14 14 # - tests are a mix of shell scripts and Python scripts
15 15 #
16 16 # If you change this script, it is recommended that you ensure you
17 17 # haven't broken it by running it in various modes with a representative
18 18 # sample of test scripts. For example:
19 19 #
20 20 # 1) serial, no coverage, temp install:
21 21 # ./run-tests.py test-s*
22 22 # 2) serial, no coverage, local hg:
23 23 # ./run-tests.py --local test-s*
24 24 # 3) serial, coverage, temp install:
25 25 # ./run-tests.py -c test-s*
26 26 # 4) serial, coverage, local hg:
27 27 # ./run-tests.py -c --local test-s* # unsupported
28 28 # 5) parallel, no coverage, temp install:
29 29 # ./run-tests.py -j2 test-s*
30 30 # 6) parallel, no coverage, local hg:
31 31 # ./run-tests.py -j2 --local test-s*
32 32 # 7) parallel, coverage, temp install:
33 33 # ./run-tests.py -j2 -c test-s* # currently broken
34 34 # 8) parallel, coverage, local install:
35 35 # ./run-tests.py -j2 -c --local test-s* # unsupported (and broken)
36 36 # 9) parallel, custom tmp dir:
37 37 # ./run-tests.py -j2 --tmpdir /tmp/myhgtests
38 38 #
39 39 # (You could use any subset of the tests: test-s* happens to match
40 40 # enough that it's worth doing parallel runs, few enough that it
41 41 # completes fairly quickly, includes both shell and Python scripts, and
42 42 # includes some scripts that run daemon processes.)
43 43
44 44 from distutils import version
45 45 import difflib
46 46 import errno
47 47 import optparse
48 48 import os
49 49 import shutil
50 50 import subprocess
51 51 import signal
52 52 import sys
53 53 import tempfile
54 54 import time
55 55 import random
56 56 import re
57 57 import threading
58 58 import killdaemons as killmod
59 59 import Queue as queue
60 60 import unittest
61 61
62 62 processlock = threading.Lock()
63 63
64 64 # subprocess._cleanup can race with any Popen.wait or Popen.poll on py24
65 65 # http://bugs.python.org/issue1731717 for details. We shouldn't be producing
66 66 # zombies but it's pretty harmless even if we do.
67 67 if sys.version_info < (2, 5):
68 68 subprocess._cleanup = lambda: None
69 69
70 70 closefds = os.name == 'posix'
71 71 def Popen4(cmd, wd, timeout, env=None):
72 72 processlock.acquire()
73 73 p = subprocess.Popen(cmd, shell=True, bufsize=-1, cwd=wd, env=env,
74 74 close_fds=closefds,
75 75 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
76 76 stderr=subprocess.STDOUT)
77 77 processlock.release()
78 78
79 79 p.fromchild = p.stdout
80 80 p.tochild = p.stdin
81 81 p.childerr = p.stderr
82 82
83 83 p.timeout = False
84 84 if timeout:
85 85 def t():
86 86 start = time.time()
87 87 while time.time() - start < timeout and p.returncode is None:
88 88 time.sleep(.1)
89 89 p.timeout = True
90 90 if p.returncode is None:
91 91 terminate(p)
92 92 threading.Thread(target=t).start()
93 93
94 94 return p
95 95
96 96 PYTHON = sys.executable.replace('\\', '/')
97 97 IMPL_PATH = 'PYTHONPATH'
98 98 if 'java' in sys.platform:
99 99 IMPL_PATH = 'JYTHONPATH'
100 100
101 101 TESTDIR = HGTMP = INST = BINDIR = TMPBINDIR = PYTHONDIR = None
102 102
103 103 defaults = {
104 104 'jobs': ('HGTEST_JOBS', 1),
105 105 'timeout': ('HGTEST_TIMEOUT', 180),
106 106 'port': ('HGTEST_PORT', 20059),
107 107 'shell': ('HGTEST_SHELL', 'sh'),
108 108 }
109 109
110 110 def parselistfiles(files, listtype, warn=True):
111 111 entries = dict()
112 112 for filename in files:
113 113 try:
114 114 path = os.path.expanduser(os.path.expandvars(filename))
115 115 f = open(path, "r")
116 116 except IOError, err:
117 117 if err.errno != errno.ENOENT:
118 118 raise
119 119 if warn:
120 120 print "warning: no such %s file: %s" % (listtype, filename)
121 121 continue
122 122
123 123 for line in f.readlines():
124 124 line = line.split('#', 1)[0].strip()
125 125 if line:
126 126 entries[line] = filename
127 127
128 128 f.close()
129 129 return entries
130 130
131 131 def getparser():
132 132 """Obtain the OptionParser used by the CLI."""
133 133 parser = optparse.OptionParser("%prog [options] [tests]")
134 134
135 135 # keep these sorted
136 136 parser.add_option("--blacklist", action="append",
137 137 help="skip tests listed in the specified blacklist file")
138 138 parser.add_option("--whitelist", action="append",
139 139 help="always run tests listed in the specified whitelist file")
140 140 parser.add_option("--changed", type="string",
141 141 help="run tests that are changed in parent rev or working directory")
142 142 parser.add_option("-C", "--annotate", action="store_true",
143 143 help="output files annotated with coverage")
144 144 parser.add_option("-c", "--cover", action="store_true",
145 145 help="print a test coverage report")
146 146 parser.add_option("-d", "--debug", action="store_true",
147 147 help="debug mode: write output of test scripts to console"
148 148 " rather than capturing and diffing it (disables timeout)")
149 149 parser.add_option("-f", "--first", action="store_true",
150 150 help="exit on the first test failure")
151 151 parser.add_option("-H", "--htmlcov", action="store_true",
152 152 help="create an HTML report of the coverage of the files")
153 153 parser.add_option("-i", "--interactive", action="store_true",
154 154 help="prompt to accept changed output")
155 155 parser.add_option("-j", "--jobs", type="int",
156 156 help="number of jobs to run in parallel"
157 157 " (default: $%s or %d)" % defaults['jobs'])
158 158 parser.add_option("--keep-tmpdir", action="store_true",
159 159 help="keep temporary directory after running tests")
160 160 parser.add_option("-k", "--keywords",
161 161 help="run tests matching keywords")
162 162 parser.add_option("-l", "--local", action="store_true",
163 163 help="shortcut for --with-hg=<testdir>/../hg")
164 164 parser.add_option("--loop", action="store_true",
165 165 help="loop tests repeatedly")
166 166 parser.add_option("-n", "--nodiff", action="store_true",
167 167 help="skip showing test changes")
168 168 parser.add_option("-p", "--port", type="int",
169 169 help="port on which servers should listen"
170 170 " (default: $%s or %d)" % defaults['port'])
171 171 parser.add_option("--compiler", type="string",
172 172 help="compiler to build with")
173 173 parser.add_option("--pure", action="store_true",
174 174 help="use pure Python code instead of C extensions")
175 175 parser.add_option("-R", "--restart", action="store_true",
176 176 help="restart at last error")
177 177 parser.add_option("-r", "--retest", action="store_true",
178 178 help="retest failed tests")
179 179 parser.add_option("-S", "--noskips", action="store_true",
180 180 help="don't report skip tests verbosely")
181 181 parser.add_option("--shell", type="string",
182 182 help="shell to use (default: $%s or %s)" % defaults['shell'])
183 183 parser.add_option("-t", "--timeout", type="int",
184 184 help="kill errant tests after TIMEOUT seconds"
185 185 " (default: $%s or %d)" % defaults['timeout'])
186 186 parser.add_option("--time", action="store_true",
187 187 help="time how long each test takes")
188 188 parser.add_option("--tmpdir", type="string",
189 189 help="run tests in the given temporary directory"
190 190 " (implies --keep-tmpdir)")
191 191 parser.add_option("--unittest", action="store_true",
192 192 help="run tests with Python's unittest package"
193 193 " (this is an experimental feature)")
194 194 parser.add_option("-v", "--verbose", action="store_true",
195 195 help="output verbose messages")
196 196 parser.add_option("--view", type="string",
197 197 help="external diff viewer")
198 198 parser.add_option("--with-hg", type="string",
199 199 metavar="HG",
200 200 help="test using specified hg script rather than a "
201 201 "temporary installation")
202 202 parser.add_option("-3", "--py3k-warnings", action="store_true",
203 203 help="enable Py3k warnings on Python 2.6+")
204 204 parser.add_option('--extra-config-opt', action="append",
205 205 help='set the given config opt in the test hgrc')
206 206 parser.add_option('--random', action="store_true",
207 207 help='run tests in random order')
208 208
209 209 for option, (envvar, default) in defaults.items():
210 210 defaults[option] = type(default)(os.environ.get(envvar, default))
211 211 parser.set_defaults(**defaults)
212 212
213 213 return parser
214 214
215 215 def parseargs(args, parser):
216 216 """Parse arguments with our OptionParser and validate results."""
217 217 (options, args) = parser.parse_args(args)
218 218
219 219 # jython is always pure
220 220 if 'java' in sys.platform or '__pypy__' in sys.modules:
221 221 options.pure = True
222 222
223 223 if options.with_hg:
224 224 options.with_hg = os.path.expanduser(options.with_hg)
225 225 if not (os.path.isfile(options.with_hg) and
226 226 os.access(options.with_hg, os.X_OK)):
227 227 parser.error('--with-hg must specify an executable hg script')
228 228 if not os.path.basename(options.with_hg) == 'hg':
229 229 sys.stderr.write('warning: --with-hg should specify an hg script\n')
230 230 if options.local:
231 231 testdir = os.path.dirname(os.path.realpath(sys.argv[0]))
232 232 hgbin = os.path.join(os.path.dirname(testdir), 'hg')
233 233 if os.name != 'nt' and not os.access(hgbin, os.X_OK):
234 234 parser.error('--local specified, but %r not found or not executable'
235 235 % hgbin)
236 236 options.with_hg = hgbin
237 237
238 238 options.anycoverage = options.cover or options.annotate or options.htmlcov
239 239 if options.anycoverage:
240 240 try:
241 241 import coverage
242 242 covver = version.StrictVersion(coverage.__version__).version
243 243 if covver < (3, 3):
244 244 parser.error('coverage options require coverage 3.3 or later')
245 245 except ImportError:
246 246 parser.error('coverage options now require the coverage package')
247 247
248 248 if options.anycoverage and options.local:
249 249 # this needs some path mangling somewhere, I guess
250 250 parser.error("sorry, coverage options do not work when --local "
251 251 "is specified")
252 252
253 253 global verbose
254 254 if options.verbose:
255 255 verbose = ''
256 256
257 257 if options.tmpdir:
258 258 options.tmpdir = os.path.expanduser(options.tmpdir)
259 259
260 260 if options.jobs < 1:
261 261 parser.error('--jobs must be positive')
262 262 if options.unittest:
263 263 if options.jobs > 1:
264 264 sys.stderr.write(
265 265 'warning: --jobs has no effect with --unittest')
266 266 if options.loop:
267 267 sys.stderr.write(
268 268 'warning: --loop has no effect with --unittest')
269 269 if options.interactive and options.debug:
270 270 parser.error("-i/--interactive and -d/--debug are incompatible")
271 271 if options.debug:
272 272 if options.timeout != defaults['timeout']:
273 273 sys.stderr.write(
274 274 'warning: --timeout option ignored with --debug\n')
275 275 options.timeout = 0
276 276 if options.py3k_warnings:
277 277 if sys.version_info[:2] < (2, 6) or sys.version_info[:2] >= (3, 0):
278 278 parser.error('--py3k-warnings can only be used on Python 2.6+')
279 279 if options.blacklist:
280 280 options.blacklist = parselistfiles(options.blacklist, 'blacklist')
281 281 if options.whitelist:
282 282 options.whitelisted = parselistfiles(options.whitelist, 'whitelist')
283 283 else:
284 284 options.whitelisted = {}
285 285
286 286 return (options, args)
287 287
288 288 def rename(src, dst):
289 289 """Like os.rename(), trade atomicity and opened files friendliness
290 290 for existing destination support.
291 291 """
292 292 shutil.copy(src, dst)
293 293 os.remove(src)
294 294
295 295 def showdiff(expected, output, ref, err):
296 296 print
297 297 servefail = False
298 298 for line in difflib.unified_diff(expected, output, ref, err):
299 299 sys.stdout.write(line)
300 300 if not servefail and line.startswith(
301 301 '+ abort: child process failed to start'):
302 302 servefail = True
303 303 return {'servefail': servefail}
304 304
305 305
306 306 verbose = False
307 307 def vlog(*msg):
308 308 if verbose is not False:
309 309 iolock.acquire()
310 310 if verbose:
311 311 print verbose,
312 312 for m in msg:
313 313 print m,
314 314 print
315 315 sys.stdout.flush()
316 316 iolock.release()
317 317
318 318 def log(*msg):
319 319 iolock.acquire()
320 320 if verbose:
321 321 print verbose,
322 322 for m in msg:
323 323 print m,
324 324 print
325 325 sys.stdout.flush()
326 326 iolock.release()
327 327
328 328 def terminate(proc):
329 329 """Terminate subprocess (with fallback for Python versions < 2.6)"""
330 330 vlog('# Terminating process %d' % proc.pid)
331 331 try:
332 332 getattr(proc, 'terminate', lambda : os.kill(proc.pid, signal.SIGTERM))()
333 333 except OSError:
334 334 pass
335 335
336 336 def killdaemons(pidfile):
337 337 return killmod.killdaemons(pidfile, tryhard=False, remove=True,
338 338 logfn=vlog)
339 339
340 340 class Test(object):
341 341 """Encapsulates a single, runnable test.
342 342
343 343 Test instances can be run multiple times via run(). However, multiple
344 344 runs cannot be run concurrently.
345 345 """
346 346
347 347 # Status code reserved for skipped tests (used by hghave).
348 348 SKIPPED_STATUS = 80
349 349
350 350 def __init__(self, runner, test, count, refpath):
351 351 path = os.path.join(runner.testdir, test)
352 352 errpath = os.path.join(runner.testdir, '%s.err' % test)
353 353
354 354 self._runner = runner
355 355 self._testdir = runner.testdir
356 356 self._test = test
357 357 self._path = path
358 358 self._options = runner.options
359 359 self._count = count
360 360 self._daemonpids = []
361 361 self._refpath = refpath
362 362 self._errpath = errpath
363 363
364 364 # If we're not in --debug mode and reference output file exists,
365 365 # check test output against it.
366 366 if runner.options.debug:
367 367 self._refout = None # to match "out is None"
368 368 elif os.path.exists(refpath):
369 369 f = open(refpath, 'r')
370 370 self._refout = f.read().splitlines(True)
371 371 f.close()
372 372 else:
373 373 self._refout = []
374 374
375 375 self._threadtmp = os.path.join(runner.hgtmp, 'child%d' % count)
376 376 os.mkdir(self._threadtmp)
377 377
378 378 def cleanup(self):
379 379 for entry in self._daemonpids:
380 380 killdaemons(entry)
381 381
382 382 if self._threadtmp and not self._options.keep_tmpdir:
383 383 shutil.rmtree(self._threadtmp, True)
384 384
385 385 def run(self):
386 386 """Run this test instance.
387 387
388 388 This will return a tuple describing the result of the test.
389 389 """
390 390 if not os.path.exists(self._path):
391 391 return self.skip("Doesn't exist")
392 392
393 393 options = self._options
394 394 if not (options.whitelisted and self._test in options.whitelisted):
395 395 if options.blacklist and self._test in options.blacklist:
396 396 return self.skip('blacklisted')
397 397
398 398 if options.retest and not os.path.exists('%s.err' % self._test):
399 399 return self.ignore('not retesting')
400 400
401 401 if options.keywords:
402 402 f = open(self._test)
403 403 t = f.read().lower() + self._test.lower()
404 404 f.close()
405 405 for k in options.keywords.lower().split():
406 406 if k in t:
407 407 break
408 408 else:
409 409 return self.ignore("doesn't match keyword")
410 410
411 411 if not os.path.basename(self._test.lower()).startswith('test-'):
412 412 return self.skip('not a test file')
413 413
414 414 # Remove any previous output files.
415 415 if os.path.exists(self._errpath):
416 416 os.remove(self._errpath)
417 417
418 418 testtmp = os.path.join(self._threadtmp, os.path.basename(self._path))
419 419 os.mkdir(testtmp)
420 420 replacements, port = self._getreplacements(testtmp)
421 421 env = self._getenv(testtmp, port)
422 422 self._daemonpids.append(env['DAEMON_PIDS'])
423 423 self._createhgrc(env['HGRCPATH'])
424 424
425 425 vlog('# Test', self._test)
426 426
427 427 starttime = time.time()
428 428 try:
429 429 ret, out = self._run(testtmp, replacements, env)
430 430 duration = time.time() - starttime
431 431 except KeyboardInterrupt:
432 432 duration = time.time() - starttime
433 433 log('INTERRUPTED: %s (after %d seconds)' % (self._test, duration))
434 434 raise
435 435 except Exception, e:
436 436 return self.fail('Exception during execution: %s' % e, 255)
437 437
438 438 killdaemons(env['DAEMON_PIDS'])
439 439
440 440 if not options.keep_tmpdir:
441 441 shutil.rmtree(testtmp)
442 442
443 443 def describe(ret):
444 444 if ret < 0:
445 445 return 'killed by signal: %d' % -ret
446 446 return 'returned error code %d' % ret
447 447
448 448 skipped = False
449 449
450 450 if ret == self.SKIPPED_STATUS:
451 451 if out is None: # Debug mode, nothing to parse.
452 452 missing = ['unknown']
453 453 failed = None
454 454 else:
455 455 missing, failed = TTest.parsehghaveoutput(out)
456 456
457 457 if not missing:
458 458 missing = ['irrelevant']
459 459
460 460 if failed:
461 461 res = self.fail('hg have failed checking for %s' % failed[-1],
462 462 ret)
463 463 else:
464 464 skipped = True
465 465 res = self.skip(missing[-1])
466 466 elif ret == 'timeout':
467 467 res = self.fail('timed out', ret)
468 468 elif out != self._refout:
469 469 info = {}
470 470 if not options.nodiff:
471 471 iolock.acquire()
472 472 if options.view:
473 473 os.system("%s %s %s" % (options.view, self._refpath,
474 474 self._errpath))
475 475 else:
476 476 info = showdiff(self._refout, out, self._refpath,
477 477 self._errpath)
478 478 iolock.release()
479 479 msg = ''
480 480 if info.get('servefail'):
481 481 msg += 'serve failed and '
482 482 if ret:
483 483 msg += 'output changed and ' + describe(ret)
484 484 else:
485 485 msg += 'output changed'
486 486
487 487 if (ret != 0 or out != self._refout) and not skipped \
488 488 and not options.debug:
489 489 f = open(self._errpath, 'wb')
490 490 for line in out:
491 491 f.write(line)
492 492 f.close()
493 493 res = self.fail(msg, ret)
494 494 elif ret:
495 495 res = self.fail(describe(ret), ret)
496 496 else:
497 497 res = self.success()
498 498
499 499
500 500 vlog("# Ret was:", ret)
501 501
502 502 if not options.verbose:
503 503 iolock.acquire()
504 504 sys.stdout.write(res[0])
505 505 sys.stdout.flush()
506 506 iolock.release()
507 507
508 508 self._runner.times.append((self._test, duration))
509 509
510 510 return res
511 511
512 512 def _run(self, testtmp, replacements, env):
513 513 # This should be implemented in child classes to run tests.
514 514 return self._skip('unknown test type')
515 515
516 516 def _getreplacements(self, testtmp):
517 517 port = self._options.port + self._count * 3
518 518 r = [
519 519 (r':%s\b' % port, ':$HGPORT'),
520 520 (r':%s\b' % (port + 1), ':$HGPORT1'),
521 521 (r':%s\b' % (port + 2), ':$HGPORT2'),
522 522 ]
523 523
524 524 if os.name == 'nt':
525 525 r.append(
526 526 (''.join(c.isalpha() and '[%s%s]' % (c.lower(), c.upper()) or
527 527 c in '/\\' and r'[/\\]' or c.isdigit() and c or '\\' + c
528 528 for c in testtmp), '$TESTTMP'))
529 529 else:
530 530 r.append((re.escape(testtmp), '$TESTTMP'))
531 531
532 532 return r, port
533 533
534 534 def _getenv(self, testtmp, port):
535 535 env = os.environ.copy()
536 536 env['TESTTMP'] = testtmp
537 537 env['HOME'] = testtmp
538 538 env["HGPORT"] = str(port)
539 539 env["HGPORT1"] = str(port + 1)
540 540 env["HGPORT2"] = str(port + 2)
541 541 env["HGRCPATH"] = os.path.join(self._threadtmp, '.hgrc')
542 542 env["DAEMON_PIDS"] = os.path.join(self._threadtmp, 'daemon.pids')
543 543 env["HGEDITOR"] = sys.executable + ' -c "import sys; sys.exit(0)"'
544 544 env["HGMERGE"] = "internal:merge"
545 545 env["HGUSER"] = "test"
546 546 env["HGENCODING"] = "ascii"
547 547 env["HGENCODINGMODE"] = "strict"
548 548
549 549 # Reset some environment variables to well-known values so that
550 550 # the tests produce repeatable output.
551 551 env['LANG'] = env['LC_ALL'] = env['LANGUAGE'] = 'C'
552 552 env['TZ'] = 'GMT'
553 553 env["EMAIL"] = "Foo Bar <foo.bar@example.com>"
554 554 env['COLUMNS'] = '80'
555 555 env['TERM'] = 'xterm'
556 556
557 557 for k in ('HG HGPROF CDPATH GREP_OPTIONS http_proxy no_proxy ' +
558 558 'NO_PROXY').split():
559 559 if k in env:
560 560 del env[k]
561 561
562 562 # unset env related to hooks
563 563 for k in env.keys():
564 564 if k.startswith('HG_'):
565 565 del env[k]
566 566
567 567 return env
568 568
569 569 def _createhgrc(self, path):
570 570 # create a fresh hgrc
571 571 hgrc = open(path, 'w')
572 572 hgrc.write('[ui]\n')
573 573 hgrc.write('slash = True\n')
574 574 hgrc.write('interactive = False\n')
575 575 hgrc.write('[defaults]\n')
576 576 hgrc.write('backout = -d "0 0"\n')
577 577 hgrc.write('commit = -d "0 0"\n')
578 578 hgrc.write('shelve = --date "0 0"\n')
579 579 hgrc.write('tag = -d "0 0"\n')
580 580 if self._options.extra_config_opt:
581 581 for opt in self._options.extra_config_opt:
582 582 section, key = opt.split('.', 1)
583 583 assert '=' in key, ('extra config opt %s must '
584 584 'have an = for assignment' % opt)
585 585 hgrc.write('[%s]\n%s\n' % (section, key))
586 586 hgrc.close()
587 587
588 588 def success(self):
589 589 return '.', self._test, ''
590 590
591 591 def fail(self, msg, ret):
592 592 warned = ret is False
593 593 if not self._options.nodiff:
594 594 log("\n%s: %s %s" % (warned and 'Warning' or 'ERROR', self._test,
595 595 msg))
596 596 if (not ret and self._options.interactive and
597 597 os.path.exists(self._errpath)):
598 598 iolock.acquire()
599 599 print 'Accept this change? [n] ',
600 600 answer = sys.stdin.readline().strip()
601 601 iolock.release()
602 602 if answer.lower() in ('y', 'yes'):
603 603 if self._test.endswith('.t'):
604 604 rename(self._errpath, self._path)
605 605 else:
606 606 rename(self._errpath, '%s.out' % self._path)
607 607
608 608 return '.', self._test, ''
609 609
610 610 return warned and '~' or '!', self._test, msg
611 611
612 612 def skip(self, msg):
613 613 if self._options.verbose:
614 614 log("\nSkipping %s: %s" % (self._path, msg))
615 615
616 616 return 's', self._test, msg
617 617
618 618 def ignore(self, msg):
619 619 return 'i', self._test, msg
620 620
621 621 class PythonTest(Test):
622 622 """A Python-based test."""
623 623 def _run(self, testtmp, replacements, env):
624 624 py3kswitch = self._options.py3k_warnings and ' -3' or ''
625 625 cmd = '%s%s "%s"' % (PYTHON, py3kswitch, self._path)
626 626 vlog("# Running", cmd)
627 627 if os.name == 'nt':
628 628 replacements.append((r'\r\n', '\n'))
629 629 return run(cmd, testtmp, self._options, replacements, env,
630 630 self._runner.abort)
631 631
632 632 class TTest(Test):
633 633 """A "t test" is a test backed by a .t file."""
634 634
635 635 SKIPPED_PREFIX = 'skipped: '
636 636 FAILED_PREFIX = 'hghave check failed: '
637 637 NEEDESCAPE = re.compile(r'[\x00-\x08\x0b-\x1f\x7f-\xff]').search
638 638
639 639 ESCAPESUB = re.compile(r'[\x00-\x08\x0b-\x1f\\\x7f-\xff]').sub
640 640 ESCAPEMAP = dict((chr(i), r'\x%02x' % i) for i in range(256)).update(
641 641 {'\\': '\\\\', '\r': r'\r'})
642 642
643 643 def _run(self, testtmp, replacements, env):
644 644 f = open(self._path)
645 645 lines = f.readlines()
646 646 f.close()
647 647
648 648 salt, script, after, expected = self._parsetest(lines, testtmp)
649 649
650 650 # Write out the generated script.
651 651 fname = '%s.sh' % testtmp
652 652 f = open(fname, 'w')
653 653 for l in script:
654 654 f.write(l)
655 655 f.close()
656 656
657 657 cmd = '%s "%s"' % (self._options.shell, fname)
658 658 vlog("# Running", cmd)
659 659
660 660 exitcode, output = run(cmd, testtmp, self._options, replacements, env,
661 661 self._runner.abort)
662 662 # Do not merge output if skipped. Return hghave message instead.
663 663 # Similarly, with --debug, output is None.
664 664 if exitcode == self.SKIPPED_STATUS or output is None:
665 665 return exitcode, output
666 666
667 667 return self._processoutput(exitcode, output, salt, after, expected)
668 668
669 669 def _hghave(self, reqs, testtmp):
670 670 # TODO do something smarter when all other uses of hghave are gone.
671 671 tdir = self._testdir.replace('\\', '/')
672 672 proc = Popen4('%s -c "%s/hghave %s"' %
673 673 (self._options.shell, tdir, ' '.join(reqs)),
674 674 testtmp, 0)
675 675 stdout, stderr = proc.communicate()
676 676 ret = proc.wait()
677 677 if wifexited(ret):
678 678 ret = os.WEXITSTATUS(ret)
679 679 if ret == 2:
680 680 print stdout
681 681 sys.exit(1)
682 682
683 683 return ret == 0
684 684
685 685 def _parsetest(self, lines, testtmp):
686 686 # We generate a shell script which outputs unique markers to line
687 687 # up script results with our source. These markers include input
688 688 # line number and the last return code.
689 689 salt = "SALT" + str(time.time())
690 690 def addsalt(line, inpython):
691 691 if inpython:
692 692 script.append('%s %d 0\n' % (salt, line))
693 693 else:
694 694 script.append('echo %s %s $?\n' % (salt, line))
695 695
696 696 script = []
697 697
698 698 # After we run the shell script, we re-unify the script output
699 699 # with non-active parts of the source, with synchronization by our
700 700 # SALT line number markers. The after table contains the non-active
701 701 # components, ordered by line number.
702 702 after = {}
703 703
704 704 # Expected shell script output.
705 705 expected = {}
706 706
707 707 pos = prepos = -1
708 708
709 709 # True or False when in a true or false conditional section
710 710 skipping = None
711 711
712 712 # We keep track of whether or not we're in a Python block so we
713 713 # can generate the surrounding doctest magic.
714 714 inpython = False
715 715
716 716 if self._options.debug:
717 717 script.append('set -x\n')
718 718 if os.getenv('MSYSTEM'):
719 719 script.append('alias pwd="pwd -W"\n')
720 720
721 721 for n, l in enumerate(lines):
722 722 if not l.endswith('\n'):
723 723 l += '\n'
724 724 if l.startswith('#if'):
725 725 lsplit = l.split()
726 726 if len(lsplit) < 2 or lsplit[0] != '#if':
727 727 after.setdefault(pos, []).append(' !!! invalid #if\n')
728 728 if skipping is not None:
729 729 after.setdefault(pos, []).append(' !!! nested #if\n')
730 730 skipping = not self._hghave(lsplit[1:], testtmp)
731 731 after.setdefault(pos, []).append(l)
732 732 elif l.startswith('#else'):
733 733 if skipping is None:
734 734 after.setdefault(pos, []).append(' !!! missing #if\n')
735 735 skipping = not skipping
736 736 after.setdefault(pos, []).append(l)
737 737 elif l.startswith('#endif'):
738 738 if skipping is None:
739 739 after.setdefault(pos, []).append(' !!! missing #if\n')
740 740 skipping = None
741 741 after.setdefault(pos, []).append(l)
742 742 elif skipping:
743 743 after.setdefault(pos, []).append(l)
744 744 elif l.startswith(' >>> '): # python inlines
745 745 after.setdefault(pos, []).append(l)
746 746 prepos = pos
747 747 pos = n
748 748 if not inpython:
749 749 # We've just entered a Python block. Add the header.
750 750 inpython = True
751 751 addsalt(prepos, False) # Make sure we report the exit code.
752 752 script.append('%s -m heredoctest <<EOF\n' % PYTHON)
753 753 addsalt(n, True)
754 754 script.append(l[2:])
755 755 elif l.startswith(' ... '): # python inlines
756 756 after.setdefault(prepos, []).append(l)
757 757 script.append(l[2:])
758 758 elif l.startswith(' $ '): # commands
759 759 if inpython:
760 760 script.append('EOF\n')
761 761 inpython = False
762 762 after.setdefault(pos, []).append(l)
763 763 prepos = pos
764 764 pos = n
765 765 addsalt(n, False)
766 766 cmd = l[4:].split()
767 767 if len(cmd) == 2 and cmd[0] == 'cd':
768 768 l = ' $ cd %s || exit 1\n' % cmd[1]
769 769 script.append(l[4:])
770 770 elif l.startswith(' > '): # continuations
771 771 after.setdefault(prepos, []).append(l)
772 772 script.append(l[4:])
773 773 elif l.startswith(' '): # results
774 774 # Queue up a list of expected results.
775 775 expected.setdefault(pos, []).append(l[2:])
776 776 else:
777 777 if inpython:
778 778 script.append('EOF\n')
779 779 inpython = False
780 780 # Non-command/result. Queue up for merged output.
781 781 after.setdefault(pos, []).append(l)
782 782
783 783 if inpython:
784 784 script.append('EOF\n')
785 785 if skipping is not None:
786 786 after.setdefault(pos, []).append(' !!! missing #endif\n')
787 787 addsalt(n + 1, False)
788 788
789 789 return salt, script, after, expected
790 790
791 791 def _processoutput(self, exitcode, output, salt, after, expected):
792 792 # Merge the script output back into a unified test.
793 793 warnonly = 1 # 1: not yet; 2: yes; 3: for sure not
794 794 if exitcode != 0:
795 795 warnonly = 3
796 796
797 797 pos = -1
798 798 postout = []
799 799 for l in output:
800 800 lout, lcmd = l, None
801 801 if salt in l:
802 802 lout, lcmd = l.split(salt, 1)
803 803
804 804 if lout:
805 805 if not lout.endswith('\n'):
806 806 lout += ' (no-eol)\n'
807 807
808 808 # Find the expected output at the current position.
809 809 el = None
810 810 if expected.get(pos, None):
811 811 el = expected[pos].pop(0)
812 812
813 813 r = TTest.linematch(el, lout)
814 814 if isinstance(r, str):
815 815 if r == '+glob':
816 816 lout = el[:-1] + ' (glob)\n'
817 817 r = '' # Warn only this line.
818 818 elif r == '-glob':
819 819 lout = ''.join(el.rsplit(' (glob)', 1))
820 820 r = '' # Warn only this line.
821 821 else:
822 822 log('\ninfo, unknown linematch result: %r\n' % r)
823 823 r = False
824 824 if r:
825 825 postout.append(' ' + el)
826 826 else:
827 827 if self.NEEDESCAPE(lout):
828 828 lout = TTest.stringescape('%s (esc)\n' %
829 829 lout.rstrip('\n'))
830 830 postout.append(' ' + lout) # Let diff deal with it.
831 831 if r != '': # If line failed.
832 832 warnonly = 3 # for sure not
833 833 elif warnonly == 1: # Is "not yet" and line is warn only.
834 834 warnonly = 2 # Yes do warn.
835 835
836 836 if lcmd:
837 837 # Add on last return code.
838 838 ret = int(lcmd.split()[1])
839 839 if ret != 0:
840 840 postout.append(' [%s]\n' % ret)
841 841 if pos in after:
842 842 # Merge in non-active test bits.
843 843 postout += after.pop(pos)
844 844 pos = int(lcmd.split()[0])
845 845
846 846 if pos in after:
847 847 postout += after.pop(pos)
848 848
849 849 if warnonly == 2:
850 850 exitcode = False # Set exitcode to warned.
851 851
852 852 return exitcode, postout
853 853
854 854 @staticmethod
855 855 def rematch(el, l):
856 856 try:
857 857 # use \Z to ensure that the regex matches to the end of the string
858 858 if os.name == 'nt':
859 859 return re.match(el + r'\r?\n\Z', l)
860 860 return re.match(el + r'\n\Z', l)
861 861 except re.error:
862 862 # el is an invalid regex
863 863 return False
864 864
865 865 @staticmethod
866 866 def globmatch(el, l):
867 867 # The only supported special characters are * and ? plus / which also
868 868 # matches \ on windows. Escaping of these characters is supported.
869 869 if el + '\n' == l:
870 870 if os.altsep:
871 871 # matching on "/" is not needed for this line
872 872 return '-glob'
873 873 return True
874 874 i, n = 0, len(el)
875 875 res = ''
876 876 while i < n:
877 877 c = el[i]
878 878 i += 1
879 879 if c == '\\' and el[i] in '*?\\/':
880 880 res += el[i - 1:i + 1]
881 881 i += 1
882 882 elif c == '*':
883 883 res += '.*'
884 884 elif c == '?':
885 885 res += '.'
886 886 elif c == '/' and os.altsep:
887 887 res += '[/\\\\]'
888 888 else:
889 889 res += re.escape(c)
890 890 return TTest.rematch(res, l)
891 891
892 892 @staticmethod
893 893 def linematch(el, l):
894 894 if el == l: # perfect match (fast)
895 895 return True
896 896 if el:
897 897 if el.endswith(" (esc)\n"):
898 898 el = el[:-7].decode('string-escape') + '\n'
899 899 if el == l or os.name == 'nt' and el[:-1] + '\r\n' == l:
900 900 return True
901 901 if el.endswith(" (re)\n"):
902 902 return TTest.rematch(el[:-6], l)
903 903 if el.endswith(" (glob)\n"):
904 904 return TTest.globmatch(el[:-8], l)
905 905 if os.altsep and l.replace('\\', '/') == el:
906 906 return '+glob'
907 907 return False
908 908
909 909 @staticmethod
910 910 def parsehghaveoutput(lines):
911 911 '''Parse hghave log lines.
912 912
913 913 Return tuple of lists (missing, failed):
914 914 * the missing/unknown features
915 915 * the features for which existence check failed'''
916 916 missing = []
917 917 failed = []
918 918 for line in lines:
919 919 if line.startswith(TTest.SKIPPED_PREFIX):
920 920 line = line.splitlines()[0]
921 921 missing.append(line[len(TTest.SKIPPED_PREFIX):])
922 922 elif line.startswith(TTest.FAILED_PREFIX):
923 923 line = line.splitlines()[0]
924 924 failed.append(line[len(TTest.FAILED_PREFIX):])
925 925
926 926 return missing, failed
927 927
928 928 @staticmethod
929 929 def _escapef(m):
930 930 return TTest.ESCAPEMAP[m.group(0)]
931 931
932 932 @staticmethod
933 933 def _stringescape(s):
934 934 return TTest.ESCAPESUB(TTest._escapef, s)
935 935
936 936
937 937 wifexited = getattr(os, "WIFEXITED", lambda x: False)
938 938 def run(cmd, wd, options, replacements, env, abort):
939 939 """Run command in a sub-process, capturing the output (stdout and stderr).
940 940 Return a tuple (exitcode, output). output is None in debug mode."""
941 941 # TODO: Use subprocess.Popen if we're running on Python 2.4
942 942 if options.debug:
943 943 proc = subprocess.Popen(cmd, shell=True, cwd=wd, env=env)
944 944 ret = proc.wait()
945 945 return (ret, None)
946 946
947 947 proc = Popen4(cmd, wd, options.timeout, env)
948 948 def cleanup():
949 949 terminate(proc)
950 950 ret = proc.wait()
951 951 if ret == 0:
952 952 ret = signal.SIGTERM << 8
953 953 killdaemons(env['DAEMON_PIDS'])
954 954 return ret
955 955
956 956 output = ''
957 957 proc.tochild.close()
958 958
959 959 try:
960 960 output = proc.fromchild.read()
961 961 except KeyboardInterrupt:
962 962 vlog('# Handling keyboard interrupt')
963 963 cleanup()
964 964 raise
965 965
966 966 ret = proc.wait()
967 967 if wifexited(ret):
968 968 ret = os.WEXITSTATUS(ret)
969 969
970 970 if proc.timeout:
971 971 ret = 'timeout'
972 972
973 973 if ret:
974 974 killdaemons(env['DAEMON_PIDS'])
975 975
976 976 if abort[0]:
977 977 raise KeyboardInterrupt()
978 978
979 979 for s, r in replacements:
980 980 output = re.sub(s, r, output)
981 981 return ret, output.splitlines(True)
982 982
983 983 iolock = threading.Lock()
984 984
985 985 class TestRunner(object):
986 986 """Holds context for executing tests.
987 987
988 988 Tests rely on a lot of state. This object holds it for them.
989 989 """
990 990
991 991 REQUIREDTOOLS = [
992 992 os.path.basename(sys.executable),
993 993 'diff',
994 994 'grep',
995 995 'unzip',
996 996 'gunzip',
997 997 'bunzip2',
998 998 'sed',
999 999 ]
1000 1000
1001 1001 TESTTYPES = [
1002 1002 ('.py', PythonTest, '.out'),
1003 1003 ('.t', TTest, ''),
1004 1004 ]
1005 1005
1006 1006 def __init__(self):
1007 1007 self.options = None
1008 1008 self.testdir = None
1009 1009 self.hgtmp = None
1010 1010 self.inst = None
1011 1011 self.bindir = None
1012 1012 self.tmpbinddir = None
1013 1013 self.pythondir = None
1014 1014 self.coveragefile = None
1015 1015 self.times = [] # Holds execution times of tests.
1016 1016 self.results = {
1017 1017 '.': [],
1018 1018 '!': [],
1019 1019 '~': [],
1020 1020 's': [],
1021 1021 'i': [],
1022 1022 }
1023 1023 self.abort = [False]
1024 1024 self._createdfiles = []
1025 1025 self._hgpath = None
1026 1026
1027 1027 def run(self, args, parser=None):
1028 1028 """Run the test suite."""
1029 1029 oldmask = os.umask(022)
1030 1030 try:
1031 1031 parser = parser or getparser()
1032 1032 options, args = parseargs(args, parser)
1033 1033 self.options = options
1034 1034
1035 1035 self._checktools()
1036 1036 tests = self.findtests(args)
1037 1037 return self._run(tests)
1038 1038 finally:
1039 1039 os.umask(oldmask)
1040 1040
1041 1041 def _run(self, tests):
1042 1042 if self.options.random:
1043 1043 random.shuffle(tests)
1044 1044 else:
1045 1045 # keywords for slow tests
1046 1046 slow = 'svn gendoc check-code-hg'.split()
1047 1047 def sortkey(f):
1048 1048 # run largest tests first, as they tend to take the longest
1049 1049 try:
1050 1050 val = -os.stat(f).st_size
1051 1051 except OSError, e:
1052 1052 if e.errno != errno.ENOENT:
1053 1053 raise
1054 1054 return -1e9 # file does not exist, tell early
1055 1055 for kw in slow:
1056 1056 if kw in f:
1057 1057 val *= 10
1058 1058 return val
1059 1059 tests.sort(key=sortkey)
1060 1060
1061 1061 self.testdir = os.environ['TESTDIR'] = os.getcwd()
1062 1062
1063 1063 if 'PYTHONHASHSEED' not in os.environ:
1064 1064 # use a random python hash seed all the time
1065 1065 # we do the randomness ourself to know what seed is used
1066 1066 os.environ['PYTHONHASHSEED'] = str(random.getrandbits(32))
1067 1067
1068 1068 if self.options.tmpdir:
1069 1069 self.options.keep_tmpdir = True
1070 1070 tmpdir = self.options.tmpdir
1071 1071 if os.path.exists(tmpdir):
1072 1072 # Meaning of tmpdir has changed since 1.3: we used to create
1073 1073 # HGTMP inside tmpdir; now HGTMP is tmpdir. So fail if
1074 1074 # tmpdir already exists.
1075 1075 print "error: temp dir %r already exists" % tmpdir
1076 1076 return 1
1077 1077
1078 1078 # Automatically removing tmpdir sounds convenient, but could
1079 1079 # really annoy anyone in the habit of using "--tmpdir=/tmp"
1080 1080 # or "--tmpdir=$HOME".
1081 1081 #vlog("# Removing temp dir", tmpdir)
1082 1082 #shutil.rmtree(tmpdir)
1083 1083 os.makedirs(tmpdir)
1084 1084 else:
1085 1085 d = None
1086 1086 if os.name == 'nt':
1087 1087 # without this, we get the default temp dir location, but
1088 1088 # in all lowercase, which causes troubles with paths (issue3490)
1089 1089 d = os.getenv('TMP')
1090 1090 tmpdir = tempfile.mkdtemp('', 'hgtests.', d)
1091 1091 self.hgtmp = os.environ['HGTMP'] = os.path.realpath(tmpdir)
1092 1092
1093 1093 if self.options.with_hg:
1094 1094 self.inst = None
1095 1095 self.bindir = os.path.dirname(os.path.realpath(
1096 1096 self.options.with_hg))
1097 1097 self.tmpbindir = os.path.join(self.hgtmp, 'install', 'bin')
1098 1098 os.makedirs(self.tmpbindir)
1099 1099
1100 1100 # This looks redundant with how Python initializes sys.path from
1101 1101 # the location of the script being executed. Needed because the
1102 1102 # "hg" specified by --with-hg is not the only Python script
1103 1103 # executed in the test suite that needs to import 'mercurial'
1104 1104 # ... which means it's not really redundant at all.
1105 1105 self.pythondir = self.bindir
1106 1106 else:
1107 1107 self.inst = os.path.join(self.hgtmp, "install")
1108 1108 self.bindir = os.environ["BINDIR"] = os.path.join(self.inst,
1109 1109 "bin")
1110 1110 self.tmpbindir = self.bindir
1111 1111 self.pythondir = os.path.join(self.inst, "lib", "python")
1112 1112
1113 1113 os.environ["BINDIR"] = self.bindir
1114 1114 os.environ["PYTHON"] = PYTHON
1115 1115
1116 1116 path = [self.bindir] + os.environ["PATH"].split(os.pathsep)
1117 1117 if self.tmpbindir != self.bindir:
1118 1118 path = [self.tmpbindir] + path
1119 1119 os.environ["PATH"] = os.pathsep.join(path)
1120 1120
1121 1121 # Include TESTDIR in PYTHONPATH so that out-of-tree extensions
1122 1122 # can run .../tests/run-tests.py test-foo where test-foo
1123 1123 # adds an extension to HGRC. Also include run-test.py directory to
1124 1124 # import modules like heredoctest.
1125 1125 pypath = [self.pythondir, self.testdir,
1126 1126 os.path.abspath(os.path.dirname(__file__))]
1127 1127 # We have to augment PYTHONPATH, rather than simply replacing
1128 1128 # it, in case external libraries are only available via current
1129 1129 # PYTHONPATH. (In particular, the Subversion bindings on OS X
1130 1130 # are in /opt/subversion.)
1131 1131 oldpypath = os.environ.get(IMPL_PATH)
1132 1132 if oldpypath:
1133 1133 pypath.append(oldpypath)
1134 1134 os.environ[IMPL_PATH] = os.pathsep.join(pypath)
1135 1135
1136 1136 self.coveragefile = os.path.join(self.testdir, '.coverage')
1137 1137
1138 1138 vlog("# Using TESTDIR", self.testdir)
1139 1139 vlog("# Using HGTMP", self.hgtmp)
1140 1140 vlog("# Using PATH", os.environ["PATH"])
1141 1141 vlog("# Using", IMPL_PATH, os.environ[IMPL_PATH])
1142 1142
1143 1143 try:
1144 1144 return self._runtests(tests) or 0
1145 1145 finally:
1146 1146 time.sleep(.1)
1147 1147 self._cleanup()
1148 1148
1149 1149 def findtests(self, args):
1150 1150 """Finds possible test files from arguments.
1151 1151
1152 1152 If you wish to inject custom tests into the test harness, this would
1153 1153 be a good function to monkeypatch or override in a derived class.
1154 1154 """
1155 1155 if not args:
1156 1156 if self.options.changed:
1157 1157 proc = Popen4('hg st --rev "%s" -man0 .' %
1158 1158 self.options.changed, None, 0)
1159 1159 stdout, stderr = proc.communicate()
1160 1160 args = stdout.strip('\0').split('\0')
1161 1161 else:
1162 1162 args = os.listdir('.')
1163 1163
1164 1164 return [t for t in args
1165 1165 if os.path.basename(t).startswith('test-')
1166 1166 and (t.endswith('.py') or t.endswith('.t'))]
1167 1167
1168 1168 def _runtests(self, tests):
1169 1169 try:
1170 1170 if self.inst:
1171 1171 self._installhg()
1172 1172 self._checkhglib("Testing")
1173 1173 else:
1174 1174 self._usecorrectpython()
1175 1175
1176 1176 if self.options.restart:
1177 1177 orig = list(tests)
1178 1178 while tests:
1179 1179 if os.path.exists(tests[0] + ".err"):
1180 1180 break
1181 1181 tests.pop(0)
1182 1182 if not tests:
1183 1183 print "running all tests"
1184 1184 tests = orig
1185 1185
1186 1186 if self.options.unittest:
1187 1187 suite = unittest.TestSuite()
1188 1188 for count, testpath in enumerate(tests):
1189 1189 suite.addTest(self._gettest(testpath, count, asunit=True))
1190 1190
1191 1191 verbosity = 1
1192 1192 if self.options.verbose:
1193 1193 verbosity = 2
1194 1194 runner = unittest.TextTestRunner(verbosity=verbosity)
1195 1195 runner.run(suite)
1196 1196 else:
1197 1197 self._executetests(tests)
1198 1198
1199 1199 failed = len(self.results['!'])
1200 1200 warned = len(self.results['~'])
1201 1201 tested = len(self.results['.']) + failed + warned
1202 1202 skipped = len(self.results['s'])
1203 1203 ignored = len(self.results['i'])
1204 1204
1205 1205 print
1206 1206 if not self.options.noskips:
1207 1207 for s in self.results['s']:
1208 1208 print "Skipped %s: %s" % s
1209 1209 for s in self.results['~']:
1210 1210 print "Warned %s: %s" % s
1211 1211 for s in self.results['!']:
1212 1212 print "Failed %s: %s" % s
1213 1213 self._checkhglib("Tested")
1214 1214 print "# Ran %d tests, %d skipped, %d warned, %d failed." % (
1215 1215 tested, skipped + ignored, warned, failed)
1216 1216 if self.results['!']:
1217 1217 print 'python hash seed:', os.environ['PYTHONHASHSEED']
1218 1218 if self.options.time:
1219 1219 self._outputtimes()
1220 1220
1221 1221 if self.options.anycoverage:
1222 1222 self._outputcoverage()
1223 1223 except KeyboardInterrupt:
1224 1224 failed = True
1225 1225 print "\ninterrupted!"
1226 1226
1227 1227 if failed:
1228 1228 return 1
1229 1229 if warned:
1230 1230 return 80
1231 1231
1232 1232 def _gettest(self, test, count, asunit=False):
1233 1233 """Obtain a Test by looking at its filename.
1234 1234
1235 1235 Returns a Test instance. The Test may not be runnable if it doesn't
1236 1236 map to a known type.
1237 1237 """
1238 1238 lctest = test.lower()
1239 1239 refpath = os.path.join(self.testdir, test)
1240 1240
1241 1241 testcls = Test
1242 1242
1243 1243 for ext, cls, out in self.TESTTYPES:
1244 1244 if lctest.endswith(ext):
1245 1245 testcls = cls
1246 1246 refpath = os.path.join(self.testdir, test + out)
1247 1247 break
1248 1248
1249 1249 t = testcls(self, test, count, refpath)
1250 1250
1251 1251 if not asunit:
1252 1252 return t
1253 1253
1254 1254 class MercurialTest(unittest.TestCase):
1255 1255 def __init__(self, name, *args, **kwargs):
1256 1256 super(MercurialTest, self).__init__(*args, **kwargs)
1257 1257 self.name = name
1258 1258
1259 1259 def shortDescription(self):
1260 1260 return self.name
1261 1261
1262 # Need to stash away the TestResult since we do custom things
1263 # with it.
1264 def run(self, result):
1265 self._result = result
1266
1267 return super(MercurialTest, self).run(result)
1268
1262 1269 def runTest(self):
1263 t.run()
1270 code, tname, msg = t.run()
1271
1272 if code == '!':
1273 self._result.failures.append((self, msg))
1274 elif code == '~':
1275 pass
1276 elif code == '.':
1277 pass
1278 elif code == 's':
1279 pass
1280 elif code == 'i':
1281 pass
1282 else:
1283 self.fail('Unknown test result code: %s' % code)
1264 1284
1265 1285 return MercurialTest(test)
1266 1286
1267 1287 def _cleanup(self):
1268 1288 """Clean up state from this test invocation."""
1269 1289
1270 1290 if self.options.keep_tmpdir:
1271 1291 return
1272 1292
1273 1293 vlog("# Cleaning up HGTMP", self.hgtmp)
1274 1294 shutil.rmtree(self.hgtmp, True)
1275 1295 for f in self._createdfiles:
1276 1296 try:
1277 1297 os.remove(f)
1278 1298 except OSError:
1279 1299 pass
1280 1300
1281 1301 def _usecorrectpython(self):
1282 1302 # Some tests run the Python interpreter. They must use the
1283 1303 # same interpreter or bad things will happen.
1284 1304 pyexename = sys.platform == 'win32' and 'python.exe' or 'python'
1285 1305 if getattr(os, 'symlink', None):
1286 1306 vlog("# Making python executable in test path a symlink to '%s'" %
1287 1307 sys.executable)
1288 1308 mypython = os.path.join(self.tmpbindir, pyexename)
1289 1309 try:
1290 1310 if os.readlink(mypython) == sys.executable:
1291 1311 return
1292 1312 os.unlink(mypython)
1293 1313 except OSError, err:
1294 1314 if err.errno != errno.ENOENT:
1295 1315 raise
1296 1316 if self._findprogram(pyexename) != sys.executable:
1297 1317 try:
1298 1318 os.symlink(sys.executable, mypython)
1299 1319 self._createdfiles.append(mypython)
1300 1320 except OSError, err:
1301 1321 # child processes may race, which is harmless
1302 1322 if err.errno != errno.EEXIST:
1303 1323 raise
1304 1324 else:
1305 1325 exedir, exename = os.path.split(sys.executable)
1306 1326 vlog("# Modifying search path to find %s as %s in '%s'" %
1307 1327 (exename, pyexename, exedir))
1308 1328 path = os.environ['PATH'].split(os.pathsep)
1309 1329 while exedir in path:
1310 1330 path.remove(exedir)
1311 1331 os.environ['PATH'] = os.pathsep.join([exedir] + path)
1312 1332 if not self._findprogram(pyexename):
1313 1333 print "WARNING: Cannot find %s in search path" % pyexename
1314 1334
1315 1335 def _installhg(self):
1316 1336 vlog("# Performing temporary installation of HG")
1317 1337 installerrs = os.path.join("tests", "install.err")
1318 1338 compiler = ''
1319 1339 if self.options.compiler:
1320 1340 compiler = '--compiler ' + self.options.compiler
1321 1341 pure = self.options.pure and "--pure" or ""
1322 1342 py3 = ''
1323 1343 if sys.version_info[0] == 3:
1324 1344 py3 = '--c2to3'
1325 1345
1326 1346 # Run installer in hg root
1327 1347 script = os.path.realpath(sys.argv[0])
1328 1348 hgroot = os.path.dirname(os.path.dirname(script))
1329 1349 os.chdir(hgroot)
1330 1350 nohome = '--home=""'
1331 1351 if os.name == 'nt':
1332 1352 # The --home="" trick works only on OS where os.sep == '/'
1333 1353 # because of a distutils convert_path() fast-path. Avoid it at
1334 1354 # least on Windows for now, deal with .pydistutils.cfg bugs
1335 1355 # when they happen.
1336 1356 nohome = ''
1337 1357 cmd = ('%(exe)s setup.py %(py3)s %(pure)s clean --all'
1338 1358 ' build %(compiler)s --build-base="%(base)s"'
1339 1359 ' install --force --prefix="%(prefix)s"'
1340 1360 ' --install-lib="%(libdir)s"'
1341 1361 ' --install-scripts="%(bindir)s" %(nohome)s >%(logfile)s 2>&1'
1342 1362 % {'exe': sys.executable, 'py3': py3, 'pure': pure,
1343 1363 'compiler': compiler,
1344 1364 'base': os.path.join(self.hgtmp, "build"),
1345 1365 'prefix': self.inst, 'libdir': self.pythondir,
1346 1366 'bindir': self.bindir,
1347 1367 'nohome': nohome, 'logfile': installerrs})
1348 1368 vlog("# Running", cmd)
1349 1369 if os.system(cmd) == 0:
1350 1370 if not self.options.verbose:
1351 1371 os.remove(installerrs)
1352 1372 else:
1353 1373 f = open(installerrs)
1354 1374 for line in f:
1355 1375 print line,
1356 1376 f.close()
1357 1377 sys.exit(1)
1358 1378 os.chdir(self.testdir)
1359 1379
1360 1380 self._usecorrectpython()
1361 1381
1362 1382 if self.options.py3k_warnings and not self.options.anycoverage:
1363 1383 vlog("# Updating hg command to enable Py3k Warnings switch")
1364 1384 f = open(os.path.join(self.bindir, 'hg'), 'r')
1365 1385 lines = [line.rstrip() for line in f]
1366 1386 lines[0] += ' -3'
1367 1387 f.close()
1368 1388 f = open(os.path.join(self.bindir, 'hg'), 'w')
1369 1389 for line in lines:
1370 1390 f.write(line + '\n')
1371 1391 f.close()
1372 1392
1373 1393 hgbat = os.path.join(self.bindir, 'hg.bat')
1374 1394 if os.path.isfile(hgbat):
1375 1395 # hg.bat expects to be put in bin/scripts while run-tests.py
1376 1396 # installation layout put it in bin/ directly. Fix it
1377 1397 f = open(hgbat, 'rb')
1378 1398 data = f.read()
1379 1399 f.close()
1380 1400 if '"%~dp0..\python" "%~dp0hg" %*' in data:
1381 1401 data = data.replace('"%~dp0..\python" "%~dp0hg" %*',
1382 1402 '"%~dp0python" "%~dp0hg" %*')
1383 1403 f = open(hgbat, 'wb')
1384 1404 f.write(data)
1385 1405 f.close()
1386 1406 else:
1387 1407 print 'WARNING: cannot fix hg.bat reference to python.exe'
1388 1408
1389 1409 if self.options.anycoverage:
1390 1410 custom = os.path.join(self.testdir, 'sitecustomize.py')
1391 1411 target = os.path.join(self.pythondir, 'sitecustomize.py')
1392 1412 vlog('# Installing coverage trigger to %s' % target)
1393 1413 shutil.copyfile(custom, target)
1394 1414 rc = os.path.join(self.testdir, '.coveragerc')
1395 1415 vlog('# Installing coverage rc to %s' % rc)
1396 1416 os.environ['COVERAGE_PROCESS_START'] = rc
1397 1417 fn = os.path.join(self.inst, '..', '.coverage')
1398 1418 os.environ['COVERAGE_FILE'] = fn
1399 1419
1400 1420 def _checkhglib(self, verb):
1401 1421 """Ensure that the 'mercurial' package imported by python is
1402 1422 the one we expect it to be. If not, print a warning to stderr."""
1403 1423 expecthg = os.path.join(self.pythondir, 'mercurial')
1404 1424 actualhg = self._gethgpath()
1405 1425 if os.path.abspath(actualhg) != os.path.abspath(expecthg):
1406 1426 sys.stderr.write('warning: %s with unexpected mercurial lib: %s\n'
1407 1427 ' (expected %s)\n'
1408 1428 % (verb, actualhg, expecthg))
1409 1429 def _gethgpath(self):
1410 1430 """Return the path to the mercurial package that is actually found by
1411 1431 the current Python interpreter."""
1412 1432 if self._hgpath is not None:
1413 1433 return self._hgpath
1414 1434
1415 1435 cmd = '%s -c "import mercurial; print (mercurial.__path__[0])"'
1416 1436 pipe = os.popen(cmd % PYTHON)
1417 1437 try:
1418 1438 self._hgpath = pipe.read().strip()
1419 1439 finally:
1420 1440 pipe.close()
1421 1441
1422 1442 return self._hgpath
1423 1443
1424 1444 def _outputtimes(self):
1425 1445 vlog('# Producing time report')
1426 1446 self.times.sort(key=lambda t: (t[1], t[0]), reverse=True)
1427 1447 cols = '%7.3f %s'
1428 1448 print '\n%-7s %s' % ('Time', 'Test')
1429 1449 for test, timetaken in self.times:
1430 1450 print cols % (timetaken, test)
1431 1451
1432 1452 def _outputcoverage(self):
1433 1453 vlog('# Producing coverage report')
1434 1454 os.chdir(self.pythondir)
1435 1455
1436 1456 def covrun(*args):
1437 1457 cmd = 'coverage %s' % ' '.join(args)
1438 1458 vlog('# Running: %s' % cmd)
1439 1459 os.system(cmd)
1440 1460
1441 1461 covrun('-c')
1442 1462 omit = ','.join(os.path.join(x, '*') for x in
1443 1463 [self.bindir, self.testdir])
1444 1464 covrun('-i', '-r', '"--omit=%s"' % omit) # report
1445 1465 if self.options.htmlcov:
1446 1466 htmldir = os.path.join(self.testdir, 'htmlcov')
1447 1467 covrun('-i', '-b', '"--directory=%s"' % htmldir,
1448 1468 '"--omit=%s"' % omit)
1449 1469 if self.options.annotate:
1450 1470 adir = os.path.join(self.testdir, 'annotated')
1451 1471 if not os.path.isdir(adir):
1452 1472 os.mkdir(adir)
1453 1473 covrun('-i', '-a', '"--directory=%s"' % adir, '"--omit=%s"' % omit)
1454 1474
1455 1475 def _executetests(self, tests):
1456 1476 jobs = self.options.jobs
1457 1477 done = queue.Queue()
1458 1478 running = 0
1459 1479 count = 0
1460 1480
1461 1481 def job(test, count):
1462 1482 try:
1463 1483 t = self._gettest(test, count)
1464 1484 done.put(t.run())
1465 1485 t.cleanup()
1466 1486 except KeyboardInterrupt:
1467 1487 pass
1468 1488 except: # re-raises
1469 1489 done.put(('!', test, 'run-test raised an error, see traceback'))
1470 1490 raise
1471 1491
1472 1492 try:
1473 1493 while tests or running:
1474 1494 if not done.empty() or running == jobs or not tests:
1475 1495 try:
1476 1496 code, test, msg = done.get(True, 1)
1477 1497 self.results[code].append((test, msg))
1478 1498 if self.options.first and code not in '.si':
1479 1499 break
1480 1500 except queue.Empty:
1481 1501 continue
1482 1502 running -= 1
1483 1503 if tests and not running == jobs:
1484 1504 test = tests.pop(0)
1485 1505 if self.options.loop:
1486 1506 tests.append(test)
1487 1507 t = threading.Thread(target=job, name=test,
1488 1508 args=(test, count))
1489 1509 t.start()
1490 1510 running += 1
1491 1511 count += 1
1492 1512 except KeyboardInterrupt:
1493 1513 self.abort[0] = True
1494 1514
1495 1515 def _findprogram(self, program):
1496 1516 """Search PATH for a executable program"""
1497 1517 for p in os.environ.get('PATH', os.defpath).split(os.pathsep):
1498 1518 name = os.path.join(p, program)
1499 1519 if os.name == 'nt' or os.access(name, os.X_OK):
1500 1520 return name
1501 1521 return None
1502 1522
1503 1523 def _checktools(self):
1504 1524 # Before we go any further, check for pre-requisite tools
1505 1525 # stuff from coreutils (cat, rm, etc) are not tested
1506 1526 for p in self.REQUIREDTOOLS:
1507 1527 if os.name == 'nt' and not p.endswith('.exe'):
1508 1528 p += '.exe'
1509 1529 found = self._findprogram(p)
1510 1530 if found:
1511 1531 vlog("# Found prerequisite", p, "at", found)
1512 1532 else:
1513 1533 print "WARNING: Did not find prerequisite tool: %s " % p
1514 1534
1515 1535 if __name__ == '__main__':
1516 1536 runner = TestRunner()
1517 1537 sys.exit(runner.run(sys.argv[1:]))
General Comments 0
You need to be logged in to leave comments. Login now