// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2020, Linaro Limited
 *
 * Tests scmi_agent uclass and the SCMI drivers implemented in other
 * uclass devices probe when a SCMI server exposes resources.
 *
 * Note in test.dts the protocol@10 node in agent 1. Protocol 0x10 is not
 * implemented in U-Boot SCMI components but the implementation is exepected
 * to not complain on unknown protocol IDs, as long as it is not used. Note
 * in test.dts tests that SCMI drivers probing does not fail for such an
 * unknown SCMI protocol ID.
 */

#include <common.h>
#include <clk.h>
#include <dm.h>
#include <reset.h>
#include <asm/scmi_test.h>
#include <dm/device-internal.h>
#include <dm/test.h>
#include <linux/kconfig.h>
#include <test/ut.h>

static int ut_assert_scmi_state_preprobe(struct unit_test_state *uts)
{
	struct sandbox_scmi_service *scmi_ctx = sandbox_scmi_service_ctx();

	ut_assertnonnull(scmi_ctx);
	if (scmi_ctx->agent_count)
		ut_asserteq(2, scmi_ctx->agent_count);

	return 0;
}

static int ut_assert_scmi_state_postprobe(struct unit_test_state *uts,
					  struct udevice *dev)
{
	struct sandbox_scmi_devices *scmi_devices;
	struct sandbox_scmi_service *scmi_ctx;

	/* Device references to check context against test sequence */
	scmi_devices = sandbox_scmi_devices_ctx(dev);

	ut_assertnonnull(scmi_devices);
	if (IS_ENABLED(CONFIG_CLK_SCMI))
		ut_asserteq(3, scmi_devices->clk_count);
	if (IS_ENABLED(CONFIG_RESET_SCMI))
		ut_asserteq(1, scmi_devices->reset_count);

	/* State of the simulated SCMI server exposed */
	scmi_ctx = sandbox_scmi_service_ctx();

	ut_asserteq(2, scmi_ctx->agent_count);

	ut_assertnonnull(scmi_ctx->agent[0]);
	ut_asserteq(2, scmi_ctx->agent[0]->clk_count);
	ut_assertnonnull(scmi_ctx->agent[0]->clk);
	ut_asserteq(1, scmi_ctx->agent[0]->reset_count);
	ut_assertnonnull(scmi_ctx->agent[0]->reset);

	ut_assertnonnull(scmi_ctx->agent[1]);
	ut_assertnonnull(scmi_ctx->agent[1]->clk);
	ut_asserteq(1, scmi_ctx->agent[1]->clk_count);

	return 0;
}

static int load_sandbox_scmi_test_devices(struct unit_test_state *uts,
					  struct udevice **dev)
{
	int ret;

	ret = ut_assert_scmi_state_preprobe(uts);
	if (ret)
		return ret;

	ut_assertok(uclass_get_device_by_name(UCLASS_MISC, "sandbox_scmi",
					      dev));
	ut_assertnonnull(*dev);

	return ut_assert_scmi_state_postprobe(uts, *dev);
}

static int release_sandbox_scmi_test_devices(struct unit_test_state *uts,
					     struct udevice *dev)
{
	ut_assertok(device_remove(dev, DM_REMOVE_NORMAL));

	/* Not sure test devices are fully removed, agent may not be visible */
	return 0;
}

/*
 * Test SCMI states when loading and releasing resources
 * related to SCMI drivers.
 */
static int dm_test_scmi_sandbox_agent(struct unit_test_state *uts)
{
	struct udevice *dev = NULL;
	int ret;

	ret = load_sandbox_scmi_test_devices(uts, &dev);
	if (!ret)
		ret = release_sandbox_scmi_test_devices(uts, dev);

	return ret;
}

DM_TEST(dm_test_scmi_sandbox_agent, UT_TESTF_SCAN_FDT);

static int dm_test_scmi_clocks(struct unit_test_state *uts)
{
	struct sandbox_scmi_devices *scmi_devices;
	struct sandbox_scmi_service *scmi_ctx;
	struct udevice *dev = NULL;
	int ret_dev;
	int ret;

	if (!IS_ENABLED(CONFIG_CLK_SCMI))
		return 0;

	ret = load_sandbox_scmi_test_devices(uts, &dev);
	if (ret)
		return ret;

	scmi_devices = sandbox_scmi_devices_ctx(dev);
	scmi_ctx = sandbox_scmi_service_ctx();

	/* Test SCMI clocks rate manipulation */
	ut_asserteq(1000, clk_get_rate(&scmi_devices->clk[0]));
	ut_asserteq(333, clk_get_rate(&scmi_devices->clk[1]));
	ut_asserteq(44, clk_get_rate(&scmi_devices->clk[2]));

	ret_dev = clk_set_rate(&scmi_devices->clk[1], 1088);
	ut_assert(!ret_dev || ret_dev == 1088);

	ut_asserteq(1000, scmi_ctx->agent[0]->clk[0].rate);
	ut_asserteq(1088, scmi_ctx->agent[0]->clk[1].rate);
	ut_asserteq(44, scmi_ctx->agent[1]->clk[0].rate);

	ut_asserteq(1000, clk_get_rate(&scmi_devices->clk[0]));
	ut_asserteq(1088, clk_get_rate(&scmi_devices->clk[1]));
	ut_asserteq(44, clk_get_rate(&scmi_devices->clk[2]));

	/* restore original rate for further tests */
	ret_dev = clk_set_rate(&scmi_devices->clk[1], 333);
	ut_assert(!ret_dev || ret_dev == 333);

	/* Test SCMI clocks gating manipulation */
	ut_assert(!scmi_ctx->agent[0]->clk[0].enabled);
	ut_assert(!scmi_ctx->agent[0]->clk[1].enabled);
	ut_assert(!scmi_ctx->agent[1]->clk[0].enabled);

	ut_asserteq(0, clk_enable(&scmi_devices->clk[1]));
	ut_asserteq(0, clk_enable(&scmi_devices->clk[2]));

	ut_assert(!scmi_ctx->agent[0]->clk[0].enabled);
	ut_assert(scmi_ctx->agent[0]->clk[1].enabled);
	ut_assert(scmi_ctx->agent[1]->clk[0].enabled);

	ut_assertok(clk_disable(&scmi_devices->clk[1]));
	ut_assertok(clk_disable(&scmi_devices->clk[2]));

	ut_assert(!scmi_ctx->agent[0]->clk[0].enabled);
	ut_assert(!scmi_ctx->agent[0]->clk[1].enabled);
	ut_assert(!scmi_ctx->agent[1]->clk[0].enabled);

	return release_sandbox_scmi_test_devices(uts, dev);
}

DM_TEST(dm_test_scmi_clocks, UT_TESTF_SCAN_FDT);

static int dm_test_scmi_resets(struct unit_test_state *uts)
{
	struct sandbox_scmi_devices *scmi_devices;
	struct sandbox_scmi_service *scmi_ctx;
	struct udevice *dev = NULL;
	int ret;

	if (!IS_ENABLED(CONFIG_RESET_SCMI))
		return 0;

	ret = load_sandbox_scmi_test_devices(uts, &dev);
	if (ret)
		return ret;

	scmi_devices = sandbox_scmi_devices_ctx(dev);
	scmi_ctx = sandbox_scmi_service_ctx();

	/* Test SCMI resect controller manipulation */
	ut_assert(!scmi_ctx->agent[0]->reset[0].asserted)

	ut_assertok(reset_assert(&scmi_devices->reset[0]));
	ut_assert(scmi_ctx->agent[0]->reset[0].asserted)

	ut_assertok(reset_deassert(&scmi_devices->reset[0]));
	ut_assert(!scmi_ctx->agent[0]->reset[0].asserted);

	return release_sandbox_scmi_test_devices(uts, dev);
}

DM_TEST(dm_test_scmi_resets, UT_TESTF_SCAN_FDT);