#!/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

def cleanup_exit():

    if verbose:

        print "# Cleaning up HGTMP", HGTMP

    shutil.rmtree(HGTMP, True)

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

if failed:

    sys.exit(1)