128 lines · c
1// SPDX-License-Identifier: GPL-2.02 3#ifndef DRM_KUNIT_HELPERS_H_4#define DRM_KUNIT_HELPERS_H_5 6#include <drm/drm_drv.h>7 8#include <linux/device.h>9 10#include <kunit/test.h>11 12struct drm_crtc_funcs;13struct drm_crtc_helper_funcs;14struct drm_device;15struct drm_plane_funcs;16struct drm_plane_helper_funcs;17struct kunit;18 19struct device *drm_kunit_helper_alloc_device(struct kunit *test);20void drm_kunit_helper_free_device(struct kunit *test, struct device *dev);21 22struct drm_device *23__drm_kunit_helper_alloc_drm_device_with_driver(struct kunit *test,24 struct device *dev,25 size_t size, size_t offset,26 const struct drm_driver *driver);27 28/**29 * drm_kunit_helper_alloc_drm_device_with_driver - Allocates a mock DRM device for KUnit tests30 * @_test: The test context object31 * @_dev: The parent device object32 * @_type: the type of the struct which contains struct &drm_device33 * @_member: the name of the &drm_device within @_type.34 * @_drv: Mocked DRM device driver features35 *36 * This function creates a struct &drm_device from @_dev and @_drv.37 *38 * @_dev should be allocated using drm_kunit_helper_alloc_device().39 *40 * The driver is tied to the @_test context and will get cleaned at the41 * end of the test. The drm_device is allocated through42 * devm_drm_dev_alloc() and will thus be freed through a device-managed43 * resource.44 *45 * Returns:46 * A pointer to the new drm_device, or an ERR_PTR() otherwise.47 */48#define drm_kunit_helper_alloc_drm_device_with_driver(_test, _dev, _type, _member, _drv) \49 ((_type *)__drm_kunit_helper_alloc_drm_device_with_driver(_test, _dev, \50 sizeof(_type), \51 offsetof(_type, _member), \52 _drv))53 54static inline struct drm_device *55__drm_kunit_helper_alloc_drm_device(struct kunit *test,56 struct device *dev,57 size_t size, size_t offset,58 u32 features)59{60 struct drm_driver *driver;61 62 driver = devm_kzalloc(dev, sizeof(*driver), GFP_KERNEL);63 KUNIT_ASSERT_NOT_NULL(test, driver);64 65 driver->driver_features = features;66 67 return __drm_kunit_helper_alloc_drm_device_with_driver(test, dev,68 size, offset,69 driver);70}71 72/**73 * drm_kunit_helper_alloc_drm_device - Allocates a mock DRM device for KUnit tests74 * @_test: The test context object75 * @_dev: The parent device object76 * @_type: the type of the struct which contains struct &drm_device77 * @_member: the name of the &drm_device within @_type.78 * @_feat: Mocked DRM device driver features79 *80 * This function creates a struct &drm_driver and will create a struct81 * &drm_device from @_dev and that driver.82 *83 * @_dev should be allocated using drm_kunit_helper_alloc_device().84 *85 * The driver is tied to the @_test context and will get cleaned at the86 * end of the test. The drm_device is allocated through87 * devm_drm_dev_alloc() and will thus be freed through a device-managed88 * resource.89 *90 * Returns:91 * A pointer to the new drm_device, or an ERR_PTR() otherwise.92 */93#define drm_kunit_helper_alloc_drm_device(_test, _dev, _type, _member, _feat) \94 ((_type *)__drm_kunit_helper_alloc_drm_device(_test, _dev, \95 sizeof(_type), \96 offsetof(_type, _member), \97 _feat))98struct drm_modeset_acquire_ctx *99drm_kunit_helper_acquire_ctx_alloc(struct kunit *test);100 101struct drm_atomic_state *102drm_kunit_helper_atomic_state_alloc(struct kunit *test,103 struct drm_device *drm,104 struct drm_modeset_acquire_ctx *ctx);105 106struct drm_plane *107drm_kunit_helper_create_primary_plane(struct kunit *test,108 struct drm_device *drm,109 const struct drm_plane_funcs *funcs,110 const struct drm_plane_helper_funcs *helper_funcs,111 const uint32_t *formats,112 unsigned int num_formats,113 const uint64_t *modifiers);114 115struct drm_crtc *116drm_kunit_helper_create_crtc(struct kunit *test,117 struct drm_device *drm,118 struct drm_plane *primary,119 struct drm_plane *cursor,120 const struct drm_crtc_funcs *funcs,121 const struct drm_crtc_helper_funcs *helper_funcs);122 123struct drm_display_mode *124drm_kunit_display_mode_from_cea_vic(struct kunit *test, struct drm_device *dev,125 u8 video_code);126 127#endif // DRM_KUNIT_HELPERS_H_128