#include "CoreSuite.h"

#include "TestSuiteTest.h"

#include <cppunit/TestResult.h>

#include "MockTestCase.h"

CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TestSuiteTest,

                                       coreSuiteName() );

TestSuiteTest::TestSuiteTest()

{

}

TestSuiteTest::~TestSuiteTest()

{

}

void 

TestSuiteTest::setUp()

{

  m_suite = new CPPUNIT_NS::TestSuite();

}

void 

TestSuiteTest::tearDown()

{

  delete m_suite;

}

void 

TestSuiteTest::testConstructor()

{

  std::string name( "MySuite" );

  CPPUNIT_NS::TestSuite suite( name );

  CPPUNIT_ASSERT_EQUAL( name, suite.getName() );

}

void 

TestSuiteTest::testCountTestCasesWithNoTest()

{

  CPPUNIT_ASSERT_EQUAL( 0, m_suite->countTestCases() );

}

void 

TestSuiteTest::testCountTestCasesWithTwoTests()

{

  MockTestCase *case1 = new MockTestCase( "test1" );

  case1->setExpectedCountTestCasesCall();

  MockTestCase *case2 = new MockTestCase( "test2" );

  case2->setExpectedCountTestCasesCall();

  m_suite->addTest( case1 );

  m_suite->addTest( case2 );

  CPPUNIT_ASSERT_EQUAL( 2, m_suite->countTestCases() );

  case1->verify();

  case2->verify();

}

void 

TestSuiteTest::testCountTestCasesWithSubSuite()

{

  MockTestCase *case1 = new MockTestCase( "test1" );

  case1->setExpectedCountTestCasesCall();

  MockTestCase *case2 = new MockTestCase( "test2" );

  case2->setExpectedCountTestCasesCall();

  MockTestCase *case3 = new MockTestCase( "test3" );

  case3->setExpectedCountTestCasesCall();

  CPPUNIT_NS::TestSuite *subSuite = new CPPUNIT_NS::TestSuite( "SubSuite");

  subSuite->addTest( case1 );

  subSuite->addTest( case2 );

  m_suite->addTest( case3 );

  m_suite->addTest( subSuite );

  CPPUNIT_ASSERT_EQUAL( 3, m_suite->countTestCases() );

  case1->verify();

  case2->verify();

  case3->verify();

}

void 

TestSuiteTest::testRunWithOneTest()

{

  MockTestCase *case1 = new MockTestCase( "test1" );

  case1->setExpectedRunTestCall();

  m_suite->addTest( case1 );

  CPPUNIT_NS::TestResult result;

  m_suite->run( &result );

  case1->verify();

}

void 

TestSuiteTest::testRunWithOneTestAndSubSuite()

{

  MockTestCase *case1 = new MockTestCase( "test1" );

  case1->setExpectedRunTestCall();

  MockTestCase *case2 = new MockTestCase( "test2" );

  case2->setExpectedRunTestCall();

  MockTestCase *case3 = new MockTestCase( "test3" );

  case3->setExpectedRunTestCall();

  CPPUNIT_NS::TestSuite *subSuite = new CPPUNIT_NS::TestSuite( "SubSuite");

  subSuite->addTest( case1 );

  subSuite->addTest( case2 );

  m_suite->addTest( case3 );

  m_suite->addTest( subSuite);

  CPPUNIT_NS::TestResult result;

  m_suite->run( &result );

  case1->verify();

  case2->verify();

  case3->verify();

}

void 

TestSuiteTest::testGetTests()

{

  m_suite->addTest( new CPPUNIT_NS::TestCase( "test1" ) );

  m_suite->addTest( new CPPUNIT_NS::TestCase( "test2" ) );

  CPPUNIT_ASSERT_EQUAL( 2, int(m_suite->getTests().size()) );

}

void 

TestSuiteTest::testDeleteContents()

{

  m_suite->addTest( new CPPUNIT_NS::TestCase( "test2" ) );

  m_suite->deleteContents();

  CPPUNIT_ASSERT_EQUAL( 0, int(m_suite->getTests().size()) );

}

void 

TestSuiteTest::testGetChildTestCount()

{

  m_suite->addTest( new CPPUNIT_NS::TestCase( "test1" ) );

  m_suite->addTest( new CPPUNIT_NS::TestCase( "test2" ) );

  CPPUNIT_ASSERT_EQUAL( 2, m_suite->getChildTestCount() );

}

void 

TestSuiteTest::testGetChildTestAt()

{

  CPPUNIT_NS::TestCase *test1 = new CPPUNIT_NS::TestCase( "test1" );

  CPPUNIT_NS::TestCase *test2 = new CPPUNIT_NS::TestCase( "test2" );

  m_suite->addTest( test1 );

  m_suite->addTest( test2 );

  CPPUNIT_ASSERT( test1 == m_suite->getChildTestAt(0) );

  CPPUNIT_ASSERT( test2 == m_suite->getChildTestAt(1) );

}

void 

TestSuiteTest::testGetChildTestAtThrow1()

{

  m_suite->getChildTestAt(-1);

}

void 

TestSuiteTest::testGetChildTestAtThrow2()

{

  m_suite->getChildTestAt(0);

}