// SPDX-License-Identifier: GPL-2.0-only/* * Copyright (C) 2025 Google LLC * * Author: Lee Jones <lee@kernel.org> */ #include <kunit/device.h>#include <kunit/test.h>#include <linux/device.h>#include <linux/leds.h> #define LED_TEST_POST_REG_BRIGHTNESS 10 struct led_test_ddata {	struct led_classdev cdev;	struct device *dev;}; static enum led_brightness led_test_brightness_get(struct led_classdev *cdev){	return LED_TEST_POST_REG_BRIGHTNESS;} static void led_test_class_register(struct kunit *test){	struct led_test_ddata *ddata = test->priv;	struct led_classdev *cdev_clash, *cdev = &ddata->cdev;	struct device *dev = ddata->dev;	int ret; 	/* Register a LED class device */	cdev->name = "led-test";	cdev->brightness_get = led_test_brightness_get;	cdev->brightness = 0; 	ret = devm_led_classdev_register(dev, cdev);	KUNIT_ASSERT_EQ(test, ret, 0); 	KUNIT_EXPECT_EQ(test, cdev->max_brightness, LED_FULL);	KUNIT_EXPECT_EQ(test, cdev->brightness, LED_TEST_POST_REG_BRIGHTNESS);	KUNIT_EXPECT_STREQ(test, cdev->dev->kobj.name, "led-test"); 	/* Register again with the same name - expect it to pass with the LED renamed */	cdev_clash = devm_kmemdup(dev, cdev, sizeof(*cdev), GFP_KERNEL);	KUNIT_ASSERT_NOT_ERR_OR_NULL(test, cdev_clash); 	ret = devm_led_classdev_register(dev, cdev_clash);	KUNIT_ASSERT_EQ(test, ret, 0); 	KUNIT_EXPECT_STREQ(test, cdev_clash->dev->kobj.name, "led-test_1");	KUNIT_EXPECT_STREQ(test, cdev_clash->name, "led-test"); 	/* Enable name conflict rejection and register with the same name again - expect failure */	cdev_clash->flags |= LED_REJECT_NAME_CONFLICT;	ret = devm_led_classdev_register(dev, cdev_clash);	KUNIT_EXPECT_EQ(test, ret, -EEXIST);} static void led_test_class_add_lookup_and_get(struct kunit *test){	struct led_test_ddata *ddata = test->priv;	struct led_classdev *cdev = &ddata->cdev, *cdev_get;	struct device *dev = ddata->dev;	struct led_lookup_data lookup;	int ret; 	/* First, register a LED class device */	cdev->name = "led-test";	ret = devm_led_classdev_register(dev, cdev);	KUNIT_ASSERT_EQ(test, ret, 0); 	/* Then make the LED available for lookup */	lookup.provider = cdev->name;	lookup.dev_id = dev_name(dev);	lookup.con_id = "led-test-1";	led_add_lookup(&lookup); 	/* Finally, attempt to look it up via the API - imagine this was an orthogonal driver */	cdev_get = devm_led_get(dev, "led-test-1");	KUNIT_ASSERT_FALSE(test, IS_ERR(cdev_get)); 	KUNIT_EXPECT_STREQ(test, cdev_get->name, cdev->name); 	led_remove_lookup(&lookup);} static struct kunit_case led_test_cases[] = {	KUNIT_CASE(led_test_class_register),	KUNIT_CASE(led_test_class_add_lookup_and_get),	{ }}; static int led_test_init(struct kunit *test){	struct led_test_ddata *ddata;	struct device *dev; 	ddata = kunit_kzalloc(test, sizeof(*ddata), GFP_KERNEL);	if (!ddata)		return -ENOMEM; 	test->priv = ddata; 	dev = kunit_device_register(test, "led_test");	if (IS_ERR(dev))		return PTR_ERR(dev); 	ddata->dev = get_device(dev); 	return 0;} static void led_test_exit(struct kunit *test){	struct led_test_ddata *ddata = test->priv; 	if (ddata && ddata->dev)		put_device(ddata->dev);} static struct kunit_suite led_test_suite = {	.name = "led",	.init = led_test_init,	.exit = led_test_exit,	.test_cases = led_test_cases,};kunit_test_suite(led_test_suite); MODULE_AUTHOR("Lee Jones <lee@kernel.org>");MODULE_DESCRIPTION("KUnit tests for the LED framework");MODULE_LICENSE("GPL");