// SPDX-License-Identifier: GPL-2.0-only OR MIT
/* Copyright (c) 2025 Imagination Technologies Ltd. */

#include "pvr_device.h"

#include <linux/errno.h>
#include <linux/stddef.h>
#include <linux/string.h>
#include <linux/types.h>

#include <kunit/test.h>
#include <kunit/visibility.h>

static void decode_gpuid_string(struct kunit *test)
{
	const struct pvr_gpu_id bad_gpuid = { 0xdead, 0xbeef, 0xcafe, 0xface };
	const u64 packed_bad_gpuid = pvr_gpu_id_to_packed_bvnc(&bad_gpuid);

#define GPUID_TEST_CASE(str_, err_, value_)					\
	do {									\
		struct pvr_gpu_id _gpuid_out = bad_gpuid;			\
		int _err;							\
		_err = pvr_gpuid_decode_string(NULL, str_, &_gpuid_out);	\
		KUNIT_EXPECT_EQ(test, _err, err_);				\
		KUNIT_EXPECT_EQ(test,						\
				pvr_gpu_id_to_packed_bvnc(&_gpuid_out),		\
				value_);					\
	} while (0)

#define GPUID_TEST_CASE_OK(str_, b_, v_, n_, c_) \
	GPUID_TEST_CASE(str_, 0, PVR_PACKED_BVNC(b_, v_, n_, c_))

#define GPUID_TEST_CASE_INVAL(str_) \
	GPUID_TEST_CASE(str_, -EINVAL, packed_bad_gpuid)

	GPUID_TEST_CASE_OK("12.34.56.78", 12, 34, 56, 78);
	GPUID_TEST_CASE_OK("0.0.0.0", 0, 0, 0, 0);

	GPUID_TEST_CASE_INVAL("");
	GPUID_TEST_CASE_INVAL("42.foobar-invalid.gpuid.bvnc");

	/* String longer than PVR_GPUID_STRING_MAX_LENGTH. */
	GPUID_TEST_CASE_INVAL("12.34.56.789012345678901234567890123456");

	/* Single value overflowing u16. */
	GPUID_TEST_CASE_INVAL("12.34.56.999999");

	/* Wrong number of parts and/or dots. */
	GPUID_TEST_CASE_INVAL("12.34.56.78.90");
	GPUID_TEST_CASE_INVAL("12.34.56..78");
	GPUID_TEST_CASE_INVAL("12.34..56");
	GPUID_TEST_CASE_INVAL("12.34.56");

#undef GPUID_TEST_CASE_INVAL
#undef GPUID_TEST_CASE_OK
#undef GPUID_TEST_CASE
}

static struct kunit_case pvr_tests_cases[] = {
	KUNIT_CASE(decode_gpuid_string),
	{},
};

static struct kunit_suite pvr_tests_suite = {
	.name = "pvr_tests",
	.test_cases = pvr_tests_cases,
};
kunit_test_suite(pvr_tests_suite);

MODULE_AUTHOR("Imagination Technologies Ltd.");
MODULE_LICENSE("Dual MIT/GPL");
MODULE_DESCRIPTION("pvr kunit tests");
MODULE_IMPORT_NS("EXPORTED_FOR_KUNIT_TESTING");