linux-imx

#include <linux/init.h>

#include <linux/kernel.h>

#include <linux/module.h>

​

#define for_each_test(i, test)  \

    for (i = 0; i < ARRAY_SIZE(test); i++)

​

struct test_fail {

    const char *str;

    unsigned int base;

};

​

#define DEFINE_TEST_FAIL(test)  \

    const struct test_fail test[] __initconst

​

#define DECLARE_TEST_OK(type, test_type)    \

    test_type {             \

        const char *str;        \

        unsigned int base;      \

        type expected_res;      \

    }

​

#define DEFINE_TEST_OK(type, test)  \

    const type test[] __initconst

​

#define TEST_FAIL(fn, type, fmt, test)                  \

{                                   \

    unsigned int i;                         \

                                    \

    for_each_test(i, test) {                    \

        const struct test_fail *t = &test[i];           \

        type tmp;                       \

        int rv;                         \

                                    \

        tmp = 0;                        \

        rv = fn(t->str, t->base, &tmp);             \

        if (rv >= 0) {                      \

            WARN(1, "str '%s', base %u, expected -E, got %d/" fmt "\n", \

                t->str, t->base, rv, tmp);      \

            continue;                   \

        }                           \

    }                               \

}

​

#define TEST_OK(fn, type, fmt, test)                    \

{                                   \

    unsigned int i;                         \

                                    \

    for_each_test(i, test) {                    \

        const typeof(test[0]) *t = &test[i];            \

        type res;                       \

        int rv;                         \

                                    \

        rv = fn(t->str, t->base, &res);             \

        if (rv != 0) {                      \

            WARN(1, "str '%s', base %u, expected 0/" fmt ", got %d\n",  \

                t->str, t->base, t->expected_res, rv);  \

            continue;                   \

        }                           \

        if (res != t->expected_res) {               \

            WARN(1, "str '%s', base %u, expected " fmt ", got " fmt "\n",   \

                t->str, t->base, t->expected_res, res); \

            continue;                   \