#ifndef CPPUNIT_EXTENSIONS_AUTOREGISTERSUITE_H

#define CPPUNIT_EXTENSIONS_AUTOREGISTERSUITE_H

#include <cppunit/extensions/TestSuiteFactory.h>

#include <cppunit/extensions/TestFactoryRegistry.h>

#include <string>

CPPUNIT_NS_BEGIN

/*! \brief (Implementation) Automatically register the test suite of the specified type.

 *

 * You should not use this class directly. Instead, use the following macros:

 * - CPPUNIT_TEST_SUITE_REGISTRATION()

 * - CPPUNIT_TEST_SUITE_NAMED_REGISTRATION()

 *

 * This object will register the test returned by TestCaseType::suite()

 * when constructed to the test registry.

 *

 * This object is intented to be used as a static variable.

 *

 *

 * \param TestCaseType Type of the test case which suite is registered.

 * \see CPPUNIT_TEST_SUITE_REGISTRATION, CPPUNIT_TEST_SUITE_NAMED_REGISTRATION

 * \see CppUnit::TestFactoryRegistry.

 */

template<class TestCaseType>

class AutoRegisterSuite

{

public:

  /** Auto-register the suite factory in the global registry.

   */

  AutoRegisterSuite()

      : m_registry( &TestFactoryRegistry::getRegistry() )

  {

    m_registry->registerFactory( &m_factory );

  }

  /** Auto-register the suite factory in the specified registry.

   * \param name Name of the registry.

   */

  AutoRegisterSuite( const std::string &name )

      : m_registry( &TestFactoryRegistry::getRegistry( name ) )

  {

    m_registry->registerFactory( &m_factory );

  }

  ~AutoRegisterSuite()

  {

    if ( TestFactoryRegistry::isValid() )

      m_registry->unregisterFactory( &m_factory );

  }

private:

  TestFactoryRegistry *m_registry;

  TestSuiteFactory<TestCaseType> m_factory;

};

/*! \brief (Implementation) Automatically adds a registry into another registry.

 *

 * Don't use this class. Use the macros CPPUNIT_REGISTRY_ADD() and

 * CPPUNIT_REGISTRY_ADD_TO_DEFAULT() instead.

 */

class AutoRegisterRegistry

{

public:

  AutoRegisterRegistry( const std::string &which,

                        const std::string &to )

  {

    TestFactoryRegistry::getRegistry( to ).addRegistry( which );

  }

  AutoRegisterRegistry( const std::string &which )

  {

    TestFactoryRegistry::getRegistry().addRegistry( which );

  }

};

CPPUNIT_NS_END

#endif  // CPPUNIT_EXTENSIONS_AUTOREGISTERSUITE_H