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