162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright 2019 NXP
462306a36Sopenharmony_ci */
562306a36Sopenharmony_ci#include <kunit/test.h>
662306a36Sopenharmony_ci#include <linux/pm_qos.h>
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci/* Basic test for aggregating two "min" requests */
962306a36Sopenharmony_cistatic void freq_qos_test_min(struct kunit *test)
1062306a36Sopenharmony_ci{
1162306a36Sopenharmony_ci	struct freq_constraints	qos;
1262306a36Sopenharmony_ci	struct freq_qos_request	req1, req2;
1362306a36Sopenharmony_ci	int ret;
1462306a36Sopenharmony_ci
1562306a36Sopenharmony_ci	freq_constraints_init(&qos);
1662306a36Sopenharmony_ci	memset(&req1, 0, sizeof(req1));
1762306a36Sopenharmony_ci	memset(&req2, 0, sizeof(req2));
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req1, FREQ_QOS_MIN, 1000);
2062306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
2162306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req2, FREQ_QOS_MIN, 2000);
2262306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 2000);
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci	ret = freq_qos_remove_request(&req2);
2762306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
2862306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 1000);
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci	ret = freq_qos_remove_request(&req1);
3162306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
3262306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
3362306a36Sopenharmony_ci			FREQ_QOS_MIN_DEFAULT_VALUE);
3462306a36Sopenharmony_ci}
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci/* Test that requests for MAX_DEFAULT_VALUE have no effect */
3762306a36Sopenharmony_cistatic void freq_qos_test_maxdef(struct kunit *test)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	struct freq_constraints	qos;
4062306a36Sopenharmony_ci	struct freq_qos_request	req1, req2;
4162306a36Sopenharmony_ci	int ret;
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci	freq_constraints_init(&qos);
4462306a36Sopenharmony_ci	memset(&req1, 0, sizeof(req1));
4562306a36Sopenharmony_ci	memset(&req2, 0, sizeof(req2));
4662306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX),
4762306a36Sopenharmony_ci			FREQ_QOS_MAX_DEFAULT_VALUE);
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req1, FREQ_QOS_MAX,
5062306a36Sopenharmony_ci			FREQ_QOS_MAX_DEFAULT_VALUE);
5162306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 0);
5262306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req2, FREQ_QOS_MAX,
5362306a36Sopenharmony_ci			FREQ_QOS_MAX_DEFAULT_VALUE);
5462306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 0);
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_ci	/* Add max 1000 */
5762306a36Sopenharmony_ci	ret = freq_qos_update_request(&req1, 1000);
5862306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
5962306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 1000);
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci	/* Add max 2000, no impact */
6262306a36Sopenharmony_ci	ret = freq_qos_update_request(&req2, 2000);
6362306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 0);
6462306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 1000);
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	/* Remove max 1000, new max 2000 */
6762306a36Sopenharmony_ci	ret = freq_qos_remove_request(&req1);
6862306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
6962306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MAX), 2000);
7062306a36Sopenharmony_ci}
7162306a36Sopenharmony_ci
7262306a36Sopenharmony_ci/*
7362306a36Sopenharmony_ci * Test that a freq_qos_request can be added again after removal
7462306a36Sopenharmony_ci *
7562306a36Sopenharmony_ci * This issue was solved by commit 05ff1ba412fd ("PM: QoS: Invalidate frequency
7662306a36Sopenharmony_ci * QoS requests after removal")
7762306a36Sopenharmony_ci */
7862306a36Sopenharmony_cistatic void freq_qos_test_readd(struct kunit *test)
7962306a36Sopenharmony_ci{
8062306a36Sopenharmony_ci	struct freq_constraints	qos;
8162306a36Sopenharmony_ci	struct freq_qos_request	req;
8262306a36Sopenharmony_ci	int ret;
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	freq_constraints_init(&qos);
8562306a36Sopenharmony_ci	memset(&req, 0, sizeof(req));
8662306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
8762306a36Sopenharmony_ci			FREQ_QOS_MIN_DEFAULT_VALUE);
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	/* Add */
9062306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req, FREQ_QOS_MIN, 1000);
9162306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
9262306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 1000);
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci	/* Remove */
9562306a36Sopenharmony_ci	ret = freq_qos_remove_request(&req);
9662306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
9762306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN),
9862306a36Sopenharmony_ci			FREQ_QOS_MIN_DEFAULT_VALUE);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	/* Add again */
10162306a36Sopenharmony_ci	ret = freq_qos_add_request(&qos, &req, FREQ_QOS_MIN, 2000);
10262306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, ret, 1);
10362306a36Sopenharmony_ci	KUNIT_EXPECT_EQ(test, freq_qos_read_value(&qos, FREQ_QOS_MIN), 2000);
10462306a36Sopenharmony_ci}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_cistatic struct kunit_case pm_qos_test_cases[] = {
10762306a36Sopenharmony_ci	KUNIT_CASE(freq_qos_test_min),
10862306a36Sopenharmony_ci	KUNIT_CASE(freq_qos_test_maxdef),
10962306a36Sopenharmony_ci	KUNIT_CASE(freq_qos_test_readd),
11062306a36Sopenharmony_ci	{},
11162306a36Sopenharmony_ci};
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_cistatic struct kunit_suite pm_qos_test_module = {
11462306a36Sopenharmony_ci	.name = "qos-kunit-test",
11562306a36Sopenharmony_ci	.test_cases = pm_qos_test_cases,
11662306a36Sopenharmony_ci};
11762306a36Sopenharmony_cikunit_test_suites(&pm_qos_test_module);
118