test/test_main.cpp

Go to the documentation of this file.

#include <getopt.h>

#include "Epetra_ConfigDefs.h"
#ifdef HAVE_MPI
#include <mpi.h>
#include "Epetra_MpiComm.h"
#else
#include "Epetra_SerialComm.h"
#endif

// include after mpi.h
#include "pbe.h"

#include "Epetra_CrsMatrix.h"
#include "Epetra_Map.h"
#include "EpetraExt_MatrixMatrix.h"

#include <tut.h>
#include <tut_reporter.h>

#include "rlog/rlog.h"
#include "rlog/RLogChannel.h"
#include "rlog/StdioNode.h"
#include "myrlog.h"

using namespace std;
using namespace rlog;

namespace tut {
    test_runner_singleton runner;
    
    Epetra_Comm* comm_world;

    Epetra_Comm& get_comm_world() {
        return *comm_world;
    }
}

void usage(const Epetra_Comm& comm) {
    if (comm.MyPID() == 0) {
        cerr << "test_runner" << endl;
        cerr << endl;
        cerr << "Driver for femaXX unit tests." << endl;
        cerr << endl;
        cerr << "Usage: test_runner [options] [test_group]" << endl;
        cerr << endl;
        cerr << "options:" << endl;
        cerr << "  -l   Print list of unit test groups and exit." << endl;
        cerr << endl;
        cerr << "If test_group is omitted, all tests are started." << endl;
        cerr << endl;
    }
}

void finalize() {
#ifdef HAVE_MPI
    MPI_Finalize();
#endif
}


int main(int argc, char *argv[]) {
#ifdef HAVE_MPI
    MPI_Init(&argc, &argv);
    atexit(finalize);
    tut::comm_world = new Epetra_MpiComm(MPI_COMM_WORLD);
#else
    tut::comm_world = new Epetra_SerialComm;
#endif

    // extern char *optarg;
    extern int optind, optopt;
    int dflg = 0;
    int lflg = 0;
    int errflg = 0;
    int c;
    while ((c = getopt(argc, argv, "dl")) != -1) {
        switch(c) {
        case 'd':
            dflg++;
            break;
        case 'l':
            lflg ++;
            break;
        case '?':
            fprintf(stderr, "Unrecognized option: -%c\n", optopt);
            errflg++;
        }
    }
    if (errflg) {
        usage(tut::get_comm_world());
        exit(1);
    }
    if (lflg) {
        tut::groupnames group_names = tut::runner.get().list_groups();
        for (tut::groupnames::iterator it = group_names.begin(); it != group_names.end(); ++ it) {
            cerr << *it << endl;
        }
        exit(0);
    }

    StdioNode stdLog;

    GetGlobalChannel("warning/all", Log_Warning);
    GetGlobalChannel("error/all", Log_Error);
    GetGlobalChannel("info/all", Log_Info);
    GetGlobalChannel("debug/all", Log_Debug);

    stdLog.subscribeTo(GetGlobalChannel("error"));
    stdLog.subscribeTo(GetGlobalChannel("warning"));
    if (dflg) {
        if (tut::comm_world->MyPID() == 0)
            stdLog.subscribeTo(GetGlobalChannel("info"));
        stdLog.subscribeTo(GetGlobalChannel("debug"));
    }

    const int nof_pbe_timers = 150;
    pbe_init("femaxx", nof_pbe_timers, 0, NULL);

    tut::reporter visi;
    if (tut::comm_world->MyPID() == 0)
        tut::runner.get().set_callback(&visi);

    if (optind < argc) {
        tut::runner.get().run_tests(argv[optind]);
    } else {
        // run all tests in all groups
        tut::runner.get().run_tests();
    }

    delete tut::comm_world;
    return 0;
}

---