Merge Mark's test code.
[awesomized/libmemcached] / tests / test.c
1 /*
2 Sample test application.
3 */
4 #include <assert.h>
5 #include <memcached.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9
10 void init_test(void)
11 {
12 memcached_st memc;
13
14 (void)memcached_init(&memc);
15 memcached_deinit(&memc);
16 }
17
18 void allocation_test(void)
19 {
20 memcached_st *memc;
21 memc= memcached_init(NULL);
22 assert(memc);
23 memcached_deinit(memc);
24 }
25
26 void connection_test(void)
27 {
28 memcached_return rc;
29 memcached_st *memc;
30 memc= memcached_init(NULL);
31 rc= memcached_server_add(memc, "localhost", 0);
32 assert(rc == MEMCACHED_SUCCESS);
33 assert(memc);
34 memcached_deinit(memc);
35 }
36
37 void set_test(void)
38 {
39 memcached_st *memc;
40 memcached_return rc;
41 char *key= "foo";
42 char *value= "when we sanitize";
43
44 memc= memcached_init(NULL);
45 assert(memc);
46 rc= memcached_set(memc, key, strlen(key),
47 value, strlen(value),
48 (time_t)0, (uint16_t)0);
49 assert(rc == MEMCACHED_SUCCESS);
50
51 memcached_deinit(memc);
52 }
53
54 void add_test(void)
55 {
56 memcached_st *memc;
57 memcached_return rc;
58 char *key= "foo";
59 char *value= "when we sanitize";
60
61 memc= memcached_init(NULL);
62 assert(memc);
63 rc= memcached_add(memc, key, strlen(key),
64 value, strlen(value),
65 (time_t)0, (uint16_t)0);
66 assert(rc == MEMCACHED_NOTSTORED);
67
68 memcached_deinit(memc);
69 }
70
71 void replace_test(void)
72 {
73 memcached_st *memc;
74 memcached_return rc;
75 char *key= "foo";
76 char *value= "when we sanitize";
77
78 memc= memcached_init(NULL);
79 assert(memc);
80 rc= memcached_replace(memc, key, strlen(key),
81 value, strlen(value),
82 (time_t)0, (uint16_t)0);
83 assert(rc == MEMCACHED_SUCCESS);
84
85 memcached_deinit(memc);
86 }
87
88 void delete_test(void)
89 {
90 memcached_st *memc;
91 memcached_return rc;
92 char *key= "foo";
93 char *value= "when we sanitize";
94
95 memc= memcached_init(NULL);
96 assert(memc);
97 rc= memcached_set(memc, key, strlen(key),
98 value, strlen(value),
99 (time_t)0, (uint16_t)0);
100 assert(rc == MEMCACHED_SUCCESS);
101
102 rc= memcached_delete(memc, key, strlen(key), (time_t)0);
103 assert(rc == MEMCACHED_SUCCESS);
104
105 memcached_deinit(memc);
106 }
107
108 void flush_test(void)
109 {
110 memcached_st *memc;
111 memcached_return rc;
112
113 memc= memcached_init(NULL);
114 assert(memc);
115 rc= memcached_flush(memc, 0);
116 assert(rc == MEMCACHED_SUCCESS);
117
118 memcached_deinit(memc);
119 }
120
121 void get_test(void)
122 {
123 memcached_st *memc;
124 memcached_return rc;
125 char *key= "foo";
126 char *string;
127 size_t string_length;
128 uint16_t flags;
129
130 memc= memcached_init(NULL);
131 assert(memc);
132
133 string= memcached_get(memc, key, strlen(key),
134 &string_length, &flags, &rc);
135
136 assert(rc == MEMCACHED_NOTFOUND);
137 assert(string_length == 0);
138 assert(!string);
139
140 memcached_deinit(memc);
141 }
142
143 void get_test2(void)
144 {
145 memcached_st *memc;
146 memcached_return rc;
147 char *key= "foo";
148 char *value= "when we sanitize";
149 char *string;
150 size_t string_length;
151 uint16_t flags;
152
153 memc= memcached_init(NULL);
154 assert(memc);
155 rc= memcached_set(memc, key, strlen(key),
156 value, strlen(value),
157 (time_t)0, (uint16_t)0);
158 assert(rc == MEMCACHED_SUCCESS);
159
160 string= memcached_get(memc, key, strlen(key),
161 &string_length, &flags, &rc);
162
163 assert(string);
164 assert(rc == MEMCACHED_SUCCESS);
165 assert(string_length == strlen(value));
166 assert(!memcmp(string, value, string_length));
167
168 free(string);
169
170 memcached_deinit(memc);
171 }
172
173 void get_test3(void)
174 {
175 memcached_st *memc;
176 memcached_return rc;
177 char *key= "foo";
178 char *value;
179 size_t value_length= 8191;
180 char *string;
181 size_t string_length;
182 uint16_t flags;
183 int i;
184
185 value = (char*) malloc(value_length);
186 for (i=0; i<value_length; i++)
187 value[i] = (char) (i % 127);
188
189 memc= memcached_init(NULL);
190 assert(memc);
191 rc= memcached_set(memc, key, strlen(key),
192 value, value_length,
193 (time_t)0, (uint16_t)0);
194 assert(rc == MEMCACHED_SUCCESS);
195
196 string= memcached_get(memc, key, strlen(key),
197 &string_length, &flags, &rc);
198
199 assert(string_length == value_length);
200 assert(!memcmp(string, value, string_length));
201
202 free(string);
203
204 memcached_deinit(memc);
205 }
206
207 void stats_hostname_test(void)
208 {
209 memcached_return rc;
210 memcached_stat_st stat;
211 rc= memcached_stat_hostname(&stat, NULL,
212 "localhost",
213 MEMCACHED_DEFAULT_PORT);
214 }
215
216 void increment_test(void)
217 {
218 memcached_st *memc;
219 unsigned int new_number;
220 memcached_return rc;
221 char *key= "number";
222 char *value= "0";
223
224 memc= memcached_init(NULL);
225 assert(memc);
226 rc= memcached_set(memc, key, strlen(key),
227 value, strlen(value),
228 (time_t)0, (uint16_t)0);
229 assert(rc == MEMCACHED_SUCCESS);
230
231 rc= memcached_increment(memc, key, strlen(key),
232 1, &new_number);
233 assert(rc == MEMCACHED_SUCCESS);
234 assert(new_number == 1);
235
236 rc= memcached_increment(memc, key, strlen(key),
237 1, &new_number);
238 assert(rc == MEMCACHED_SUCCESS);
239 assert(new_number == 2);
240
241 memcached_deinit(memc);
242 }
243
244 void decrement_test(void)
245 {
246 memcached_st *memc;
247 unsigned int new_number;
248 memcached_return rc;
249 char *key= "number";
250 char *value= "3";
251
252 memc= memcached_init(NULL);
253 assert(memc);
254 rc= memcached_set(memc, key, strlen(key),
255 value, strlen(value),
256 (time_t)0, (uint16_t)0);
257 assert(rc == MEMCACHED_SUCCESS);
258
259 rc= memcached_decrement(memc, key, strlen(key),
260 1, &new_number);
261 assert(rc == MEMCACHED_SUCCESS);
262 assert(new_number == 2);
263
264 rc= memcached_decrement(memc, key, strlen(key),
265 1, &new_number);
266 assert(rc == MEMCACHED_SUCCESS);
267 assert(new_number == 1);
268
269 memcached_deinit(memc);
270 }
271
272 void quit_test(void)
273 {
274 memcached_st *memc;
275 memcached_return rc;
276 char *key= "fudge";
277 char *value= "sanford and sun";
278
279 memc= memcached_init(NULL);
280 assert(memc);
281 rc= memcached_set(memc, key, strlen(key),
282 value, strlen(value),
283 (time_t)10, (uint16_t)3);
284 assert(rc == MEMCACHED_SUCCESS);
285 memcached_quit(memc);
286
287 rc= memcached_set(memc, key, strlen(key),
288 value, strlen(value),
289 (time_t)50, (uint16_t)9);
290 assert(rc == MEMCACHED_SUCCESS);
291
292 memcached_deinit(memc);
293 }
294
295 void mget_test(void)
296 {
297 memcached_st *memc;
298 memcached_return rc;
299 char *keys[]= {"fudge", "son", "food"};
300 size_t key_length[]= {5, 3, 4};
301 unsigned int x;
302 uint16_t flags;
303
304 char return_key[MEMCACHED_MAX_KEY];
305 size_t return_key_length;
306 char *return_value;
307 size_t return_value_length;
308
309 memc= memcached_init(NULL);
310 assert(memc);
311
312 rc= memcached_mget(memc, keys, key_length, 3);
313 assert(rc == MEMCACHED_SUCCESS);
314
315 while (return_value= memcached_fetch(memc, return_key, &return_key_length,
316 &return_value_length, &flags, &rc))
317 {
318 assert(return_value);
319 }
320 assert(rc == MEMCACHED_NOTFOUND);
321
322 for (x= 0; x < 3; x++)
323 {
324 rc= memcached_set(memc, keys[x], key_length[x],
325 keys[x], key_length[x],
326 (time_t)50, (uint16_t)9);
327 assert(rc == MEMCACHED_SUCCESS);
328 }
329
330 rc= memcached_mget(memc, keys, key_length, 3);
331 assert(rc == MEMCACHED_SUCCESS);
332
333 x= 0;
334 while (return_value= memcached_fetch(memc, return_key, &return_key_length,
335 &return_value_length, &flags, &rc))
336 {
337 assert(return_value);
338 assert(rc == MEMCACHED_SUCCESS);
339 assert(key_length[x] == return_value_length);
340 assert(!memcmp(return_value, keys[x], return_value_length));
341 x++;
342 }
343
344 memcached_deinit(memc);
345 }
346
347
348 int main(void)
349 {
350 /* Clean the server before beginning testing */
351 flush_test();
352 init_test();
353 allocation_test();
354 connection_test();
355 set_test();
356 add_test();
357 replace_test();
358 flush_test();
359 delete_test();
360 flush_test();
361 get_test();
362 get_test2();
363 get_test3();
364 stats_hostname_test();
365
366 increment_test();
367 decrement_test();
368 quit_test();
369 mget_test();
370
371 /* Clean up whatever we might have left */
372 flush_test();
373 return 0;
374 }