Quelle  overlay_test.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * KUnit tests for device tree overlays
 */
#include <linux/device/bus.h>
#include <linux/kconfig.h>
#include <linux/of.h>
#include <linux/of_platform.h>
#include <linux/platform_device.h>

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

#include "of_private.h"

static const char * const kunit_node_name = "kunit-test";
static const char * const kunit_compatible = "test,empty";

/* Test that of_overlay_apply_kunit() adds a node to the live tree */
static void of_overlay_apply_kunit_apply(struct kunit *test)
{
 struct device_node *np;

 KUNIT_ASSERT_EQ(test, 0,
   of_overlay_apply_kunit(test, kunit_overlay_test));

 np = of_find_node_by_name(NULL, kunit_node_name);
 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, np);
 of_node_put(np);
}

/*
 * Test that of_overlay_apply_kunit() creates platform devices with the
 * expected device_node
 */
static void of_overlay_apply_kunit_platform_device(struct kunit *test)
{
 struct platform_device *pdev;
 struct device_node *np;

 KUNIT_ASSERT_EQ(test, 0,
   of_overlay_apply_kunit(test, kunit_overlay_test));

 np = of_find_node_by_name(NULL, kunit_node_name);
 of_node_put_kunit(test, np);
 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, np);

 pdev = of_find_device_by_node(np);
 KUNIT_EXPECT_NOT_ERR_OR_NULL(test, pdev);
 if (pdev)
  put_device(&pdev->dev);
}

static int of_overlay_bus_match_compatible(struct device *dev, const void *data)
{
 return of_device_is_compatible(dev->of_node, data);
}

/* Test that of_overlay_apply_kunit() cleans up after the test is finished */
static void of_overlay_apply_kunit_cleanup(struct kunit *test)
{
 struct kunit fake;
 struct platform_device *pdev;
 struct device *dev;
 struct device_node *np;

 of_root_kunit_skip(test);
 if (!IS_ENABLED(CONFIG_OF_OVERLAY))
  kunit_skip(test, "requires CONFIG_OF_OVERLAY to apply overlay");
 if (!IS_ENABLED(CONFIG_OF_EARLY_FLATTREE))
  kunit_skip(test, "requires CONFIG_OF_EARLY_FLATTREE for root node");

 kunit_init_test(&fake, "fake test", NULL);
 KUNIT_ASSERT_EQ(test, fake.status, KUNIT_SUCCESS);

 KUNIT_ASSERT_EQ(test, 0,
   of_overlay_apply_kunit(&fake, kunit_overlay_test));

 np = of_find_node_by_name(NULL, kunit_node_name);
 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, np);
 of_node_put_kunit(&fake, np);

 pdev = of_find_device_by_node(np);
 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, pdev);
 put_device(&pdev->dev); /* Not derefing 'pdev' after this */

 /* Remove overlay */
 kunit_cleanup(&fake);

 /* The node and device should be removed */
 np = of_find_node_by_name(NULL, kunit_node_name);
 KUNIT_EXPECT_PTR_EQ(test, NULL, np);
 of_node_put(np);

 dev = bus_find_device(&platform_bus_type, NULL, kunit_compatible,
         of_overlay_bus_match_compatible);
 KUNIT_EXPECT_PTR_EQ(test, NULL, dev);
 put_device(dev);
}

static struct kunit_case of_overlay_apply_kunit_test_cases[] = {
 KUNIT_CASE(of_overlay_apply_kunit_apply),
 KUNIT_CASE(of_overlay_apply_kunit_platform_device),
 KUNIT_CASE(of_overlay_apply_kunit_cleanup),
 {}
};

/*
 * Test suite for test managed device tree overlays.
 */
static struct kunit_suite of_overlay_apply_kunit_suite = {
 .name = "of_overlay_apply_kunit",
 .test_cases = of_overlay_apply_kunit_test_cases,
};

kunit_test_suites(
 &of_overlay_apply_kunit_suite,
);
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("KUnit tests for device tree overlays");