130 lines · c
1// SPDX-License-Identifier: GPL-2.0 AND MIT2/*3 * Copyright © 2023 Intel Corporation4 */5 6#include <kunit/test.h>7#include <kunit/static_stub.h>8#include <kunit/visibility.h>9 10#include <drm/drm_drv.h>11#include <drm/drm_kunit_helpers.h>12 13#include "tests/xe_kunit_helpers.h"14#include "tests/xe_pci_test.h"15#include "xe_device.h"16#include "xe_device_types.h"17#include "xe_pm.h"18 19/**20 * xe_kunit_helper_alloc_xe_device - Allocate a &xe_device for a KUnit test.21 * @test: the &kunit where this &xe_device will be used22 * @dev: The parent device object23 *24 * This function allocates xe_device using drm_kunit_helper_alloc_device().25 * The xe_device allocation is managed by the test.26 *27 * @dev should be allocated using drm_kunit_helper_alloc_device().28 *29 * This function uses KUNIT_ASSERT to detect any allocation failures.30 *31 * Return: A pointer to the new &xe_device.32 */33struct xe_device *xe_kunit_helper_alloc_xe_device(struct kunit *test,34 struct device *dev)35{36 struct xe_device *xe;37 38 xe = drm_kunit_helper_alloc_drm_device(test, dev,39 struct xe_device,40 drm, DRIVER_GEM);41 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, xe);42 return xe;43}44EXPORT_SYMBOL_IF_KUNIT(xe_kunit_helper_alloc_xe_device);45 46static void kunit_action_restore_priv(void *priv)47{48 struct kunit *test = kunit_get_current_test();49 50 test->priv = priv;51}52 53/**54 * xe_kunit_helper_xe_device_test_init - Prepare a &xe_device for a KUnit test.55 * @test: the &kunit where this fake &xe_device will be used56 *57 * This function allocates and initializes a fake &xe_device and stores its58 * pointer as &kunit.priv to allow the test code to access it.59 *60 * This function can be directly used as custom implementation of61 * &kunit_suite.init.62 *63 * It is possible to prepare specific variant of the fake &xe_device by passing64 * in &kunit.priv pointer to the struct xe_pci_fake_data supplemented with65 * desired parameters prior to calling this function.66 *67 * This function uses KUNIT_ASSERT to detect any failures.68 *69 * Return: Always 0.70 */71int xe_kunit_helper_xe_device_test_init(struct kunit *test)72{73 struct xe_device *xe;74 struct device *dev;75 int err;76 77 dev = drm_kunit_helper_alloc_device(test);78 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, dev);79 80 xe = xe_kunit_helper_alloc_xe_device(test, dev);81 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, xe);82 83 err = xe_pci_fake_device_init(xe);84 KUNIT_ASSERT_EQ(test, err, 0);85 86 err = kunit_add_action_or_reset(test, kunit_action_restore_priv, test->priv);87 KUNIT_ASSERT_EQ(test, err, 0);88 89 test->priv = xe;90 return 0;91}92EXPORT_SYMBOL_IF_KUNIT(xe_kunit_helper_xe_device_test_init);93 94KUNIT_DEFINE_ACTION_WRAPPER(put_xe_pm_runtime, xe_pm_runtime_put, struct xe_device *);95 96/**97 * xe_kunit_helper_xe_device_live_test_init - Prepare a &xe_device for98 * use in a live KUnit test.99 * @test: the &kunit where live &xe_device will be used100 *101 * This function expects pointer to the &xe_device in the &test.param_value,102 * like it is prepared by the &xe_pci_live_device_gen_param and stores that103 * pointer as &kunit.priv to allow the test code to access it.104 *105 * This function makes sure that device is not wedged and then resumes it106 * to avoid waking up the device inside the test. It uses deferred cleanup107 * action to release a runtime_pm reference.108 *109 * This function can be used as custom implementation of &kunit_suite.init.110 *111 * This function uses KUNIT_ASSERT to detect any failures.112 *113 * Return: Always 0.114 */115int xe_kunit_helper_xe_device_live_test_init(struct kunit *test)116{117 struct xe_device *xe = xe_device_const_cast(test->param_value);118 119 KUNIT_ASSERT_NOT_ERR_OR_NULL(test, xe);120 kunit_info(test, "running on %s device\n", xe->info.platform_name);121 122 KUNIT_ASSERT_FALSE(test, xe_device_wedged(xe));123 xe_pm_runtime_get(xe);124 KUNIT_ASSERT_EQ(test, 0, kunit_add_action_or_reset(test, put_xe_pm_runtime, xe));125 126 test->priv = xe;127 return 0;128}129EXPORT_SYMBOL_IF_KUNIT(xe_kunit_helper_xe_device_live_test_init);130