1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright (C) 2022 Intel Corporation.
3 * All rights reserved.
4 */
5
6 #include "spdk/stdinc.h"
7
8 #include "spdk_internal/cunit.h"
9 #include "common/lib/test_env.c"
10
11 #include "ftl/utils/ftl_mempool.c"
12
13 #define COUNT 16
14 #define ALIGNMENT 64
15 #define SIZE (ALIGNMENT * 2)
16 #define SOCKET_ID_ANY -1
17
18 DEFINE_STUB(ftl_bitmap_create, struct ftl_bitmap *, (void *buf, size_t size), NULL);
19 DEFINE_STUB_V(ftl_bitmap_destroy, (struct ftl_bitmap *bitmap));
20 DEFINE_STUB(ftl_bitmap_get, bool, (const struct ftl_bitmap *bitmap, uint64_t bit), true);
21 DEFINE_STUB_V(ftl_bitmap_set, (struct ftl_bitmap *bitmap, uint64_t bit));
22 DEFINE_STUB_V(ftl_bitmap_clear, (struct ftl_bitmap *bitmap, uint64_t bit));
23
24 static struct ftl_mempool *g_mpool;
25
26 static void
test_ftl_mempool_create(void)27 test_ftl_mempool_create(void)
28 {
29 struct ftl_mempool *mpool;
30
31 /* improper value of alignment */
32 mpool = ftl_mempool_create(COUNT, SIZE, ALIGNMENT + 1, SOCKET_ID_ANY);
33 CU_ASSERT_EQUAL(mpool, NULL);
34 }
35
36 static void
test_ftl_mempool_get_put(void)37 test_ftl_mempool_get_put(void)
38 {
39 void *elem[COUNT];
40 void *elem_empty;
41 void *elem_first = SLIST_FIRST(&g_mpool->list);
42 struct ftl_mempool_element *ftl_elem;
43 int i;
44 for (i = 0; i < COUNT; i++) {
45 elem[i] = ftl_mempool_get(g_mpool);
46 ftl_elem = elem[i];
47 CU_ASSERT_EQUAL(ftl_elem->entry.sle_next, SLIST_FIRST(&g_mpool->list));
48 }
49
50 CU_ASSERT(SLIST_EMPTY(&g_mpool->list));
51
52 elem_empty = ftl_mempool_get(g_mpool);
53 CU_ASSERT_EQUAL(elem_empty, NULL);
54
55 for (i = COUNT - 1; i >= 0; i--) {
56 ftl_mempool_put(g_mpool, elem[i]);
57 CU_ASSERT_EQUAL(SLIST_FIRST(&g_mpool->list), elem[i]);
58 }
59
60 CU_ASSERT_EQUAL(SLIST_FIRST(&g_mpool->list), elem_first);
61 }
62
63 static int
test_setup(void)64 test_setup(void)
65 {
66 g_mpool = ftl_mempool_create(COUNT, SIZE, ALIGNMENT, SOCKET_ID_ANY);
67 if (!g_mpool) {
68 return -ENOMEM;
69 }
70
71 return 0;
72 }
73
74 static int
test_cleanup(void)75 test_cleanup(void)
76 {
77 ftl_mempool_destroy(g_mpool);
78 g_mpool = NULL;
79 return 0;
80 }
81
82 int
main(int argc,char ** argv)83 main(int argc, char **argv)
84 {
85 CU_pSuite suite = NULL;
86 unsigned int num_failures;
87
88 CU_initialize_registry();
89
90 suite = CU_add_suite("ftl_mempool", test_setup, test_cleanup);
91 CU_ADD_TEST(suite, test_ftl_mempool_create);
92 CU_ADD_TEST(suite, test_ftl_mempool_get_put);
93
94 num_failures = spdk_ut_run_tests(argc, argv, NULL);
95 CU_cleanup_registry();
96
97 return num_failures;
98 }
99