64 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			64 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* SPDX-License-Identifier: GPL-2.0 AND MIT */
 | |
| /*
 | |
|  * Copyright © 2022 Intel Corporation
 | |
|  */
 | |
| 
 | |
| #ifndef _XE_TEST_H_
 | |
| #define _XE_TEST_H_
 | |
| 
 | |
| #include <linux/types.h>
 | |
| 
 | |
| #if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST)
 | |
| #include <kunit/test.h>
 | |
| #include <kunit/test-bug.h>
 | |
| 
 | |
| /*
 | |
|  * Each test that provides a kunit private test structure, place a test id
 | |
|  * here and point the kunit->priv to an embedded struct xe_test_priv.
 | |
|  */
 | |
| enum xe_test_priv_id {
 | |
| 	XE_TEST_LIVE_DMA_BUF,
 | |
| 	XE_TEST_LIVE_MIGRATE,
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * struct xe_test_priv - Base class for test private info
 | |
|  * @id: enum xe_test_priv_id to identify the subclass.
 | |
|  */
 | |
| struct xe_test_priv {
 | |
| 	enum xe_test_priv_id id;
 | |
| };
 | |
| 
 | |
| #define XE_TEST_DECLARE(x) x
 | |
| #define XE_TEST_ONLY(x) unlikely(x)
 | |
| 
 | |
| /**
 | |
|  * xe_cur_kunit_priv - Obtain the struct xe_test_priv pointed to by
 | |
|  * current->kunit->priv if it exists and is embedded in the expected subclass.
 | |
|  * @id: Id of the expected subclass.
 | |
|  *
 | |
|  * Return: NULL if the process is not a kunit test, and NULL if the
 | |
|  * current kunit->priv pointer is not pointing to an object of the expected
 | |
|  * subclass. A pointer to the embedded struct xe_test_priv otherwise.
 | |
|  */
 | |
| static inline struct xe_test_priv *
 | |
| xe_cur_kunit_priv(enum xe_test_priv_id id)
 | |
| {
 | |
| 	struct xe_test_priv *priv;
 | |
| 
 | |
| 	if (!kunit_get_current_test())
 | |
| 		return NULL;
 | |
| 
 | |
| 	priv = kunit_get_current_test()->priv;
 | |
| 	return priv->id == id ? priv : NULL;
 | |
| }
 | |
| 
 | |
| #else /* if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST) */
 | |
| 
 | |
| #define XE_TEST_DECLARE(x)
 | |
| #define XE_TEST_ONLY(x) 0
 | |
| #define xe_cur_kunit_priv(_id) NULL
 | |
| 
 | |
| #endif
 | |
| #endif
 |