1 #include <check.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4
5 #include "src/common/slurmdb_pack.h"
6 #include "src/common/xmalloc.h"
7 #include "src/common/slurm_protocol_common.h"
8 #include "src/common/slurm_protocol_defs.h"
9 #include "src/common/list.h"
10 #include "src/common/pack.h"
11
12 #ifndef NDEBUG
START_TEST(pack_null_usage)13 START_TEST(pack_null_usage)
14 {
15 int rc;
16 uint32_t x;
17
18 slurmdb_assoc_rec_t *assoc_rec = xmalloc(sizeof(slurmdb_assoc_rec_t));
19 assoc_rec->usage = NULL;
20
21 Buf buf = init_buf(1024);
22
23 pack32(22, buf);
24 set_buf_offset(buf, 0);
25
26 slurmdb_assoc_rec_t *acr;
27
28 /* Should assert */
29 slurmdb_pack_assoc_rec_with_usage((void *)assoc_rec, 0, buf);
30 unpack32(&x, buf);
31 rc = slurmdb_unpack_assoc_rec_with_usage((void **)&acr, 0, buf);
32 ck_assert_int_eq(rc, SLURM_ERROR);
33 ck_assert(x == 22);
34 free_buf(buf);
35 slurmdb_destroy_assoc_rec(assoc_rec);
36 }
37 END_TEST
38 #endif
39
START_TEST(invalid_protocol)40 START_TEST(invalid_protocol)
41 {
42 int rc;
43 uint32_t x;
44
45 slurmdb_assoc_rec_t *assoc_rec = xmalloc(sizeof(slurmdb_assoc_rec_t));
46 assoc_rec->usage = xmalloc(sizeof(slurmdb_assoc_usage_t));
47
48 Buf buf = init_buf(1024);
49
50 pack32(22, buf);
51 set_buf_offset(buf, 0);
52
53 slurmdb_assoc_rec_t *acr;
54
55 slurmdb_pack_assoc_rec_with_usage((void *)assoc_rec, 0, buf);
56 unpack32(&x, buf);
57 rc = slurmdb_unpack_assoc_rec_with_usage((void **)&acr, 0, buf);
58 ck_assert_int_eq(rc, SLURM_ERROR);
59 ck_assert(x == 22);
60 free_buf(buf);
61 slurmdb_destroy_assoc_rec(assoc_rec);
62 }
63 END_TEST
64
65
START_TEST(pack_1702_assoc_rec)66 START_TEST(pack_1702_assoc_rec)
67 {
68 int rc;
69 Buf buf = init_buf(1024);
70 slurmdb_assoc_rec_t *pack_arwu = xmalloc(sizeof(slurmdb_assoc_rec_t));
71 slurmdb_accounting_rec_t *art = xmalloc(sizeof(slurmdb_accounting_rec_t));
72
73 art->id = 66;
74 char *qos = xstrdup("Eusebius");
75
76 pack_arwu->accounting_list = list_create(slurmdb_destroy_accounting_rec);
77 pack_arwu->acct = xstrdup("Socrates");
78 pack_arwu->assoc_next = xmalloc(sizeof(slurmdb_assoc_rec_t));
79 pack_arwu->assoc_next->rgt = 11;
80 pack_arwu->assoc_next->grp_jobs = 22;
81 pack_arwu->assoc_next_id = xmalloc(sizeof(slurmdb_assoc_rec_t));
82 pack_arwu->assoc_next_id->grp_jobs = 33;
83 pack_arwu->cluster = xstrdup("Parmenides");
84 pack_arwu->def_qos_id = 1;
85 pack_arwu->grp_jobs = 2;
86 pack_arwu->grp_submit_jobs = 3;
87 pack_arwu->grp_tres = xstrdup("Parmenides");
88 pack_arwu->grp_tres_ctld = NULL;
89 pack_arwu->grp_tres_mins = xstrdup("Parmenides");
90 pack_arwu->grp_tres_run_mins = xstrdup("Parmenides");
91 pack_arwu->grp_tres_run_mins_ctld = NULL;
92 pack_arwu->grp_wall = 6;
93 pack_arwu->id = 7;
94 pack_arwu->is_def = 8;
95 pack_arwu->lft = 9;
96 pack_arwu->max_jobs = 1;
97 pack_arwu->max_submit_jobs = 2;
98 pack_arwu->max_tres_mins_pj = xstrdup("Parmenides");
99 pack_arwu->max_tres_mins_ctld = NULL;
100 pack_arwu->max_tres_run_mins = xstrdup("Parmenides");
101 pack_arwu->max_tres_run_mins_ctld = NULL;
102 pack_arwu->max_tres_pj = xstrdup("Parmenides");
103 pack_arwu->max_tres_ctld = NULL;
104 pack_arwu->max_tres_pn = xstrdup("Parmenides");
105 pack_arwu->max_tres_pn_ctld = NULL;
106 pack_arwu->max_wall_pj = 7;
107 pack_arwu->parent_acct = xstrdup("Parmenides");
108 pack_arwu->parent_id = 8;
109 pack_arwu->partition = xstrdup("Parmenides");
110 pack_arwu->qos_list = list_create(xfree_ptr);
111 pack_arwu->rgt = 9;
112 pack_arwu->shares_raw = 1;
113 pack_arwu->uid = 2;
114 pack_arwu->usage = xmalloc(sizeof(slurmdb_assoc_usage_t));
115 pack_arwu->user = xstrdup("Parmenides");
116
117
118 pack_arwu->usage->children_list = NULL;
119 pack_arwu->usage->grp_used_tres = NULL;
120 pack_arwu->usage->grp_used_tres_run_secs = NULL;
121 pack_arwu->usage->grp_used_wall = 77;
122 pack_arwu->usage->fs_factor = 0;
123 pack_arwu->usage->level_shares = 0;
124 pack_arwu->usage->parent_assoc_ptr = NULL;
125 pack_arwu->usage->fs_assoc_ptr = NULL;
126 pack_arwu->usage->shares_norm = 0;
127 pack_arwu->usage->tres_cnt = 0;
128 pack_arwu->usage->usage_efctv = 123123;
129 pack_arwu->usage->usage_norm = 4857;
130 pack_arwu->usage->usage_raw = 4747;
131 pack_arwu->usage->usage_tres_raw = NULL;
132 pack_arwu->usage->used_jobs = 234;
133 pack_arwu->usage->used_submit_jobs = 433;
134 pack_arwu->usage->level_fs = 3333;
135 pack_arwu->usage->valid_qos = NULL;
136
137
138 list_append(pack_arwu->accounting_list, (void *)art);
139 list_append(pack_arwu->qos_list , (void *)qos);
140
141
142 slurmdb_pack_assoc_rec_with_usage(pack_arwu, SLURM_MIN_PROTOCOL_VERSION, buf);
143 set_buf_offset(buf, 0);
144 slurmdb_assoc_rec_t *unpack_arwu;
145 rc = slurmdb_unpack_assoc_rec_with_usage((void **)&unpack_arwu, SLURM_MIN_PROTOCOL_VERSION, buf);
146 ck_assert(rc == SLURM_SUCCESS);
147 ck_assert_str_eq(pack_arwu->acct, unpack_arwu->acct);
148 ck_assert(NULL == unpack_arwu->assoc_next);
149 ck_assert(NULL == unpack_arwu->assoc_next_id);
150 ck_assert_str_eq(pack_arwu->cluster, unpack_arwu->cluster);
151 ck_assert(pack_arwu->def_qos_id == unpack_arwu->def_qos_id);
152 ck_assert(pack_arwu->grp_jobs == unpack_arwu->grp_jobs);
153 ck_assert(pack_arwu->grp_submit_jobs == unpack_arwu->grp_submit_jobs);
154 ck_assert_str_eq(pack_arwu->grp_tres, unpack_arwu->grp_tres);
155 ck_assert(pack_arwu->grp_tres_ctld == unpack_arwu->grp_tres_ctld);
156 ck_assert_str_eq(pack_arwu->grp_tres_run_mins, unpack_arwu->grp_tres_run_mins);
157 ck_assert(pack_arwu->grp_wall == unpack_arwu->grp_wall);
158 ck_assert(pack_arwu->id == unpack_arwu->id);
159 ck_assert(pack_arwu->is_def == unpack_arwu->is_def);
160 ck_assert(pack_arwu->lft == unpack_arwu->lft);
161 ck_assert(pack_arwu->max_jobs == unpack_arwu->max_jobs);
162 ck_assert(pack_arwu->max_submit_jobs == unpack_arwu->max_submit_jobs);
163 ck_assert_str_eq(pack_arwu->max_tres_mins_pj, unpack_arwu->max_tres_mins_pj);
164 ck_assert(pack_arwu->max_tres_mins_ctld == unpack_arwu->max_tres_mins_ctld);
165 ck_assert_str_eq(pack_arwu->max_tres_run_mins, unpack_arwu->max_tres_run_mins);
166 ck_assert(pack_arwu->max_tres_run_mins_ctld == unpack_arwu->max_tres_run_mins_ctld);
167 ck_assert_str_eq(pack_arwu->max_tres_pj, unpack_arwu->max_tres_pj);
168 ck_assert(pack_arwu->max_tres_ctld == unpack_arwu->max_tres_ctld);
169 ck_assert_str_eq(pack_arwu->max_tres_pn, unpack_arwu->max_tres_pn);
170 ck_assert(pack_arwu->max_tres_pn_ctld == unpack_arwu->max_tres_pn_ctld);
171 ck_assert(pack_arwu->max_wall_pj == unpack_arwu->max_wall_pj);
172 ck_assert_str_eq(pack_arwu->parent_acct, unpack_arwu->parent_acct);
173 ck_assert(pack_arwu->parent_id == unpack_arwu->parent_id);
174 ck_assert_str_eq(pack_arwu->partition, unpack_arwu->partition);
175 ck_assert(pack_arwu->rgt == unpack_arwu->rgt);
176 ck_assert(pack_arwu->shares_raw == unpack_arwu->shares_raw);
177 ck_assert(pack_arwu->uid == unpack_arwu->uid);
178 ck_assert_str_eq(pack_arwu->user, unpack_arwu->user);
179
180 slurmdb_accounting_rec_t *b = (slurmdb_accounting_rec_t *)list_peek(pack_arwu->accounting_list);
181 slurmdb_accounting_rec_t *a = (slurmdb_accounting_rec_t *)list_peek(pack_arwu->accounting_list);
182
183 char *before = (char *)list_peek(pack_arwu->qos_list);
184 char *after = (char *)list_peek(unpack_arwu->qos_list);
185
186 ck_assert(b->id == a->id);
187
188 ck_assert_str_eq(before, after);
189
190 ck_assert(pack_arwu->usage->children_list == unpack_arwu->usage->children_list);
191 ck_assert(pack_arwu->usage->grp_used_tres == unpack_arwu->usage->grp_used_tres);
192 ck_assert(pack_arwu->usage->grp_used_tres_run_secs == unpack_arwu->usage->grp_used_tres_run_secs);
193 ck_assert(pack_arwu->usage->grp_used_wall == unpack_arwu->usage->grp_used_wall);
194 ck_assert(pack_arwu->usage->fs_factor == unpack_arwu->usage->fs_factor);
195 ck_assert(pack_arwu->usage->level_shares == unpack_arwu->usage->level_shares);
196 ck_assert(pack_arwu->usage->parent_assoc_ptr == unpack_arwu->usage->parent_assoc_ptr);
197 ck_assert(pack_arwu->usage->fs_assoc_ptr == unpack_arwu->usage->fs_assoc_ptr);
198 ck_assert(pack_arwu->usage->shares_norm == unpack_arwu->usage->shares_norm);
199 ck_assert(pack_arwu->usage->tres_cnt == unpack_arwu->usage->tres_cnt);
200 ck_assert(pack_arwu->usage->usage_efctv == unpack_arwu->usage->usage_efctv);
201 ck_assert(pack_arwu->usage->usage_norm == unpack_arwu->usage->usage_norm);
202 ck_assert(pack_arwu->usage->usage_raw == unpack_arwu->usage->usage_raw);
203 ck_assert(pack_arwu->usage->usage_tres_raw == unpack_arwu->usage->usage_tres_raw);
204 ck_assert(pack_arwu->usage->used_jobs == unpack_arwu->usage->used_jobs);
205 ck_assert(pack_arwu->usage->used_submit_jobs == unpack_arwu->usage->used_submit_jobs);
206 ck_assert(pack_arwu->usage->level_fs == unpack_arwu->usage->level_fs);
207 ck_assert(pack_arwu->usage->valid_qos == unpack_arwu->usage->valid_qos);
208
209
210
211 free_buf(buf);
212 xfree(pack_arwu->assoc_next);
213 xfree(pack_arwu->assoc_next_id);
214 xfree(pack_arwu->usage);
215 slurmdb_destroy_assoc_rec(pack_arwu);
216 slurmdb_destroy_assoc_rec(unpack_arwu);
217 }
218 END_TEST
219
220 /*****************************************************************************
221 * TEST SUITE *
222 ****************************************************************************/
223
suite(SRunner * sr)224 Suite *suite(SRunner *sr)
225 {
226 Suite *s = suite_create("Pack slurmdb_assoc_rec_t");
227 TCase *tc_core = tcase_create("Pack slurmdb_assoc_rec_t");
228 tcase_add_test(tc_core, invalid_protocol);
229
230 #ifndef NDEBUG
231 if (srunner_fork_status(sr) != CK_NOFORK)
232 tcase_add_test_raise_signal(tc_core, pack_null_usage, SIGABRT);
233 #endif
234
235 tcase_add_test(tc_core, pack_1702_assoc_rec);
236 suite_add_tcase(s, tc_core);
237 return s;
238 }
239
240 /*****************************************************************************
241 * TEST RUNNER *
242 ****************************************************************************/
243
main(void)244 int main(void)
245 {
246 int number_failed;
247 SRunner *sr = srunner_create(NULL);
248 //srunner_set_fork_status(sr, CK_NOFORK);
249 srunner_add_suite(sr, suite(sr));
250
251 srunner_run_all(sr, CK_VERBOSE);
252 //srunner_run_all(sr, CK_NORMAL);
253 number_failed = srunner_ntests_failed(sr);
254 srunner_free(sr);
255
256 return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
257 }
258