#include <rz_core.h>
#include "minunit.h"

static void *my_function(RzCore *core, void *user) {
	size_t val = (size_t)user;
	int i;
	for (i = 0; i < 5; i++) {
		rz_cons_printf ("%u, %d\n", (unsigned int)val, i);
		rz_core_task_yield (&core->tasks);
	}
	return strdup (rz_cons_get_buffer ());
}

static bool test_core_task(void) {
	RzCore *core = rz_core_new ();
	rz_config_set_i (core->config, "scr.interactive", 0);
	rz_core_task_sync_begin (&core->tasks);

	RzCoreTask *a = rz_core_cmd_task_new (core, "?e hello; ?e world; ?e from; ?e a; ?e task", NULL, NULL);
	rz_core_task_enqueue (&core->tasks, a);

	RzCoreTask *b = rz_core_function_task_new (core, my_function, (void *)(size_t)1337);
	rz_core_task_enqueue (&core->tasks, b);

	rz_cons_printf ("Hello\n");
	rz_cons_printf ("this\n");
	rz_cons_printf ("is\n");
	rz_cons_printf ("the\n");
	rz_cons_printf ("main\n");
	rz_cons_printf ("task!\n");

	rz_core_task_join (&core->tasks, rz_core_task_self (&core->tasks), a->id);
	rz_core_task_join (&core->tasks, rz_core_task_self (&core->tasks), b->id);

	void *fcn_result = rz_core_function_task_get_result (a);
	mu_assert_null (fcn_result, "fcn result");
	const char *cmd_result = rz_core_cmd_task_get_result (a);
	mu_assert_streq (cmd_result, "hello\nworld\nfrom\na\ntask\n", "cmd result");

	fcn_result = rz_core_function_task_get_result (b);
	mu_assert_streq ((const char *)fcn_result, "1337, 0\n1337, 1\n1337, 2\n1337, 3\n1337, 4\n", "fcn result");
	free (fcn_result);
	cmd_result = rz_core_cmd_task_get_result (b);
	mu_assert_null (cmd_result, "cmd result");

	rz_core_task_del (&core->tasks, a->id);
	rz_core_task_del (&core->tasks, b->id);

	mu_assert_streq (rz_cons_get_buffer (), "Hello\nthis\nis\nthe\nmain\ntask!\n", "main buffer");

	rz_core_task_sync_end (&core->tasks);
	rz_core_free (core);
	mu_end;
}

static void finished_cb(const char *res, void *user) {
	*(char **)user = strdup (res);
}

static bool test_core_task_finished_cb(void) {
	RzCore *core = rz_core_new ();
	rz_config_set_i (core->config, "scr.interactive", 0);
	rz_core_task_sync_begin (&core->tasks);

	char *res_indir = NULL; // finished_cb puts the result in here too
	RzCoreTask *a = rz_core_cmd_task_new (core, "?e amor; ?e vincit; ?e omnia", finished_cb, &res_indir);
	rz_core_task_enqueue (&core->tasks, a);

	rz_core_task_join (&core->tasks, rz_core_task_self (&core->tasks), a->id);

	const char *cmd_result = rz_core_cmd_task_get_result (a);
	mu_assert_streq (cmd_result, "amor\nvincit\nomnia\n", "cmd result");
	mu_assert_streq (res_indir, "amor\nvincit\nomnia\n", "cmd result");
	free (res_indir);

	rz_core_task_del (&core->tasks, a->id);

	rz_core_task_sync_end (&core->tasks);
	rz_core_free (core);
	mu_end;

}

// This test is best served with helgrind
static int all_tests(void) {
	mu_run_test (test_core_task);
	mu_run_test (test_core_task_finished_cb);
	return tests_passed != tests_run;
}

mu_main (all_tests)