ctdb-protocol: Fix marshalling for ctdb_notify_data
[vlendec/samba-autobuild/.git] / ctdb / tests / src / protocol_types_test.c
1 /*
2    protocol types tests
3
4    Copyright (C) Amitay Isaacs  2015
5
6    This program is free software; you can redistribute it and/or modify
7    it under the terms of the GNU General Public License as published by
8    the Free Software Foundation; either version 3 of the License, or
9    (at your option) any later version.
10
11    This program is distributed in the hope that it will be useful,
12    but WITHOUT ANY WARRANTY; without even the implied warranty of
13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14    GNU General Public License for more details.
15
16    You should have received a copy of the GNU General Public License
17    along with this program; if not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include "replace.h"
21 #include "system/filesys.h"
22
23 #include <assert.h>
24
25 #include "protocol/protocol_basic.c"
26 #include "protocol/protocol_types.c"
27
28 #include "tests/src/protocol_common.h"
29
30 PROTOCOL_TYPE2_TEST(TDB_DATA, ctdb_tdb_data);
31 PROTOCOL_TYPE2_TEST(TDB_DATA, ctdb_tdb_datan);
32 PROTOCOL_TYPE1_TEST(struct ctdb_latency_counter, ctdb_latency_counter);
33
34 static void test_ctdb_g_lock(void)
35 {
36         TALLOC_CTX *mem_ctx = talloc_new(NULL);
37         struct ctdb_g_lock p1, p2;
38         size_t buflen;
39         int ret;
40
41         fill_ctdb_g_lock(mem_ctx, &p1);
42         buflen = ctdb_g_lock_len(&p1);
43         ctdb_g_lock_push(&p1, BUFFER);
44         ret = ctdb_g_lock_pull(BUFFER, buflen, &p2);
45         assert(ret == 0);
46         verify_ctdb_g_lock(&p1, &p2);
47         talloc_free(mem_ctx);
48 }
49
50 PROTOCOL_TYPE3_TEST(struct ctdb_statistics, ctdb_statistics);
51 PROTOCOL_TYPE3_TEST(struct ctdb_vnn_map, ctdb_vnn_map);
52 PROTOCOL_TYPE3_TEST(struct ctdb_dbid, ctdb_dbid);
53 PROTOCOL_TYPE3_TEST(struct ctdb_dbid_map, ctdb_dbid_map);
54 PROTOCOL_TYPE3_TEST(struct ctdb_pulldb, ctdb_pulldb);
55 PROTOCOL_TYPE3_TEST(struct ctdb_pulldb_ext, ctdb_pulldb_ext);
56 PROTOCOL_TYPE1_TEST(struct ctdb_ltdb_header, ctdb_ltdb_header);
57 PROTOCOL_TYPE3_TEST(struct ctdb_rec_data, ctdb_rec_data);
58 PROTOCOL_TYPE3_TEST(struct ctdb_rec_buffer, ctdb_rec_buffer);
59 PROTOCOL_TYPE3_TEST(struct ctdb_traverse_start, ctdb_traverse_start);
60 PROTOCOL_TYPE3_TEST(struct ctdb_traverse_all, ctdb_traverse_all);
61 PROTOCOL_TYPE3_TEST(struct ctdb_traverse_start_ext, ctdb_traverse_start_ext);
62 PROTOCOL_TYPE3_TEST(struct ctdb_traverse_all_ext, ctdb_traverse_all_ext);
63 PROTOCOL_TYPE3_TEST(ctdb_sock_addr, ctdb_sock_addr);
64 PROTOCOL_TYPE3_TEST(struct ctdb_connection, ctdb_connection);
65 PROTOCOL_TYPE3_TEST(struct ctdb_tunable, ctdb_tunable);
66 PROTOCOL_TYPE3_TEST(struct ctdb_node_flag_change, ctdb_node_flag_change);
67 PROTOCOL_TYPE3_TEST(struct ctdb_var_list, ctdb_var_list);
68 PROTOCOL_TYPE3_TEST(struct ctdb_tunable_list, ctdb_tunable_list);
69 PROTOCOL_TYPE3_TEST(struct ctdb_tickle_list, ctdb_tickle_list);
70 PROTOCOL_TYPE3_TEST(struct ctdb_addr_info, ctdb_addr_info);
71 PROTOCOL_TYPE3_TEST(struct ctdb_transdb, ctdb_transdb);
72 PROTOCOL_TYPE3_TEST(struct ctdb_uptime, ctdb_uptime);
73 PROTOCOL_TYPE3_TEST(struct ctdb_public_ip, ctdb_public_ip);
74 PROTOCOL_TYPE3_TEST(struct ctdb_public_ip_list, ctdb_public_ip_list);
75 PROTOCOL_TYPE3_TEST(struct ctdb_node_and_flags, ctdb_node_and_flags);
76 PROTOCOL_TYPE3_TEST(struct ctdb_node_map, ctdb_node_map);
77 PROTOCOL_TYPE3_TEST(struct ctdb_script, ctdb_script);
78 PROTOCOL_TYPE3_TEST(struct ctdb_script_list, ctdb_script_list);
79 PROTOCOL_TYPE3_TEST(struct ctdb_ban_state, ctdb_ban_state);
80 PROTOCOL_TYPE3_TEST(struct ctdb_notify_data, ctdb_notify_data);
81 DEFINE_TEST(struct ctdb_iface, ctdb_iface);
82 DEFINE_TEST(struct ctdb_iface_list, ctdb_iface_list);
83 DEFINE_TEST(struct ctdb_public_ip_info, ctdb_public_ip_info);
84 DEFINE_TEST(struct ctdb_statistics_list, ctdb_statistics_list);
85 DEFINE_TEST(struct ctdb_key_data, ctdb_key_data);
86 DEFINE_TEST(struct ctdb_db_statistics, ctdb_db_statistics);
87 DEFINE_TEST(struct ctdb_election_message, ctdb_election_message);
88 DEFINE_TEST(struct ctdb_srvid_message, ctdb_srvid_message);
89 DEFINE_TEST(struct ctdb_disable_message, ctdb_disable_message);
90 DEFINE_TEST(struct ctdb_g_lock_list, ctdb_g_lock_list);
91
92 static void test_ctdb_rec_buffer_read_write(void)
93 {
94         TALLOC_CTX *mem_ctx = talloc_new(NULL);
95         struct ctdb_rec_buffer *p1, **p2;
96         const char *filename = "ctdb_rec_buffer_test.dat";
97         int count = 100;
98         int fd, i, ret;
99         off_t offset;
100
101         p1 = talloc_array(mem_ctx, struct ctdb_rec_buffer, count);
102         assert(p1 != NULL);
103         for (i=0; i<count; i++) {
104                 fill_ctdb_rec_buffer(mem_ctx, &p1[i]);
105         }
106
107         fd = open(filename, O_RDWR|O_CREAT, 0600);
108         assert(fd != -1);
109         unlink(filename);
110
111         for (i=0; i<count; i++) {
112                 ret = ctdb_rec_buffer_write(&p1[i], fd);
113                 assert(ret == 0);
114         }
115
116         offset = lseek(fd, 0, SEEK_CUR);
117         assert(offset != -1);
118         offset = lseek(fd, -offset, SEEK_CUR);
119         assert(offset == 0);
120
121         p2 = talloc_array(mem_ctx, struct ctdb_rec_buffer *, count);
122         assert(p2 != NULL);
123
124         for (i=0; i<count; i++) {
125                 ret = ctdb_rec_buffer_read(fd, mem_ctx, &p2[i]);
126                 assert(ret == 0);
127         }
128
129         close(fd);
130
131         for (i=0; i<count; i++) {
132                 verify_ctdb_rec_buffer(&p1[i], p2[i]);
133         }
134
135         talloc_free(mem_ctx);
136 }
137
138 int main(int argc, char *argv[])
139 {
140         if (argc == 2) {
141                 int seed = atoi(argv[1]);
142                 srandom(seed);
143         }
144
145         TEST_FUNC(ctdb_tdb_data)();
146         TEST_FUNC(ctdb_tdb_datan)();
147         TEST_FUNC(ctdb_latency_counter)();
148         test_ctdb_g_lock();
149
150         TEST_FUNC(ctdb_statistics)();
151         TEST_FUNC(ctdb_vnn_map)();
152         TEST_FUNC(ctdb_dbid)();
153         TEST_FUNC(ctdb_dbid_map)();
154         TEST_FUNC(ctdb_pulldb)();
155         TEST_FUNC(ctdb_pulldb_ext)();
156         TEST_FUNC(ctdb_ltdb_header)();
157         TEST_FUNC(ctdb_rec_data)();
158         TEST_FUNC(ctdb_rec_buffer)();
159         TEST_FUNC(ctdb_traverse_start)();
160         TEST_FUNC(ctdb_traverse_all)();
161         TEST_FUNC(ctdb_traverse_start_ext)();
162         TEST_FUNC(ctdb_traverse_all_ext)();
163         TEST_FUNC(ctdb_sock_addr)();
164         TEST_FUNC(ctdb_connection)();
165         TEST_FUNC(ctdb_tunable)();
166         TEST_FUNC(ctdb_node_flag_change)();
167         TEST_FUNC(ctdb_var_list)();
168         TEST_FUNC(ctdb_tunable_list)();
169         TEST_FUNC(ctdb_tickle_list)();
170         TEST_FUNC(ctdb_addr_info)();
171         TEST_FUNC(ctdb_transdb)();
172         TEST_FUNC(ctdb_uptime)();
173         TEST_FUNC(ctdb_public_ip)();
174         TEST_FUNC(ctdb_public_ip_list)();
175         TEST_FUNC(ctdb_node_and_flags)();
176         TEST_FUNC(ctdb_node_map)();
177         TEST_FUNC(ctdb_script)();
178         TEST_FUNC(ctdb_script_list)();
179         TEST_FUNC(ctdb_ban_state)();
180         TEST_FUNC(ctdb_notify_data)();
181         TEST_FUNC(ctdb_iface)();
182         TEST_FUNC(ctdb_iface_list)();
183         TEST_FUNC(ctdb_public_ip_info)();
184         TEST_FUNC(ctdb_statistics_list)();
185         TEST_FUNC(ctdb_key_data)();
186         TEST_FUNC(ctdb_db_statistics)();
187         TEST_FUNC(ctdb_election_message)();
188         TEST_FUNC(ctdb_srvid_message)();
189         TEST_FUNC(ctdb_disable_message)();
190         TEST_FUNC(ctdb_g_lock_list)();
191
192         test_ctdb_rec_buffer_read_write();
193
194         return 0;
195 }