#include "ExampleTest.h"

#include <TestUtils.h>
#include <ThreadedTestCaller.h>
#include <cppunit/Test.h>
#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>
#include <iostream>
#include <kernel/OS.h>
#include <stdio.h>


ExampleTest::ExampleTest(std::string name)
	: BThreadedTestCase(name), fLocker(new BLocker())
{
}


CppUnit::Test*
ExampleTest::Suite()
{
	CppUnit::TestSuite* suite = new CppUnit::TestSuite("Yo");
	BThreadedTestCaller<ExampleTest>* caller;

	// Add a multithreaded test
	ExampleTest* test
		= new ExampleTest("This name is never used, just so you know :-)");
	caller = new BThreadedTestCaller<ExampleTest>(
		"ExampleTests::MultiThreaded Test #1", test);
	caller->addThread("A", &ExampleTest::TestFunc1);
	caller->addThread("B", &ExampleTest::TestFunc2);
	caller->addThread("C", &ExampleTest::TestFunc3);
	suite->addTest(caller);

	// And another
	caller = new BThreadedTestCaller<ExampleTest>(
		"ExampleTests::MultiThreaded Test #2");
	caller->addThread("Thread1", &ExampleTest::TestFunc1);
	caller->addThread("Thread2", &ExampleTest::TestFunc1);
	caller->addThread("Thread3", &ExampleTest::TestFunc1);
	suite->addTest(caller);

	// And one that fails, if you're so inclined
	caller = new BThreadedTestCaller<ExampleTest>(
		"ExampleTests::MultiThreaded Failing Test");
	caller->addThread("GoodThread1", &ExampleTest::TestFunc1);
	caller->addThread("GoodThread2", &ExampleTest::TestFunc2);
	caller->addThread("BadThread", &ExampleTest::FailureFunc);
	suite->addTest(caller);

	// And some single threaded ones
	suite->addTest(new CppUnit::TestCaller<ExampleTest>(
		"ExampleTests::SingleThreaded Test #1", &ExampleTest::TestFunc1));
	suite->addTest(new CppUnit::TestCaller<ExampleTest>(
		"ExampleTests::SingleThreaded Test #2", &ExampleTest::TestFunc2));

	return suite;
}


const int sleeptime = 10000;


void
ExampleTest::TestFunc1()
{
	for (int i = 0; i < 10; i++) {
		// Get the lock and do our business
		NextSubTest();
		fLocker->Lock();
		fNum += 10;
		fLocker->Unlock();
		snooze(sleeptime);
		//		Outputf("(1:%d)", i);
	}
}


void
ExampleTest::TestFunc2()
{
	for (int i = 0; i < 13; i++) {
		// Get the lock and do our business
		NextSubTest();
		fLocker->Lock();
		fNum *= 2;
		fLocker->Unlock();
		snooze(sleeptime);
		//		Outputf("(2:%d)", i);
	}
}


void
ExampleTest::TestFunc3()
{
	for (int i = 0; i < 15; i++) {
		// Get the lock and do our business
		NextSubTest();
		fLocker->Lock();
		fNum += 10;
		fLocker->Unlock();
		snooze(sleeptime);
		//		Outputf("(3:%d)", i);
	}
}


void
ExampleTest::FailureFunc()
{
	CHK(true == false);
}