To run automated tests for coroutines: make check-coroutine ./check-coroutine
On success the program terminates with exit status 0. On failure an error message is written to stderr and the program exits with exit status 1. Signed-off-by: Stefan Hajnoczi <stefa...@linux.vnet.ibm.com> --- Makefile | 3 +- check-coroutine.c | 188 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 190 insertions(+), 1 deletions(-) create mode 100644 check-coroutine.c diff --git a/Makefile b/Makefile index 2b0438c..69c08c2 100644 --- a/Makefile +++ b/Makefile @@ -132,7 +132,7 @@ qemu-io$(EXESUF): qemu-io.o cmd.o qemu-tool.o qemu-error.o $(oslib-obj-y) $(trac qemu-img-cmds.h: $(SRC_PATH)/qemu-img-cmds.hx $(call quiet-command,sh $(SRC_PATH)/scripts/hxtool -h < $< > $@," GEN $@") -check-qint.o check-qstring.o check-qdict.o check-qlist.o check-qfloat.o check-qjson.o: $(GENERATED_HEADERS) +check-qint.o check-qstring.o check-qdict.o check-qlist.o check-qfloat.o check-qjson.o check-coroutine.o: $(GENERATED_HEADERS) CHECK_PROG_DEPS = qemu-malloc.o $(oslib-obj-y) $(trace-obj-y) @@ -142,6 +142,7 @@ check-qdict: check-qdict.o qdict.o qfloat.o qint.o qstring.o qbool.o qlist.o $(C check-qlist: check-qlist.o qlist.o qint.o $(CHECK_PROG_DEPS) check-qfloat: check-qfloat.o qfloat.o $(CHECK_PROG_DEPS) check-qjson: check-qjson.o qfloat.o qint.o qdict.o qstring.o qlist.o qbool.o qjson.o json-streamer.o json-lexer.o json-parser.o $(CHECK_PROG_DEPS) +check-coroutine: check-coroutine.o $(coroutine-obj-y) $(CHECK_PROG_DEPS) QEMULIBS=libhw32 libhw64 libuser libdis libdis-user diff --git a/check-coroutine.c b/check-coroutine.c new file mode 100644 index 0000000..f65ac2e --- /dev/null +++ b/check-coroutine.c @@ -0,0 +1,188 @@ +/* + * Coroutine tests + * + * Copyright IBM, Corp. 2011 + * + * Authors: + * Stefan Hajnoczi <stefa...@linux.vnet.ibm.com> + * + * This work is licensed under the terms of the GNU LGPL, version 2 or later. + * See the COPYING.LIB file in the top-level directory. + * + */ + +#include <stdlib.h> +#include <stdio.h> +#include "qemu-coroutine.h" + +static const char *cur_test_name; + +static void test_assert(bool condition, const char *msg) +{ + if (!condition) { + fprintf(stderr, "%s: %s\n", cur_test_name, msg); + exit(EXIT_FAILURE); + } +} + +/* + * Check that qemu_in_coroutine() works + */ + +static void coroutine_fn verify_in_coroutine(void *opaque) +{ + test_assert(qemu_in_coroutine(), "expected coroutine context"); +} + +static void test_in_coroutine(void) +{ + Coroutine *coroutine; + + test_assert(!qemu_in_coroutine(), "expected no coroutine context"); + + coroutine = qemu_coroutine_create(verify_in_coroutine); + qemu_coroutine_enter(coroutine, NULL); +} + +/* + * Check that qemu_coroutine_self() works + */ + +static void coroutine_fn verify_self(void *opaque) +{ + test_assert(qemu_coroutine_self() == opaque, + "qemu_coroutine_self() did not return this coroutine"); +} + +static void test_self(void) +{ + Coroutine *coroutine; + + coroutine = qemu_coroutine_create(verify_self); + qemu_coroutine_enter(coroutine, coroutine); +} + +/* + * Check that coroutines may nest multiple levels + */ + +typedef struct { + unsigned int n_enter; /* num coroutines entered */ + unsigned int n_return; /* num coroutines returned */ + unsigned int max; /* maximum level of nesting */ +} NestData; + +static void coroutine_fn nest(void *opaque) +{ + NestData *nd = opaque; + + nd->n_enter++; + + if (nd->n_enter < nd->max) { + Coroutine *child; + + child = qemu_coroutine_create(nest); + qemu_coroutine_enter(child, nd); + } + + nd->n_return++; +} + +static void test_nesting(void) +{ + Coroutine *root; + NestData nd = { + .n_enter = 0, + .n_return = 0, + .max = 1, + }; + + root = qemu_coroutine_create(nest); + qemu_coroutine_enter(root, &nd); + + test_assert(nd.n_enter == nd.max, + "failed entering to max nesting level"); + test_assert(nd.n_return == nd.max, + "failed returning from max nesting level"); +} + +/* + * Check that yield/enter transfer control correctly + */ + +static void coroutine_fn yield_5_times(void *opaque) +{ + bool *done = opaque; + int i; + + for (i = 0; i < 5; i++) { + qemu_coroutine_yield(); + } + *done = true; +} + +static void test_yield(void) +{ + Coroutine *coroutine; + bool done = false; + int i = -1; /* one extra time to return from coroutine */ + + coroutine = qemu_coroutine_create(yield_5_times); + while (!done) { + qemu_coroutine_enter(coroutine, &done); + i++; + } + test_assert(i == 5, "coroutine did not yield 5 times"); +} + +/* + * Check that creation, enter, and return work + */ + +static void coroutine_fn set_and_exit(void *opaque) +{ + bool *done = opaque; + + *done = true; +} + +static void test_lifecycle(void) +{ + Coroutine *coroutine; + bool done = false; + + /* Create, enter, and return from coroutine */ + coroutine = qemu_coroutine_create(set_and_exit); + qemu_coroutine_enter(coroutine, &done); + test_assert(done, "expected done to be true (first time)"); + + /* Repeat to check that no state affects this test */ + done = false; + coroutine = qemu_coroutine_create(set_and_exit); + qemu_coroutine_enter(coroutine, &done); + test_assert(done, "expected done to be true (second time)"); +} + +#define TESTCASE(fn) { #fn, fn } +int main(int argc, char **argv) +{ + static struct { + const char *name; + void (*run)(void); + } testcases[] = { + TESTCASE(test_lifecycle), + TESTCASE(test_yield), + TESTCASE(test_nesting), + TESTCASE(test_self), + TESTCASE(test_in_coroutine), + {}, + }; + int i; + + for (i = 0; testcases[i].name; i++) { + cur_test_name = testcases[i].name; + printf("%s\n", testcases[i].name); + testcases[i].run(); + } + return EXIT_SUCCESS; +} -- 1.7.4.4