// SPDX-License-Identifier: GPL-2.0-only
/*
 * KUnit tests for NVM parse
 *
 * Copyright (C) 2025 Intel Corporation
 */
#include <kunit/static_stub.h>
#include <kunit/test.h>
#include <iwl-nvm-parse.h>

MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");

static const struct nvm_flag_case {
        const char *desc;
        u16 nvm_flags;
        u32 reg_rule_flags;
        u32 set_reg_rule_flags;
        u32 clear_reg_rule_flags;
} nvm_flag_cases[] = {
        {
                .desc = "Restricting VLP client and AP access",
                .nvm_flags = 0,
                .set_reg_rule_flags = NL80211_RRF_NO_6GHZ_VLP_CLIENT,
                .clear_reg_rule_flags = NL80211_RRF_ALLOW_6GHZ_VLP_AP,
        },
        {
                .desc = "Allow VLP client and AP access",
                .nvm_flags = NVM_CHANNEL_VLP,
                .set_reg_rule_flags = NL80211_RRF_ALLOW_6GHZ_VLP_AP,
                .clear_reg_rule_flags = NL80211_RRF_NO_6GHZ_VLP_CLIENT,
        },
        {
                .desc = "Allow VLP client access, while restricting AP access",
                .nvm_flags = NVM_CHANNEL_VLP | NVM_CHANNEL_VLP_AP_NOT_ALLOWED,
                .set_reg_rule_flags = 0,
                .clear_reg_rule_flags = NL80211_RRF_ALLOW_6GHZ_VLP_AP |
                                        NL80211_RRF_NO_6GHZ_VLP_CLIENT,
        },
};

KUNIT_ARRAY_PARAM_DESC(nvm_flag, nvm_flag_cases, desc)

static void test_nvm_flags(struct kunit *test)
{
        const struct nvm_flag_case *params = test->param_value;
        struct iwl_reg_capa reg_capa = {};
        u32 flags = 0;

        flags = iwl_nvm_get_regdom_bw_flags(NULL, 0, params->nvm_flags,
                                            reg_capa);

        if ((params->set_reg_rule_flags & flags) != params->set_reg_rule_flags)
                KUNIT_FAIL(test, "Expected set bits:0x%08x flags:0x%08x\n",
                           params->set_reg_rule_flags, flags);

        if (params->clear_reg_rule_flags & flags)
                KUNIT_FAIL(test, "Expected clear bits:0x%08x flags:0x%08x\n",
                           params->clear_reg_rule_flags, flags);
}

static struct kunit_case nvm_flags_test_cases[] = {
        KUNIT_CASE_PARAM(test_nvm_flags,
                         nvm_flag_gen_params),
        {},
};

static struct kunit_suite nvm_flags_suite = {
        .name = "iwlwifi-nvm_flags",
        .test_cases = nvm_flags_test_cases,
};

kunit_test_suite(nvm_flags_suite);