/*
-------------------------------------------------------------------------
 CxxTest: A lightweight C++ unit testing library.
 Copyright (c) 2008 Sandia Corporation.
 This software is distributed under the LGPL License v3
 For more information, see the COPYING file in the top CxxTest directory.
 Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
 the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------
*/

#ifndef __cxxtest_TestRunner_h__
#define __cxxtest_TestRunner_h__

//
// TestRunner is the class that runs all the tests.
// To use it, create an object that implements the TestListener
// interface and call TestRunner::runAllTests( myListener );
//

#include <cxxtest/TestListener.h>
#include <cxxtest/RealDescriptions.h>
#include <cxxtest/TestSuite.h>
#include <cxxtest/TestTracker.h>

namespace CxxTest
{
extern bool g_RunDisabled;

class TestRunner
{
public:

    static void setListener(TestListener* listener)
    {
        tracker().setListener(listener);
    }

    static void runAllTests(TestListener &listener)
    {
        tracker().setListener(&listener);
        _TS_TRY { TestRunner().runWorld(); }
        _TS_LAST_CATCH( { tracker().failedTest(__FILE__, __LINE__, "Exception thrown from world"); });
        tracker().setListener(0);
    }

    static void runAllTests(TestListener *listener)
    {
        if (listener)
        {
            listener->warning(__FILE__, __LINE__, "Deprecated; Use runAllTests( TestListener & )");
            runAllTests(*listener);
        }
    }

private:
    void runWorld()
    {
        RealWorldDescription wd;
        WorldGuard sg;

        tracker().enterWorld(wd);
        if (wd.setUp())
        {
            for (SuiteDescription *sd = wd.firstSuite(); sd; sd = sd->next())
            {
                if (sd->active())
                {
                    runSuite(*sd);
                }
            }

            wd.tearDown();
        }
        tracker().leaveWorld(wd);
    }

    void runSuite(SuiteDescription &sd)
    {
        StateGuard sg;

        tracker().enterSuite(sd);
        if (sd.setUp())
        {
            for (TestDescription *td = sd.firstTest(); td; td = td->next())
            {
                if ((g_RunDisabled || !strstr(td->testName(), "DISABLED")) && td->active())
                {
                    runTest(*td);
                }
            }

            sd.tearDown();
        }
        tracker().leaveSuite(sd);
    }

    void runTest(TestDescription &td)
    {
        StateGuard sg;

        tracker().enterTest(td);
        if (td.setUp())
        {
            td.run();
            td.tearDown();
        }
        tracker().leaveTest(td);
    }

    class StateGuard
    {
#ifdef _CXXTEST_HAVE_EH
        bool _abortTestOnFail;
#endif // _CXXTEST_HAVE_EH
        unsigned _maxDumpSize;

    public:
        StateGuard()
        {
#ifdef _CXXTEST_HAVE_EH
            _abortTestOnFail = abortTestOnFail();
#endif // _CXXTEST_HAVE_EH
            _maxDumpSize = maxDumpSize();
        }

        ~StateGuard()
        {
#ifdef _CXXTEST_HAVE_EH
            setAbortTestOnFail(_abortTestOnFail);
#endif // _CXXTEST_HAVE_EH
            setMaxDumpSize(_maxDumpSize);
        }
    };

    class WorldGuard : public StateGuard
    {
    public:
        WorldGuard() : StateGuard()
        {
#ifdef _CXXTEST_HAVE_EH
            setAbortTestOnFail(CXXTEST_DEFAULT_ABORT);
#endif // _CXXTEST_HAVE_EH
            setMaxDumpSize(CXXTEST_MAX_DUMP_SIZE);
        }
    };
};

//
// For --no-static-init
//
void initialize();
}


#endif // __cxxtest_TestRunner_h__