/*
 * _test_main.cpp
 * Copyright 2013 Google Inc. All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include "test.hpp"
#include <cstdio>
#include <cstdlib>
#ifndef __APPLE__
#include <error.h>
#endif
#include <unistd.h>
#include <sys/wait.h>

#ifdef __APPLE__
#include <stdarg.h>

void error(int status, int errnum, const char *format, ...) {
	fflush(stdout);
	fprintf(stderr, "%s: ", getprogname());

	va_list args;
	va_start(args, format);
	vfprintf(stderr, format, args);
	va_end(args);

	if (errnum != 0) {
		fprintf(stderr, ": %s\n", strerror(errnum));
	} else {
		fprintf(stderr, "\n");
	}
	if (status != 0) {
		exit(status);
	}
}
#endif

test::test_registry_t test::test_registry;
using namespace test;

bool run_test(TestBase& test) {
	printf("-- running test case: %s\n", test.name);

	fflush(stdout);
	pid_t child_pid = fork();
	if (child_pid == 0) {
		exit(static_cast<int>(test.run()));
	}
	if (child_pid == -1) {
		error(EXIT_FAILURE, 0, "unable to fork");
	}

	int child_status = 0;
	waitpid(child_pid, &child_status, 0);

	test::TestStatus status;

	if (WIFEXITED(child_status)) {
		int exit_status = WEXITSTATUS(child_status);
		if (exit_status & ~test::STATUS_MASK) {
			status = test::FAILED;
		} else {
			status = static_cast<test::TestStatus>(exit_status);
		}
	} else if (WIFSIGNALED(child_status)) {
		const int signum = WTERMSIG(child_status);
		printf("!! signal (%d) %s\n", signum, strsignal(signum));
		switch (signum) {
			case SIGABRT:
				status = test::SIGNAL_ABORT; break;
			case SIGSEGV:
			case SIGBUS:
				status = test::SIGNAL_SEGFAULT; break;
			case SIGFPE:
				status = test::SIGNAL_DIVZERO; break;
			default:
				status = test::SIGNAL_UNCAUGHT;
		}
	} else {
		status = test::SUCCESS;
	}

	if (test.expected_status == test::FAILED) {
		return (status & test::FAILED);
	}

	if (test.expected_status == test::SIGNAL_UNCAUGHT) {
		return (status & test::SIGNAL_UNCAUGHT);
	}

	return status == test.expected_status;
}

int main(int argc, const char* const argv[]) {

	size_t success_cnt = 0;
	size_t total_cnt = 0;
	for (test_registry_t::iterator it = test_registry.begin();
			it != test_registry.end(); ++it) {
		TestBase& test = **it;

		bool consider_test = (argc <= 1);
		for (int i = 1; i < argc; ++i) {
			if (strcasecmp(argv[i], test.name) == 0) {
				consider_test = true;
				break;
			}
		}
		if (not consider_test) {
			continue;
		}

		total_cnt += 1;
		if (run_test(test)) {
			printf("-- test case success: %s\n", test.name);
			success_cnt += 1;
		} else {
			printf("** test case FAILED : %s\n", test.name);
		}
	}
	printf("-- tests passing: %lu/%lu", success_cnt, total_cnt);
	if (total_cnt) {
		printf(" (%lu%%)\n", success_cnt * 100 / total_cnt);
	} else {
		printf("\n");
	}
	return (success_cnt == total_cnt) ? EXIT_SUCCESS : EXIT_FAILURE;
}