mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +03:00
193 lines
5.3 KiB
Python
193 lines
5.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
#
|
||
|
# run-tests.py - Run a set of tests on Mercurial
|
||
|
#
|
||
|
# Copyright 2006 Matt Mackall <mpm@selenic.com>
|
||
|
#
|
||
|
# This software may be used and distributed according to the terms
|
||
|
# of the GNU General Public License, incorporated herein by reference.
|
||
|
|
||
|
import os, sys, shutil, re
|
||
|
import tempfile
|
||
|
import difflib
|
||
|
import popen2
|
||
|
from optparse import OptionParser
|
||
|
|
||
|
required_tools = ["python", "diff", "grep", "unzip", "gunzip", "bunzip2", "sed"]
|
||
|
|
||
|
parser = OptionParser()
|
||
|
parser.add_option("-v", "--verbose", action="store_true", dest="verbose",
|
||
|
default=False, help="output verbose messages")
|
||
|
(options, args) = parser.parse_args()
|
||
|
verbose = options.verbose
|
||
|
|
||
|
def vlog(*msg):
|
||
|
if verbose:
|
||
|
for m in msg:
|
||
|
print m,
|
||
|
print
|
||
|
|
||
|
def show_diff(expected, output):
|
||
|
for line in difflib.unified_diff(expected, output,
|
||
|
"Expected output", "Test output", lineterm=''):
|
||
|
print line
|
||
|
|
||
|
def find_program(program):
|
||
|
"""Search PATH for a executable program"""
|
||
|
for p in os.environ.get('PATH', os.defpath).split(os.pathsep):
|
||
|
name = os.path.join(p, program)
|
||
|
if os.access(name, os.X_OK):
|
||
|
return name
|
||
|
return None
|
||
|
|
||
|
# Before we go any further, check for pre-requisite tools
|
||
|
# stuff from coreutils (cat, rm, etc) are not tested
|
||
|
for p in required_tools:
|
||
|
if os.name == 'nt':
|
||
|
p += '.exe'
|
||
|
found = find_program(p)
|
||
|
if found:
|
||
|
vlog("# Found prerequisite", p, "at", found)
|
||
|
else:
|
||
|
print "WARNING: Did not find prerequisite tool: "+p
|
||
|
|
||
|
# Reset some environment variables to well-known values
|
||
|
os.environ['LANG'] = os.environ['LC_ALL'] = 'C'
|
||
|
os.environ['TZ'] = 'GMT'
|
||
|
|
||
|
os.environ["HGEDITOR"] = sys.executable + ' -c "import sys; sys.exit(0)"'
|
||
|
os.environ["HGMERGE"] = sys.executable + ' -c "import sys; sys.exit(0)"'
|
||
|
os.environ["HGUSER"] = "test"
|
||
|
os.environ["HGRCPATH"] = ""
|
||
|
|
||
|
TESTDIR = os.environ["TESTDIR"] = os.getcwd()
|
||
|
HGTMP = os.environ["HGTMP"] = tempfile.mkdtemp("", "hgtests.")
|
||
|
|
||
|
def cleanup_exit():
|
||
|
if verbose:
|
||
|
print "# Cleaning up HGTMP", HGTMP
|
||
|
shutil.rmtree(HGTMP, True)
|
||
|
|
||
|
vlog("# Using TESTDIR", TESTDIR)
|
||
|
vlog("# Using HGTMP", HGTMP)
|
||
|
|
||
|
os.umask(022)
|
||
|
|
||
|
vlog("# Performing temporary installation of HG")
|
||
|
INST = os.path.join(HGTMP, "install")
|
||
|
BINDIR = os.path.join(INST, "bin")
|
||
|
PYTHONDIR = os.path.join(INST, "lib", "python")
|
||
|
installerrs = os.path.join("tests", "install.err")
|
||
|
|
||
|
os.chdir("..") # Get back to hg root
|
||
|
cmd = '%s setup.py install --home="%s" --install-lib="%s" >%s 2>&1' % \
|
||
|
(sys.executable, INST, PYTHONDIR, installerrs)
|
||
|
vlog("# Running", cmd)
|
||
|
if os.system(cmd) == 0:
|
||
|
if not verbose:
|
||
|
os.remove(installerrs)
|
||
|
else:
|
||
|
f = open(installerrs)
|
||
|
for line in f:
|
||
|
print line,
|
||
|
f.close()
|
||
|
cleanup_exit()
|
||
|
sys.exit(1)
|
||
|
os.chdir(TESTDIR)
|
||
|
|
||
|
os.environ["PATH"] = "%s%s%s" % (BINDIR, os.pathsep, os.environ["PATH"])
|
||
|
os.environ["PYTHONPATH"] = PYTHONDIR
|
||
|
|
||
|
tests = 0
|
||
|
failed = 0
|
||
|
|
||
|
def run(cmd, split_lines=True):
|
||
|
"""Run command in a sub-process, capturing the output (stdout and stderr).
|
||
|
Return the exist code, and output."""
|
||
|
# TODO: Use subprocess.Popen if we're running on Python 2.4
|
||
|
if os.name == 'nt':
|
||
|
tochild, fromchild = os.popen4(cmd)
|
||
|
tochild.close()
|
||
|
output = fromchild.read()
|
||
|
ret = fromchild.close()
|
||
|
if ret == None:
|
||
|
ret = 0
|
||
|
else:
|
||
|
proc = popen2.Popen4(cmd)
|
||
|
proc.tochild.close()
|
||
|
output = proc.fromchild.read()
|
||
|
ret = proc.wait()
|
||
|
if split_lines:
|
||
|
output = output.splitlines()
|
||
|
return ret, output
|
||
|
|
||
|
def run_one(test):
|
||
|
vlog("# Test", test)
|
||
|
if not verbose:
|
||
|
sys.stdout.write('.')
|
||
|
sys.stdout.flush()
|
||
|
|
||
|
err = os.path.join(TESTDIR, test+".err")
|
||
|
ref = os.path.join(TESTDIR, test+".out")
|
||
|
|
||
|
if os.path.exists(err):
|
||
|
os.remove(err) # Remove any previous output files
|
||
|
|
||
|
# Make a tmp subdirectory to work in
|
||
|
tmpd = os.path.join(HGTMP, test)
|
||
|
os.mkdir(tmpd)
|
||
|
os.chdir(tmpd)
|
||
|
|
||
|
if test.endswith(".py"):
|
||
|
cmd = '%s "%s"' % (sys.executable, os.path.join(TESTDIR, test))
|
||
|
else:
|
||
|
cmd = '"%s"' % (os.path.join(TESTDIR, test))
|
||
|
|
||
|
# To reliably get the error code from batch files on WinXP,
|
||
|
# the "cmd /c call" prefix is needed. Grrr
|
||
|
if os.name == 'nt' and test.endswith(".bat"):
|
||
|
cmd = 'cmd /c call "%s"' % (os.path.join(TESTDIR, test))
|
||
|
|
||
|
vlog("# Running", cmd)
|
||
|
ret, out = run(cmd)
|
||
|
vlog("# Ret was:", ret)
|
||
|
|
||
|
if ret == 0:
|
||
|
# If reference output file exists, check test output against it
|
||
|
if os.path.exists(ref):
|
||
|
f = open(ref, "r")
|
||
|
ref_out = f.read().splitlines()
|
||
|
f.close()
|
||
|
if out != ref_out:
|
||
|
ret = 1
|
||
|
print "\nERROR: %s output changed" % (test)
|
||
|
show_diff(ref_out, out)
|
||
|
else:
|
||
|
print "\nERROR: %s failed with error code %d" % (test, ret)
|
||
|
|
||
|
if ret != 0: # Save errors to a file for diagnosis
|
||
|
f = open(err, "w")
|
||
|
for line in out:
|
||
|
f.write(line)
|
||
|
f.write("\n")
|
||
|
f.close()
|
||
|
|
||
|
os.chdir(TESTDIR)
|
||
|
shutil.rmtree(tmpd, True)
|
||
|
return ret == 0
|
||
|
|
||
|
for test in os.listdir("."):
|
||
|
if test.startswith("test-"):
|
||
|
if '~' in test or re.search(r'\.(out|err)$', test):
|
||
|
continue
|
||
|
if not run_one(test):
|
||
|
failed += 1
|
||
|
tests += 1
|
||
|
|
||
|
print "# Ran %d tests, %d failed." % (tests, failed)
|
||
|
|
||
|
cleanup_exit()
|
||
|
|
||
|
if failed:
|
||
|
sys.exit(1)
|