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