1/* SPDX-License-Identifier: GPL-2.0 AND MIT */ 2/* 3 * Copyright �� 2022 Intel Corporation 4 */ 5 6#ifndef _XE_TEST_H_ 7#define _XE_TEST_H_ 8 9#include <linux/types.h> 10 11#if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST) 12#include <linux/sched.h> 13#include <kunit/test.h> 14 15/* 16 * Each test that provides a kunit private test structure, place a test id 17 * here and point the kunit->priv to an embedded struct xe_test_priv. 18 */ 19enum xe_test_priv_id { 20 XE_TEST_LIVE_DMA_BUF, 21 XE_TEST_LIVE_MIGRATE, 22}; 23 24/** 25 * struct xe_test_priv - Base class for test private info 26 * @id: enum xe_test_priv_id to identify the subclass. 27 */ 28struct xe_test_priv { 29 enum xe_test_priv_id id; 30}; 31 32#define XE_TEST_DECLARE(x) x 33#define XE_TEST_ONLY(x) unlikely(x) 34#define XE_TEST_EXPORT 35#define xe_cur_kunit() current->kunit_test 36 37/** 38 * xe_cur_kunit_priv - Obtain the struct xe_test_priv pointed to by 39 * current->kunit->priv if it exists and is embedded in the expected subclass. 40 * @id: Id of the expected subclass. 41 * 42 * Return: NULL if the process is not a kunit test, and NULL if the 43 * current kunit->priv pointer is not pointing to an object of the expected 44 * subclass. A pointer to the embedded struct xe_test_priv otherwise. 45 */ 46static inline struct xe_test_priv * 47xe_cur_kunit_priv(enum xe_test_priv_id id) 48{ 49 struct xe_test_priv *priv; 50 51 if (!xe_cur_kunit()) 52 return NULL; 53 54 priv = xe_cur_kunit()->priv; 55 return priv->id == id ? priv : NULL; 56} 57 58#else /* if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST) */ 59 60#define XE_TEST_DECLARE(x) 61#define XE_TEST_ONLY(x) 0 62#define XE_TEST_EXPORT static 63#define xe_cur_kunit() NULL 64#define xe_cur_kunit_priv(_id) NULL 65 66#endif 67#endif 68