We don't have a set of kunit tests for the functions under drm_atomic.h.
Let's use the introduction of drm_atomic_get_connector_for_encoder() to
create some tests for it and thus create that set.

Reviewed-by: Dmitry Baryshkov <dmitry.barysh...@linaro.org>
Signed-off-by: Maxime Ripard <mrip...@kernel.org>
---
 drivers/gpu/drm/tests/Makefile          |   1 +
 drivers/gpu/drm/tests/drm_atomic_test.c | 153 ++++++++++++++++++++++++++++++++
 2 files changed, 154 insertions(+)

diff --git a/drivers/gpu/drm/tests/Makefile b/drivers/gpu/drm/tests/Makefile
index 
6691c577d2d4581a4185bac2ce89a6b14b339b35..3afd6587df0831a0d10e91ff0a584af6ba1813e4
 100644
--- a/drivers/gpu/drm/tests/Makefile
+++ b/drivers/gpu/drm/tests/Makefile
@@ -2,10 +2,11 @@
 
 obj-$(CONFIG_DRM_KUNIT_TEST_HELPERS) += \
        drm_kunit_helpers.o
 
 obj-$(CONFIG_DRM_KUNIT_TEST) += \
+       drm_atomic_test.o \
        drm_atomic_state_test.o \
        drm_bridge_test.o \
        drm_buddy_test.o \
        drm_cmdline_parser_test.o \
        drm_connector_test.o \
diff --git a/drivers/gpu/drm/tests/drm_atomic_test.c 
b/drivers/gpu/drm/tests/drm_atomic_test.c
new file mode 100644
index 
0000000000000000000000000000000000000000..ea91bec6569e688ad33149d8ee6ecc3562b61254
--- /dev/null
+++ b/drivers/gpu/drm/tests/drm_atomic_test.c
@@ -0,0 +1,153 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * Kunit test for drm_atomic functions
+ */
+#include <drm/drm_atomic.h>
+#include <drm/drm_atomic_state_helper.h>
+#include <drm/drm_atomic_uapi.h>
+#include <drm/drm_encoder.h>
+#include <drm/drm_kunit_helpers.h>
+#include <drm/drm_modeset_helper_vtables.h>
+
+#include <kunit/test.h>
+
+struct drm_atomic_test_priv {
+       struct drm_device drm;
+       struct drm_plane *plane;
+       struct drm_crtc *crtc;
+       struct drm_encoder encoder;
+       struct drm_connector connector;
+};
+
+static const struct drm_connector_helper_funcs 
drm_atomic_init_connector_helper_funcs = {
+};
+
+static const struct drm_connector_funcs drm_atomic_init_connector_funcs = {
+       .atomic_destroy_state   = drm_atomic_helper_connector_destroy_state,
+       .atomic_duplicate_state = drm_atomic_helper_connector_duplicate_state,
+       .reset                  = drm_atomic_helper_connector_reset,
+};
+
+static struct drm_atomic_test_priv *create_device(struct kunit *test)
+{
+       struct drm_atomic_test_priv *priv;
+       struct drm_connector *connector;
+       struct drm_encoder *enc;
+       struct drm_device *drm;
+       struct drm_plane *plane;
+       struct drm_crtc *crtc;
+       struct device *dev;
+       int ret;
+
+       dev = drm_kunit_helper_alloc_device(test);
+       if (IS_ERR(dev))
+               return ERR_CAST(dev);
+
+       priv = drm_kunit_helper_alloc_drm_device(test, dev,
+                                                struct drm_atomic_test_priv, 
drm,
+                                                DRIVER_MODESET | 
DRIVER_ATOMIC);
+       if (IS_ERR(priv))
+               return ERR_CAST(priv);
+
+       drm = &priv->drm;
+       plane = drm_kunit_helper_create_primary_plane(test, drm,
+                                                     NULL,
+                                                     NULL,
+                                                     NULL, 0,
+                                                     NULL);
+       if (IS_ERR(plane))
+               return ERR_CAST(plane);
+       priv->plane = plane;
+
+       crtc = drm_kunit_helper_create_crtc(test, drm,
+                                           plane, NULL,
+                                           NULL,
+                                           NULL);
+       if (IS_ERR(crtc))
+               return ERR_CAST(crtc);
+       priv->crtc = crtc;
+
+       enc = &priv->encoder;
+       ret = drmm_encoder_init(drm, enc, NULL, DRM_MODE_ENCODER_TMDS, NULL);
+       if (ret)
+               return ERR_PTR(ret);
+
+       enc->possible_crtcs = drm_crtc_mask(crtc);
+
+       connector = &priv->connector;
+       ret = drmm_connector_init(drm, connector,
+                                 &drm_atomic_init_connector_funcs,
+                                 DRM_MODE_CONNECTOR_VIRTUAL,
+                                 NULL);
+       if (ret)
+               return ERR_PTR(ret);
+
+       drm_connector_helper_add(connector, 
&drm_atomic_init_connector_helper_funcs);
+
+       drm_connector_attach_encoder(connector, enc);
+
+       drm_mode_config_reset(drm);
+
+       return priv;
+}
+
+static void drm_test_drm_atomic_get_connector_for_encoder(struct kunit *test)
+{
+       struct drm_modeset_acquire_ctx ctx;
+       struct drm_atomic_test_priv *priv;
+       struct drm_display_mode *mode;
+       struct drm_connector *curr_connector;
+       int ret;
+
+       priv = create_device(test);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, priv);
+
+       mode = drm_kunit_display_mode_from_cea_vic(test, &priv->drm, 16);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, mode);
+
+       drm_modeset_acquire_init(&ctx, 0);
+
+retry_enable:
+       ret = drm_kunit_helper_enable_crtc_connector(test, &priv->drm,
+                                                    priv->crtc, 
&priv->connector,
+                                                    mode, &ctx);
+       if (ret == -EDEADLK) {
+               drm_modeset_backoff(&ctx);
+               goto retry_enable;
+       }
+       KUNIT_ASSERT_EQ(test, ret, 0);
+
+       drm_modeset_drop_locks(&ctx);
+       drm_modeset_acquire_fini(&ctx);
+
+       drm_modeset_acquire_init(&ctx, 0);
+
+retry_conn:
+       curr_connector = drm_atomic_get_connector_for_encoder(&priv->encoder,
+                                                             &ctx);
+       if (PTR_ERR(curr_connector) == -EDEADLK) {
+               drm_modeset_backoff(&ctx);
+               goto retry_conn;
+       }
+       KUNIT_EXPECT_PTR_EQ(test, curr_connector, &priv->connector);
+
+       drm_modeset_drop_locks(&ctx);
+       drm_modeset_acquire_fini(&ctx);
+}
+
+static struct kunit_case drm_atomic_get_connector_for_encoder_tests[] = {
+       KUNIT_CASE(drm_test_drm_atomic_get_connector_for_encoder),
+       { }
+};
+
+
+static struct kunit_suite drm_atomic_get_connector_for_encoder_test_suite = {
+       .name = "drm_test_atomic_get_connector_for_encoder",
+       .test_cases = drm_atomic_get_connector_for_encoder_tests,
+};
+
+kunit_test_suite(drm_atomic_get_connector_for_encoder_test_suite);
+
+MODULE_AUTHOR("Maxime Ripard <mrip...@kernel.org>");
+MODULE_DESCRIPTION("Kunit test for drm_atomic functions");
+MODULE_LICENSE("GPL");

-- 
2.48.1

Reply via email to