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