linux/drivers/gpu/drm/xe/tests/xe_test.h

/* 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 <linux/sched.h>
#include <kunit/test.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 { … };

/**
 * struct xe_test_priv - Base class for test private info
 * @id: enum xe_test_priv_id to identify the subclass.
 */
struct xe_test_priv { … };

#define XE_TEST_DECLARE(x) …
#define XE_TEST_ONLY(x) …
#define XE_TEST_EXPORT
#define xe_cur_kunit() …

/**
 * 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)
{ … }

#else /* if IS_ENABLED(CONFIG_DRM_XE_KUNIT_TEST) */

#define XE_TEST_DECLARE …
#define XE_TEST_ONLY …
#define XE_TEST_EXPORT …
#define xe_cur_kunit …
#define xe_cur_kunit_priv …

#endif
#endif