1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3 * Libmemcached Client and Server
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are
13 * * Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
16 * * Redistributions in binary form must reproduce the above
17 * copyright notice, this list of conditions and the following disclaimer
18 * in the documentation and/or other materials provided with the
21 * * The names of its contributors may not be used to endorse or
22 * promote products derived from this software without specific prior
25 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
28 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
29 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
35 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
44 /* The structure we use for the test system */
45 struct libmemcached_test_container_st
47 libtest::server_startup_st
& construct
;
51 libmemcached_test_container_st(libtest::server_startup_st
&construct_arg
) :
52 construct(construct_arg
),
58 static void *world_create(libtest::server_startup_st
& servers
, test_return_t
& error
)
60 if (HAVE_MEMCACHED_BINARY
== 0)
66 if (servers
.sasl() and (LIBMEMCACHED_WITH_SASL_SUPPORT
== 0 or MEMCACHED_SASL_BINARY
== 0))
72 // Assume we are running under valgrind, and bail
73 if (servers
.sasl() and getenv("TESTS_ENVIRONMENT"))
79 in_port_t max_port
= TEST_PORT_BASE
;
80 for (uint32_t x
= 0; x
< servers
.count(); x
++)
84 char variable_buffer
[1024];
85 snprintf(variable_buffer
, sizeof(variable_buffer
), "LIBMEMCACHED_PORT_%u", x
);
88 if ((var
= getenv(variable_buffer
)))
90 port
= in_port_t(atoi(var
));
94 port
= in_port_t(TEST_PORT_BASE
+x
);
98 const char *argv
[1]= { "memcached" };
101 if (not server_startup(servers
, "memcached-sasl", port
, 1, argv
))
109 if (not server_startup(servers
, "memcached", port
, 1, argv
))
117 if (servers
.socket())
121 const char *argv
[1]= { "memcached" };
122 if (not servers
.start_socket_server("memcached-sasl", max_port
+1, 1, argv
))
130 const char *argv
[1]= { "memcached" };
131 if (not servers
.start_socket_server("memcached", max_port
+1, 1, argv
))
140 libmemcached_test_container_st
*global_container
= new libmemcached_test_container_st(servers
);
141 if (global_container
== NULL
)
143 error
= TEST_MEMORY_ALLOCATION_FAILURE
;
149 return global_container
;
152 static test_return_t
world_container_startup(libmemcached_test_container_st
*container
)
156 test_compare_got(MEMCACHED_SUCCESS
,
157 libmemcached_check_configuration(container
->construct
.option_string().c_str(), container
->construct
.option_string().size(),
158 buffer
, sizeof(buffer
)),
159 container
->construct
.option_string().c_str());
161 test_null(container
->parent
);
162 container
->parent
= memcached(container
->construct
.option_string().c_str(), container
->construct
.option_string().size());
163 test_true(container
->parent
);
164 test_compare(MEMCACHED_SUCCESS
, memcached_version(container
->parent
));
166 if (container
->construct
.sasl())
168 if (memcached_failed(memcached_behavior_set(container
->parent
, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
, 1)))
170 memcached_free(container
->parent
);
174 if (memcached_failed(memcached_set_sasl_auth_data(container
->parent
, container
->construct
.username().c_str(), container
->construct
.password().c_str())))
176 memcached_free(container
->parent
);
181 for (uint32_t host
= 0; host
< memcached_server_count(container
->parent
); ++host
)
183 memcached_server_instance_st instance
=
184 memcached_server_instance_by_position(container
->parent
, host
);
186 if (instance
->type
== MEMCACHED_CONNECTION_TCP
)
188 test_true_got(memcached_server_port(instance
) >= TEST_PORT_BASE
, memcached_server_port(instance
));
195 static test_return_t
world_container_shutdown(libmemcached_test_container_st
*container
)
197 memcached_free(container
->parent
);
198 container
->parent
= NULL
;
203 static test_return_t
world_test_startup(libmemcached_test_container_st
*container
)
205 test_true(container
);
206 test_null(container
->memc
);
207 test_true(container
->parent
);
208 container
->memc
= memcached_clone(NULL
, container
->parent
);
209 test_true(container
->memc
);
214 test_return_t
world_flush(libmemcached_test_container_st
*container
);
215 test_return_t
world_flush(libmemcached_test_container_st
*container
)
217 test_true(container
->memc
);
218 memcached_flush(container
->memc
, 0);
219 memcached_quit(container
->memc
);
224 static test_return_t
world_pre_run(libmemcached_test_container_st
*container
)
226 test_true(container
->memc
);
227 for (uint32_t loop
= 0; loop
< memcached_server_list_count(container
->memc
->servers
); loop
++)
229 memcached_server_instance_st instance
=
230 memcached_server_instance_by_position(container
->memc
, loop
);
232 test_compare(-1, instance
->fd
);
233 test_compare(0U, instance
->cursor_active
);
240 static test_return_t
world_post_run(libmemcached_test_container_st
*container
)
242 test_true(container
->memc
);
247 static test_return_t
world_on_error(test_return_t
, libmemcached_test_container_st
*container
)
249 test_true(container
->memc
);
250 memcached_free(container
->memc
);
251 container
->memc
= NULL
;
256 static bool world_destroy(void *object
)
258 libmemcached_test_container_st
*container
= (libmemcached_test_container_st
*)object
;
259 #if defined(LIBMEMCACHED_WITH_SASL_SUPPORT) && LIBMEMCACHED_WITH_SASL_SUPPORT
260 if (LIBMEMCACHED_WITH_SASL_SUPPORT
)
271 typedef test_return_t (*libmemcached_test_callback_fn
)(memcached_st
*);
273 static test_return_t
_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
277 test_true(container
);
278 test_true(container
->memc
);
281 ret
= func(container
->memc
);
283 catch (std::exception
& e
)
285 libtest::Error
<< e
.what();
295 static test_return_t
_pre_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
299 return func(container
->parent
);
305 static test_return_t
_post_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
309 return func(container
->parent
);
315 class LibmemcachedRunner
: public libtest::Runner
{
317 test_return_t
run(test_callback_fn
* func
, void *object
)
319 return _runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
322 test_return_t
pre(test_callback_fn
* func
, void *object
)
324 return _pre_runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
327 test_return_t
post(test_callback_fn
* func
, void *object
)
329 return _post_runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
333 static LibmemcachedRunner defualt_libmemcached_runner
;