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 <kunit/test.h> 13 #include <kunit/test-bug.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 */ 19 enum 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 */ 28 struct 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 35 /** 36 * xe_cur_kunit_priv - Obtain the struct xe_test_priv pointed to by 37 * current->kunit->priv if it exists and is embedded in the expected subclass. 38 * @id: Id of the expected subclass. 39 * 40 * Return: NULL if the process is not a kunit test, and NULL if the 41 * current kunit->priv pointer is not pointing to an object of the expected 42 * subclass. A pointer to the embedded struct xe_test_priv otherwise. 43 */ 44 static inline struct xe_test_priv * 45 xe_cur_kunit_priv(enum xe_test_priv_id id) 46 { 47 struct xe_test_priv *priv; 48 49 if (!kunit_get_current_test()) 50 return NULL; 51 52 priv = kunit_get_current_test()->priv; 53 return priv->id == id ? priv : NULL; 54 } 55 56 #else /* if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST) */ 57 58 #define XE_TEST_DECLARE(x) 59 #define XE_TEST_ONLY(x) 0 60 #define xe_cur_kunit_priv(_id) NULL 61 62 #endif 63 #endif 64