/*
 * Copyright 2009, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Distributed under the terms of the MIT License.
 */


#include "TestSuite.h"

#include <new>

#include <string.h>

#include "TestVisitor.h"


TestSuite::TestSuite(const char* name)
	:
	Test(name),
	fTests(NULL),
	fTestCount(0)
{
}


TestSuite::~TestSuite()
{
	for (int32 i = 0; i < fTestCount; i++)
		delete fTests[i];
	delete[] fTests;
}


int32
TestSuite::CountTests() const
{
	return fTestCount;
}


Test*
TestSuite::TestAt(int32 index) const
{
	return index >= 0 && index < fTestCount ? fTests[index] : NULL;
}


Test*
TestSuite::FindTest(const char* name, int32 nameLength) const
{
	if (nameLength < 0)
		nameLength = strlen(name);

	for (int32 i = 0; Test* test = TestAt(i); i++) {
		if (strlen(test->Name()) == (size_t)nameLength
			&& strncmp(test->Name(), name, nameLength) == 0) {
			return test;
		}
	}

	return NULL;
}


bool
TestSuite::AddTest(Test* test)
{
	if (test == NULL)
		return test;

	Test** tests = new(std::nothrow) Test*[fTestCount + 1];
	if (tests == NULL) {
		delete test;
		return false;
	}

	if (fTestCount > 0)
		memcpy(tests, fTests, sizeof(Test*) * fTestCount);

	delete[] fTests;

	fTests = tests;
	fTests[fTestCount++] = test;

	test->SetSuite(this);

	return true;
}


bool
TestSuite::IsLeafTest() const
{
	return false;
}


bool
TestSuite::Run(TestContext& context)
{
	for (int32 i = 0; Test* test = TestAt(i); i++) {
		bool result = _Run(context, test, NULL);
		if (!result && context.Options().quitAfterFailure)
			return false;
	}

	return true;
}


bool
TestSuite::Run(TestContext& context, const char* name)
{
	const char* separator = strstr(name, "::");
	Test* test = FindTest(name, separator != NULL ? separator - name : -1);
	if (test == NULL) {
		context.Print("No such test: \"%.*s\"\n",
			int(separator != NULL ? separator - name : strlen(name)), name);
		return !context.Options().quitAfterFailure;
	}

	return _Run(context, test, separator != NULL ? separator + 2 : NULL)
		|| !context.Options().quitAfterFailure;
}


Test*
TestSuite::Visit(TestVisitor& visitor)
{
	if (visitor.VisitTestSuitePre(this))
		return this;

	for (int32 i = 0; Test* test = TestAt(i); i++) {
		if (Test* foundTest = test->Visit(visitor))
			return foundTest;
	}

	return visitor.VisitTestSuitePost(this) ? this : NULL;
}


bool
TestSuite::_Run(TestContext& context, Test* test, const char* name)
{
	TestContext subContext(context, test);

	status_t error = test->Setup(subContext);
	if (error != B_OK) {
		subContext.Error("setup failed\n");
		test->Cleanup(subContext, false);
		subContext.TestDone(false);
		return false;
	}

	bool result = name != NULL
		? test->Run(subContext, name) : test->Run(subContext);
	test->Cleanup(subContext, true);

	subContext.TestDone(result);

	return result;
}