#include <TestSuite.h>
#include <cppunit/Test.h>
#include <cppunit/TestResult.h>

using std::map;
using std::string;

// Default constructor
_EXPORT
BTestSuite::BTestSuite( string name )
	: fName(name)
{
}

// Destructor
_EXPORT
BTestSuite::~BTestSuite() {
	deleteContents();
}


// Deletes all tests in the suite.
_EXPORT
void
BTestSuite::deleteContents() {
	for ( map<string, CppUnit::Test*>::iterator it = fTests.begin();
		   it != fTests.end();
		     ++it)
		delete it->second;
	fTests.clear();
}


/// Runs the tests and collects their result in a TestResult.
_EXPORT
void
BTestSuite::run( CppUnit::TestResult *result ) {
	for ( map<string, CppUnit::Test*>::iterator it = fTests.begin();
    	   it != fTests.end();
		     ++it )
	{
		if ( result->shouldStop() )
			break;

		Test *test = it->second;
		test->run( result );
	}
}


// Counts the number of test cases that will be run by this test.
_EXPORT
int
BTestSuite::countTestCases() const {
	int count = 0;

	for ( map<string, CppUnit::Test *>::const_iterator it = fTests.begin();
		   it != fTests.end();
		     ++it )
		count += it->second->countTestCases();

	return count;
}


// Adds a test to the suite.
_EXPORT
void
BTestSuite::addTest(string name, CppUnit::Test *test) {
	fTests[name] = test;
}


// Returns a string representation of the test suite.
_EXPORT
string
BTestSuite::toString() const {
	return "suite " + getName();
}


// Returns the name of the test suite.
_EXPORT
string
BTestSuite::getName() const {
	return fName;
}


_EXPORT
const map<string, CppUnit::Test*> &
BTestSuite::getTests() const {
	return fTests;
}