// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2019 NXP
 */
#include <kunit/test.h>
#include <linux/pm_qos.h>

/* Basic test for aggregating two "min" requests */
static void freq_qos_test_min(struct kunit *test)
{
 struct freq_constraints qos;
 struct freq_qos_request req1, req2;
 int ret;

 freq_constraints_init(&qos);
 memset(&req1, 0, sizeof(req1));
 memset(&req2, 0, sizeof(req2));

 ret = freq_qos_add_request(&qos, &req1, FREQ_QOS_MIN, 1000);
 KUNIT_EXPECT_EQ(test, ret, 1);
 ret = freq_qos_add_request(&qos, &req2, FREQ_QOS_MIN, 2000);
 KUNIT_EXPECT_EQ(test, ret, 1);

 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 2000);

 ret = freq_qos_remove_request(&req2);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 1000);

 ret = freq_qos_remove_request(&req1);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
   FREQ_QOS_MIN_DEFAULT_VALUE);
}

/* Test that requests for MAX_DEFAULT_VALUE have no effect */
static void freq_qos_test_maxdef(struct kunit *test)
{
 struct freq_constraints qos;
 struct freq_qos_request req1, req2;
 int ret;

 freq_constraints_init(&qos);
 memset(&req1, 0, sizeof(req1));
 memset(&req2, 0, sizeof(req2));
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX),
   FREQ_QOS_MAX_DEFAULT_VALUE);

 ret = freq_qos_add_request(&qos, &req1, FREQ_QOS_MAX,
   FREQ_QOS_MAX_DEFAULT_VALUE);
 KUNIT_EXPECT_EQ(test, ret, 0);
 ret = freq_qos_add_request(&qos, &req2, FREQ_QOS_MAX,
   FREQ_QOS_MAX_DEFAULT_VALUE);
 KUNIT_EXPECT_EQ(test, ret, 0);

 /* Add max 1000 */
 ret = freq_qos_update_request(&req1, 1000);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 1000);

 /* Add max 2000, no impact */
 ret = freq_qos_update_request(&req2, 2000);
 KUNIT_EXPECT_EQ(test, ret, 0);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 1000);

 /* Remove max 1000, new max 2000 */
 ret = freq_qos_remove_request(&req1);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 2000);
}

/*
 * Test that a freq_qos_request can be added again after removal
 *
 * This issue was solved by commit 05ff1ba412fd ("PM: QoS: Invalidate frequency
 * QoS requests after removal")
 */
static void freq_qos_test_readd(struct kunit *test)
{
 struct freq_constraints qos;
 struct freq_qos_request req;
 int ret;

 freq_constraints_init(&qos);
 memset(&req, 0, sizeof(req));
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
   FREQ_QOS_MIN_DEFAULT_VALUE);

 /* Add */
 ret = freq_qos_add_request(&qos, &req, FREQ_QOS_MIN, 1000);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 1000);

 /* Remove */
 ret = freq_qos_remove_request(&req);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
   FREQ_QOS_MIN_DEFAULT_VALUE);

 /* Add again */
 ret = freq_qos_add_request(&qos, &req, FREQ_QOS_MIN, 2000);
 KUNIT_EXPECT_EQ(test, ret, 1);
 KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 2000);
}

static struct kunit_case pm_qos_test_cases[] = {
 KUNIT_CASE(freq_qos_test_min),
 KUNIT_CASE(freq_qos_test_maxdef),
 KUNIT_CASE(freq_qos_test_readd),
 {},
};

static struct kunit_suite pm_qos_test_module = {
 .name = "qos-kunit-test",
 .test_cases = pm_qos_test_cases,
};
kunit_test_suites(&pm_qos_test_module);