/*

	9unit
	Copyright (C) Jonathan Lamothe <jonathan@jlamothe.net>

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Lesser General Public License as
	published by the Free Software Foundation, either version 3 of the
	License, or (at your option) any later version.

	This program is distributed in the hope that it will be useful, but
	WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
	Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public
	License along with this program.  If not, see
	<http://www.gnu.org/licenses/>.

*/

#include <u.h>
#include <libc.h>
#include <stdio.h>

#include "9unit.h"

// Internal Types

typedef struct RunTestWith RunTestWith;
typedef struct RunTestCompare RunTestCompare;
typedef struct ContextData ContextData;
typedef struct TestContextWith TestContextWith;
typedef struct TestContextCompare TestContextCompare;
typedef struct SingleTestContextWith SingleTestContextWith;
typedef struct SingleTestContextCompare SingleTestContextCompare;

// data required by run_test_with()
struct RunTestWith
{
	void *ptr; // the state's previous ptr value
	TestResult (*test)(TestState *, void *); // the test
	void *val; // the value passed in
};

// data needed by run_test_compare()
struct RunTestCompare
{
	TestResult (*test)(TestState *, void *, void *);
	void *val1;
	void *val2;
};

struct ContextData
{
	const char *old_c; // previous context
	const char *old_fc; // previous full context
	const char *new_c; // new context
	char *new_fc; // new full context
};

// data needed by test_context_with()
struct TestContextWith
{
	void *ptr; // state's original ptr value
	void (*test)(TestState *, void *); // the test function
	void *val; // the value being passed in
};

// data needed by test_context_compare()
struct TestContextCompare
{
	void (*test)(TestState *, void *, void *);
	void *val1;
	void *val2;
};

// data needed by single_test_context_with()
struct SingleTestContextWith
{
	TestResult (*test)(TestState *, void *);
	void *val;
};

// data needed by single_test_context_compare()
struct SingleTestContextCompare
{
	TestResult (*test)(TestState *, void *, void *);
	void *val1;
	void *val2;
};

// Internal Prototypes

static void init_TestState(TestState *);
static void print_log(TestState *);
static void clear_log(TestState *);
static void reindex(TestState *);
static void report(const char *);
static void build_new_context(TestState *, ContextData *);
static void display_context(TestState *);
static void restore_context(TestState *, ContextData *);
static TestResult run_test_with_test(TestState *);
static TestResult run_test_compare_test(TestState *, void *);
static void test_context_with_test(TestState *);
static void test_context_compare_test(TestState *, void *);
static void single_test_context_test(TestState *, void *);
static void single_test_context_with_test(TestState *, void *);
static TestResult single_test_context_compare_test(TestState *, void *);

// Public Functions

void
run_tests(void (*tests)(TestState *))
{
	if (!tests) return;
	TestState s;
	memset(&s, 0, sizeof(TestState));
	s.report = report;
	(*tests)(&s);
	print_log(&s);
	print("\n\nTests run: %d\n", s.run);
	print("Tests passed: %d\n", s.passed);
	print("Tests failed: %d\n", s.failed);
	print("Tests pending: %d\n", s.pending);
	clear_log(&s);
	if (s.failed) exits("test(s) failed");
}

void
append_test_log(TestState *s, const char *msg)
{
	if (!(s && msg)) return;

	// build a new entry:
	TestLogEntry *entry = malloc(sizeof(TestLogEntry));
	entry->text = malloc(strlen(msg) + 1);
	strcpy(entry->text, msg);
	entry->next = 0;

	// add it to the list:
	if (!s->last_log)
	{
		if (s->first_log) // no last entry but we have a first?
		{
			reindex(s);
			s->last_log->next = entry;
		}
		else s->first_log = entry;
	}
	else // there's already a last entry
	{
		if (!s->first_log) // no first entry but we have a last?
			reindex(s); // do our best to fix that
		s->last_log->next = entry;
	}
	s->last_log = entry;
}

void
log_test_context(TestState *s)
{
	if (!s) return;
	if (s->full_context) append_test_log(s, s->full_context);
	else append_test_log(s, "<no context>");
}

void
run_test(TestState *s, TestResult (*t)(TestState *))
{
	if (!s) return;
	s->run++;

	// a null test function is a pending test
	if (!t)
	{
		s->pending++;
		return;
	}

	switch ((*t)(s))
	{
	case test_success:
		s->passed++;
		s->report(" [PASS]");
		break;
	case test_failure:
		s->failed++;
		s->report(" [FAIL]");
		break;
	case test_pending:
		s->pending++;
		s->report(" [PEND]");
		break;
	default:
		exits("test returned an invalid response");
	}
}

void
run_test_with(
	TestState *s, // the state
	TestResult (*test)(TestState *, void *), // the test
	void *val // the value being passed in
)
{
	if (!s) return;
	if (!test)
	{
		run_test(s, 0);
		return;
	}
	RunTestWith d;
	d.ptr = s->ptr;
	d.test = test;
	d.val = val;
	s->ptr = &d;
	run_test(s, run_test_with_test);
}

void
run_test_compare(
	TestState *s,
	TestResult (*test)(TestState *, void *, void *),
	void *val1,
	void *val2
)
{
	RunTestCompare d;
	d.test = test;
	d.val1 = val1;
	d.val2 = val2;
	run_test_with(s, run_test_compare_test, &d);
}

void
test_context(
	TestState *s,
	const char *label,
	void (*test)(TestState *)
)
{
	if (!(s && test)) return;
	if (label)
	{
		ContextData cd;
		cd.new_c = label;
		build_new_context(s, &cd);
		display_context(s);
		(*test)(s);
		restore_context(s, &cd);
	}
	else (*test)(s);
}

void
test_context_with(
	TestState *s,
	const char *context,
	void (*test)(TestState *, void *),
	void *val
)
{
	if (!s) return;
	TestContextWith d;
	d.ptr = s->ptr;
	d.test = test;
	d.val = val;
	s->ptr = &d;
	test_context(s, context, test_context_with_test);
}

void
test_context_compare(
	TestState *s,
	const char *context,
	void (*test)(TestState *, void *, void *),
	void *val1,
	void *val2
)
{
	TestContextCompare d;
	d.test = test;
	d.val1 = val1;
	d.val2 = val2;
	test_context_with(
		s,
		context,
		test_context_compare_test,
		&d
	);
}

void
single_test_context(
	TestState *s,
	const char *context,
	TestResult (*test)(TestState *)
)
{
	test_context_with(s, context, single_test_context_test, test);
}

void
single_test_context_with(
	TestState *s,
	const char *context,
	TestResult (*test)(TestState *, void *),
	void *val
)
{
	SingleTestContextWith d;
	d.test = test;
	d.val = val;
	test_context_with(
		s,
		context,
		single_test_context_with_test,
		&d
	);
}

void
single_test_context_compare(
	TestState *s,
	const char *context,
	TestResult (*test)(TestState *s, void *, void *),
	void *val1,
	void *val2
)
{
	SingleTestContextCompare d;
	d.test = test;
	d.val1 = val1;
	d.val2 = val2;
	single_test_context_with(
		s,
		context,
		single_test_context_compare_test,
		&d
	);
}

// Internal Functions

static void
print_log(TestState *s)
{
	if (!s) return;
	TestLogEntry *e = s->first_log;
	while(e)
	{
		if(e->text) print("\n%s", e->text);
		else print("(empty message)\n");
		e = e->next;
	}
}

static void
clear_log(TestState *s)
{
	if (!s) return;
	if(s->last_log && !s->first_log) reindex(s); // fix if broken
	TestLogEntry *e = s->first_log, *next;
	s->first_log = 0;
	s->last_log = 0;
	while (e)
	{
		next = e->next;
		free(e->text);
		free(e);
		e = next;
	}
}

static void
reindex(TestState *s)
{
	if (!s) return;
	if (s->first_log)
	{
		TestLogEntry *e = s->first_log;
		while (e)
		{
			s->last_log = e;
			e = e->next;
		}
	}
	else if (s->last_log) // we have a last log but no first?
	{
		s->first_log = s->last_log;
		log_test_context(s);
		append_test_log(s, "potential memory leak in test log");
	}
}

static void
report(const char *str)
{
	print("%s", str);
}

static void
build_new_context(TestState *s, ContextData *cd)
{
	cd->old_c = s->context;
	cd->old_fc = s->full_context;
	if (s->full_context)
	{
		cd->new_fc = malloc(strlen(cd->old_fc) + strlen(cd->new_c) + 3);
		sprintf(cd->new_fc, "%s: %s", cd->old_fc, cd->new_c);
	}
	else
	{
		cd->new_fc = malloc(strlen(cd->new_c) + 1);
		strcpy(cd->new_fc, cd->new_c);
	}
	s->context = cd->new_c;
	s->full_context = cd->new_fc;
	s->depth++;
}

static void
display_context(TestState *s)
{
	s->report("\n");
	for (int i = 1; i < s->depth; i++)
		s->report("\t");
	s->report(s->context);
}

static void
restore_context(TestState *s, ContextData *cd)
{
	s->context = cd->old_c;
	s->full_context = cd->old_fc;
	s->depth--;
	free(cd->new_fc);
}

static TestResult
run_test_with_test(TestState *s)
{
	RunTestWith *d = s->ptr;
	s->ptr = d->ptr;
	return (*d->test)(s, d->val);
}

static TestResult
run_test_compare_test(
	TestState *s,
	void *ptr
)
{
	RunTestCompare *d = ptr;
	return (*d->test)(s, d->val1, d->val2);
}

static void
test_context_with_test(TestState *s)
{
	TestContextWith *d = s->ptr;
	s->ptr = d->ptr;
	(*d->test)(s, d->val);
}

static void
test_context_compare_test(
	TestState *s,
	void *ptr
)
{
	TestContextCompare *d = ptr;
	(*d->test)(s, d->val1, d->val2);
}

static void
single_test_context_test(TestState *s, void *test)
{
	run_test(s, test);
}

static void
single_test_context_with_test(TestState *s, void *ptr)
{
	SingleTestContextWith *d = ptr;
	run_test_with(s, d->test, d->val);
}

static TestResult
single_test_context_compare_test(TestState *s, void *ptr)
{
	SingleTestContextCompare *d = ptr;
	return (*d->test)(s, d->val1, d->val2);
}

//jl