2 * Copyright (C) 2006-2009 Brian Aker
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
13 Sample test application.
15 #include "libmemcached/common.h"
22 #include <sys/types.h>
27 #include "../clients/generator.h"
28 #include "../clients/execute.h"
31 #define INT64_MAX LONG_MAX
34 #define INT32_MAX INT_MAX
40 /* Number of items generated for tests */
41 #define GLOBAL_COUNT 100000
43 /* Number of times to run the test loop */
44 #define TEST_COUNTER 500000
45 static uint32_t global_count
;
47 static pairs_st
*global_pairs
;
48 static char *global_keys
[GLOBAL_COUNT
];
49 static size_t global_keys_length
[GLOBAL_COUNT
];
51 static test_return_t
cleanup_pairs(memcached_st
*memc
__attribute__((unused
)))
53 pairs_free(global_pairs
);
58 static test_return_t
generate_pairs(memcached_st
*memc
__attribute__((unused
)))
61 global_pairs
= pairs_generate(GLOBAL_COUNT
, 400);
62 global_count
= GLOBAL_COUNT
;
64 for (x
= 0; x
< global_count
; x
++)
66 global_keys
[x
]= global_pairs
[x
].key
;
67 global_keys_length
[x
]= global_pairs
[x
].key_length
;
73 static test_return_t
drizzle(memcached_st
*memc
)
76 memcached_return_t rc
;
78 size_t return_value_length
;
82 for (x
= 0; x
< TEST_COUNTER
; x
++)
87 test_bit
= (uint32_t)(random() % GLOBAL_COUNT
);
88 which
= (uint8_t)(random() % 2);
92 return_value
= memcached_get(memc
, global_keys
[test_bit
], global_keys_length
[test_bit
],
93 &return_value_length
, &flags
, &rc
);
94 if (rc
== MEMCACHED_SUCCESS
&& return_value
)
96 else if (rc
== MEMCACHED_NOTFOUND
)
100 WATCHPOINT_ERROR(rc
);
101 WATCHPOINT_ASSERT(rc
);
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
,
111 if (rc
!= MEMCACHED_SUCCESS
&& rc
!= MEMCACHED_BUFFERED
)
113 WATCHPOINT_ERROR(rc
);
114 WATCHPOINT_ASSERT(0);
119 if (getenv("MEMCACHED_ATOM_BURIN_IN"))
125 static memcached_return_t
pre_nonblock(memcached_st
*memc
)
127 memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_NO_BLOCK
, 0);
129 return MEMCACHED_SUCCESS
;
132 static memcached_return_t
pre_md5(memcached_st
*memc
)
134 memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_HASH
, (uint64_t)MEMCACHED_HASH_MD5
);
136 return MEMCACHED_SUCCESS
;
139 static memcached_return_t
pre_hsieh(memcached_st
*memc
)
141 memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_HASH
, (uint64_t)MEMCACHED_HASH_HSIEH
);
143 return MEMCACHED_SUCCESS
;
146 static memcached_return_t
enable_consistent(memcached_st
*memc
)
148 memcached_server_distribution_t value
= MEMCACHED_DISTRIBUTION_CONSISTENT
;
149 memcached_hash_t hash
;
150 memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_DISTRIBUTION
, value
);
153 value
= (memcached_server_distribution_t
)memcached_behavior_get(memc
, MEMCACHED_BEHAVIOR_DISTRIBUTION
);
154 assert(value
== MEMCACHED_DISTRIBUTION_CONSISTENT
);
156 hash
= (memcached_hash_t
)memcached_behavior_get(memc
, MEMCACHED_BEHAVIOR_HASH
);
157 assert(hash
== MEMCACHED_HASH_HSIEH
);
160 return MEMCACHED_SUCCESS
;
164 Set the value, then quit to make sure it is flushed.
165 Come back in and test that add fails.
167 static test_return_t
add_test(memcached_st
*memc
)
169 memcached_return_t rc
;
170 const char *key
= "foo";
171 const char *value
= "when we sanitize";
172 unsigned long long setting_value
;
174 setting_value
= memcached_behavior_get(memc
, MEMCACHED_BEHAVIOR_NO_BLOCK
);
176 rc
= memcached_set(memc
, key
, strlen(key
),
177 value
, strlen(value
),
178 (time_t)0, (uint32_t)0);
179 test_truth(rc
== MEMCACHED_SUCCESS
|| rc
== MEMCACHED_BUFFERED
);
180 memcached_quit(memc
);
181 rc
= memcached_add(memc
, key
, strlen(key
),
182 value
, strlen(value
),
183 (time_t)0, (uint32_t)0);
185 /* Too many broken OS'es have broken loopback in async, so we can't be sure of the result */
188 test_truth(rc
== MEMCACHED_NOTSTORED
|| rc
== MEMCACHED_STORED
);
192 test_truth(rc
== MEMCACHED_NOTSTORED
);
199 * repeating add_tests many times
200 * may show a problem in timing
202 static test_return_t
many_adds(memcached_st
*memc
)
205 for (i
= 0; i
< TEST_COUNTER
; i
++)
212 test_st smash_tests
[] ={
213 {"generate_pairs", 1, (test_callback_fn
)generate_pairs
},
214 {"drizzle", 1, (test_callback_fn
)drizzle
},
215 {"cleanup", 1, (test_callback_fn
)cleanup_pairs
},
216 {"many_adds", 1, (test_callback_fn
)many_adds
},
221 collection_st collection
[] ={
222 {"smash", 0, 0, smash_tests
},
223 {"smash_hsieh", (test_callback_fn
)pre_hsieh
, 0, smash_tests
},
224 {"smash_hsieh_consistent", (test_callback_fn
)enable_consistent
, 0, smash_tests
},
225 {"smash_md5", (test_callback_fn
)pre_md5
, 0, smash_tests
},
226 {"smash_nonblock", (test_callback_fn
)pre_nonblock
, 0, smash_tests
},
231 #define SERVERS_TO_CREATE 5
233 #include "libmemcached_world.h"
235 void get_world(world_st
*world
)
237 world
->collections
= collection
;
239 world
->create
= (test_callback_create_fn
)world_create
;
240 world
->destroy
= (test_callback_fn
)world_destroy
;
242 world
->test
.startup
= (test_callback_fn
)world_test_startup
;
243 world
->test
.flush
= (test_callback_fn
)world_flush
;
244 world
->test
.pre_run
= (test_callback_fn
)world_pre_run
;
245 world
->test
.post_run
= (test_callback_fn
)world_post_run
;
246 world
->test
.on_error
= (test_callback_error_fn
)world_on_error
;
248 world
->runner
= &defualt_libmemcached_runner
;