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