test/drm-lease: Create helper functions to reduce boilerplate

[src/drm-lease-manager.git] / drm-lease-manager / test / test-drm-device.c

diff --git a/drm-lease-manager/test/test-drm-device.c b/drm-lease-manager/test/test-drm-device.c
index 844599a..1d1e096 100644 (file)
--- a/drm-lease-manager/test/test-drm-device.c
+++ b/drm-lease-manager/test/test-drm-device.c
@@ -17,6 +17,7 @@
 #include <stdbool.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
 #include <xf86drmMode.h>
 
 #include "test-drm-device.h"
@@ -75,6 +76,13 @@ void reset_drm_test_device(void)
        free(test_device.resources.encoders);
        free(test_device.plane_resources.planes);
        free(test_device.leases.lessee_ids);
+
+       if (test_device.layout.free_on_reset) {
+               free(test_device.layout.connectors);
+               free(test_device.layout.encoders);
+               free(test_device.layout.planes);
+       }
+
        memset(&test_device, 0, sizeof(test_device));
 }
 
@@ -86,6 +94,42 @@ void setup_test_device_layout(drmModeConnector *connectors,
        test_device.layout.planes = planes;
 }
 
+void setup_layout_simple_test_device(int conn_cnt, int plane_cnt)
+{
+       drmModeConnector *connectors;
+       drmModeEncoder *encoders;
+       drmModePlane *planes = NULL;
+
+       ck_assert_int_ge(conn_cnt, 1);
+
+       setup_drm_test_device(conn_cnt, conn_cnt, conn_cnt, plane_cnt);
+
+       ck_assert_ptr_ne(
+           connectors = calloc(sizeof(drmModeConnector), conn_cnt), NULL);
+       ck_assert_ptr_ne(encoders = calloc(sizeof(drmModeEncoder), conn_cnt),
+                        NULL);
+
+       if (plane_cnt > 0)
+               ck_assert_ptr_ne(
+                   planes = calloc(sizeof(drmModePlane), plane_cnt), NULL);
+
+       int crtc_mask = (1 << conn_cnt) - 1;
+       for (int i = 0; i < conn_cnt; i++) {
+               connectors[i] = (drmModeConnector)CONNECTOR(
+                   CONNECTOR_ID(i), ENCODER_ID(i), &ENCODER_ID(i), 1);
+               encoders[i] = (drmModeEncoder)ENCODER(ENCODER_ID(i), CRTC_ID(i),
+                                                     crtc_mask);
+       }
+
+       for (int i = 0; i < plane_cnt; i++) {
+               planes[i] =
+                   (drmModePlane)PLANE(PLANE_ID(i), 1 << (i % conn_cnt));
+       }
+
+       setup_test_device_layout(connectors, encoders, planes);
+       test_device.layout.free_on_reset = true;
+}
+
 #define GET_DRM_RESOURCE_FN(Res, res, RES, container)                       \
        drmMode##Res##Ptr get_##res(int fd, uint32_t id)                    \
        {                                                                   \
@@ -105,7 +149,6 @@ GET_DRM_RESOURCE_FN(Plane, plane, PLANE, plane_resources)
 int create_lease(int fd, const uint32_t *objects, int num_objects, int flags,
                 uint32_t *lessee_id)
 {
-       UNUSED(fd);
        UNUSED(objects);
        UNUSED(num_objects);
        UNUSED(flags);
@@ -118,5 +161,5 @@ int create_lease(int fd, const uint32_t *objects, int num_objects, int flags,
 
        test_device.leases.count++;
 
-       return 0;
+       return dup(fd);
 }