#include <TestCase.h>
#include <TestShell.h>
#include <unistd.h>
#include <stdio.h>
#include <stdarg.h>

using std::string;

_EXPORT
BTestCase::BTestCase(string name)
	: CppUnit::TestCase(name)
	, fValidCWD(false)
	, fSubTestNum(0)
{
}

_EXPORT
void
BTestCase::tearDown() {
	if (fSubTestNum != 0)
		NextSubTestBlock();
}

_EXPORT
void
BTestCase::NextSubTest() {
	if (BTestShell::GlobalBeVerbose()) {
		printf("[%" B_PRId32 "]", fSubTestNum++);
		fflush(stdout);
	}
}

_EXPORT
void
BTestCase::NextSubTestBlock() {
	if (BTestShell::GlobalBeVerbose())
		printf("\n");
}

_EXPORT
void
BTestCase::Outputf(const char *str, ...) {
	if (BTestShell::GlobalBeVerbose()) {
		va_list args;
		va_start(args, str);
		vprintf(str, args);
		va_end(args);
		fflush(stdout);
	}
}

/*! To return to the last saved working directory, call RestoreCWD(). */
_EXPORT
void
BTestCase::SaveCWD() {
	fValidCWD = getcwd(fCurrentWorkingDir, B_PATH_NAME_LENGTH);
}

/*	If SaveCWD() has not been called and an alternate
	directory is specified by alternate, the current working directory is
	changed to alternate. If alternate is null, the current working directory
	is not modified.
*/
_EXPORT
void
BTestCase::RestoreCWD(const char *alternate) {
	if (fValidCWD)
		chdir(fCurrentWorkingDir);
	else if (alternate != NULL)
		chdir(alternate);
}