[PATCH v2 07/11] drm/tests: Add test for drm_framebuffer_lookup()

[Carlos Eduardo Gallo Filho]

Add a single KUnit test case for the drm_framebuffer_lookup function.

Signed-off-by: Carlos Eduardo Gallo Filho <gcar...@disroot.org>
---
v2:
  - Reorder kunit cases alphabetically.
  - Replace drm_mode_object_add() call to drm_framebuffer_init().
  - Rely on drm_kunit_helper_alloc_device() for mock initialization.
---
 drivers/gpu/drm/tests/drm_framebuffer_test.c | 26 ++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/drivers/gpu/drm/tests/drm_framebuffer_test.c 
b/drivers/gpu/drm/tests/drm_framebuffer_test.c
index a63f30985f75..fb9589dd8aed 100644
--- a/drivers/gpu/drm/tests/drm_framebuffer_test.c
+++ b/drivers/gpu/drm/tests/drm_framebuffer_test.c
@@ -526,10 +526,36 @@ static void drm_test_framebuffer_cleanup(struct kunit 
*test)
        KUNIT_ASSERT_EQ(test, dev->mode_config.num_fb, 0);
 }
 
+static void drm_test_framebuffer_lookup(struct kunit *test)
+{
+       struct drm_framebuffer_test_priv *priv = test->priv;
+       struct drm_device *dev = &priv->dev;
+       struct drm_format_info format = { };
+       struct drm_framebuffer fb1 = { .dev = dev, .format = &format };
+       struct drm_framebuffer *fb2;
+       uint32_t id = 0;
+       int ret;
+
+       ret = drm_framebuffer_init(dev, &fb1, NULL);
+       KUNIT_ASSERT_EQ(test, ret, 0);
+       id = fb1.base.id;
+
+       /* Looking for fb1 */
+       fb2 = drm_framebuffer_lookup(dev, NULL, id);
+       KUNIT_EXPECT_PTR_EQ(test, fb2, &fb1);
+
+       /* Looking for an inexistent framebuffer */
+       fb2 = drm_framebuffer_lookup(dev, NULL, id + 1);
+       KUNIT_EXPECT_NULL(test, fb2);
+
+       drm_framebuffer_cleanup(&fb1);
+}
+
 static struct kunit_case drm_framebuffer_tests[] = {
        KUNIT_CASE_PARAM(drm_test_framebuffer_check_src_coords, 
check_src_coords_gen_params),
        KUNIT_CASE(drm_test_framebuffer_cleanup),
        KUNIT_CASE_PARAM(drm_test_framebuffer_create, 
drm_framebuffer_create_gen_params),
+       KUNIT_CASE(drm_test_framebuffer_lookup),
        KUNIT_CASE(drm_test_framebuffer_modifiers_not_supported),
        { }
 };
-- 
2.41.0