xe_guc_id_mgr_test.c

  1// SPDX-License-Identifier: GPL-2.0 AND MIT
  2/*
  3 * Copyright © 2024 Intel Corporation
  4 */
  5
  6#include <kunit/test.h>
  7
  8#include "xe_device.h"
  9#include "xe_kunit_helpers.h"
 10
 11static int guc_id_mgr_test_init(struct kunit *test)
 12{
 13	struct xe_guc_id_mgr *idm;
 14
 15	xe_kunit_helper_xe_device_test_init(test);
 16	idm = &xe_device_get_gt(test->priv, 0)->uc.guc.submission_state.idm;
 17
 18	mutex_init(idm_mutex(idm));
 19	test->priv = idm;
 20	return 0;
 21}
 22
 23static void bad_init(struct kunit *test)
 24{
 25	struct xe_guc_id_mgr *idm = test->priv;
 26
 27	KUNIT_EXPECT_EQ(test, -EINVAL, xe_guc_id_mgr_init(idm, 0));
 28	KUNIT_EXPECT_EQ(test, -ERANGE, xe_guc_id_mgr_init(idm, GUC_ID_MAX + 1));
 29}
 30
 31static void no_init(struct kunit *test)
 32{
 33	struct xe_guc_id_mgr *idm = test->priv;
 34
 35	mutex_lock(idm_mutex(idm));
 36	KUNIT_EXPECT_EQ(test, -ENODATA, xe_guc_id_mgr_reserve_locked(idm, 0));
 37	mutex_unlock(idm_mutex(idm));
 38
 39	KUNIT_EXPECT_EQ(test, -ENODATA, xe_guc_id_mgr_reserve(idm, 1, 1));
 40}
 41
 42static void init_fini(struct kunit *test)
 43{
 44	struct xe_guc_id_mgr *idm = test->priv;
 45
 46	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, -1));
 47	KUNIT_EXPECT_NOT_NULL(test, idm->bitmap);
 48	KUNIT_EXPECT_EQ(test, idm->total, GUC_ID_MAX);
 49	__fini_idm(NULL, idm);
 50	KUNIT_EXPECT_NULL(test, idm->bitmap);
 51	KUNIT_EXPECT_EQ(test, idm->total, 0);
 52}
 53
 54static void check_used(struct kunit *test)
 55{
 56	struct xe_guc_id_mgr *idm = test->priv;
 57	unsigned int n;
 58
 59	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, 2));
 60
 61	mutex_lock(idm_mutex(idm));
 62
 63	for (n = 0; n < idm->total; n++) {
 64		kunit_info(test, "n=%u", n);
 65		KUNIT_EXPECT_EQ(test, idm->used, n);
 66		KUNIT_EXPECT_GE(test, idm_reserve_chunk_locked(idm, 1, 0), 0);
 67		KUNIT_EXPECT_EQ(test, idm->used, n + 1);
 68	}
 69	KUNIT_EXPECT_EQ(test, idm->used, idm->total);
 70	idm_release_chunk_locked(idm, 0, idm->used);
 71	KUNIT_EXPECT_EQ(test, idm->used, 0);
 72
 73	mutex_unlock(idm_mutex(idm));
 74}
 75
 76static void check_quota(struct kunit *test)
 77{
 78	struct xe_guc_id_mgr *idm = test->priv;
 79	unsigned int n;
 80
 81	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, 2));
 82
 83	mutex_lock(idm_mutex(idm));
 84
 85	for (n = 0; n < idm->total - 1; n++) {
 86		kunit_info(test, "n=%u", n);
 87		KUNIT_EXPECT_EQ(test, idm_reserve_chunk_locked(idm, 1, idm->total), -EDQUOT);
 88		KUNIT_EXPECT_EQ(test, idm_reserve_chunk_locked(idm, 1, idm->total - n), -EDQUOT);
 89		KUNIT_EXPECT_EQ(test, idm_reserve_chunk_locked(idm, idm->total - n, 1), -EDQUOT);
 90		KUNIT_EXPECT_GE(test, idm_reserve_chunk_locked(idm, 1, 1), 0);
 91	}
 92	KUNIT_EXPECT_LE(test, 0, idm_reserve_chunk_locked(idm, 1, 0));
 93	KUNIT_EXPECT_EQ(test, idm->used, idm->total);
 94	idm_release_chunk_locked(idm, 0, idm->total);
 95	KUNIT_EXPECT_EQ(test, idm->used, 0);
 96
 97	mutex_unlock(idm_mutex(idm));
 98}
 99
100static void check_all(struct kunit *test)
101{
102	struct xe_guc_id_mgr *idm = test->priv;
103	unsigned int n;
104
105	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, -1));
106
107	mutex_lock(idm_mutex(idm));
108
109	for (n = 0; n < idm->total; n++)
110		KUNIT_EXPECT_LE(test, 0, idm_reserve_chunk_locked(idm, 1, 0));
111	KUNIT_EXPECT_EQ(test, idm->used, idm->total);
112	for (n = 0; n < idm->total; n++)
113		idm_release_chunk_locked(idm, n, 1);
114
115	mutex_unlock(idm_mutex(idm));
116}
117
118static struct kunit_case guc_id_mgr_test_cases[] = {
119	KUNIT_CASE(bad_init),
120	KUNIT_CASE(no_init),
121	KUNIT_CASE(init_fini),
122	KUNIT_CASE(check_used),
123	KUNIT_CASE(check_quota),
124	KUNIT_CASE_SLOW(check_all),
125	{}
126};
127
128static struct kunit_suite guc_id_mgr_suite = {
129	.name = "guc_idm",
130	.test_cases = guc_id_mgr_test_cases,
131
132	.init = guc_id_mgr_test_init,
133	.exit = NULL,
134};
135
136kunit_test_suites(&guc_id_mgr_suite);