// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
/*
 * KUnit tests for channel helper functions
 *
 * Copyright (C) 2024-2025 Intel Corporation
 */
#include <kunit/static_stub.h>

#include "utils.h"
#include "mld.h"
#include "link.h"
#include "iface.h"
#include "fw/api/mac-cfg.h"

static const struct missed_beacon_test_case {
        const char *desc;
        struct {
                struct iwl_missed_beacons_notif notif;
                bool emlsr;
        } input;
        struct {
                bool disconnected;
                bool emlsr;
        } output;
} missed_beacon_cases[] = {
        {
                .desc = "no EMLSR, no disconnect",
                .input.notif = {
                        .consec_missed_beacons = cpu_to_le32(4),
                },
        },
        {
                .desc = "no EMLSR, no beacon loss since Rx, no disconnect",
                .input.notif = {
                        .consec_missed_beacons = cpu_to_le32(20),
                },
        },
        {
                .desc = "no EMLSR, beacon loss since Rx, disconnect",
                .input.notif = {
                        .consec_missed_beacons = cpu_to_le32(20),
                        .consec_missed_beacons_since_last_rx =
                                cpu_to_le32(10),
                },
                .output.disconnected = true,
        },
};

KUNIT_ARRAY_PARAM_DESC(test_missed_beacon, missed_beacon_cases, desc);

static void fake_ieee80211_connection_loss(struct ieee80211_vif *vif)
{
        vif->cfg.assoc = false;
}

static void test_missed_beacon(struct kunit *test)
{
        struct iwl_mld *mld = test->priv;
        struct iwl_missed_beacons_notif *notif;
        const struct missed_beacon_test_case *test_param =
                (const void *)(test->param_value);
        struct ieee80211_vif *vif;
        struct iwl_rx_packet *pkt;
        struct iwl_mld_kunit_link link1 = {
                .id = 0,
                .chandef = &chandef_6ghz_160mhz,
        };
        struct iwl_mld_kunit_link link2 = {
                .id = 1,
                .chandef = &chandef_5ghz_80mhz,
        };

        kunit_activate_static_stub(test, ieee80211_connection_loss,
                                   fake_ieee80211_connection_loss);
        pkt = iwl_mld_kunit_create_pkt(test_param->input.notif);
        notif = (void *)pkt->data;

        if (test_param->input.emlsr) {
                vif = iwlmld_kunit_assoc_emlsr(&link1, &link2);
        } else {
                struct iwl_mld_vif *mld_vif;

                vif = iwlmld_kunit_setup_non_mlo_assoc(&link1);
                mld_vif = iwl_mld_vif_from_mac80211(vif);
                notif->link_id = cpu_to_le32(mld_vif->deflink.fw_id);
        }

        wiphy_lock(mld->wiphy);

        iwl_mld_handle_missed_beacon_notif(mld, pkt);

        wiphy_unlock(mld->wiphy);

        KUNIT_ASSERT_NE(test, vif->cfg.assoc, test_param->output.disconnected);

        /* TODO: add test cases for esr and check */
}

static struct kunit_case link_cases[] = {
        KUNIT_CASE_PARAM(test_missed_beacon, test_missed_beacon_gen_params),
        {},
};

static struct kunit_suite link = {
        .name = "iwlmld-link",
        .test_cases = link_cases,
        .init = iwlmld_kunit_test_init,
};

kunit_test_suite(link);