// 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);