/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2015 Joyent, Inc.
 */

/*
 * mergeq testing routines
 */

#include <mergeq.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>

const char *
_umem_debug_init()
{
	return ("default,verbose");
}

const char *
_umem_logging_init(void)
{
	return ("fail,contents");
}

void *
mergeq_alloc(size_t size)
{
	return (malloc(size));
}

/*ARGSUSED*/
void
mergeq_free(void *buf, size_t size)
{
	free(buf);
}

static int
mqt_int(void *first, void *second, void **outp, void *arg)
{
	uintptr_t a, b, c;
	a = (uintptr_t)first;
	b = (uintptr_t)second;
	c = a + b;
	*outp = (void *)c;

	return (0);
}

static int
mqt_append(void *first, void *second, void **outp, void *arg)
{
	char *out;

	/* Yes, this leaks, don't worry about it for the test */
	if (asprintf(&out, "%s%s", first, second) != -1) {
		*outp = out;
		return (0);
	}
	return (-1);
}

static int
mqt_fatal(void *first, void *second, void **outp, void *arg)
{
	return (-1);
}

/*
 * Test structures and cases. We really want mq_args to be a flexible array
 * member, but then we cant initialize it. Thus we set a fixed size number of
 * entries.
 */
typedef struct mq_test {
	const char	*mq_desc;	/* test description/name */
	mergeq_proc_f	*mq_proc;	/* processing function */
	int		mq_rval;	/* mergeq_merge return value */
	int		mq_uerr;	/* user error, if any */
	boolean_t	mq_strcmp;	/* use strcmp rather than == */
	void		*mq_result;	/* expected result */
	void		**mq_args;	/* argument array */
} mq_test_t;

static void *mqt_empty_args[] = { NULL };
static void *mqt_single_args[] = { (void *)42, NULL };
static void *mqt_double_args[] = { (void *)42, (void *)27, NULL };
static void *mqt_wrap_args[] = {
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, NULL
};
static void *mqt_grow_args[] = {
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, (void *)1, (void *)1, (void *)1, (void *)1,
	(void *)1, (void *)1, NULL
};
static void *mqt_order_args[] = { "l", "e", "g", "e", "n", "d", " ", "o", "f",
	" ", "z", "e", "l", "d", "a", NULL };


static mq_test_t mq_tests[] = {
	{ "empty", mqt_int, 0, 0, B_FALSE, NULL, mqt_empty_args },
	{ "single", mqt_int, 0, 0, B_FALSE, (void *)42, mqt_single_args },
	{ "double", mqt_int, 0, 0, B_FALSE, (void *)69, mqt_double_args },
	{ "wrap", mqt_int, 0, 0, B_FALSE, (void *)64, mqt_wrap_args },
	{ "grow", mqt_int, 0, 0, B_FALSE, (void *)92, mqt_grow_args },
	{ "fatal", mqt_fatal, MERGEQ_UERROR, -1, B_FALSE, NULL,
	    mqt_double_args },
	{ "order", mqt_append, 0, 0, B_TRUE, "alegend of zeld", mqt_order_args }
};

#define	NMQ_TESTS (sizeof (mq_tests) / sizeof (mq_test_t))

static void
mq_test_run(mergeq_t *mqp, mq_test_t *mqt)
{
	int ret, err;
	void **itemp = mqt->mq_args;
	void *out;

	while (*itemp != NULL) {
		if ((ret = mergeq_add(mqp, *itemp)) != 0) {
			(void) fprintf(stderr,
			    "test %s: failed to add item: %s\n",
			    mqt->mq_desc, strerror(errno));
			exit(1);
		}
		itemp++;
	}

	ret = mergeq_merge(mqp, mqt->mq_proc, NULL, &out, &err);
	if (ret != mqt->mq_rval) {
		(void) fprintf(stderr, "test %s: got incorrect rval. "
		    "Expected %d, got %d\n", mqt->mq_desc, mqt->mq_rval, ret);
		exit(1);
	}

	if (ret == MERGEQ_UERROR && err != mqt->mq_uerr) {
		(void) fprintf(stderr, "test %s: got incorrect user error. "
		    "Expected %d, got %d\n", mqt->mq_desc, mqt->mq_uerr, err);
		exit(1);
	}

	if (ret == 0) {
		if (mqt->mq_strcmp == B_TRUE &&
		    strcmp(out, mqt->mq_result) != 0) {
			(void) fprintf(stderr, "test %s: got unexpected "
			    "result: %s, expected %s\n", mqt->mq_desc, out,
			    mqt->mq_result);
			exit(1);
		} else if (mqt->mq_strcmp == B_FALSE && out != mqt->mq_result) {
			(void) fprintf(stderr, "test %s: got unexpected "
			    "result: %p, expected %p\n", mqt->mq_desc, out,
			    mqt->mq_result);
			exit(1);
		}
	}
}

int
main(void)
{
	int ret, i, t;
	mergeq_t *mqp;
	int nthreads[] = { 1, 2, 4, 8, 16, 32, 64, 128, 256, -1 };

	for (t = 0; nthreads[t] != -1; t++) {
		printf("Beginning tests with %d threads\n", nthreads[t]);
		if ((ret = mergeq_init(&mqp, nthreads[t])) != 0) {
			fprintf(stderr, "failed to init mergeq: %s\n",
			    strerror(errno));
			return (1);
		}

		for (i = 0; i < NMQ_TESTS; i++) {
			mq_test_run(mqp, &mq_tests[i]);
		}

		mergeq_fini(mqp);
	}

	return (0);
}