AutoRegisterSuite.h

Go to the documentation of this file.

#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
 
 
template<class TestCaseType>
class AutoRegisterSuite
{
public:
  AutoRegisterSuite()
      : m_registry( &TestFactoryRegistry::getRegistry() )
  {
    m_registry->registerFactory( &m_factory );
  }
 
  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;
};
 
 
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