xe/tests/xe_guc_id_mgr_test.c

6 #include <kunit/test.h>
11 static int guc_id_mgr_test_init(struct kunit *test)
15 	xe_kunit_helper_xe_device_test_init(test);
16 	idm = &xe_device_get_gt(test->priv, 0)->uc.guc.submission_state.idm;
19 	test->priv = idm;
23 static void bad_init(struct kunit *test)
25 	struct xe_guc_id_mgr *idm = test->priv;
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));
31 static void no_init(struct kunit *test)
33 	struct xe_guc_id_mgr *idm = test->priv;
36 	KUNIT_EXPECT_EQ(test, -ENODATA, xe_guc_id_mgr_reserve_locked(idm, 0));
39 	KUNIT_EXPECT_EQ(test, -ENODATA, xe_guc_id_mgr_reserve(idm, 1, 1));
42 static void init_fini(struct kunit *test)
44 	struct xe_guc_id_mgr *idm = test->priv;
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);
50 	KUNIT_EXPECT_NULL(test, idm->bitmap);
51 	KUNIT_EXPECT_EQ(test, idm->total, 0);
54 static void check_used(struct kunit *test)
56 	struct xe_guc_id_mgr *idm = test->priv;
59 	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, 2));
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);
69 	KUNIT_EXPECT_EQ(test, idm->used, idm->total);
71 	KUNIT_EXPECT_EQ(test, idm->used, 0);
76 static void check_quota(struct kunit *test)
78 	struct xe_guc_id_mgr *idm = test->priv;
81 	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, 2));
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);
92 	KUNIT_EXPECT_LE(test, 0, idm_reserve_chunk_locked(idm, 1, 0));
93 	KUNIT_EXPECT_EQ(test, idm->used, idm->total);
95 	KUNIT_EXPECT_EQ(test, idm->used, 0);
100 static void check_all(struct kunit *test)
102 	struct xe_guc_id_mgr *idm = test->priv;
105 	KUNIT_ASSERT_EQ(test, 0, xe_guc_id_mgr_init(idm, -1));
110 		KUNIT_EXPECT_LE(test, 0, idm_reserve_chunk_locked(idm, 1, 0));
111 	KUNIT_EXPECT_EQ(test, idm->used, idm->total);