git.m6w6.name Git - m6w6/libmemcached/blob - tests/libmemcached-1.0/setup_and_teardowns.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached Client and Server
   4  *
   5  *  Copyright (C) 2012 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #include "mem_config.h"
  39 #include "libtest/test.hpp"
  40
  41 #include "libmemcachedutil-1.0/util.h"
  42
  43 #include "tests/print.h"
  44 #include "tests/libmemcached-1.0/setup_and_teardowns.h"
  45
  46 #include <sys/stat.h>
  47
  48 using namespace libtest;
  49
  50 memcached_return_t return_value_based_on_buffering(memcached_st *memc)
  51 {
  52   if (memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS))
  53   {
  54     return MEMCACHED_BUFFERED;
  55   }
  56
  57   return MEMCACHED_SUCCESS;
  58 }
  59
  60 /**
  61   @note This should be testing to see if the server really supports the binary protocol.
  62 */
  63 test_return_t pre_binary(memcached_st *memc)
  64 {
  65   test_true(memcached_server_count(memc) > 0);
  66   test_skip(true, libmemcached_util_version_check(memc, 1, 4, 4));
  67   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL, true));
  68
  69   return TEST_SUCCESS;
  70 }
  71
  72 test_return_t pre_buffer(memcached_st *memc)
  73 {
  74   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, true));
  75
  76   return TEST_SUCCESS;
  77 }
  78
  79 test_return_t pre_unix_socket(memcached_st *memc)
  80 {
  81   struct stat buf;
  82
  83   memcached_servers_reset(memc);
  84   const char *socket_file= libtest::default_socket();
  85   test_skip(true, bool(socket_file));
  86
  87   test_skip(0, stat(socket_file, &buf));
  88
  89   test_compare(MEMCACHED_SUCCESS,
  90                memcached_server_add_unix_socket_with_weight(memc, socket_file, 0));
  91
  92   return TEST_SUCCESS;
  93 }
  94
  95 test_return_t pre_nodelay(memcached_st *memc)
  96 {
  97   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
  98   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, 0);
  99
 100   return TEST_SUCCESS;
 101 }
 102
 103 test_return_t pre_settimer(memcached_st *memc)
 104 {
 105   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SND_TIMEOUT, 1000);
 106   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_RCV_TIMEOUT, 1000);
 107
 108   return TEST_SUCCESS;
 109 }
 110
 111 test_return_t pre_murmur(memcached_st *memc)
 112 {
 113   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_MURMUR));
 114   return TEST_SUCCESS;
 115 }
 116
 117 test_return_t pre_jenkins(memcached_st *memc)
 118 {
 119   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_JENKINS);
 120
 121   return TEST_SKIPPED;
 122 }
 123
 124
 125 test_return_t pre_md5(memcached_st *memc)
 126 {
 127   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_MD5);
 128
 129   return TEST_SUCCESS;
 130 }
 131
 132 test_return_t pre_crc(memcached_st *memc)
 133 {
 134   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_CRC);
 135
 136   return TEST_SUCCESS;
 137 }
 138
 139 test_return_t pre_hsieh(memcached_st *memc)
 140 {
 141   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_HSIEH));
 142   return TEST_SUCCESS;
 143 }
 144
 145 test_return_t pre_hash_fnv1_64(memcached_st *memc)
 146 {
 147   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_MURMUR));
 148
 149   return TEST_SUCCESS;
 150 }
 151
 152 test_return_t pre_hash_fnv1a_64(memcached_st *memc)
 153 {
 154   test_skip(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1A_64));
 155
 156   return TEST_SUCCESS;
 157 }
 158
 159 test_return_t pre_hash_fnv1_32(memcached_st *memc)
 160 {
 161   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1_32);
 162
 163   return TEST_SUCCESS;
 164 }
 165
 166 test_return_t pre_hash_fnv1a_32(memcached_st *memc)
 167 {
 168   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, (uint64_t)MEMCACHED_HASH_FNV1A_32);
 169
 170   return TEST_SUCCESS;
 171 }
 172
 173 test_return_t memcached_servers_reset_SETUP(memcached_st *memc)
 174 {
 175   memcached_servers_reset(memc);
 176   test_compare(0U, memcached_server_count(memc));
 177   return TEST_SUCCESS;
 178 }
 179
 180 test_return_t memcached_servers_reset_CONTINUUM(memcached_st *memc)
 181 {
 182   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, MEMCACHED_DISTRIBUTION_CONSISTENT);
 183   memcached_servers_reset(memc);
 184   test_zero(memc->ketama.continuum_count);
 185   // If memc->ketama.continuum_count is non-zero at this point, any call to
 186   // memcached_server_add will cause a segfault.
 187   return TEST_SUCCESS;
 188 }
 189
 190 test_return_t memcached_servers_reset_MEMCACHED_DISTRIBUTION_CONSISTENT_SETUP(memcached_st *memc)
 191 {
 192   test_compare(TEST_SUCCESS, memcached_servers_reset_SETUP(memc));
 193
 194   test_compare(MEMCACHED_SUCCESS, memcached_behavior_set_distribution(memc, MEMCACHED_DISTRIBUTION_CONSISTENT));
 195   test_compare(memcached_behavior_get_distribution(memc), MEMCACHED_DISTRIBUTION_CONSISTENT);
 196
 197   return TEST_SUCCESS;
 198 }
 199
 200 test_return_t memcached_servers_reset_MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED_SETUP(memcached_st *memc)
 201 {
 202   test_compare(TEST_SUCCESS, memcached_servers_reset_SETUP(memc));
 203   ASSERT_EQ(0U, memcached_server_count(0));
 204
 205   test_compare(MEMCACHED_SUCCESS, memcached_behavior_set_distribution(memc, MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED));
 206   test_compare(memcached_behavior_get_distribution(memc), MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED);
 207
 208   return TEST_SUCCESS;
 209 }
 210
 211 test_return_t pre_behavior_ketama(memcached_st *memc)
 212 {
 213   test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_KETAMA, 1));
 214
 215   test_compare(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_KETAMA), uint64_t(1));
 216
 217   return TEST_SUCCESS;
 218 }
 219
 220 test_return_t pre_behavior_ketama_weighted(memcached_st *memc)
 221 {
 222   test_compare(memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED, true), MEMCACHED_SUCCESS);
 223
 224   test_compare(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED), uint64_t(1));
 225
 226   test_compare(memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_KETAMA_HASH, MEMCACHED_HASH_MD5), MEMCACHED_SUCCESS);
 227
 228   test_compare(memcached_hash_t(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_KETAMA_HASH)), MEMCACHED_HASH_MD5);
 229
 230   return TEST_SUCCESS;
 231 }
 232
 233 test_return_t pre_replication(memcached_st *memc)
 234 {
 235   test_skip(TEST_SUCCESS, pre_binary(memc));
 236
 237   /*
 238    * Make sure that we store the item on all servers
 239    * (master + replicas == number of servers)
 240  */
 241   test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS, memcached_server_count(memc) - 1));
 242   test_compare(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS), uint64_t(memcached_server_count(memc) - 1));
 243
 244   return TEST_SUCCESS;
 245 }
 246
 247
 248 test_return_t pre_replication_noblock(memcached_st *memc)
 249 {
 250   test_skip(TEST_SUCCESS, pre_replication(memc));
 251
 252   return pre_nonblock(memc);
 253 }
 254
 255 test_return_t pre_nonblock(memcached_st *memc)
 256 {
 257   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
 258
 259   return TEST_SUCCESS;
 260 }
 261
 262 test_return_t pre_nonblock_binary(memcached_st *memc)
 263 {
 264   memcached_st *memc_clone= memcached_clone(NULL, memc);
 265   test_true(memc_clone);
 266
 267   // The memcached_version needs to be done on a clone, because the server
 268   // will not toggle protocol on an connection.
 269   memcached_version(memc_clone);
 270
 271   memcached_return_t rc= MEMCACHED_FAILURE;
 272   if (libmemcached_util_version_check(memc_clone, 1, 4, 4))
 273   {
 274     memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
 275     test_compare(MEMCACHED_SUCCESS,
 276                  memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL, 1));
 277     test_compare(uint64_t(1), memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL));
 278   }
 279   else
 280   {
 281     memcached_free(memc_clone);
 282     return TEST_SKIPPED;
 283   }
 284
 285   memcached_free(memc_clone);
 286
 287   return rc == MEMCACHED_SUCCESS ? TEST_SUCCESS : TEST_SKIPPED;
 288 }