#include <cppunit/config/SourcePrefix.h>
#include <cppunit/TestRunner.h>
#include <cppunit/TestPath.h>
#include <cppunit/TestResult.h>


CPPUNIT_NS_BEGIN


TestRunner::WrappingSuite::WrappingSuite( const std::string &name ) 
    : TestSuite( name )
{
}


int 
TestRunner::WrappingSuite::getChildTestCount() const
{
  if ( hasOnlyOneTest() )
    return getUniqueChildTest()->getChildTestCount();
  return TestSuite::getChildTestCount();
}


std::string 
TestRunner::WrappingSuite::getName() const
{
  if ( hasOnlyOneTest() )
    return getUniqueChildTest()->getName();
  return TestSuite::getName();
}


Test *
TestRunner::WrappingSuite::doGetChildTestAt( int index ) const
{
  if ( hasOnlyOneTest() )
    return getUniqueChildTest()->getChildTestAt( index );
  return TestSuite::doGetChildTestAt( index );
}


void 
TestRunner::WrappingSuite::run( TestResult *result )
{
  if ( hasOnlyOneTest() )
    getUniqueChildTest()->run( result );
  else
    TestSuite::run( result );
}


bool 
TestRunner::WrappingSuite::hasOnlyOneTest() const
{
  return TestSuite::getChildTestCount() == 1;
}


Test *
TestRunner::WrappingSuite::getUniqueChildTest() const
{
  return TestSuite::doGetChildTestAt( 0 );
}





TestRunner::TestRunner()
    : m_suite( new WrappingSuite() )
{
}


TestRunner::~TestRunner()
{
  delete m_suite;
}


void 
TestRunner::addTest( Test *test )
{
  m_suite->addTest( test ); 
}


void 
TestRunner::run( TestResult &controller,
                 const std::string &testPath )
{
  TestPath path = m_suite->resolveTestPath( testPath );
  Test *testToRun = path.getChildTest();

  controller.runTest( testToRun );
}


CPPUNIT_NS_END