/*
 * 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.
 */

/*
 * workq testing routines
 *
 * What we want to guarantee is that every function is executed exactly once. To
 * that end we have the callback function basically increment a global in the
 * test around a mutex.
 */

#include <workq.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <thread.h>
#include <synch.h>

mutex_t wqt_lock = ERRORCHECKMUTEX;
uintptr_t wqt_count;

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

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

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

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

/*ARGSUSED*/
int
wqt_fatal(void *item, void *arg)
{
        return (-1);
}

int
wqt_add(void *item, void *arg)
{
        uintptr_t a = (uintptr_t)item;

        mutex_enter(&wqt_lock);
        wqt_count += a;
        mutex_exit(&wqt_lock);

        return (0);
}

typedef struct wq_test {
        const char      *wq_desc;       /* test description/name */
        workq_proc_f    *wq_proc;       /* processing function */
        int             wq_rval;        /* workq_work return value */
        int             wq_uerr;        /* user error, if any */
        uintptr_t       wq_sum;         /* expected sum */
        void            **wq_args;      /* argument array */
} wq_test_t;

static void *wqt_empty_args[] = { NULL };
static void *wqt_single_args[] = { (void *)42, NULL };
static void *wqt_double_args[] = { (void *)42, (void *)27, NULL };
static void *wqt_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 *wqt_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 wq_test_t wq_tests[] = {
        { "empty", wqt_add, 0, 0, 0, wqt_empty_args },
        { "single", wqt_add, 0, 0, 42, wqt_single_args },
        { "double", wqt_add, 0, 0, 69, wqt_double_args },
        { "wrap", wqt_add, 0, 0, 64, wqt_wrap_args },
        { "grow", wqt_add, 0, 0, 92, wqt_grow_args },
        { "fatal", wqt_fatal, WORKQ_UERROR, -1, -1, wqt_double_args }
};

#define NWQ_TESTS (sizeof (wq_tests) / sizeof (wq_test_t))

static void
wq_test_run(workq_t *wqp, wq_test_t *wqt)
{
        int ret, err;
        void **itemp = wqt->wq_args;

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

        wqt_count = 0;
        ret = workq_work(wqp, wqt->wq_proc, NULL, &err);
        if (ret != wqt->wq_rval) {
                (void) fprintf(stderr, "test %s: got incorrect rval. "
                    "Expected %d, got %d (%d)\n", wqt->wq_desc, wqt->wq_rval,
                    ret, errno);
                exit(1);
        }

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

        if (ret == 0 && wqt_count != wqt->wq_sum) {
                (void) fprintf(stderr, "test %s: got unexpected "
                    "result: %d, expected %d\n", wqt->wq_desc, wqt_count,
                    wqt->wq_sum);
                exit(1);
        }
}

int
main(void)
{
        int ret, i, t;
        workq_t *wqp;
        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 = workq_init(&wqp, nthreads[t])) != 0) {
                        fprintf(stderr, "failed to init workq: %s\n",
                            strerror(errno));
                        return (1);
                }

                for (i = 0; i < NWQ_TESTS; i++) {
                        wq_test_run(wqp, &wq_tests[i]);
                }

                workq_fini(wqp);
        }


        return (0);
}