// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright © 2024 Intel Corporation
 */

#include <kunit/test.h>

#include "xe_args.h"

static void call_args_example(struct kunit *test)
{
#define foo     X, Y, Z, Q
#define bar     COUNT_ARGS(foo)
#define buz     CALL_ARGS(COUNT_ARGS, foo)

        KUNIT_EXPECT_EQ(test, bar, 1);
        KUNIT_EXPECT_EQ(test, buz, 4);

#undef foo
#undef bar
#undef buz
}

static void drop_first_arg_example(struct kunit *test)
{
#define foo     X, Y, Z, Q
#define bar     CALL_ARGS(COUNT_ARGS, DROP_FIRST_ARG(foo))

        KUNIT_EXPECT_EQ(test, bar, 3);

#undef foo
#undef bar
}

static void first_arg_example(struct kunit *test)
{
        int X = 1;

#define foo     X, Y, Z, Q
#define bar     FIRST_ARG(foo)

        KUNIT_EXPECT_EQ(test, bar, X);
        KUNIT_EXPECT_STREQ(test, __stringify(bar), "X");

#undef foo
#undef bar
}

static void last_arg_example(struct kunit *test)
{
        int Q = 1;

#define foo     X, Y, Z, Q
#define bar     LAST_ARG(foo)

        KUNIT_EXPECT_EQ(test, bar, Q);
        KUNIT_EXPECT_STREQ(test, __stringify(bar), "Q");

#undef foo
#undef bar
}

static void pick_arg_example(struct kunit *test)
{
        int Y = 1, Z = 2;

#define foo     X, Y, Z, Q
#define bar     PICK_ARG(2, foo)
#define buz     PICK_ARG3(foo)

        KUNIT_EXPECT_EQ(test, bar, Y);
        KUNIT_EXPECT_STREQ(test, __stringify(bar), "Y");
        KUNIT_EXPECT_EQ(test, buz, Z);
        KUNIT_EXPECT_STREQ(test, __stringify(buz), "Z");

#undef foo
#undef bar
#undef buz
}

static void if_args_example(struct kunit *test)
{
        enum { Z = 1, Q };

#define foo     X, Y
#define bar     IF_ARGS(Z, Q, foo)
#define buz     IF_ARGS(Z, Q, DROP_FIRST_ARG(FIRST_ARG(foo)))

        KUNIT_EXPECT_EQ(test, bar, Z);
        KUNIT_EXPECT_EQ(test, buz, Q);
        KUNIT_EXPECT_STREQ(test, __stringify(bar), "Z");
        KUNIT_EXPECT_STREQ(test, __stringify(buz), "Q");

#undef foo
#undef bar
#undef buz
}

static void sep_comma_example(struct kunit *test)
{
#define foo(f)  f(X) f(Y) f(Z) f(Q)
#define bar     DROP_FIRST_ARG(foo(ARGS_SEP_COMMA __stringify))
#define buz     CALL_ARGS(COUNT_ARGS, DROP_FIRST_ARG(foo(ARGS_SEP_COMMA)))

        static const char * const a[] = { bar };

        KUNIT_EXPECT_STREQ(test, a[0], "X");
        KUNIT_EXPECT_STREQ(test, a[1], "Y");
        KUNIT_EXPECT_STREQ(test, a[2], "Z");
        KUNIT_EXPECT_STREQ(test, a[3], "Q");

        KUNIT_EXPECT_EQ(test, buz, 4);

#undef foo
#undef bar
#undef buz
}

#define NO_ARGS
#define FOO_ARGS        X, Y, Z, Q
#define MAX_ARGS        -1, -2, -3, -4, -5, -6, -7, -8, -9, -10, -11, -12

static void count_args_test(struct kunit *test)
{
        int count;

        /* COUNT_ARGS() counts to 12 */

        count = COUNT_ARGS();
        KUNIT_EXPECT_EQ(test, count, 0);

        count = COUNT_ARGS(1);
        KUNIT_EXPECT_EQ(test, count, 1);

        count = COUNT_ARGS(a, b, c, d, e);
        KUNIT_EXPECT_EQ(test, count, 5);

        count = COUNT_ARGS(a, b, c, d, e, f, g, h, i, j, k, l);
        KUNIT_EXPECT_EQ(test, count, 12);

        /* COUNT_ARGS() does not expand params */

        count = COUNT_ARGS(NO_ARGS);
        KUNIT_EXPECT_EQ(test, count, 1);

        count = COUNT_ARGS(FOO_ARGS);
        KUNIT_EXPECT_EQ(test, count, 1);
}

static void call_args_test(struct kunit *test)
{
        int count;

        count = CALL_ARGS(COUNT_ARGS, NO_ARGS);
        KUNIT_EXPECT_EQ(test, count, 0);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, NO_ARGS), 0);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, FOO_ARGS), 4);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, FOO_ARGS, FOO_ARGS), 8);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, MAX_ARGS), 12);
}

static void drop_first_arg_test(struct kunit *test)
{
        int Y = -2, Z = -3, Q = -4;
        int a[] = { DROP_FIRST_ARG(FOO_ARGS) };

        KUNIT_EXPECT_EQ(test, DROP_FIRST_ARG(0, -1), -1);
        KUNIT_EXPECT_EQ(test, DROP_FIRST_ARG(DROP_FIRST_ARG(0, -1, -2)), -2);

        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, DROP_FIRST_ARG(FOO_ARGS)), 3);
        KUNIT_EXPECT_EQ(test, DROP_FIRST_ARG(DROP_FIRST_ARG(DROP_FIRST_ARG(FOO_ARGS))), -4);
        KUNIT_EXPECT_EQ(test, a[0], -2);
        KUNIT_EXPECT_EQ(test, a[1], -3);
        KUNIT_EXPECT_EQ(test, a[2], -4);

#define foo     DROP_FIRST_ARG(FOO_ARGS)
#define bar     DROP_FIRST_ARG(DROP_FIRST_ARG(FOO_ARGS))
#define buz     DROP_FIRST_ARG(DROP_FIRST_ARG(DROP_FIRST_ARG(FOO_ARGS)))

        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, foo), 3);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, bar), 2);
        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, buz), 1);
        KUNIT_EXPECT_STREQ(test, __stringify(buz), "Q");

#undef foo
#undef bar
#undef buz
}

static void first_arg_test(struct kunit *test)
{
        int X = -1;
        int a[] = { FIRST_ARG(FOO_ARGS) };

        KUNIT_EXPECT_EQ(test, FIRST_ARG(-1, -2), -1);

        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, FIRST_ARG(FOO_ARGS)), 1);
        KUNIT_EXPECT_EQ(test, FIRST_ARG(FOO_ARGS), -1);
        KUNIT_EXPECT_EQ(test, a[0], -1);
        KUNIT_EXPECT_STREQ(test, __stringify(FIRST_ARG(FOO_ARGS)), "X");
}

static void last_arg_test(struct kunit *test)
{
        int Q = -4;
        int a[] = { LAST_ARG(FOO_ARGS) };

        KUNIT_EXPECT_EQ(test, LAST_ARG(-1, -2), -2);

        KUNIT_EXPECT_EQ(test, CALL_ARGS(COUNT_ARGS, LAST_ARG(FOO_ARGS)), 1);
        KUNIT_EXPECT_EQ(test, LAST_ARG(FOO_ARGS), -4);
        KUNIT_EXPECT_EQ(test, a[0], -4);
        KUNIT_EXPECT_STREQ(test, __stringify(LAST_ARG(FOO_ARGS)), "Q");

        KUNIT_EXPECT_EQ(test, LAST_ARG(MAX_ARGS), -12);
        KUNIT_EXPECT_STREQ(test, __stringify(LAST_ARG(MAX_ARGS)), "-12");
}

static void if_args_test(struct kunit *test)
{
        bool with_args = true;
        bool no_args = false;
        enum { X = 100 };

        KUNIT_EXPECT_TRUE(test, IF_ARGS(true, false, FOO_ARGS));
        KUNIT_EXPECT_FALSE(test, IF_ARGS(true, false, NO_ARGS));

        KUNIT_EXPECT_TRUE(test, CONCATENATE(IF_ARGS(with, no, FOO_ARGS), _args));
        KUNIT_EXPECT_FALSE(test, CONCATENATE(IF_ARGS(with, no, NO_ARGS), _args));

        KUNIT_EXPECT_STREQ(test, __stringify(IF_ARGS(yes, no, FOO_ARGS)), "yes");
        KUNIT_EXPECT_STREQ(test, __stringify(IF_ARGS(yes, no, NO_ARGS)), "no");

        KUNIT_EXPECT_EQ(test, IF_ARGS(CALL_ARGS(COUNT_ARGS, FOO_ARGS), -1, FOO_ARGS), 4);
        KUNIT_EXPECT_EQ(test, IF_ARGS(CALL_ARGS(COUNT_ARGS, FOO_ARGS), -1, NO_ARGS), -1);
        KUNIT_EXPECT_EQ(test, IF_ARGS(CALL_ARGS(COUNT_ARGS, NO_ARGS), -1, FOO_ARGS), 0);
        KUNIT_EXPECT_EQ(test, IF_ARGS(CALL_ARGS(COUNT_ARGS, NO_ARGS), -1, NO_ARGS), -1);

        KUNIT_EXPECT_EQ(test,
                        CALL_ARGS(FIRST_ARG,
                                  CALL_ARGS(CONCATENATE, IF_ARGS(FOO, MAX, FOO_ARGS), _ARGS)), X);
        KUNIT_EXPECT_EQ(test,
                        CALL_ARGS(FIRST_ARG,
                                  CALL_ARGS(CONCATENATE, IF_ARGS(FOO, MAX, NO_ARGS), _ARGS)), -1);
        KUNIT_EXPECT_EQ(test,
                        CALL_ARGS(COUNT_ARGS,
                                  CALL_ARGS(CONCATENATE, IF_ARGS(FOO, MAX, FOO_ARGS), _ARGS)), 4);
        KUNIT_EXPECT_EQ(test,
                        CALL_ARGS(COUNT_ARGS,
                                  CALL_ARGS(CONCATENATE, IF_ARGS(FOO, MAX, NO_ARGS), _ARGS)), 12);
}

static struct kunit_case args_tests[] = {
        KUNIT_CASE(count_args_test),
        KUNIT_CASE(call_args_example),
        KUNIT_CASE(call_args_test),
        KUNIT_CASE(drop_first_arg_example),
        KUNIT_CASE(drop_first_arg_test),
        KUNIT_CASE(first_arg_example),
        KUNIT_CASE(first_arg_test),
        KUNIT_CASE(last_arg_example),
        KUNIT_CASE(last_arg_test),
        KUNIT_CASE(pick_arg_example),
        KUNIT_CASE(if_args_example),
        KUNIT_CASE(if_args_test),
        KUNIT_CASE(sep_comma_example),
        {}
};

static struct kunit_suite args_test_suite = {
        .name = "args",
        .test_cases = args_tests,
};

kunit_test_suite(args_test_suite);