Updating for 1.0.2 release
[awesomized/libmemcached] / tests / atomsmasher.cc
1 /* LibMemcached
2 * Copyright (C) 2006-2009 Brian Aker
3 * All rights reserved.
4 *
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
7 *
8 * Summary:
9 *
10 */
11
12 /*
13 Sample test application.
14 */
15 #include <config.h>
16
17 #include <libtest/test.hpp>
18
19 #include <libmemcached/memcached.h>
20
21 #include <cstdio>
22 #include <cstdlib>
23 #include <stdint.h>
24 #include <cstring>
25 #include <sys/time.h>
26 #include <sys/types.h>
27 #include <sys/stat.h>
28 #include <unistd.h>
29 #include <ctime>
30 #include <clients/generator.h>
31 #include <clients/execute.h>
32
33 #include <libtest/server.h>
34
35 #include <tests/debug.h>
36
37 using namespace libtest;
38
39 /* Number of items generated for tests */
40 #define GLOBAL_COUNT 100000
41
42 /* Number of times to run the test loop */
43 #define TEST_COUNTER 500000
44 static uint32_t global_count;
45
46 static pairs_st *global_pairs;
47 static char *global_keys[GLOBAL_COUNT];
48 static size_t global_keys_length[GLOBAL_COUNT];
49
50 static test_return_t cleanup_pairs(memcached_st *)
51 {
52 pairs_free(global_pairs);
53
54 return TEST_SUCCESS;
55 }
56
57 static test_return_t generate_pairs(memcached_st *)
58 {
59 global_pairs= pairs_generate(GLOBAL_COUNT, 400);
60 global_count= GLOBAL_COUNT;
61
62 for (size_t x= 0; x < global_count; x++)
63 {
64 global_keys[x]= global_pairs[x].key;
65 global_keys_length[x]= global_pairs[x].key_length;
66 }
67
68 return TEST_SUCCESS;
69 }
70
71 static test_return_t drizzle(memcached_st *memc)
72 {
73 infinite:
74 for (size_t x= 0; x < TEST_COUNTER; x++)
75 {
76 memcached_return_t rc;
77 char *return_value;
78 size_t return_value_length;
79 uint32_t flags;
80
81 uint32_t test_bit;
82 uint8_t which;
83
84 test_bit= (uint32_t)(random() % GLOBAL_COUNT);
85 which= (uint8_t)(random() % 2);
86
87 if (which == 0)
88 {
89 return_value= memcached_get(memc, global_keys[test_bit], global_keys_length[test_bit],
90 &return_value_length, &flags, &rc);
91 if (rc == MEMCACHED_SUCCESS && return_value)
92 {
93 free(return_value);
94 }
95 else if (rc == MEMCACHED_NOTFOUND)
96 {
97 continue;
98 }
99 else
100 {
101 test_compare(MEMCACHED_SUCCESS, rc);
102 }
103 }
104 else
105 {
106 rc= memcached_set(memc, global_pairs[test_bit].key,
107 global_pairs[test_bit].key_length,
108 global_pairs[test_bit].value,
109 global_pairs[test_bit].value_length,
110 0, 0);
111 if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_BUFFERED)
112 {
113 test_compare(MEMCACHED_SUCCESS, rc);
114 }
115 }
116 }
117
118 if (getenv("MEMCACHED_ATOM_BURIN_IN"))
119 {
120 goto infinite;
121 }
122
123 return TEST_SUCCESS;
124 }
125
126 static test_return_t pre_nonblock(memcached_st *memc)
127 {
128 memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
129
130 return TEST_SUCCESS;
131 }
132
133 /*
134 Set the value, then quit to make sure it is flushed.
135 Come back in and test that add fails.
136 */
137 static test_return_t add_test(memcached_st *memc)
138 {
139 const char *key= "foo";
140 const char *value= "when we sanitize";
141
142 memcached_return_t rc;
143 rc= memcached_set(memc, key, strlen(key),
144 value, strlen(value),
145 (time_t)0, (uint32_t)0);
146 test_true_got(rc == MEMCACHED_SUCCESS or rc == MEMCACHED_BUFFERED, memcached_strerror(NULL, rc));
147 memcached_quit(memc);
148 rc= memcached_add(memc, key, strlen(key),
149 value, strlen(value),
150 (time_t)0, (uint32_t)0);
151
152 if (rc == MEMCACHED_CONNECTION_FAILURE)
153 {
154 print_servers(memc);
155 }
156
157 /* Too many broken OS'es have broken loopback in async, so we can't be sure of the result */
158 if (memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK))
159 {
160 test_true(rc == MEMCACHED_NOTSTORED or rc == MEMCACHED_STORED);
161 }
162 else
163 {
164 test_compare_got(MEMCACHED_NOTSTORED, rc, memcached_strerror(NULL, rc));
165 }
166
167 return TEST_SUCCESS;
168 }
169
170 /*
171 * repeating add_tests many times
172 * may show a problem in timing
173 */
174 static test_return_t many_adds(memcached_st *memc)
175 {
176 test_true(memc);
177 for (size_t x= 0; x < TEST_COUNTER; x++)
178 {
179 test_compare_got(TEST_SUCCESS, add_test(memc), x);
180 }
181 return TEST_SUCCESS;
182 }
183
184 test_st smash_tests[] ={
185 {"generate_pairs", true, (test_callback_fn*)generate_pairs },
186 {"drizzle", true, (test_callback_fn*)drizzle },
187 {"cleanup", true, (test_callback_fn*)cleanup_pairs },
188 {"many_adds", true, (test_callback_fn*)many_adds },
189 {0, 0, 0}
190 };
191
192 #define BENCHMARK_TEST_LOOP 20000
193
194 struct benchmark_state_st
195 {
196 bool create_init;
197 bool clone_init;
198 memcached_st *create;
199 memcached_st *clone;
200 } benchmark_state;
201
202 static test_return_t memcached_create_benchmark(memcached_st *)
203 {
204 benchmark_state.create_init= true;
205
206 for (size_t x= 0; x < BENCHMARK_TEST_LOOP; x++)
207 {
208 memcached_st *ptr;
209 ptr= memcached_create(&benchmark_state.create[x]);
210
211 test_true(ptr);
212 }
213
214 return TEST_SUCCESS;
215 }
216
217 static test_return_t memcached_clone_benchmark(memcached_st *memc)
218 {
219 benchmark_state.clone_init= true;
220
221 for (size_t x= 0; x < BENCHMARK_TEST_LOOP; x++)
222 {
223 memcached_st *ptr;
224 ptr= memcached_clone(&benchmark_state.clone[x], memc);
225
226 test_true(ptr);
227 }
228
229 return TEST_SUCCESS;
230 }
231
232 static test_return_t pre_allocate(memcached_st *)
233 {
234 memset(&benchmark_state, 0, sizeof(benchmark_state));
235
236 benchmark_state.create= (memcached_st *)calloc(BENCHMARK_TEST_LOOP, sizeof(memcached_st));
237 test_true(benchmark_state.create);
238 benchmark_state.clone= (memcached_st *)calloc(BENCHMARK_TEST_LOOP, sizeof(memcached_st));
239 test_true(benchmark_state.clone);
240
241 return TEST_SUCCESS;
242 }
243
244 static test_return_t post_allocate(memcached_st *)
245 {
246 for (size_t x= 0; x < BENCHMARK_TEST_LOOP; x++)
247 {
248 if (benchmark_state.create_init)
249 {
250 memcached_free(&benchmark_state.create[x]);
251 }
252
253 if (benchmark_state.clone_init)
254 {
255 memcached_free(&benchmark_state.clone[x]);
256 }
257 }
258
259 free(benchmark_state.create);
260 free(benchmark_state.clone);
261
262 return TEST_SUCCESS;
263 }
264
265
266 test_st micro_tests[] ={
267 {"memcached_create", 1, (test_callback_fn*)memcached_create_benchmark },
268 {"memcached_clone", 1, (test_callback_fn*)memcached_clone_benchmark },
269 {0, 0, 0}
270 };
271
272
273 collection_st collection[] ={
274 {"smash", 0, 0, smash_tests},
275 {"smash_nonblock", (test_callback_fn*)pre_nonblock, 0, smash_tests},
276 {"micro-benchmark", (test_callback_fn*)pre_allocate, (test_callback_fn*)post_allocate, micro_tests},
277 {0, 0, 0, 0}
278 };
279
280
281 #define TEST_PORT_BASE MEMCACHED_DEFAULT_PORT +10
282
283 #include "libmemcached_world.h"
284
285 void get_world(Framework *world)
286 {
287 world->collections= collection;
288
289 world->_create= (test_callback_create_fn*)world_create;
290 world->_destroy= (test_callback_destroy_fn*)world_destroy;
291
292 world->item._startup= (test_callback_fn*)world_test_startup;
293 world->item._flush= (test_callback_fn*)world_flush;
294 world->item.set_pre((test_callback_fn*)world_pre_run);
295 world->item.set_post((test_callback_fn*)world_post_run);
296 world->_on_error= (test_callback_error_fn*)world_on_error;
297
298 world->collection_startup= (test_callback_fn*)world_container_startup;
299 world->collection_shutdown= (test_callback_fn*)world_container_shutdown;
300
301 world->set_runner(&defualt_libmemcached_runner);
302 }