137 lines · c
1// SPDX-License-Identifier: GPL-2.0 AND MIT2/*3 * Copyright © 2024 Intel Corporation4 */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);137