-/* libMemcached Functions Test
- * Copyright (C) 2006-2009 Brian Aker
- * All rights reserved.
+/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
+ *
+ * Libmemcached Client and Server
*
- * Use and distribution licensed under the BSD license. See
- * the COPYING file in the parent directory for full text.
+ * Copyright (C) 2012 Data Differential, http://datadifferential.com/
+ * Copyright (C) 2006-2009 Brian Aker
+ * All rights reserved.
*
- * Description: This is the startup bits for any libmemcached test.
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ * * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following disclaimer
+ * in the documentation and/or other materials provided with the
+ * distribution.
+ *
+ * * The names of its contributors may not be used to endorse or
+ * promote products derived from this software without specific prior
+ * written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*/
+
#pragma once
#include <cassert>
/* The structure we use for the test system */
struct libmemcached_test_container_st
{
- server_startup_st construct;
+ libtest::server_startup_st& construct;
memcached_st *parent;
memcached_st *memc;
- libmemcached_test_container_st() :
+ libmemcached_test_container_st(libtest::server_startup_st &construct_arg) :
+ construct(construct_arg),
parent(NULL),
memc(NULL)
{ }
};
-#ifdef __cplusplus
-extern "C" {
-#endif
+static void *world_create(libtest::server_startup_st& servers, test_return_t& error)
+{
+ if (HAVE_MEMCACHED_BINARY == 0)
+ {
+ error= TEST_SKIPPED;
+ return NULL;
+ }
-/* Prototypes for functions we will pass to test framework */
-libmemcached_test_container_st *world_create(test_return_t *error);
-test_return_t world_test_startup(libmemcached_test_container_st *);
-test_return_t world_flush(libmemcached_test_container_st *container);
-test_return_t world_pre_run(libmemcached_test_container_st *);
+ if (servers.sasl() and (LIBMEMCACHED_WITH_SASL_SUPPORT == 0 or MEMCACHED_SASL_BINARY == 0))
+ {
+ error= TEST_SKIPPED;
+ return NULL;
+ }
-test_return_t world_post_run(libmemcached_test_container_st *);
-test_return_t world_on_error(test_return_t, libmemcached_test_container_st *);
-test_return_t world_destroy(libmemcached_test_container_st *);
+ // Assume we are running under valgrind, and bail
+ if (servers.sasl() and getenv("TESTS_ENVIRONMENT"))
+ {
+ error= TEST_SKIPPED;
+ return NULL;
+ }
-/**
- @note generic shutdown/startup for libmemcached tests.
-*/
-test_return_t world_container_startup(libmemcached_test_container_st *container);
-test_return_t world_container_shutdown(libmemcached_test_container_st *container);
+ for (uint32_t x= 0; x < servers.count(); x++)
+ {
+ in_port_t port;
+
+ char variable_buffer[1024];
+ snprintf(variable_buffer, sizeof(variable_buffer), "LIBMEMCACHED_PORT_%u", x);
+
+ char *var;
+ if ((var= getenv(variable_buffer)))
+ {
+ port= in_port_t(atoi(var));
+ }
+ else
+ {
+ port= in_port_t(TEST_PORT_BASE +x);
+ }
+
+ const char *argv[1]= { "memcached" };
+ if (servers.sasl())
+ {
+ if (not server_startup(servers, "memcached-sasl", port, 1, argv))
+ {
+ error= TEST_FATAL;
+ return NULL;
+ }
+ }
+ else
+ {
+ if (not server_startup(servers, "memcached", port, 1, argv))
+ {
+ error= TEST_FATAL;
+ return NULL;
+ }
+ }
+ }
-libmemcached_test_container_st *world_create(test_return_t *error)
-{
- libmemcached_test_container_st *global_container= new libmemcached_test_container_st();
- if (not server_startup(&global_container->construct))
+ libmemcached_test_container_st *global_container= new libmemcached_test_container_st(servers);
+ if (global_container == NULL)
{
- *error= TEST_FAILURE;
+ error= TEST_MEMORY_ALLOCATION_FAILURE;
return NULL;
}
- *error= TEST_SUCCESS;
+ error= TEST_SUCCESS;
return global_container;
}
-test_return_t world_container_startup(libmemcached_test_container_st *container)
+static test_return_t world_container_startup(libmemcached_test_container_st *container)
{
char buffer[BUFSIZ];
test_compare_got(MEMCACHED_SUCCESS,
- libmemcached_check_configuration(container->construct.server_list.c_str(), container->construct.server_list.size(),
+ libmemcached_check_configuration(container->construct.option_string().c_str(), container->construct.option_string().size(),
buffer, sizeof(buffer)),
- buffer);
+ container->construct.option_string().c_str());
- assert(not container->parent);
- container->parent= memcached(container->construct.server_list.c_str(), container->construct.server_list.size());
+ test_null(container->parent);
+ container->parent= memcached(container->construct.option_string().c_str(), container->construct.option_string().size());
test_true(container->parent);
+ test_compare(MEMCACHED_SUCCESS, memcached_version(container->parent));
+
+ if (container->construct.sasl())
+ {
+ if (memcached_failed(memcached_behavior_set(container->parent, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL, 1)))
+ {
+ memcached_free(container->parent);
+ return TEST_FAILURE;
+ }
+
+ if (memcached_failed(memcached_set_sasl_auth_data(container->parent, container->construct.username().c_str(), container->construct.password().c_str())))
+ {
+ memcached_free(container->parent);
+ return TEST_FAILURE;
+ }
+ }
+
+ for (uint32_t host= 0; host < memcached_server_count(container->parent); ++host)
+ {
+ memcached_server_instance_st instance=
+ memcached_server_instance_by_position(container->parent, host);
+
+ if (instance->type == MEMCACHED_CONNECTION_TCP)
+ {
+ test_true_got(memcached_server_port(instance) >= TEST_PORT_BASE, memcached_server_port(instance));
+ }
+ }
return TEST_SUCCESS;
}
-test_return_t world_container_shutdown(libmemcached_test_container_st *container)
+static test_return_t world_container_shutdown(libmemcached_test_container_st *container)
{
memcached_free(container->parent);
container->parent= NULL;
return TEST_SUCCESS;
}
-test_return_t world_test_startup(libmemcached_test_container_st *container)
+static test_return_t world_test_startup(libmemcached_test_container_st *container)
{
- assert(container);
- assert(not container->memc);
- assert(container->parent);
+ test_true(container);
+ test_null(container->memc);
+ test_true(container->parent);
container->memc= memcached_clone(NULL, container->parent);
test_true(container->memc);
return TEST_SUCCESS;
}
+test_return_t world_flush(libmemcached_test_container_st *container);
test_return_t world_flush(libmemcached_test_container_st *container)
{
- assert(container->memc);
+ test_true(container->memc);
memcached_flush(container->memc, 0);
memcached_quit(container->memc);
return TEST_SUCCESS;
}
-test_return_t world_pre_run(libmemcached_test_container_st *container)
+static test_return_t world_pre_run(libmemcached_test_container_st *container)
{
- assert(container->memc);
+ test_true(container->memc);
for (uint32_t loop= 0; loop < memcached_server_list_count(container->memc->servers); loop++)
{
memcached_server_instance_st instance=
memcached_server_instance_by_position(container->memc, loop);
test_compare(-1, instance->fd);
- test_compare(0, instance->cursor_active);
+ test_compare(0U, instance->cursor_active);
}
return TEST_SUCCESS;
}
-test_return_t world_post_run(libmemcached_test_container_st *container)
+static test_return_t world_post_run(libmemcached_test_container_st *container)
{
test_true(container->memc);
return TEST_SUCCESS;
}
-test_return_t world_on_error(test_return_t test_state, libmemcached_test_container_st *container)
+static test_return_t world_on_error(test_return_t , libmemcached_test_container_st *container)
{
- (void)test_state;
- assert(container->memc);
+ test_true(container->memc);
memcached_free(container->memc);
container->memc= NULL;
return TEST_SUCCESS;
}
-test_return_t world_destroy(libmemcached_test_container_st *container)
+static bool world_destroy(void *object)
{
- server_startup_st *construct= &container->construct;
-
- server_shutdown(construct);
-
-#ifdef LIBMEMCACHED_WITH_SASL_SUPPORT
- sasl_done();
+ libmemcached_test_container_st *container= (libmemcached_test_container_st *)object;
+#if defined(LIBMEMCACHED_WITH_SASL_SUPPORT) && LIBMEMCACHED_WITH_SASL_SUPPORT
+ if (LIBMEMCACHED_WITH_SASL_SUPPORT)
+ {
+ sasl_done();
+ }
#endif
delete container;
}
typedef test_return_t (*libmemcached_test_callback_fn)(memcached_st *);
+
static test_return_t _runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
{
if (func)
{
- assert(container);
- assert(container->memc);
- return func(container->memc);
- }
- else
- {
- return TEST_SUCCESS;
+ test_true(container);
+ test_true(container->memc);
+ test_return_t ret;
+ try {
+ ret= func(container->memc);
+ }
+ catch (std::exception& e)
+ {
+ libtest::Error << e.what();
+ return TEST_FAILURE;
+ }
+
+ return ret;
}
+
+ return TEST_SUCCESS;
}
static test_return_t _pre_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
{
return func(container->parent);
}
- else
- {
- return TEST_SUCCESS;
- }
+
+ return TEST_SUCCESS;
}
static test_return_t _post_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
{
return func(container->parent);
}
- else
- {
- return TEST_SUCCESS;
- }
-}
-#ifdef __cplusplus
+ return TEST_SUCCESS;
}
-#endif
-#ifdef __cplusplus
-
-static Runner defualt_libmemcached_runner= {
- reinterpret_cast<test_callback_runner_fn*>(_pre_runner_default),
- reinterpret_cast<test_callback_runner_fn*>(_runner_default),
- reinterpret_cast<test_callback_runner_fn*>(_post_runner_default)
-};
+class LibmemcachedRunner : public libtest::Runner {
+public:
+ test_return_t run(test_callback_fn* func, void *object)
+ {
+ return _runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
+ }
-#else
+ test_return_t pre(test_callback_fn* func, void *object)
+ {
+ return _pre_runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
+ }
-static Runner defualt_libmemcached_runner= {
- (test_callback_runner_fn)_pre_runner_default,
- (test_callback_runner_fn)_runner_default,
- (test_callback_runner_fn)_post_runner_default
+ test_return_t post(test_callback_fn* func, void *object)
+ {
+ return _post_runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
+ }
};
-#endif
+static LibmemcachedRunner defualt_libmemcached_runner;