// SPDX-License-Identifier: GPL-2.0-only
/*
 * KUnit tests for utilities
 *
 * Copyright (C) 2024-2025 Intel Corporation
 */
#include "../iwl-utils.h"
#include <kunit/test.h>

MODULE_IMPORT_NS("IWLWIFI");

static const struct average_neg_db_case {
 const char *desc;
 u8 neg_dbm[22];
 s8 result;
} average_neg_db_cases[] = {
 {
  .desc = "Smallest possible value, all filled",
  .neg_dbm = {
   128, 128, 128, 128, 128, 128, 128, 128, 128, 128,
   128, 128, 128, 128, 128, 128, 128, 128, 128, 128,
   128, 128
  },
  .result = -128,
 },
 {
  .desc = "Biggest possible value, all filled",
  .neg_dbm = {
   0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
   0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
   0, 0,
  },
  .result = 0,
 },
 {
  .desc = "Smallest possible value, partial filled",
  .neg_dbm = {
   128, 128, 128, 128, 128, 128, 128, 128, 128, 128,
   0xff, 0xff, 0xff, 0xff, 0xff,
   0xff, 0xff, 0xff, 0xff, 0xff,
   0xff, 0xff,
  },
  .result = -128,
 },
 {
  .desc = "Biggest possible value, partial filled",
  .neg_dbm = {
   0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
   0xff, 0xff, 0xff, 0xff, 0xff,
   0xff, 0xff, 0xff, 0xff, 0xff,
   0xff, 0xff,
  },
  .result = 0,
 },
 {
  .desc = "Adding -80dBm to -75dBm until it is still rounded to -79dBm",
  .neg_dbm = {
   75, 80, 80, 80, 80, 80, 80, 80, 80, 80,
   80, 80, 80, 80, 80, 80, 80, 0xff, 0xff, 0xff,
   0xff, 0xff,
  },
  .result = -79,
 },
 {
  .desc = "Adding -80dBm to -75dBm until it is just rounded to -80dBm",
  .neg_dbm = {
   75, 80, 80, 80, 80, 80, 80, 80, 80, 80,
   80, 80, 80, 80, 80, 80, 80, 80, 0xff, 0xff,
   0xff, 0xff,
  },
  .result = -80,
 },
};

KUNIT_ARRAY_PARAM_DESC(average_neg_db, average_neg_db_cases, desc)

static void test_average_neg_db(struct kunit *test)
{
 const struct average_neg_db_case *params = test->param_value;
 u8 reversed[ARRAY_SIZE(params->neg_dbm)];
 int i;

 /* Test the values in the given order */
 KUNIT_ASSERT_EQ(test,
   iwl_average_neg_dbm(params->neg_dbm,
         ARRAY_SIZE(params->neg_dbm)),
   params->result);

 /* Test in reverse order */
 for (i = 0; i < ARRAY_SIZE(params->neg_dbm); i++)
  reversed[ARRAY_SIZE(params->neg_dbm) - i - 1] =
   params->neg_dbm[i];
 KUNIT_ASSERT_EQ(test,
   iwl_average_neg_dbm(reversed,
         ARRAY_SIZE(params->neg_dbm)),
   params->result);
}

static struct kunit_case average_db_case[] = {
 KUNIT_CASE_PARAM(test_average_neg_db, average_neg_db_gen_params),
 {}
};

static struct kunit_suite average_db = {
 .name = "iwl-average-db",
 .test_cases = average_db_case,
};

kunit_test_suite(average_db);