run_qmtest Namespace Reference

Classes
class	Options

Functions
def	parseOptions
def	main

Function Documentation

def run_qmtest.main

(

argv = None

)

Definition at line 75 of file run_qmtest.py.

def main(argv = None):
    opts = parseOptions(argv)

    print "==========> Running tests for package %s" % opts.package

    if not os.path.isdir(opts.qmtest_dir):
        print "==========> No test directory, exiting"
        return
        
    # create the destination directory if necessary
    if opts.output:
        results_dest_dir = os.path.realpath(os.path.join(opts.qmtest_dir, os.path.dirname(opts.output)))
        if not os.path.exists(results_dest_dir):
            print "==========> Creating '%s'" % results_dest_dir
            os.makedirs(results_dest_dir, 0755)

    print "==========> Entering '%s'" % opts.qmtest_dir
    os.chdir(opts.qmtest_dir)

    qmtest_cmd = ["qmtest"]

    if not os.path.isdir("QMTest"):
        # If the test-db directory is not initialized, we have to initialize it
        try:
            os.mkdir("QMTest")
        except OSError:
            # we cannot create the directory... we need a temporary test database
            from tempfile import mkdtemp
            from shutil import copytree
            # the extra level is needed because of a limitation in copytree
            tdb = os.path.join(mkdtemp("GaudiTest"), "qmtest")
            copytree(".", tdb)
            os.chmod(tdb, 0700) # This is needed because copytree will copy also the restrictive permissions
            qmtest_cmd += ["--tdb", tdb]
        print "==========> Initializing QMTest database"
        os.system(" ".join(qmtest_cmd + ["create-tdb"]))

    if opts.xml_output:
        opts.qmtest_args.insert(0, '''--result-stream "GaudiTest.XMLResultStream(dir='%s',prefix='%s_')"'''
                                   % (opts.xml_output.replace("\\","\\\\"), opts.package))

    if opts.html_output:
        opts.qmtest_args.insert(0, '''--result-stream "GaudiTest.HTMLResultStream(dir='%s')"'''
                                  % opts.html_output.replace("\\","\\\\"))

    # prepare the qmtest command
    cmd = " ".join(qmtest_cmd + ["run"] + opts.qmtest_args)

    if not opts.have_user_options:
        # If there are no user options, we should check for a default test suite
        # to run.
        # The suite to run is the first one available in the list passed via
        # the environment variable GAUDI_QMTEST_DEFAULT_SUITE (comma-separated list)
        # or the suite with the same name as the package.
        # If none of them is present, no argument is passed to qmtest, so all the tests
        # are run.
        suites = []
        if "GAUDI_QMTEST_DEFAULT_SUITE" in os.environ:
            suites.extend([s.strip().lower()
                           for s in os.environ["GAUDI_QMTEST_DEFAULT_SUITE"].split(",")])
        suites.append(opts.package.lower())
        for s in suites:
            real_name = os.path.join(*[x + '.qms' for x in s.split('.')])
            if os.path.exists(real_name):
                cmd += " %s" % s
                break

    if opts.dry_run:
        print "==========> Would run '%s'"%cmd
    else:
        print "==========> Running '%s'"%cmd
        os.system(cmd)

    # drop the temporary test DB if needed
    if "--tdb" in qmtest_cmd:
        from shutil import rmtree
        def onerror(func, path, exc_info):
            """
            Change the permissions and try again to remove a file or a dir.
            Ignore failures.
            """
            if func in [os.remove, os.rmdir]:
                # change permissions and try again
                try:
                    os.chmod(path, 0600)
                    os.chmod(os.path.dirname(path), 0700)
                    func(path)
                except os.error:
                    pass
        rmtree(os.path.dirname(qmtest_cmd[-1]), onerror = onerror)

# Note: the return code of qmtest is not propagated to avoid that