+++ /dev/null
-/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
- *
- * Libmemcached library
- *
- * Copyright (C) 2011 Data Differential, http://datadifferential.com/
- *
- * 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.
- *
- */
-
-#include "mem_config.h"
-
-/*
- C++ interface test
-*/
-#include "libmemcached-1.0/memcached.hpp"
-#include "libmemcached/server_instance.h"
-#include "libtest/test.hpp"
-
-#include <cstdio>
-#include <cstdlib>
-#include <cstring>
-#include <sys/time.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <unistd.h>
-#include <ctime>
-
-#include <string>
-#include <iostream>
-
-using namespace std;
-using namespace memcache;
-using namespace libtest;
-
-libtest::Framework *global_framework= NULL;
-
-static test_return_t shutdown_servers(memcached_st *memc)
-{
- test_compare(memcached_server_count(memc), 1U);
-
- // Disable a single server, just the first
- test_true(global_framework->servers().shutdown(0));
-
- return TEST_SUCCESS;
-}
-
-static test_return_t add_shutdown_servers(memcached_st *memc)
-{
- while (memcached_server_count(memc) < 2)
- {
- const char *argv[2]= { "add_shutdown_server", NULL };
- in_port_t port = libtest::default_port();
- test_true(global_framework->servers().start_socket_server("memcached", port, argv));
- test_compare(MEMCACHED_SUCCESS, memcached_server_add(memc, "localhost", port));
- }
-
- // Disable a single server, just the first
- test_true(global_framework->servers().shutdown(0));
-
- return TEST_SUCCESS;
-}
-
-static test_return_t restart_servers(memcached_st *)
-{
- // Restart the servers
- global_framework->servers().restart();
-
- return TEST_SUCCESS;
-}
-
-#include "libmemcached/instance.hpp"
-static test_return_t cull_TEST(memcached_st *memc)
-{
- shutdown_servers(memc);
-
- uint32_t count= memcached_server_count(memc);
-
- // Do not do this in your code, it is not supported.
- memc->servers[0].options.is_dead= true;
- memc->state.is_time_for_rebuild= true;
-
- uint32_t new_count= memcached_server_count(memc);
- test_compare(count, new_count);
-
- return restart_servers(memc);
-}
-
-static test_return_t MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST(memcached_st *memc)
-{
- shutdown_servers(memc);
-
- test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT, 30));
- test_compare_got(MEMCACHED_CONNECTION_FAILURE,
- memcached_set(memc,
- test_literal_param("foo"),
- NULL, 0, time_t(0), uint32_t(0)),
- memcached_last_error_message(memc));
-
- /*
- Setting server_failure_counter==0 should not influence the timeout that we set above,
- since we check the timeout that is created by the failure before we check how many times
- a server has failed.
- */
- test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED,
- memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0)));
-
- return restart_servers(memc);
-}
-
-static test_return_t MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST(memcached_st *memc)
-{
- shutdown_servers(memc);
-
- test_compare_got(MEMCACHED_CONNECTION_FAILURE,
- memcached_set(memc,
- test_literal_param("foo"),
- NULL, 0, time_t(0), uint32_t(0)),
- memcached_last_error_message(memc));
-
- /*
- Setting server_failure_counter==0 should not influence the timeout that we set above,
- since we check the timeout that is created by the failure before we check how many times
- a server has failed.
- */
- test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED,
- memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0)));
-
- global_framework->servers().restart();
-
- int limit= 10;
- memcached_return_t ret;
- do {
- libtest::dream(1, 0);
- ret= memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0));
- } while (ret == MEMCACHED_SERVER_TEMPORARILY_DISABLED and --limit);
-
- test_true(limit);
-
- test_compare_got(MEMCACHED_SUCCESS, ret, memcached_last_error_message(memc));
-
- return restart_servers(memc);
-}
-
-static test_return_t MEMCACHED_SERVER_MARKED_DEAD_TEST(memcached_st *memc)
-{
- add_shutdown_servers(memc);
-
- test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT, 2));
- test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT, 1));
- test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS, true));
-
- memcached_return_t ret;
- do {
- ret= memcached_set(memc,
- test_literal_param("foo"),
- NULL, 0, time_t(0), uint32_t(0));
- } while (ret == MEMCACHED_SUCCESS or ret == MEMCACHED_CONNECTION_FAILURE);
- test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED, ret);
-
- int limit= 10;
- do {
- libtest::dream(1, 0);
- ret= memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0));
- } while ((ret == MEMCACHED_SERVER_TEMPORARILY_DISABLED or ret == MEMCACHED_SUCCESS) and --limit);
-
- test_true(limit);
-
- test_compare_got(MEMCACHED_SERVER_MARKED_DEAD, ret, memcached_last_error_message(memc));
-
- return restart_servers(memc);
-}
-
-test_st cull_TESTS[] ={
- { "cull servers", true, (test_callback_fn*)cull_TEST },
- { 0, 0, 0 }
-};
-
-test_st server_temporarily_disabled_TESTS[] ={
- { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED -> MEMCACHED_SUCCESS)", true, (test_callback_fn*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST },
- { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED)", true, (test_callback_fn*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST },
- { 0, 0, 0 }
-};
-
-test_st server_permanently_disabled_TESTS[] ={
- { "memcached_set(MEMCACHED_SERVER_MARKED_DEAD)", true, (test_callback_fn*)MEMCACHED_SERVER_MARKED_DEAD_TEST },
- { 0, 0, 0 }
-};
-
-collection_st collection[] ={
- { "cull", NULL, NULL, cull_TESTS },
- { "server failed", NULL, NULL, server_temporarily_disabled_TESTS },
- { "server eject", NULL, NULL, server_permanently_disabled_TESTS },
- { 0, 0, 0, 0 }
-};
-
-#include "tests/libmemcached_world.h"
-
-void get_world(libtest::Framework* world)
-{
- world->servers().set_servers_to_run(1);
-
- world->collections(collection);
-
- world->create((test_callback_create_fn*)world_create);
- world->destroy((test_callback_destroy_fn*)world_destroy);
-
- world->set_runner(new LibmemcachedRunner);
-
- global_framework= world;
-}
projects / awesomized / libmemcached / blobdiff