##// END OF EJS Templates
run-tests: move keyword checking into runone
Matt Mackall -
r13991:8cfe191e default
parent child Browse files
Show More
@@ -1,1121 +1,1118 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 re
56 56
57 57 closefds = os.name == 'posix'
58 58 def Popen4(cmd, bufsize=-1):
59 59 p = subprocess.Popen(cmd, shell=True, bufsize=bufsize,
60 60 close_fds=closefds,
61 61 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
62 62 stderr=subprocess.STDOUT)
63 63 p.fromchild = p.stdout
64 64 p.tochild = p.stdin
65 65 p.childerr = p.stderr
66 66 return p
67 67
68 68 # reserved exit code to skip test (used by hghave)
69 69 SKIPPED_STATUS = 80
70 70 SKIPPED_PREFIX = 'skipped: '
71 71 FAILED_PREFIX = 'hghave check failed: '
72 72 PYTHON = sys.executable
73 73 IMPL_PATH = 'PYTHONPATH'
74 74 if 'java' in sys.platform:
75 75 IMPL_PATH = 'JYTHONPATH'
76 76
77 77 requiredtools = ["python", "diff", "grep", "unzip", "gunzip", "bunzip2", "sed"]
78 78
79 79 defaults = {
80 80 'jobs': ('HGTEST_JOBS', 1),
81 81 'timeout': ('HGTEST_TIMEOUT', 180),
82 82 'port': ('HGTEST_PORT', 20059),
83 83 }
84 84
85 85 def parseargs():
86 86 parser = optparse.OptionParser("%prog [options] [tests]")
87 87
88 88 # keep these sorted
89 89 parser.add_option("--blacklist", action="append",
90 90 help="skip tests listed in the specified blacklist file")
91 91 parser.add_option("-C", "--annotate", action="store_true",
92 92 help="output files annotated with coverage")
93 93 parser.add_option("--child", type="int",
94 94 help="run as child process, summary to given fd")
95 95 parser.add_option("-c", "--cover", action="store_true",
96 96 help="print a test coverage report")
97 97 parser.add_option("-d", "--debug", action="store_true",
98 98 help="debug mode: write output of test scripts to console"
99 99 " rather than capturing and diff'ing it (disables timeout)")
100 100 parser.add_option("-f", "--first", action="store_true",
101 101 help="exit on the first test failure")
102 102 parser.add_option("--inotify", action="store_true",
103 103 help="enable inotify extension when running tests")
104 104 parser.add_option("-i", "--interactive", action="store_true",
105 105 help="prompt to accept changed output")
106 106 parser.add_option("-j", "--jobs", type="int",
107 107 help="number of jobs to run in parallel"
108 108 " (default: $%s or %d)" % defaults['jobs'])
109 109 parser.add_option("--keep-tmpdir", action="store_true",
110 110 help="keep temporary directory after running tests")
111 111 parser.add_option("-k", "--keywords",
112 112 help="run tests matching keywords")
113 113 parser.add_option("-l", "--local", action="store_true",
114 114 help="shortcut for --with-hg=<testdir>/../hg")
115 115 parser.add_option("-n", "--nodiff", action="store_true",
116 116 help="skip showing test changes")
117 117 parser.add_option("-p", "--port", type="int",
118 118 help="port on which servers should listen"
119 119 " (default: $%s or %d)" % defaults['port'])
120 120 parser.add_option("--pure", action="store_true",
121 121 help="use pure Python code instead of C extensions")
122 122 parser.add_option("-R", "--restart", action="store_true",
123 123 help="restart at last error")
124 124 parser.add_option("-r", "--retest", action="store_true",
125 125 help="retest failed tests")
126 126 parser.add_option("-S", "--noskips", action="store_true",
127 127 help="don't report skip tests verbosely")
128 128 parser.add_option("-t", "--timeout", type="int",
129 129 help="kill errant tests after TIMEOUT seconds"
130 130 " (default: $%s or %d)" % defaults['timeout'])
131 131 parser.add_option("--tmpdir", type="string",
132 132 help="run tests in the given temporary directory"
133 133 " (implies --keep-tmpdir)")
134 134 parser.add_option("-v", "--verbose", action="store_true",
135 135 help="output verbose messages")
136 136 parser.add_option("--view", type="string",
137 137 help="external diff viewer")
138 138 parser.add_option("--with-hg", type="string",
139 139 metavar="HG",
140 140 help="test using specified hg script rather than a "
141 141 "temporary installation")
142 142 parser.add_option("-3", "--py3k-warnings", action="store_true",
143 143 help="enable Py3k warnings on Python 2.6+")
144 144
145 145 for option, default in defaults.items():
146 146 defaults[option] = int(os.environ.get(*default))
147 147 parser.set_defaults(**defaults)
148 148 (options, args) = parser.parse_args()
149 149
150 150 # jython is always pure
151 151 if 'java' in sys.platform or '__pypy__' in sys.modules:
152 152 options.pure = True
153 153
154 154 if options.with_hg:
155 155 if not (os.path.isfile(options.with_hg) and
156 156 os.access(options.with_hg, os.X_OK)):
157 157 parser.error('--with-hg must specify an executable hg script')
158 158 if not os.path.basename(options.with_hg) == 'hg':
159 159 sys.stderr.write('warning: --with-hg should specify an hg script')
160 160 if options.local:
161 161 testdir = os.path.dirname(os.path.realpath(sys.argv[0]))
162 162 hgbin = os.path.join(os.path.dirname(testdir), 'hg')
163 163 if not os.access(hgbin, os.X_OK):
164 164 parser.error('--local specified, but %r not found or not executable'
165 165 % hgbin)
166 166 options.with_hg = hgbin
167 167
168 168 options.anycoverage = options.cover or options.annotate
169 169 if options.anycoverage:
170 170 try:
171 171 import coverage
172 172 covver = version.StrictVersion(coverage.__version__).version
173 173 if covver < (3, 3):
174 174 parser.error('coverage options require coverage 3.3 or later')
175 175 except ImportError:
176 176 parser.error('coverage options now require the coverage package')
177 177
178 178 if options.anycoverage and options.local:
179 179 # this needs some path mangling somewhere, I guess
180 180 parser.error("sorry, coverage options do not work when --local "
181 181 "is specified")
182 182
183 183 global vlog
184 184 if options.verbose:
185 185 if options.jobs > 1 or options.child is not None:
186 186 pid = "[%d]" % os.getpid()
187 187 else:
188 188 pid = None
189 189 def vlog(*msg):
190 190 if pid:
191 191 print pid,
192 192 for m in msg:
193 193 print m,
194 194 print
195 195 sys.stdout.flush()
196 196 else:
197 197 vlog = lambda *msg: None
198 198
199 199 if options.tmpdir:
200 200 options.tmpdir = os.path.expanduser(options.tmpdir)
201 201
202 202 if options.jobs < 1:
203 203 parser.error('--jobs must be positive')
204 204 if options.interactive and options.jobs > 1:
205 205 print '(--interactive overrides --jobs)'
206 206 options.jobs = 1
207 207 if options.interactive and options.debug:
208 208 parser.error("-i/--interactive and -d/--debug are incompatible")
209 209 if options.debug:
210 210 if options.timeout != defaults['timeout']:
211 211 sys.stderr.write(
212 212 'warning: --timeout option ignored with --debug\n')
213 213 options.timeout = 0
214 214 if options.py3k_warnings:
215 215 if sys.version_info[:2] < (2, 6) or sys.version_info[:2] >= (3, 0):
216 216 parser.error('--py3k-warnings can only be used on Python 2.6+')
217 217 if options.blacklist:
218 218 blacklist = dict()
219 219 for filename in options.blacklist:
220 220 try:
221 221 path = os.path.expanduser(os.path.expandvars(filename))
222 222 f = open(path, "r")
223 223 except IOError, err:
224 224 if err.errno != errno.ENOENT:
225 225 raise
226 226 print "warning: no such blacklist file: %s" % filename
227 227 continue
228 228
229 229 for line in f.readlines():
230 230 line = line.split('#', 1)[0].strip()
231 231 if line:
232 232 blacklist[line] = filename
233 233
234 234 f.close()
235 235
236 236 options.blacklist = blacklist
237 237
238 238 return (options, args)
239 239
240 240 def rename(src, dst):
241 241 """Like os.rename(), trade atomicity and opened files friendliness
242 242 for existing destination support.
243 243 """
244 244 shutil.copy(src, dst)
245 245 os.remove(src)
246 246
247 247 def splitnewlines(text):
248 248 '''like str.splitlines, but only split on newlines.
249 249 keep line endings.'''
250 250 i = 0
251 251 lines = []
252 252 while True:
253 253 n = text.find('\n', i)
254 254 if n == -1:
255 255 last = text[i:]
256 256 if last:
257 257 lines.append(last)
258 258 return lines
259 259 lines.append(text[i:n + 1])
260 260 i = n + 1
261 261
262 262 def parsehghaveoutput(lines):
263 263 '''Parse hghave log lines.
264 264 Return tuple of lists (missing, failed):
265 265 * the missing/unknown features
266 266 * the features for which existence check failed'''
267 267 missing = []
268 268 failed = []
269 269 for line in lines:
270 270 if line.startswith(SKIPPED_PREFIX):
271 271 line = line.splitlines()[0]
272 272 missing.append(line[len(SKIPPED_PREFIX):])
273 273 elif line.startswith(FAILED_PREFIX):
274 274 line = line.splitlines()[0]
275 275 failed.append(line[len(FAILED_PREFIX):])
276 276
277 277 return missing, failed
278 278
279 279 def showdiff(expected, output, ref, err):
280 280 for line in difflib.unified_diff(expected, output, ref, err):
281 281 sys.stdout.write(line)
282 282
283 283 def findprogram(program):
284 284 """Search PATH for a executable program"""
285 285 for p in os.environ.get('PATH', os.defpath).split(os.pathsep):
286 286 name = os.path.join(p, program)
287 287 if os.access(name, os.X_OK):
288 288 return name
289 289 return None
290 290
291 291 def checktools():
292 292 # Before we go any further, check for pre-requisite tools
293 293 # stuff from coreutils (cat, rm, etc) are not tested
294 294 for p in requiredtools:
295 295 if os.name == 'nt':
296 296 p += '.exe'
297 297 found = findprogram(p)
298 298 if found:
299 299 vlog("# Found prerequisite", p, "at", found)
300 300 else:
301 301 print "WARNING: Did not find prerequisite tool: "+p
302 302
303 303 def killdaemons():
304 304 # Kill off any leftover daemon processes
305 305 try:
306 306 fp = open(DAEMON_PIDS)
307 307 for line in fp:
308 308 try:
309 309 pid = int(line)
310 310 except ValueError:
311 311 continue
312 312 try:
313 313 os.kill(pid, 0)
314 314 vlog('# Killing daemon process %d' % pid)
315 315 os.kill(pid, signal.SIGTERM)
316 316 time.sleep(0.25)
317 317 os.kill(pid, 0)
318 318 vlog('# Daemon process %d is stuck - really killing it' % pid)
319 319 os.kill(pid, signal.SIGKILL)
320 320 except OSError, err:
321 321 if err.errno != errno.ESRCH:
322 322 raise
323 323 fp.close()
324 324 os.unlink(DAEMON_PIDS)
325 325 except IOError:
326 326 pass
327 327
328 328 def cleanup(options):
329 329 if not options.keep_tmpdir:
330 330 vlog("# Cleaning up HGTMP", HGTMP)
331 331 shutil.rmtree(HGTMP, True)
332 332
333 333 def usecorrectpython():
334 334 # some tests run python interpreter. they must use same
335 335 # interpreter we use or bad things will happen.
336 336 exedir, exename = os.path.split(sys.executable)
337 337 if exename == 'python':
338 338 path = findprogram('python')
339 339 if os.path.dirname(path) == exedir:
340 340 return
341 341 vlog('# Making python executable in test path use correct Python')
342 342 mypython = os.path.join(BINDIR, 'python')
343 343 try:
344 344 os.symlink(sys.executable, mypython)
345 345 except AttributeError:
346 346 # windows fallback
347 347 shutil.copyfile(sys.executable, mypython)
348 348 shutil.copymode(sys.executable, mypython)
349 349
350 350 def installhg(options):
351 351 vlog("# Performing temporary installation of HG")
352 352 installerrs = os.path.join("tests", "install.err")
353 353 pure = options.pure and "--pure" or ""
354 354
355 355 # Run installer in hg root
356 356 script = os.path.realpath(sys.argv[0])
357 357 hgroot = os.path.dirname(os.path.dirname(script))
358 358 os.chdir(hgroot)
359 359 nohome = '--home=""'
360 360 if os.name == 'nt':
361 361 # The --home="" trick works only on OS where os.sep == '/'
362 362 # because of a distutils convert_path() fast-path. Avoid it at
363 363 # least on Windows for now, deal with .pydistutils.cfg bugs
364 364 # when they happen.
365 365 nohome = ''
366 366 cmd = ('%s setup.py %s clean --all'
367 367 ' build --build-base="%s"'
368 368 ' install --force --prefix="%s" --install-lib="%s"'
369 369 ' --install-scripts="%s" %s >%s 2>&1'
370 370 % (sys.executable, pure, os.path.join(HGTMP, "build"),
371 371 INST, PYTHONDIR, BINDIR, nohome, installerrs))
372 372 vlog("# Running", cmd)
373 373 if os.system(cmd) == 0:
374 374 if not options.verbose:
375 375 os.remove(installerrs)
376 376 else:
377 377 f = open(installerrs)
378 378 for line in f:
379 379 print line,
380 380 f.close()
381 381 sys.exit(1)
382 382 os.chdir(TESTDIR)
383 383
384 384 usecorrectpython()
385 385
386 386 vlog("# Installing dummy diffstat")
387 387 f = open(os.path.join(BINDIR, 'diffstat'), 'w')
388 388 f.write('#!' + sys.executable + '\n'
389 389 'import sys\n'
390 390 'files = 0\n'
391 391 'for line in sys.stdin:\n'
392 392 ' if line.startswith("diff "):\n'
393 393 ' files += 1\n'
394 394 'sys.stdout.write("files patched: %d\\n" % files)\n')
395 395 f.close()
396 396 os.chmod(os.path.join(BINDIR, 'diffstat'), 0700)
397 397
398 398 if options.py3k_warnings and not options.anycoverage:
399 399 vlog("# Updating hg command to enable Py3k Warnings switch")
400 400 f = open(os.path.join(BINDIR, 'hg'), 'r')
401 401 lines = [line.rstrip() for line in f]
402 402 lines[0] += ' -3'
403 403 f.close()
404 404 f = open(os.path.join(BINDIR, 'hg'), 'w')
405 405 for line in lines:
406 406 f.write(line + '\n')
407 407 f.close()
408 408
409 409 if options.anycoverage:
410 410 custom = os.path.join(TESTDIR, 'sitecustomize.py')
411 411 target = os.path.join(PYTHONDIR, 'sitecustomize.py')
412 412 vlog('# Installing coverage trigger to %s' % target)
413 413 shutil.copyfile(custom, target)
414 414 rc = os.path.join(TESTDIR, '.coveragerc')
415 415 vlog('# Installing coverage rc to %s' % rc)
416 416 os.environ['COVERAGE_PROCESS_START'] = rc
417 417 fn = os.path.join(INST, '..', '.coverage')
418 418 os.environ['COVERAGE_FILE'] = fn
419 419
420 420 def outputcoverage(options):
421 421
422 422 vlog('# Producing coverage report')
423 423 os.chdir(PYTHONDIR)
424 424
425 425 def covrun(*args):
426 426 cmd = 'coverage %s' % ' '.join(args)
427 427 vlog('# Running: %s' % cmd)
428 428 os.system(cmd)
429 429
430 430 if options.child:
431 431 return
432 432
433 433 covrun('-c')
434 434 omit = ','.join([BINDIR, TESTDIR])
435 435 covrun('-i', '-r', '"--omit=%s"' % omit) # report
436 436 if options.annotate:
437 437 adir = os.path.join(TESTDIR, 'annotated')
438 438 if not os.path.isdir(adir):
439 439 os.mkdir(adir)
440 440 covrun('-i', '-a', '"--directory=%s"' % adir, '"--omit=%s"' % omit)
441 441
442 442 class Timeout(Exception):
443 443 pass
444 444
445 445 def alarmed(signum, frame):
446 446 raise Timeout
447 447
448 448 def pytest(test, options, replacements):
449 449 py3kswitch = options.py3k_warnings and ' -3' or ''
450 450 cmd = '%s%s "%s"' % (PYTHON, py3kswitch, test)
451 451 vlog("# Running", cmd)
452 452 return run(cmd, options, replacements)
453 453
454 454 def shtest(test, options, replacements):
455 455 cmd = '"%s"' % test
456 456 vlog("# Running", cmd)
457 457 return run(cmd, options, replacements)
458 458
459 459 needescape = re.compile(r'[\x00-\x08\x0b-\x1f\x7f-\xff]').search
460 460 escapesub = re.compile(r'[\x00-\x08\x0b-\x1f\\\x7f-\xff]').sub
461 461 escapemap = dict((chr(i), r'\x%02x' % i) for i in range(256))
462 462 escapemap.update({'\\': '\\\\', '\r': r'\r'})
463 463 def escapef(m):
464 464 return escapemap[m.group(0)]
465 465 def stringescape(s):
466 466 return escapesub(escapef, s)
467 467
468 468 def tsttest(test, options, replacements):
469 469 t = open(test)
470 470 out = []
471 471 script = []
472 472 salt = "SALT" + str(time.time())
473 473
474 474 pos = prepos = -1
475 475 after = {}
476 476 expected = {}
477 477 for n, l in enumerate(t):
478 478 if not l.endswith('\n'):
479 479 l += '\n'
480 480 if l.startswith(' $ '): # commands
481 481 after.setdefault(pos, []).append(l)
482 482 prepos = pos
483 483 pos = n
484 484 script.append('echo %s %s $?\n' % (salt, n))
485 485 script.append(l[4:])
486 486 elif l.startswith(' > '): # continuations
487 487 after.setdefault(prepos, []).append(l)
488 488 script.append(l[4:])
489 489 elif l.startswith(' '): # results
490 490 # queue up a list of expected results
491 491 expected.setdefault(pos, []).append(l[2:])
492 492 else:
493 493 # non-command/result - queue up for merged output
494 494 after.setdefault(pos, []).append(l)
495 495
496 496 t.close()
497 497
498 498 script.append('echo %s %s $?\n' % (salt, n + 1))
499 499
500 500 fd, name = tempfile.mkstemp(suffix='hg-tst')
501 501
502 502 try:
503 503 for l in script:
504 504 os.write(fd, l)
505 505 os.close(fd)
506 506
507 507 cmd = '/bin/sh "%s"' % name
508 508 vlog("# Running", cmd)
509 509 exitcode, output = run(cmd, options, replacements)
510 510 # do not merge output if skipped, return hghave message instead
511 511 # similarly, with --debug, output is None
512 512 if exitcode == SKIPPED_STATUS or output is None:
513 513 return exitcode, output
514 514 finally:
515 515 os.remove(name)
516 516
517 517 def rematch(el, l):
518 518 try:
519 519 # ensure that the regex matches to the end of the string
520 520 return re.match(el + r'\Z', l)
521 521 except re.error:
522 522 # el is an invalid regex
523 523 return False
524 524
525 525 def globmatch(el, l):
526 526 # The only supported special characters are * and ?. Escaping is
527 527 # supported.
528 528 i, n = 0, len(el)
529 529 res = ''
530 530 while i < n:
531 531 c = el[i]
532 532 i += 1
533 533 if c == '\\' and el[i] in '*?\\':
534 534 res += el[i - 1:i + 1]
535 535 i += 1
536 536 elif c == '*':
537 537 res += '.*'
538 538 elif c == '?':
539 539 res += '.'
540 540 else:
541 541 res += re.escape(c)
542 542 return rematch(res, l)
543 543
544 544 pos = -1
545 545 postout = []
546 546 ret = 0
547 547 for n, l in enumerate(output):
548 548 lout, lcmd = l, None
549 549 if salt in l:
550 550 lout, lcmd = l.split(salt, 1)
551 551
552 552 if lout:
553 553 if lcmd:
554 554 lout += ' (no-eol)\n'
555 555
556 556 el = None
557 557 if pos in expected and expected[pos]:
558 558 el = expected[pos].pop(0)
559 559
560 560 if el == lout: # perfect match (fast)
561 561 postout.append(" " + lout)
562 562 elif (el and
563 563 (el.endswith(" (re)\n") and rematch(el[:-6] + '\n', lout) or
564 564 el.endswith(" (glob)\n") and globmatch(el[:-8] + '\n', lout)
565 565 or el.endswith(" (esc)\n") and
566 566 el.decode('string-escape') == l)):
567 567 postout.append(" " + el) # fallback regex/glob/esc match
568 568 else:
569 569 if needescape(lout):
570 570 lout = stringescape(lout.rstrip('\n')) + " (esc)\n"
571 571 postout.append(" " + lout) # let diff deal with it
572 572
573 573 if lcmd:
574 574 # add on last return code
575 575 ret = int(lcmd.split()[1])
576 576 if ret != 0:
577 577 postout.append(" [%s]\n" % ret)
578 578 if pos in after:
579 579 postout += after.pop(pos)
580 580 pos = int(lcmd.split()[0])
581 581
582 582 if pos in after:
583 583 postout += after.pop(pos)
584 584
585 585 return exitcode, postout
586 586
587 587 wifexited = getattr(os, "WIFEXITED", lambda x: False)
588 588 def run(cmd, options, replacements):
589 589 """Run command in a sub-process, capturing the output (stdout and stderr).
590 590 Return a tuple (exitcode, output). output is None in debug mode."""
591 591 # TODO: Use subprocess.Popen if we're running on Python 2.4
592 592 if options.debug:
593 593 proc = subprocess.Popen(cmd, shell=True)
594 594 ret = proc.wait()
595 595 return (ret, None)
596 596
597 597 if os.name == 'nt' or sys.platform.startswith('java'):
598 598 tochild, fromchild = os.popen4(cmd)
599 599 tochild.close()
600 600 output = fromchild.read()
601 601 ret = fromchild.close()
602 602 if ret is None:
603 603 ret = 0
604 604 else:
605 605 proc = Popen4(cmd)
606 606 def cleanup():
607 607 os.kill(proc.pid, signal.SIGTERM)
608 608 ret = proc.wait()
609 609 if ret == 0:
610 610 ret = signal.SIGTERM << 8
611 611 killdaemons()
612 612 return ret
613 613
614 614 try:
615 615 output = ''
616 616 proc.tochild.close()
617 617 output = proc.fromchild.read()
618 618 ret = proc.wait()
619 619 if wifexited(ret):
620 620 ret = os.WEXITSTATUS(ret)
621 621 except Timeout:
622 622 vlog('# Process %d timed out - killing it' % proc.pid)
623 623 cleanup()
624 624 ret = 'timeout'
625 625 output += ("\n### Abort: timeout after %d seconds.\n"
626 626 % options.timeout)
627 627 except KeyboardInterrupt:
628 628 vlog('# Handling keyboard interrupt')
629 629 cleanup()
630 630 raise
631 631
632 632 for s, r in replacements:
633 633 output = re.sub(s, r, output)
634 634 return ret, splitnewlines(output)
635 635
636 636 def runone(options, test, skips, fails, ignores):
637 637 '''tristate output:
638 638 None -> skipped
639 639 True -> passed
640 640 False -> failed'''
641 641
642 642 testpath = os.path.join(TESTDIR, test)
643 643
644 644 def skip(msg):
645 645 if not options.verbose:
646 646 skips.append((test, msg))
647 647 else:
648 648 print "\nSkipping %s: %s" % (testpath, msg)
649 649 return None
650 650
651 651 def fail(msg, ret):
652 652 if not options.nodiff:
653 653 print "\nERROR: %s %s" % (testpath, msg)
654 654 if not ret and options.interactive:
655 655 print "Accept this change? [n] ",
656 656 answer = sys.stdin.readline().strip()
657 657 if answer.lower() in "y yes".split():
658 658 if test.endswith(".t"):
659 659 rename(test + ".err", test)
660 660 else:
661 661 rename(test + ".err", test + ".out")
662 662 return
663 663 fails.append((test, msg))
664 664
665 665 if (test.startswith("test-") and '~' not in test and
666 666 ('.' not in test or test.endswith('.py') or
667 667 test.endswith('.bat') or test.endswith('.t'))):
668 668 if not os.path.exists(test):
669 669 skip("doesn't exist")
670 670 return None
671 671 else:
672 672 return None # not a supported test, don't record
673 673
674 if options.keywords:
675 fp = open(test)
676 t = fp.read().lower() + test.lower()
677 fp.close()
678 for k in options.keywords.lower().split():
679 if k in t:
680 break
681 else:
682 ignores.append((test, "doesn't match keyword"))
683 return None
684
674 685 vlog("# Test", test)
675 686
676 687 # create a fresh hgrc
677 688 hgrc = open(HGRCPATH, 'w+')
678 689 hgrc.write('[ui]\n')
679 690 hgrc.write('slash = True\n')
680 691 hgrc.write('[defaults]\n')
681 692 hgrc.write('backout = -d "0 0"\n')
682 693 hgrc.write('commit = -d "0 0"\n')
683 694 hgrc.write('tag = -d "0 0"\n')
684 695 if options.inotify:
685 696 hgrc.write('[extensions]\n')
686 697 hgrc.write('inotify=\n')
687 698 hgrc.write('[inotify]\n')
688 699 hgrc.write('pidfile=%s\n' % DAEMON_PIDS)
689 700 hgrc.write('appendpid=True\n')
690 701 hgrc.close()
691 702
692 703 ref = os.path.join(TESTDIR, test+".out")
693 704 err = os.path.join(TESTDIR, test+".err")
694 705 if os.path.exists(err):
695 706 os.remove(err) # Remove any previous output files
696 707 try:
697 708 tf = open(testpath)
698 709 firstline = tf.readline().rstrip()
699 710 tf.close()
700 711 except:
701 712 firstline = ''
702 713 lctest = test.lower()
703 714
704 715 if lctest.endswith('.py') or firstline == '#!/usr/bin/env python':
705 716 runner = pytest
706 717 elif lctest.endswith('.t'):
707 718 runner = tsttest
708 719 ref = testpath
709 720 else:
710 721 # do not try to run non-executable programs
711 722 if not os.access(testpath, os.X_OK):
712 723 return skip("not executable")
713 724 runner = shtest
714 725
715 726 # Make a tmp subdirectory to work in
716 727 testtmp = os.environ["TESTTMP"] = os.environ["HOME"] = \
717 728 os.path.join(HGTMP, test)
718 729
719 730 os.mkdir(testtmp)
720 731 os.chdir(testtmp)
721 732
722 733 if options.timeout > 0:
723 734 signal.alarm(options.timeout)
724 735
725 736 ret, out = runner(testpath, options, [
726 737 (re.escape(testtmp), '$TESTTMP'),
727 738 (r':%s\b' % options.port, ':$HGPORT'),
728 739 (r':%s\b' % (options.port + 1), ':$HGPORT1'),
729 740 (r':%s\b' % (options.port + 2), ':$HGPORT2'),
730 741 ])
731 742 vlog("# Ret was:", ret)
732 743
733 744 if options.timeout > 0:
734 745 signal.alarm(0)
735 746
736 747 mark = '.'
737 748
738 749 skipped = (ret == SKIPPED_STATUS)
739 750
740 751 # If we're not in --debug mode and reference output file exists,
741 752 # check test output against it.
742 753 if options.debug:
743 754 refout = None # to match "out is None"
744 755 elif os.path.exists(ref):
745 756 f = open(ref, "r")
746 757 refout = splitnewlines(f.read())
747 758 f.close()
748 759 else:
749 760 refout = []
750 761
751 762 if (ret != 0 or out != refout) and not skipped and not options.debug:
752 763 # Save errors to a file for diagnosis
753 764 f = open(err, "wb")
754 765 for line in out:
755 766 f.write(line)
756 767 f.close()
757 768
758 769 if skipped:
759 770 mark = 's'
760 771 if out is None: # debug mode: nothing to parse
761 772 missing = ['unknown']
762 773 failed = None
763 774 else:
764 775 missing, failed = parsehghaveoutput(out)
765 776 if not missing:
766 777 missing = ['irrelevant']
767 778 if failed:
768 779 fail("hghave failed checking for %s" % failed[-1], ret)
769 780 skipped = False
770 781 else:
771 782 skip(missing[-1])
772 783 elif out != refout:
773 784 mark = '!'
774 785 if ret == 'timeout':
775 786 fail("timed out", ret)
776 787 elif ret:
777 788 fail("output changed and returned error code %d" % ret, ret)
778 789 else:
779 790 fail("output changed", ret)
780 791 if ret != 'timeout' and not options.nodiff:
781 792 if options.view:
782 793 os.system("%s %s %s" % (options.view, ref, err))
783 794 else:
784 795 showdiff(refout, out, ref, err)
785 796 ret = 1
786 797 elif ret:
787 798 mark = '!'
788 799 fail("returned error code %d" % ret, ret)
789 800
790 801 if not options.verbose:
791 802 sys.stdout.write(mark)
792 803 sys.stdout.flush()
793 804
794 805 killdaemons()
795 806
796 807 os.chdir(TESTDIR)
797 808 if not options.keep_tmpdir:
798 809 shutil.rmtree(testtmp, True)
799 810 if skipped:
800 811 return None
801 812 return ret == 0
802 813
803 814 _hgpath = None
804 815
805 816 def _gethgpath():
806 817 """Return the path to the mercurial package that is actually found by
807 818 the current Python interpreter."""
808 819 global _hgpath
809 820 if _hgpath is not None:
810 821 return _hgpath
811 822
812 823 cmd = '%s -c "import mercurial; print mercurial.__path__[0]"'
813 824 pipe = os.popen(cmd % PYTHON)
814 825 try:
815 826 _hgpath = pipe.read().strip()
816 827 finally:
817 828 pipe.close()
818 829 return _hgpath
819 830
820 831 def _checkhglib(verb):
821 832 """Ensure that the 'mercurial' package imported by python is
822 833 the one we expect it to be. If not, print a warning to stderr."""
823 834 expecthg = os.path.join(PYTHONDIR, 'mercurial')
824 835 actualhg = _gethgpath()
825 836 if actualhg != expecthg:
826 837 sys.stderr.write('warning: %s with unexpected mercurial lib: %s\n'
827 838 ' (expected %s)\n'
828 839 % (verb, actualhg, expecthg))
829 840
830 841 def runchildren(options, tests):
831 842 if INST:
832 843 installhg(options)
833 844 _checkhglib("Testing")
834 845
835 846 optcopy = dict(options.__dict__)
836 847 optcopy['jobs'] = 1
837 848 del optcopy['blacklist']
838 849 if optcopy['with_hg'] is None:
839 850 optcopy['with_hg'] = os.path.join(BINDIR, "hg")
840 851 optcopy.pop('anycoverage', None)
841 852
842 853 opts = []
843 854 for opt, value in optcopy.iteritems():
844 855 name = '--' + opt.replace('_', '-')
845 856 if value is True:
846 857 opts.append(name)
847 858 elif value is not None:
848 859 opts.append(name + '=' + str(value))
849 860
850 861 tests.reverse()
851 862 jobs = [[] for j in xrange(options.jobs)]
852 863 while tests:
853 864 for job in jobs:
854 865 if not tests:
855 866 break
856 867 job.append(tests.pop())
857 868 fps = {}
858 869
859 870 for j, job in enumerate(jobs):
860 871 if not job:
861 872 continue
862 873 rfd, wfd = os.pipe()
863 874 childopts = ['--child=%d' % wfd, '--port=%d' % (options.port + j * 3)]
864 875 childtmp = os.path.join(HGTMP, 'child%d' % j)
865 876 childopts += ['--tmpdir', childtmp]
866 877 cmdline = [PYTHON, sys.argv[0]] + opts + childopts + job
867 878 vlog(' '.join(cmdline))
868 879 fps[os.spawnvp(os.P_NOWAIT, cmdline[0], cmdline)] = os.fdopen(rfd, 'r')
869 880 os.close(wfd)
870 881 signal.signal(signal.SIGINT, signal.SIG_IGN)
871 882 failures = 0
872 883 tested, skipped, failed = 0, 0, 0
873 884 skips = []
874 885 fails = []
875 886 while fps:
876 887 pid, status = os.wait()
877 888 fp = fps.pop(pid)
878 889 l = fp.read().splitlines()
879 890 try:
880 891 test, skip, fail = map(int, l[:3])
881 892 except ValueError:
882 893 test, skip, fail = 0, 0, 0
883 894 split = -fail or len(l)
884 895 for s in l[3:split]:
885 896 skips.append(s.split(" ", 1))
886 897 for s in l[split:]:
887 898 fails.append(s.split(" ", 1))
888 899 tested += test
889 900 skipped += skip
890 901 failed += fail
891 902 vlog('pid %d exited, status %d' % (pid, status))
892 903 failures |= status
893 904 print
894 905 if not options.noskips:
895 906 for s in skips:
896 907 print "Skipped %s: %s" % (s[0], s[1])
897 908 for s in fails:
898 909 print "Failed %s: %s" % (s[0], s[1])
899 910
900 911 _checkhglib("Tested")
901 912 print "# Ran %d tests, %d skipped, %d failed." % (
902 913 tested, skipped, failed)
903 914
904 915 if options.anycoverage:
905 916 outputcoverage(options)
906 917 sys.exit(failures != 0)
907 918
908 919 def runtests(options, tests):
909 920 global DAEMON_PIDS, HGRCPATH
910 921 DAEMON_PIDS = os.environ["DAEMON_PIDS"] = os.path.join(HGTMP, 'daemon.pids')
911 922 HGRCPATH = os.environ["HGRCPATH"] = os.path.join(HGTMP, '.hgrc')
912 923
913 924 try:
914 925 if INST:
915 926 installhg(options)
916 927 _checkhglib("Testing")
917 928
918 929 if options.timeout > 0:
919 930 try:
920 931 signal.signal(signal.SIGALRM, alarmed)
921 932 vlog('# Running each test with %d second timeout' %
922 933 options.timeout)
923 934 except AttributeError:
924 935 print 'WARNING: cannot run tests with timeouts'
925 936 options.timeout = 0
926 937
927 938 tested = 0
928 939 failed = 0
929 940 skipped = 0
930 941
931 942 if options.restart:
932 943 orig = list(tests)
933 944 while tests:
934 945 if os.path.exists(tests[0] + ".err"):
935 946 break
936 947 tests.pop(0)
937 948 if not tests:
938 949 print "running all tests"
939 950 tests = orig
940 951
941 952 skips = []
942 953 fails = []
943 954 ignores = []
944 955
945 956 for test in tests:
946 957 if options.blacklist:
947 958 filename = options.blacklist.get(test)
948 959 if filename is not None:
949 960 skipped.append((test, "blacklisted (%s)" % filename))
950 961 skipped += 1
951 962 continue
952 963
953 964 if options.retest and not os.path.exists(test + ".err"):
954 965 ignores.append((test, "not retesting"))
955 966 continue
956 967
957 if options.keywords:
958 try:
959 fp = open(test)
960 except IOError:
961 continue
962 t = fp.read().lower() + test.lower()
963 fp.close()
964 for k in options.keywords.lower().split():
965 if k in t:
966 break
967 else:
968 ignores.append((test, "doesn't match keyword"))
969 continue
970
971 968 ret = runone(options, test, skips, fails, ignores)
972 969 if ret is None:
973 970 skipped += 1
974 971 elif not ret:
975 972 failed += 1
976 973 if options.first:
977 974 break
978 975 tested += 1
979 976
980 977 if options.child:
981 978 fp = os.fdopen(options.child, 'w')
982 979 fp.write('%d\n%d\n%d\n' % (tested, skipped, failed))
983 980 for s in skips:
984 981 fp.write("%s %s\n" % s)
985 982 for s in fails:
986 983 fp.write("%s %s\n" % s)
987 984 fp.close()
988 985 else:
989 986 print
990 987 for s in skips:
991 988 print "Skipped %s: %s" % s
992 989 for s in fails:
993 990 print "Failed %s: %s" % s
994 991 _checkhglib("Tested")
995 992 print "# Ran %d tests, %d skipped, %d failed." % (
996 993 tested, len(skips) + len(ignores), failed)
997 994
998 995 if options.anycoverage:
999 996 outputcoverage(options)
1000 997 except KeyboardInterrupt:
1001 998 failed = True
1002 999 print "\ninterrupted!"
1003 1000
1004 1001 if failed:
1005 1002 sys.exit(1)
1006 1003
1007 1004 def main():
1008 1005 (options, args) = parseargs()
1009 1006 if not options.child:
1010 1007 os.umask(022)
1011 1008
1012 1009 checktools()
1013 1010
1014 1011 if len(args) == 0:
1015 1012 args = os.listdir(".")
1016 1013 args.sort()
1017 1014
1018 1015 tests = args
1019 1016
1020 1017 # Reset some environment variables to well-known values so that
1021 1018 # the tests produce repeatable output.
1022 1019 os.environ['LANG'] = os.environ['LC_ALL'] = os.environ['LANGUAGE'] = 'C'
1023 1020 os.environ['TZ'] = 'GMT'
1024 1021 os.environ["EMAIL"] = "Foo Bar <foo.bar@example.com>"
1025 1022 os.environ['CDPATH'] = ''
1026 1023 os.environ['COLUMNS'] = '80'
1027 1024 os.environ['GREP_OPTIONS'] = ''
1028 1025 os.environ['http_proxy'] = ''
1029 1026
1030 1027 # unset env related to hooks
1031 1028 for k in os.environ.keys():
1032 1029 if k.startswith('HG_'):
1033 1030 # can't remove on solaris
1034 1031 os.environ[k] = ''
1035 1032 del os.environ[k]
1036 1033
1037 1034 global TESTDIR, HGTMP, INST, BINDIR, PYTHONDIR, COVERAGE_FILE
1038 1035 TESTDIR = os.environ["TESTDIR"] = os.getcwd()
1039 1036 if options.tmpdir:
1040 1037 options.keep_tmpdir = True
1041 1038 tmpdir = options.tmpdir
1042 1039 if os.path.exists(tmpdir):
1043 1040 # Meaning of tmpdir has changed since 1.3: we used to create
1044 1041 # HGTMP inside tmpdir; now HGTMP is tmpdir. So fail if
1045 1042 # tmpdir already exists.
1046 1043 sys.exit("error: temp dir %r already exists" % tmpdir)
1047 1044
1048 1045 # Automatically removing tmpdir sounds convenient, but could
1049 1046 # really annoy anyone in the habit of using "--tmpdir=/tmp"
1050 1047 # or "--tmpdir=$HOME".
1051 1048 #vlog("# Removing temp dir", tmpdir)
1052 1049 #shutil.rmtree(tmpdir)
1053 1050 os.makedirs(tmpdir)
1054 1051 else:
1055 1052 tmpdir = tempfile.mkdtemp('', 'hgtests.')
1056 1053 HGTMP = os.environ['HGTMP'] = os.path.realpath(tmpdir)
1057 1054 DAEMON_PIDS = None
1058 1055 HGRCPATH = None
1059 1056
1060 1057 os.environ["HGEDITOR"] = sys.executable + ' -c "import sys; sys.exit(0)"'
1061 1058 os.environ["HGMERGE"] = "internal:merge"
1062 1059 os.environ["HGUSER"] = "test"
1063 1060 os.environ["HGENCODING"] = "ascii"
1064 1061 os.environ["HGENCODINGMODE"] = "strict"
1065 1062 os.environ["HGPORT"] = str(options.port)
1066 1063 os.environ["HGPORT1"] = str(options.port + 1)
1067 1064 os.environ["HGPORT2"] = str(options.port + 2)
1068 1065
1069 1066 if options.with_hg:
1070 1067 INST = None
1071 1068 BINDIR = os.path.dirname(os.path.realpath(options.with_hg))
1072 1069
1073 1070 # This looks redundant with how Python initializes sys.path from
1074 1071 # the location of the script being executed. Needed because the
1075 1072 # "hg" specified by --with-hg is not the only Python script
1076 1073 # executed in the test suite that needs to import 'mercurial'
1077 1074 # ... which means it's not really redundant at all.
1078 1075 PYTHONDIR = BINDIR
1079 1076 else:
1080 1077 INST = os.path.join(HGTMP, "install")
1081 1078 BINDIR = os.environ["BINDIR"] = os.path.join(INST, "bin")
1082 1079 PYTHONDIR = os.path.join(INST, "lib", "python")
1083 1080
1084 1081 os.environ["BINDIR"] = BINDIR
1085 1082 os.environ["PYTHON"] = PYTHON
1086 1083
1087 1084 if not options.child:
1088 1085 path = [BINDIR] + os.environ["PATH"].split(os.pathsep)
1089 1086 os.environ["PATH"] = os.pathsep.join(path)
1090 1087
1091 1088 # Include TESTDIR in PYTHONPATH so that out-of-tree extensions
1092 1089 # can run .../tests/run-tests.py test-foo where test-foo
1093 1090 # adds an extension to HGRC
1094 1091 pypath = [PYTHONDIR, TESTDIR]
1095 1092 # We have to augment PYTHONPATH, rather than simply replacing
1096 1093 # it, in case external libraries are only available via current
1097 1094 # PYTHONPATH. (In particular, the Subversion bindings on OS X
1098 1095 # are in /opt/subversion.)
1099 1096 oldpypath = os.environ.get(IMPL_PATH)
1100 1097 if oldpypath:
1101 1098 pypath.append(oldpypath)
1102 1099 os.environ[IMPL_PATH] = os.pathsep.join(pypath)
1103 1100
1104 1101 COVERAGE_FILE = os.path.join(TESTDIR, ".coverage")
1105 1102
1106 1103 vlog("# Using TESTDIR", TESTDIR)
1107 1104 vlog("# Using HGTMP", HGTMP)
1108 1105 vlog("# Using PATH", os.environ["PATH"])
1109 1106 vlog("# Using", IMPL_PATH, os.environ[IMPL_PATH])
1110 1107
1111 1108 try:
1112 1109 if len(tests) > 1 and options.jobs > 1:
1113 1110 runchildren(options, tests)
1114 1111 else:
1115 1112 runtests(options, tests)
1116 1113 finally:
1117 1114 time.sleep(1)
1118 1115 cleanup(options)
1119 1116
1120 1117 if __name__ == '__main__':
1121 1118 main()
General Comments 0
You need to be logged in to leave comments. Login now