// SPDX-License-Identifier: GPL-2.0

#include <kunit/test.h>
#include <kunit/device.h>
#include <linux/coresight.h>

#include "coresight-priv.h"

static struct coresight_device *coresight_test_device(struct device *dev)
{
 struct coresight_device *csdev = devm_kcalloc(dev, 1,
           sizeof(struct coresight_device),
           GFP_KERNEL);
 csdev->pdata = devm_kcalloc(dev, 1,
       sizeof(struct coresight_platform_data),
       GFP_KERNEL);
 return csdev;
}

static void test_default_sink(struct kunit *test)
{
 /*
 * Source -> ETF -> ETR -> CATU
 *                   ^
 *                   | default
 */
 struct device *dev = kunit_device_register(test, "coresight_kunit");
 struct coresight_device *src = coresight_test_device(dev),
    *etf = coresight_test_device(dev),
    *etr = coresight_test_device(dev),
    *catu = coresight_test_device(dev);
 struct coresight_connection conn = {};

 src->type = CORESIGHT_DEV_TYPE_SOURCE;
 /*
 * Don't use CORESIGHT_DEV_SUBTYPE_SOURCE_PROC, that would always return
 * a TRBE sink if one is registered.
 */
 src->subtype.source_subtype = CORESIGHT_DEV_SUBTYPE_SOURCE_BUS;
 etf->type = CORESIGHT_DEV_TYPE_LINKSINK;
 etf->subtype.sink_subtype = CORESIGHT_DEV_SUBTYPE_SINK_BUFFER;
 etr->type = CORESIGHT_DEV_TYPE_SINK;
 etr->subtype.sink_subtype = CORESIGHT_DEV_SUBTYPE_SINK_SYSMEM;
 catu->type = CORESIGHT_DEV_TYPE_HELPER;

 conn.src_dev = src;
 conn.dest_dev = etf;
 coresight_add_out_conn(dev, src->pdata, &conn);

 conn.src_dev = etf;
 conn.dest_dev = etr;
 coresight_add_out_conn(dev, etf->pdata, &conn);

 conn.src_dev = etr;
 conn.dest_dev = catu;
 coresight_add_out_conn(dev, etr->pdata, &conn);

 KUNIT_ASSERT_PTR_EQ(test, coresight_find_default_sink(src), etr);
}

static struct kunit_case coresight_testcases[] = {
 KUNIT_CASE(test_default_sink),
 {}
};

static struct kunit_suite coresight_test_suite = {
 .name = "coresight_test_suite",
 .test_cases = coresight_testcases,
};

kunit_test_suites(&coresight_test_suite);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("James Clark <james.clark@linaro.org>");
MODULE_DESCRIPTION("Arm CoreSight KUnit tests");