source: mainline/kernel/test/synch/workqueue2.c@ 14c9aa6

/*
 * Copyright (c) 2012 Adam Hraska
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <test.h>
#include <arch.h>
#include <print.h>
#include <memstr.h>
#include <synch/workqueue.h>


#define WAVES 10
#define COUNT_POW 12
#define COUNT ((1 << COUNT_POW) - 1)
#define WAVE_SLEEP_MS 100
#define MAIN_POLL_SLEEP_MS 100
#define MAIN_MAX_SLEEP_SEC 40

/*
 * Include the test implementation.
 */
#include "workq-test-core.h"


/*-------------------------------------------------------------------*/

static work_t basic_work;
static int basic_done = 0;

static void basic_test_work(work_t *work_item)
{
        basic_done = 1;
        TPRINTF("basic_test_work()");
}


static void basic_test(void)
{
        TPRINTF("Issue a single work item.\n");
        basic_done = 0;
        workq_global_enqueue(&basic_work, basic_test_work);
        
        while (!basic_done) {
                TPRINTF(".");
                thread_sleep(1);
        }

        TPRINTF("\nBasic test done\n");
}

/*-------------------------------------------------------------------*/


struct work_queue *workq = NULL;

static int core_workq_enqueue(work_t *work_item, work_func_t func)
{
        return workq_enqueue(workq, work_item, func);
}
/*-------------------------------------------------------------------*/


static const char *test_custom_workq_impl(bool stop, const char *qname)
{
        workq = workq_create(qname);
        
        if (!workq) {
                return "Failed to create a work queue.\n";
        }
        
        const char *ret = run_workq_core(stop);
        
        TPRINTF("Stopping work queue...\n");
        workq_stop(workq);
        
        TPRINTF("Destroying work queue...\n");
        workq_destroy(workq);
        return ret;
}

static const char *test_custom_workq(void)
{
        TPRINTF("Stress testing a custom queue.\n");
        return test_custom_workq_impl(false, "test-workq");
}


static const char *test_custom_workq_stop(void)
{
        TPRINTF("Stress testing a custom queue. Stops prematurely. "
                "Errors are expected.\n");
        test_custom_workq_impl(true, "test-workq-stop");
        /* Errors are expected. */
        return 0;
}


const char *test_workqueue_all(void)
{
        const char *err = 0;
        const char *res;
        
        basic_test();
        
        res = test_custom_workq();
        if (res) {
                TPRINTF(res);
                err = res;
        }
        
        res = test_custom_workq_stop();
        if (res) {
                TPRINTF(res);
                err = res;
        }
        
        res = test_workqueue3();
        if (res) {
                TPRINTF(res);
                err = res;
        }

        return err;
}