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/list.h"
9 #include "src/common/pack.h"
10
START_TEST(invalid_protocol)11 START_TEST(invalid_protocol)
12 {
13 int rc;
14 uint32_t x;
15
16 slurmdb_event_rec_t *event_rec = xmalloc(sizeof(slurmdb_event_rec_t));
17 Buf buf = init_buf(1024);
18
19 pack32(22, buf);
20 set_buf_offset(buf, 0);
21
22 slurmdb_event_rec_t *acr;
23
24 slurmdb_pack_event_rec((void **)&event_rec, 0, buf);
25 unpack32(&x, buf);
26 rc = slurmdb_unpack_event_rec((void **)&acr, 0, buf);
27 ck_assert_int_eq(rc, SLURM_ERROR);
28 ck_assert(x == 22);
29 free_buf(buf);
30 slurmdb_destroy_event_rec(event_rec);
31 }
32 END_TEST
33
34 // char *cluster; /* Name of associated cluster */
35 // char *cluster_nodes; /* node list in cluster during time
36 // * period (only set in a cluster event) */
37 // uint16_t event_type; /* type of event (slurmdb_event_type_t) */
38 // char *node_name; /* Name of node (only set in a node event) */
39 // time_t period_end; /* End of period */
40 // time_t period_start; /* Start of period */
41 // char *reason; /* reason node is in state during time
42 // period (only set in a node event) */
43 // uint32_t reason_uid; /* uid of that who set the reason */
44 // uint16_t state; /* State of node during time
45 // period (only set in a node event) */
46 // char *tres_str; /* TRES touched by this event */
47
START_TEST(pack_1702_null_event_rec)48 START_TEST(pack_1702_null_event_rec)
49 {
50 int rc;
51 Buf buf = init_buf(1024);
52 slurmdb_event_rec_t pack_er = {0};
53
54 slurmdb_pack_event_rec(NULL, SLURM_MIN_PROTOCOL_VERSION, buf);
55
56 set_buf_offset(buf, 0);
57
58 slurmdb_event_rec_t *unpack_er;
59 rc = slurmdb_unpack_event_rec((void **)&unpack_er, SLURM_MIN_PROTOCOL_VERSION, buf);
60 ck_assert(rc == SLURM_SUCCESS);
61 ck_assert(pack_er.cluster == unpack_er->cluster);
62 ck_assert(pack_er.cluster_nodes == unpack_er->cluster_nodes);
63 ck_assert(pack_er.event_type == unpack_er->event_type);
64 ck_assert(pack_er.node_name == unpack_er->node_name);
65 ck_assert(pack_er.period_end == unpack_er->period_end);
66 ck_assert(pack_er.period_start == unpack_er->period_start);
67 ck_assert(pack_er.reason == unpack_er->reason);
68 ck_assert(NO_VAL == unpack_er->reason_uid);
69 ck_assert(NO_VAL16 == unpack_er->state);
70 ck_assert(pack_er.tres_str == unpack_er->tres_str);
71
72 free_buf(buf);
73 slurmdb_destroy_event_rec(unpack_er);
74 }
75 END_TEST
76
START_TEST(pack_1702_event_rec)77 START_TEST(pack_1702_event_rec)
78 {
79 int rc;
80
81 slurmdb_event_rec_t *pack_er = xmalloc(sizeof(slurmdb_event_rec_t));
82 pack_er->cluster = xstrdup("Joseph Butler");
83 pack_er->cluster_nodes = xstrdup("David Hume");
84 pack_er->event_type = 3;
85 pack_er->node_name = xstrdup("Baruch Spinoza");
86 pack_er->period_end = 0;
87 pack_er->period_start = 10;
88 pack_er->reason = xstrdup("Gottfried Leibniz");
89 pack_er->reason_uid = 66;
90 pack_er->state = 33;
91 pack_er->tres_str = xstrdup("Karl Marx");
92
93 Buf buf = init_buf(1024);
94 slurmdb_pack_event_rec(pack_er, SLURM_MIN_PROTOCOL_VERSION, buf);
95
96 set_buf_offset(buf, 0);
97
98 slurmdb_event_rec_t *unpack_er;
99 rc = slurmdb_unpack_event_rec((void **)&unpack_er, SLURM_MIN_PROTOCOL_VERSION, buf);
100 ck_assert(rc == SLURM_SUCCESS);
101 ck_assert_str_eq(pack_er->cluster, unpack_er->cluster);
102 ck_assert_str_eq(pack_er->cluster_nodes, unpack_er->cluster_nodes);
103 ck_assert(pack_er->event_type == unpack_er->event_type);
104 ck_assert_str_eq(pack_er->node_name, unpack_er->node_name);
105 ck_assert(pack_er->period_end == unpack_er->period_end);
106 ck_assert(pack_er->period_start == unpack_er->period_start);
107 ck_assert_str_eq(pack_er->reason, unpack_er->reason);
108 ck_assert(pack_er->reason_uid == unpack_er->reason_uid);
109 ck_assert(pack_er->state == unpack_er->state);
110 ck_assert_str_eq(pack_er->tres_str, unpack_er->tres_str);
111
112 free_buf(buf);
113 slurmdb_destroy_event_rec(pack_er);
114 slurmdb_destroy_event_rec(unpack_er);
115 }
116 END_TEST
117
118
119 /*****************************************************************************
120 * TEST SUITE *
121 ****************************************************************************/
122
suite(void)123 Suite *suite(void)
124 {
125 Suite *s = suite_create("Pack slurmdb_event_rec_t");
126 TCase *tc_core = tcase_create("Pack slurmdb_event_rec_t");
127 tcase_add_test(tc_core, invalid_protocol);
128 tcase_add_test(tc_core, pack_1702_event_rec);
129 tcase_add_test(tc_core, pack_1702_null_event_rec);
130 suite_add_tcase(s, tc_core);
131 return s;
132 }
133
134 /*****************************************************************************
135 * TEST RUNNER *
136 ****************************************************************************/
137
main(void)138 int main(void)
139 {
140 int number_failed;
141 SRunner *sr = srunner_create(suite());
142
143 //srunner_set_fork_status(sr, CK_NOFORK);
144
145 srunner_run_all(sr, CK_VERBOSE);
146 //srunner_run_all(sr, CK_NORMAL);
147 number_failed = srunner_ntests_failed(sr);
148 srunner_free(sr);
149
150 return (number_failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
151 }
152