1 /* libMemcached Functions Test
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.
8 * Description: This is the startup bits for any libmemcached test.
16 /* The structure we use for the test system */
17 struct libmemcached_test_container_st
19 server_startup_st
& construct
;
23 libmemcached_test_container_st(server_startup_st
&construct_arg
) :
24 construct(construct_arg
),
30 static void *world_create(server_startup_st
& servers
, test_return_t
& error
)
32 if (LIBMEMCACHED_WITH_SASL_SUPPORT
== 0)
38 // Assume we are running under valgrind, and bail
39 if (servers
.sasl() and getenv("TESTS_ENVIRONMENT"))
46 in_port_t max_port
= TEST_PORT_BASE
;
47 for (uint32_t x
= 0; x
< servers
.count(); x
++)
51 char variable_buffer
[1024];
52 snprintf(variable_buffer
, sizeof(variable_buffer
), "LIBMEMCACHED_PORT_%u", x
);
55 if ((var
= getenv(variable_buffer
)))
57 port
= in_port_t(atoi(var
));
61 port
= in_port_t(TEST_PORT_BASE
+x
);
65 const char *argv
[1]= { "memcached" };
68 if (not server_startup(servers
, "memcached-sasl", port
, 1, argv
))
76 if (not server_startup(servers
, "memcached", port
, 1, argv
))
88 const char *argv
[1]= { "memcached" };
89 if (not servers
.start_socket_server("memcached-sasl", max_port
+1, 1, argv
))
97 const char *argv
[1]= { "memcached" };
98 if (not servers
.start_socket_server("memcached", max_port
+1, 1, argv
))
107 libmemcached_test_container_st
*global_container
= new libmemcached_test_container_st(servers
);
108 if (global_container
== NULL
)
110 error
= TEST_MEMORY_ALLOCATION_FAILURE
;
116 return global_container
;
119 static test_return_t
world_container_startup(libmemcached_test_container_st
*container
)
123 test_compare_got(MEMCACHED_SUCCESS
,
124 libmemcached_check_configuration(container
->construct
.option_string().c_str(), container
->construct
.option_string().size(),
125 buffer
, sizeof(buffer
)),
126 container
->construct
.option_string().c_str());
128 test_true(not container
->parent
);
129 container
->parent
= memcached(container
->construct
.option_string().c_str(), container
->construct
.option_string().size());
130 test_true(container
->parent
);
132 if (container
->construct
.sasl())
134 if (memcached_failed(memcached_behavior_set(container
->parent
, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
, 1)))
136 memcached_free(container
->parent
);
140 if (memcached_failed(memcached_set_sasl_auth_data(container
->parent
, container
->construct
.username().c_str(), container
->construct
.password().c_str())))
142 memcached_free(container
->parent
);
147 for (uint32_t host
= 0; host
< memcached_server_count(container
->parent
); ++host
)
149 memcached_server_instance_st instance
=
150 memcached_server_instance_by_position(container
->parent
, host
);
152 if (instance
->type
== MEMCACHED_CONNECTION_TCP
)
154 test_true_got(memcached_server_port(instance
) >= TEST_PORT_BASE
, memcached_server_port(instance
));
161 static test_return_t
world_container_shutdown(libmemcached_test_container_st
*container
)
163 memcached_free(container
->parent
);
164 container
->parent
= NULL
;
169 static test_return_t
world_test_startup(libmemcached_test_container_st
*container
)
171 test_true(container
);
172 test_true(not container
->memc
);
173 test_true(container
->parent
);
174 container
->memc
= memcached_clone(NULL
, container
->parent
);
175 test_true(container
->memc
);
180 test_return_t
world_flush(libmemcached_test_container_st
*container
);
181 test_return_t
world_flush(libmemcached_test_container_st
*container
)
183 test_true(container
->memc
);
184 memcached_flush(container
->memc
, 0);
185 memcached_quit(container
->memc
);
190 static test_return_t
world_pre_run(libmemcached_test_container_st
*container
)
192 test_true(container
->memc
);
193 for (uint32_t loop
= 0; loop
< memcached_server_list_count(container
->memc
->servers
); loop
++)
195 memcached_server_instance_st instance
=
196 memcached_server_instance_by_position(container
->memc
, loop
);
198 test_compare(-1, instance
->fd
);
199 test_compare(0U, instance
->cursor_active
);
206 static test_return_t
world_post_run(libmemcached_test_container_st
*container
)
208 test_true(container
->memc
);
213 static test_return_t
world_on_error(test_return_t test_state
, libmemcached_test_container_st
*container
)
216 test_true(container
->memc
);
217 memcached_free(container
->memc
);
218 container
->memc
= NULL
;
223 static bool world_destroy(void *object
)
225 libmemcached_test_container_st
*container
= (libmemcached_test_container_st
*)object
;
226 #if defined(LIBMEMCACHED_WITH_SASL_SUPPORT) && LIBMEMCACHED_WITH_SASL_SUPPORT
227 if (LIBMEMCACHED_WITH_SASL_SUPPORT
)
238 typedef test_return_t (*libmemcached_test_callback_fn
)(memcached_st
*);
240 static test_return_t
_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
244 test_true(container
);
245 test_true(container
->memc
);
246 return func(container
->memc
);
252 static test_return_t
_pre_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
256 return func(container
->parent
);
262 static test_return_t
_post_runner_default(libmemcached_test_callback_fn func
, libmemcached_test_container_st
*container
)
266 return func(container
->parent
);
272 class LibmemcachedRunner
: public Runner
{
274 test_return_t
run(test_callback_fn
* func
, void *object
)
276 return _runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
279 test_return_t
pre(test_callback_fn
* func
, void *object
)
281 return _pre_runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
284 test_return_t
post(test_callback_fn
* func
, void *object
)
286 return _post_runner_default(libmemcached_test_callback_fn(func
), (libmemcached_test_container_st
*)object
);
290 static LibmemcachedRunner defualt_libmemcached_runner
;