changeset 2110:25a8d116ab6a

Add a pure python version of run-tests. If this works well for most people, it should replace the shell version of run-test.
author Stephen Darnell <stephen@darnell.plus.com>
date Fri, 21 Apr 2006 18:47:55 +0200
parents b03de24ee2ec
children 2f3e644decd7 2b03c6733efa
files tests/run-tests.py
diffstat 1 files changed, 192 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/run-tests.py	Fri Apr 21 18:47:55 2006 +0200
@@ -0,0 +1,192 @@
+#!/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)