#! /usr/bin/env python

"""Generic trace-comparison-type regression test."""

import os

import shutil

def run(verbose, generate):

    """Execute a test."""

    testName = "csma-multicast"

    repoName = "ns-3-ref-traces/"

    refDirName = testName + ".ref"

    if not os.path.exists(repoName):

        print"No reference trace repository"

        return 1

    if generate:

        if not os.path.exists(repoName + refDirName):

            print "creating new " + repoName + refDirName

            os.mkdir(repoName + refDirName)

        os.system("./waf --cwd regression/" + repoName + refDirName +

            " --run " + testName + " >& /dev/null")

        print "Remember to commit " + repoName + refDirName

        return 0

    else:

        if not os.path.exists(repoName + refDirName):

            print "Cannot locate reference traces"

            return 1

        shutil.rmtree("traces");

        os.mkdir("traces")

        os.system("./waf --cwd regression/traces --run " +

          testName + " >& /dev/null")

        return os.system("diff -q traces " + repoName + refDirName +

         " >& /dev/null")