Re: [PATCH v3 05/11] of: Add a KUnit test for overlays and test managed APIs

Rob Herring
2023-04-06T15:00:04Z

On Mon, Mar 27, 2023 at 03:21:53PM -0700, Stephen Boyd wrote:
> Test the KUnit test managed overlay APIs. Confirm that platform devices
> are created and destroyed properly. This provides us confidence that the
> test managed work correctly and can be relied upon to provide tests with
> fake platform devices and device nodes via overlays compiled into the
> kernel image.
>
> Cc: Rob Herring <robh+dt@kernel.org>
> Cc: Frank Rowand <frowand.list@gmail.com>
> Signed-off-by: Stephen Boyd <sboyd@kernel.org>
> ---
> drivers/of/.kunitconfig | 2 +
> drivers/of/Kconfig | 10 +++
> drivers/of/Makefile | 1 +
> drivers/of/kunit_overlay_test.dtso | 9 +++
> drivers/of/overlay_test.c | 110 +++++++++++++++++++++++++++++
> 5 files changed, 132 insertions(+)
> create mode 100644 drivers/of/kunit_overlay_test.dtso
> create mode 100644 drivers/of/overlay_test.c
>
> diff --git a/drivers/of/.kunitconfig b/drivers/of/.kunitconfig
> index 5a8fee11978c..7d570cb922a1 100644
> --- a/drivers/of/.kunitconfig
> +++ b/drivers/of/.kunitconfig
> @@ -1,3 +1,5 @@
> CONFIG_KUNIT=y
> CONFIG_OF=y
> CONFIG_OF_KUNIT_TEST=y
> +CONFIG_OF_OVERLAY=y
> +CONFIG_OF_OVERLAY_KUNIT_TEST=y
> diff --git a/drivers/of/Kconfig b/drivers/of/Kconfig
> index 1b995cecf5be..5bdeba11268d 100644
> --- a/drivers/of/Kconfig
> +++ b/drivers/of/Kconfig
> @@ -113,6 +113,16 @@ config OF_OVERLAY
> While this option is selected automatically when needed, you can
> enable it manually to improve device tree unit test coverage.
>
> +config OF_OVERLAY_KUNIT_TEST
> + tristate "Device Tree overlay KUnit tests" if !KUNIT_ALL_TESTS
> + depends on OF_OVERLAY
> + depends on KUNIT
> + default KUNIT_ALL_TESTS
> + help
> + This option builds KUnit unit tests for the device tree overlay code.
> +
> + If unsure, say N here, but this option is safe to enable.
> +
> config OF_NUMA
> bool
>
> diff --git a/drivers/of/Makefile b/drivers/of/Makefile
> index c694f998b9f5..2ad60d5b87ac 100644
> --- a/drivers/of/Makefile
> +++ b/drivers/of/Makefile
> @@ -21,5 +21,6 @@ endif
>
> obj-$(CONFIG_KUNIT) += of_kunit.o
> obj-$(CONFIG_OF_KUNIT_TEST) += of_test.o
> +obj-$(CONFIG_OF_OVERLAY_KUNIT_TEST) += overlay_test.o kunit_overlay_test.dtbo.o
>
> obj-$(CONFIG_OF_UNITTEST) += unittest-data/
> diff --git a/drivers/of/kunit_overlay_test.dtso b/drivers/of/kunit_overlay_test.dtso
> new file mode 100644
> index 000000000000..85f20b4b4c16
> --- /dev/null
> +++ b/drivers/of/kunit_overlay_test.dtso
> @@ -0,0 +1,9 @@
> +// SPDX-License-Identifier: GPL-2.0
> +/dts-v1/;
> +/plugin/;
> +
> +&{/} {
> + kunit-test {
> + compatible = "test,empty";
> + };
> +};
> diff --git a/drivers/of/overlay_test.c b/drivers/of/overlay_test.c
> new file mode 100644
> index 000000000000..66b1dceea568
> --- /dev/null
> +++ b/drivers/of/overlay_test.c
> @@ -0,0 +1,110 @@
> +// 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/platform_device.h>
> +
> +#include <kunit/of.h>
> +#include <kunit/test.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);
> +}
> +
> +static int bus_match_np(struct device *dev, const void *data)
> +{
> + const struct device_node *np = data;
> +
> + return np == dev->of_node;
> +}
> +
> +/*
> + * 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 device *dev;
> + 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);
> +
> + dev = bus_find_device(&platform_bus_type, NULL, np, bus_match_np);

of_find_device_by_node()

> + KUNIT_EXPECT_NOT_ERR_OR_NULL(test, dev);
> + put_device(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 device *dev;
> + 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(np); /* Not derefing 'np' after this */
> + KUNIT_ASSERT_NOT_ERR_OR_NULL(test, np);
> +
> + dev = bus_find_device(&platform_bus_type, NULL, np, bus_match_np);

And here.

> + put_device(dev); /* Not derefing 'device' after this */
> + KUNIT_ASSERT_NOT_ERR_OR_NULL(test, dev);
> +
> + /* Remove overlay */
> + kunit_cleanup(test);

I guess this is testing the cleanup. It's not that obvious given the
overlay is removed by kunit_cleanup(). Not sure if anything can be done
though...

> +
> + 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");
> --
> https://git.kernel.org/pub/scm/linux/kernel/git/clk/linux.git/
> https://git.kernel.org/pub/scm/linux/kernel/git/sboyd/spmi.git
>