git.m6w6.name Git - m6w6/libmemcached/blob - tests/parser.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached Client and Server
   4  *
   5  *  Copyright (C) 2011 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 <libtest/common.h>
  39
  40 #include <vector>
  41 #include <iostream>
  42 #include <string>
  43 #include <cerrno>
  44 #include <cassert>
  45
  46 #define BUILDING_LIBMEMCACHED
  47 // !NEVER use common.h, always use memcached.h in your own apps
  48 #include <libmemcached/common.h>
  49 #include <libmemcached/util.h>
  50
  51 #include <tests/parser.h>
  52 #include <tests/print.h>
  53
  54 enum scanner_type_t
  55 {
  56   NIL,
  57   UNSIGNED,
  58   SIGNED,
  59   ARRAY
  60 };
  61
  62
  63 struct scanner_string_st {
  64   const char *c_str;
  65   size_t size;
  66 };
  67
  68 static inline scanner_string_st scanner_string(const char *arg, size_t arg_size)
  69 {
  70   scanner_string_st local= { arg, arg_size };
  71   return local;
  72 }
  73
  74 #define make_scanner_string(X) scanner_string((X), static_cast<size_t>(sizeof(X) - 1))
  75
  76 static struct scanner_string_st scanner_string_null= { 0, 0};
  77
  78 struct scanner_variable_t {
  79   enum scanner_type_t type;
  80   struct scanner_string_st option;
  81   struct scanner_string_st result;
  82   test_return_t (*check_func)(memcached_st *memc, const scanner_string_st &hostname);
  83 };
  84
  85 // Check and make sure the first host is what we expect it to be
  86 static test_return_t __check_host(memcached_st *memc, const scanner_string_st &hostname)
  87 {
  88   memcached_server_instance_st instance=
  89     memcached_server_instance_by_position(memc, 0);
  90
  91   test_true(instance);
  92
  93   const char *first_hostname = memcached_server_name(instance);
  94   test_true(first_hostname);
  95   test_strcmp(first_hostname, hostname.c_str);
  96
  97   return TEST_SUCCESS;
  98 }
  99
 100 // Check and make sure the prefix_key is what we expect it to be
 101 static test_return_t __check_namespace(memcached_st *memc, const scanner_string_st &arg)
 102 {
 103   const char *_namespace = memcached_get_namespace(memc);
 104   test_true(_namespace);
 105   test_strcmp(_namespace, arg.c_str);
 106
 107   return TEST_SUCCESS;
 108 }
 109
 110 static test_return_t __check_IO_MSG_WATERMARK(memcached_st *memc, const scanner_string_st &value)
 111 {
 112   uint64_t value_number;
 113
 114   value_number= atoll(value.c_str);
 115
 116   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK) == value_number);
 117   return TEST_SUCCESS;
 118 }
 119
 120 static test_return_t __check_REMOVE_FAILED_SERVERS(memcached_st *memc, const scanner_string_st &)
 121 {
 122   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS));
 123   return TEST_SUCCESS;
 124 }
 125
 126 static test_return_t __check_NOREPLY(memcached_st *memc, const scanner_string_st &)
 127 {
 128   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NOREPLY));
 129   return TEST_SUCCESS;
 130 }
 131
 132 static test_return_t __check_VERIFY_KEY(memcached_st *memc, const scanner_string_st &)
 133 {
 134   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_VERIFY_KEY));
 135   return TEST_SUCCESS;
 136 }
 137
 138 static test_return_t __check_distribution_RANDOM(memcached_st *memc, const scanner_string_st &)
 139 {
 140   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION) == MEMCACHED_DISTRIBUTION_RANDOM);
 141   return TEST_SUCCESS;
 142 }
 143
 144 scanner_variable_t test_server_strings[]= {
 145   { ARRAY, make_scanner_string("--server=localhost"), make_scanner_string("localhost"), __check_host },
 146   { ARRAY, make_scanner_string("--server=10.0.2.1"), make_scanner_string("10.0.2.1"), __check_host },
 147   { ARRAY, make_scanner_string("--server=example.com"), make_scanner_string("example.com"), __check_host },
 148   { ARRAY, make_scanner_string("--server=localhost:30"), make_scanner_string("localhost"), __check_host },
 149   { ARRAY, make_scanner_string("--server=10.0.2.1:20"), make_scanner_string("10.0.2.1"), __check_host },
 150   { ARRAY, make_scanner_string("--server=example.com:1024"), make_scanner_string("example.com"), __check_host },
 151   { NIL, scanner_string_null, scanner_string_null, NULL }
 152 };
 153
 154 scanner_variable_t test_server_strings_with_weights[]= {
 155   { ARRAY, make_scanner_string("--server=10.0.2.1:30/?40"), make_scanner_string("10.0.2.1"), __check_host },
 156   { ARRAY, make_scanner_string("--server=example.com:1024/?30"), make_scanner_string("example.com"), __check_host },
 157   { ARRAY, make_scanner_string("--server=10.0.2.1/?20"), make_scanner_string("10.0.2.1"), __check_host },
 158   { ARRAY, make_scanner_string("--server=example.com/?10"), make_scanner_string("example.com"), __check_host },
 159   { NIL, scanner_string_null, scanner_string_null, NULL }
 160 };
 161
 162 scanner_variable_t bad_test_strings[]= {
 163   { ARRAY, make_scanner_string("-servers=localhost:11221,localhost:11222,localhost:11223,localhost:11224,localhost:11225"), scanner_string_null, NULL },
 164   { ARRAY, make_scanner_string("-- servers=a.example.com:81,localhost:82,b.example.com"), scanner_string_null, NULL },
 165   { ARRAY, make_scanner_string("--servers=localhost:+80"), scanner_string_null, NULL},
 166   { ARRAY, make_scanner_string("--servers=localhost.com."), scanner_string_null, NULL},
 167   { ARRAY, make_scanner_string("--server=localhost.com."), scanner_string_null, NULL},
 168   { ARRAY, make_scanner_string("--server=localhost.com.:80"), scanner_string_null, NULL},
 169   { NIL, scanner_string_null, scanner_string_null, NULL}
 170 };
 171
 172 scanner_variable_t test_number_options[]= {
 173   { ARRAY,  make_scanner_string("--CONNECT-TIMEOUT=456"), scanner_string_null, NULL },
 174   { ARRAY,  make_scanner_string("--IO-BYTES-WATERMARK=456"), scanner_string_null, NULL },
 175   { ARRAY,  make_scanner_string("--IO-KEY-PREFETCH=456"), scanner_string_null, NULL },
 176   { ARRAY,  make_scanner_string("--IO-MSG-WATERMARK=456"), make_scanner_string("456"), __check_IO_MSG_WATERMARK },
 177   { ARRAY,  make_scanner_string("--NUMBER-OF-REPLICAS=456"), scanner_string_null, NULL },
 178   { ARRAY,  make_scanner_string("--POLL-TIMEOUT=456"), scanner_string_null, NULL },
 179   { ARRAY,  make_scanner_string("--RCV-TIMEOUT=456"), scanner_string_null, NULL },
 180   { ARRAY,  make_scanner_string("--REMOVE-FAILED-SERVERS=3"), scanner_string_null, __check_REMOVE_FAILED_SERVERS },
 181   { ARRAY,  make_scanner_string("--RETRY-TIMEOUT=456"), scanner_string_null, NULL },
 182   { ARRAY,  make_scanner_string("--SND-TIMEOUT=456"), scanner_string_null, NULL },
 183   { ARRAY,  make_scanner_string("--SOCKET-RECV-SIZE=456"), scanner_string_null, NULL },
 184   { ARRAY,  make_scanner_string("--SOCKET-SEND-SIZE=456"), scanner_string_null, NULL },
 185   { NIL, scanner_string_null, scanner_string_null, NULL}
 186 };
 187
 188 scanner_variable_t test_boolean_options[]= {
 189   { ARRAY,  make_scanner_string("--BINARY-PROTOCOL"), scanner_string_null, NULL },
 190   { ARRAY,  make_scanner_string("--BUFFER-REQUESTS"), scanner_string_null, NULL },
 191   { ARRAY,  make_scanner_string("--HASH-WITH-NAMESPACE"), scanner_string_null, NULL },
 192   { ARRAY,  make_scanner_string("--NOREPLY"), scanner_string_null, __check_NOREPLY },
 193   { ARRAY,  make_scanner_string("--RANDOMIZE-REPLICA-READ"), scanner_string_null, NULL },
 194   { ARRAY,  make_scanner_string("--SORT-HOSTS"), scanner_string_null, NULL },
 195   { ARRAY,  make_scanner_string("--SUPPORT-CAS"), scanner_string_null, NULL },
 196   { ARRAY,  make_scanner_string("--TCP-NODELAY"), scanner_string_null, NULL },
 197   { ARRAY,  make_scanner_string("--TCP-KEEPALIVE"), scanner_string_null, NULL },
 198   { ARRAY,  make_scanner_string("--TCP-KEEPIDLE"), scanner_string_null, NULL },
 199   { ARRAY,  make_scanner_string("--USE-UDP"), scanner_string_null, NULL },
 200   { ARRAY,  make_scanner_string("--VERIFY-KEY"), scanner_string_null, __check_VERIFY_KEY },
 201   { NIL, scanner_string_null, scanner_string_null, NULL}
 202 };
 203
 204 scanner_variable_t namespace_strings[]= {
 205   { ARRAY, make_scanner_string("--NAMESPACE=foo"), make_scanner_string("foo"), __check_namespace },
 206   { ARRAY, make_scanner_string("--NAMESPACE=\"foo\""), make_scanner_string("foo"), __check_namespace },
 207   { ARRAY, make_scanner_string("--NAMESPACE=\"This_is_a_very_long_key\""), make_scanner_string("This_is_a_very_long_key"), __check_namespace },
 208   { NIL, scanner_string_null, scanner_string_null, NULL}
 209 };
 210
 211 scanner_variable_t distribution_strings[]= {
 212   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent"), scanner_string_null, NULL },
 213   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent,CRC"), scanner_string_null, NULL },
 214   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent,MD5"), scanner_string_null, NULL },
 215   { ARRAY,  make_scanner_string("--DISTRIBUTION=random"), scanner_string_null, __check_distribution_RANDOM },
 216   { ARRAY,  make_scanner_string("--DISTRIBUTION=modula"), scanner_string_null, NULL },
 217   { NIL, scanner_string_null, scanner_string_null, NULL}
 218 };
 219
 220 scanner_variable_t hash_strings[]= {
 221   { ARRAY,  make_scanner_string("--HASH=CRC"), scanner_string_null, NULL },
 222   { ARRAY,  make_scanner_string("--HASH=FNV1A_32"), scanner_string_null, NULL },
 223   { ARRAY,  make_scanner_string("--HASH=FNV1A_64"), scanner_string_null, NULL },
 224   { ARRAY,  make_scanner_string("--HASH=FNV1_32"), scanner_string_null, NULL },
 225   { ARRAY,  make_scanner_string("--HASH=FNV1_64"), scanner_string_null, NULL },
 226   { ARRAY,  make_scanner_string("--HASH=JENKINS"), scanner_string_null, NULL },
 227   { ARRAY,  make_scanner_string("--HASH=MD5"), scanner_string_null, NULL },
 228   { ARRAY,  make_scanner_string("--HASH=MURMUR"), scanner_string_null, NULL },
 229   { NIL, scanner_string_null, scanner_string_null, NULL}
 230 };
 231
 232
 233 static test_return_t _test_option(scanner_variable_t *scanner, bool test_true_opt= true)
 234 {
 235   for (scanner_variable_t *ptr= scanner; ptr->type != NIL; ptr++)
 236   {
 237     memcached_st *memc= memcached(ptr->option.c_str, ptr->option.size);
 238     if (test_true_opt)
 239     {
 240       char buffer[2048];
 241       if (not memc)
 242       {
 243         libmemcached_check_configuration(ptr->option.c_str, ptr->option.size, buffer, sizeof(buffer));
 244       }
 245
 246       std::string temp(buffer);
 247       temp+= " with option string:";
 248       temp+= ptr->option.c_str;
 249
 250       test_true_got(memc, temp.c_str());
 251
 252       if (ptr->check_func)
 253       {
 254         test_return_t test_rc= (*ptr->check_func)(memc, ptr->result);
 255         if (test_rc != TEST_SUCCESS)
 256         {
 257           memcached_free(memc);
 258           return test_rc;
 259         }
 260       }
 261
 262       memcached_free(memc);
 263     }
 264     else
 265     {
 266       test_false_with(memc, ptr->option.c_str);
 267     }
 268   }
 269
 270   return TEST_SUCCESS;
 271 }
 272
 273 test_return_t server_test(memcached_st *)
 274 {
 275   return _test_option(test_server_strings);
 276 }
 277
 278 test_return_t server_with_weight_test(memcached_st *)
 279 {
 280   return _test_option(test_server_strings_with_weights);
 281 }
 282
 283 test_return_t servers_bad_test(memcached_st *)
 284 {
 285   test_return_t rc;
 286   if ((rc= _test_option(bad_test_strings, false)) != TEST_SUCCESS)
 287   {
 288     return rc;
 289   }
 290
 291   return TEST_SUCCESS;
 292 }
 293
 294 test_return_t parser_number_options_test(memcached_st*)
 295 {
 296   return _test_option(test_number_options);
 297 }
 298
 299 test_return_t parser_boolean_options_test(memcached_st*)
 300 {
 301   return _test_option(test_boolean_options);
 302 }
 303
 304 test_return_t behavior_parser_test(memcached_st*)
 305 {
 306   return TEST_SUCCESS;
 307 }
 308
 309 test_return_t parser_hash_test(memcached_st*)
 310 {
 311   return _test_option(hash_strings);
 312 }
 313
 314 test_return_t parser_distribution_test(memcached_st*)
 315 {
 316   return _test_option(distribution_strings);
 317 }
 318
 319 test_return_t parser_key_prefix_test(memcached_st*)
 320 {
 321   return _test_option(distribution_strings);
 322 }
 323
 324 test_return_t test_namespace_keyword(memcached_st*)
 325 {
 326   return _test_option(namespace_strings);
 327 }
 328
 329 #define SUPPORT_EXAMPLE_CNF "support/example.cnf"
 330
 331 test_return_t memcached_create_with_options_with_filename(memcached_st*)
 332 {
 333   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 334     return TEST_SKIPPED;
 335
 336   memcached_st *memc_ptr;
 337   memc_ptr= memcached(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""));
 338   test_true_got(memc_ptr, "memcached() failed");
 339   test_strcmp(SUPPORT_EXAMPLE_CNF, memcached_array_string(memc_ptr->configure.filename));
 340   memcached_free(memc_ptr);
 341
 342   return TEST_SUCCESS;
 343 }
 344
 345 test_return_t libmemcached_check_configuration_with_filename_test(memcached_st*)
 346 {
 347   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 348     return TEST_SKIPPED;
 349
 350   memcached_return_t rc;
 351   char buffer[BUFSIZ];
 352
 353   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""), buffer, sizeof(buffer));
 354   test_true_got(rc == MEMCACHED_SUCCESS, (rc == MEMCACHED_ERRNO) ? strerror(errno) : memcached_strerror(NULL, rc));
 355
 356   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=support/example.cnf"), buffer, sizeof(buffer));
 357   test_false_with(rc == MEMCACHED_SUCCESS, memcached_strerror(NULL, rc));
 358
 359   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"bad-path/example.cnf\""), buffer, sizeof(buffer));
 360   test_true_got(rc == MEMCACHED_ERRNO, memcached_strerror(NULL, rc));
 361
 362   return TEST_SUCCESS;
 363 }
 364
 365 test_return_t libmemcached_check_configuration_test(memcached_st*)
 366 {
 367   memcached_return_t rc;
 368   char buffer[BUFSIZ];
 369
 370   rc= libmemcached_check_configuration(test_literal_param("--server=localhost"), buffer, sizeof(buffer));
 371   test_true_got(rc == MEMCACHED_SUCCESS, buffer);
 372
 373   rc= libmemcached_check_configuration(test_literal_param("--dude=localhost"), buffer, sizeof(buffer));
 374   test_false_with(rc == MEMCACHED_SUCCESS, buffer);
 375   test_true(rc == MEMCACHED_PARSE_ERROR);
 376
 377   return TEST_SUCCESS;
 378 }
 379
 380 test_return_t memcached_create_with_options_test(memcached_st*)
 381 {
 382   memcached_st *memc_ptr;
 383   memc_ptr= memcached(test_literal_param("--server=localhost"));
 384   test_true_got(memc_ptr, memcached_last_error_message(memc_ptr));
 385   memcached_free(memc_ptr);
 386
 387   memc_ptr= memcached(test_literal_param("--dude=localhost"));
 388   test_false_with(memc_ptr, memcached_last_error_message(memc_ptr));
 389
 390   return TEST_SUCCESS;
 391 }
 392
 393 test_return_t test_include_keyword(memcached_st*)
 394 {
 395   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 396     return TEST_SKIPPED;
 397
 398   char buffer[BUFSIZ];
 399   memcached_return_t rc;
 400   rc= libmemcached_check_configuration(test_literal_param("INCLUDE \"support/example.cnf\""), buffer, sizeof(buffer));
 401   test_true_got(rc == MEMCACHED_SUCCESS, buffer);
 402
 403   return TEST_SUCCESS;
 404 }
 405
 406 test_return_t test_end_keyword(memcached_st*)
 407 {
 408   char buffer[BUFSIZ];
 409   memcached_return_t rc;
 410   rc= libmemcached_check_configuration(test_literal_param("--server=localhost END bad keywords"), buffer, sizeof(buffer));
 411   test_true_got(rc == MEMCACHED_SUCCESS, buffer);
 412
 413   return TEST_SUCCESS;
 414 }
 415
 416 test_return_t test_reset_keyword(memcached_st*)
 417 {
 418   char buffer[BUFSIZ];
 419   memcached_return_t rc;
 420   rc= libmemcached_check_configuration(test_literal_param("--server=localhost reset --server=bad.com"), buffer, sizeof(buffer));
 421   test_true_got(rc == MEMCACHED_SUCCESS, buffer);
 422
 423   return TEST_SUCCESS;
 424 }
 425
 426 test_return_t test_error_keyword(memcached_st*)
 427 {
 428   char buffer[BUFSIZ];
 429   memcached_return_t rc;
 430   rc= libmemcached_check_configuration(test_literal_param("--server=localhost ERROR --server=bad.com"), buffer, sizeof(buffer));
 431   test_true_got(rc != MEMCACHED_SUCCESS, buffer);
 432
 433   return TEST_SUCCESS;
 434 }
 435
 436 #define RANDOM_STRINGS 100
 437 test_return_t random_statement_build_test(memcached_st*)
 438 {
 439   std::vector<scanner_string_st *> option_list;
 440
 441   for (scanner_variable_t *ptr= test_server_strings; ptr->type != NIL; ptr++)
 442     option_list.push_back(&ptr->option);
 443
 444   for (scanner_variable_t *ptr= test_number_options; ptr->type != NIL; ptr++)
 445     option_list.push_back(&ptr->option);
 446
 447   for (scanner_variable_t *ptr= test_boolean_options; ptr->type != NIL; ptr++)
 448     option_list.push_back(&ptr->option);
 449
 450   for (scanner_variable_t *ptr= namespace_strings; ptr->type != NIL; ptr++)
 451     option_list.push_back(&ptr->option);
 452
 453   for (scanner_variable_t *ptr= distribution_strings; ptr->type != NIL; ptr++)
 454     option_list.push_back(&ptr->option);
 455
 456   for (scanner_variable_t *ptr= hash_strings; ptr->type != NIL; ptr++)
 457     option_list.push_back(&ptr->option);
 458
 459   for (uint32_t x= 0; x < RANDOM_STRINGS; x++)
 460   {
 461     std::string random_options;
 462
 463     uint32_t number_of= random() % option_list.size();
 464     for (uint32_t options= 0; options < number_of; options++)
 465     {
 466       random_options+= option_list[random() % option_list.size()]->c_str;
 467       random_options+= " ";
 468     }
 469
 470     memcached_st *memc_ptr= memcached(random_options.c_str(), random_options.size() -1);
 471     if (not memc_ptr)
 472     {
 473       switch (errno)
 474       {
 475       case EINVAL:
 476 #if 0 // Testing framework is not smart enough for this just yet.
 477         {
 478           // We will try to find the specific error
 479           char buffer[2048];
 480           memcached_return_t rc= libmemcached_check_configuration(random_options.c_str(), random_options.size(), buffer, sizeof(buffer));
 481           test_true_got(rc != MEMCACHED_SUCCESS, "memcached_create_with_options() failed whiled libmemcached_check_configuration() was successful");
 482           std::cerr << "Error occured on " << random_options.c_str() << " : " << buffer << std::endl;
 483           return TEST_FAILURE;
 484         }
 485 #endif
 486         break;
 487       case ENOMEM:
 488         std::cerr << "Failed to allocate memory for memcached_create_with_options()" << std::endl;
 489         memcached_free(memc_ptr);
 490         return TEST_FAILURE;
 491       default:
 492         std::cerr << "Unknown error from memcached_create_with_options?!!" << std::endl;
 493         memcached_free(memc_ptr);
 494         return TEST_FAILURE;
 495       }
 496     }
 497     memcached_free(memc_ptr);
 498   }
 499
 500   return TEST_SUCCESS;
 501 }
 502
 503 static memcached_return_t dump_server_information(const memcached_st *,
 504                                                   const memcached_server_st *instance,
 505                                                   void *)
 506 {
 507   if (strcmp(memcached_server_name(instance), "localhost"))
 508   {
 509     assert(not memcached_server_name(instance));
 510     return MEMCACHED_FAILURE;
 511   }
 512
 513   if (memcached_server_port(instance) < 8888 or memcached_server_port(instance) > 8892)
 514   {
 515     assert(not memcached_server_port(instance));
 516     return MEMCACHED_FAILURE;
 517   }
 518
 519   if (instance->weight > 5 or instance->weight < 2)
 520   {
 521     assert(not instance->weight);
 522     return MEMCACHED_FAILURE;
 523   }
 524
 525   return MEMCACHED_SUCCESS;
 526 }
 527
 528
 529 test_return_t test_hostname_port_weight(memcached_st *)
 530 {
 531   const char *server_string= "--server=localhost:8888/?2 --server=localhost:8889/?3 --server=localhost:8890/?4 --server=localhost:8891/?5 --server=localhost:8892/?3";
 532   char buffer[BUFSIZ];
 533
 534   memcached_return_t rc;
 535   test_compare_got(MEMCACHED_SUCCESS,
 536                    rc= libmemcached_check_configuration(server_string, strlen(server_string), buffer, sizeof(buffer)),
 537                    memcached_strerror(NULL, rc));
 538
 539   memcached_st *memc= memcached(server_string, strlen(server_string));
 540   test_true(memc);
 541
 542   memcached_server_fn callbacks[]= { dump_server_information };
 543   test_true(memcached_success(memcached_server_cursor(memc, callbacks, NULL, 1)));
 544
 545   memcached_free(memc);
 546
 547   return TEST_SUCCESS;
 548 }
 549
 550 struct socket_weight_t {
 551   const char *socket;
 552   size_t weight;
 553 };
 554
 555 static memcached_return_t dump_socket_information(const memcached_st *,
 556                                                   const memcached_server_st *instance,
 557                                                   void *context)
 558 {
 559   socket_weight_t *check= (socket_weight_t *)context;
 560
 561   if (strcmp(memcached_server_name(instance), check->socket))
 562   {
 563     std::cerr << std::endl << __FILE__ << ":" << __LINE__ << " " << memcached_server_name(instance) << " != " << check->socket << std::endl;
 564     return MEMCACHED_FAILURE;
 565   }
 566
 567   if (instance->weight == check->weight)
 568   {
 569     return MEMCACHED_SUCCESS;
 570   }
 571
 572   return MEMCACHED_FAILURE;
 573 }
 574
 575 test_return_t test_parse_socket(memcached_st *)
 576 {
 577   char buffer[BUFSIZ];
 578
 579   memcached_server_fn callbacks[]= { dump_socket_information };
 580   {
 581     test_compare_got(MEMCACHED_SUCCESS,
 582                      libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\""), buffer, sizeof(buffer)),
 583                      buffer);
 584
 585     memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\""));
 586     test_true(memc);
 587     socket_weight_t check= { "/tmp/foo", 1 };
 588     test_compare(MEMCACHED_SUCCESS,
 589                  memcached_server_cursor(memc, callbacks, &check, 1));
 590     memcached_free(memc);
 591   }
 592
 593   {
 594     test_compare_got(MEMCACHED_SUCCESS,
 595                      libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\"/?23"), buffer, sizeof(buffer)),
 596                      buffer);
 597
 598     memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\"/?23"));
 599     test_true(memc);
 600     socket_weight_t check= { "/tmp/foo", 23 };
 601     test_compare(MEMCACHED_SUCCESS,
 602                  memcached_server_cursor(memc, callbacks, &check, 1));
 603     memcached_free(memc);
 604   }
 605
 606   return TEST_SUCCESS;
 607 }
 608
 609 /*
 610   By setting the timeout value to zero, we force poll() to return immediatly.
 611 */
 612 test_return_t regression_bug_71231153_connect(memcached_st *)
 613 {
 614   if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
 615     return TEST_SKIPPED;
 616
 617   { // Test the connect-timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
 618     memcached_st *memc= memcached(memcached_literal_param("--SERVER=10.0.2.252 --CONNECT-TIMEOUT=0"));
 619     test_true(memc);
 620     test_compare(0, memc->connect_timeout);
 621     test_compare(MEMCACHED_DEFAULT_TIMEOUT, memc->poll_timeout);
 622
 623     memcached_return_t rc;
 624     size_t value_len;
 625     char *value= memcached_get(memc, memcached_literal_param("test"), &value_len, NULL, &rc);
 626     test_false(value);
 627     test_compare(0, value_len);
 628     test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_strerror(NULL, rc));
 629
 630     memcached_free(memc);
 631   }
 632
 633   return TEST_SUCCESS;
 634 }
 635
 636 test_return_t regression_bug_71231153_poll(memcached_st *)
 637 {
 638   if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
 639     return TEST_SKIPPED;
 640
 641   { // Test the poll timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
 642     memcached_st *memc= memcached(memcached_literal_param("--SERVER=10.0.2.252 --POLL-TIMEOUT=0"));
 643     test_true(memc);
 644     test_compare(MEMCACHED_DEFAULT_CONNECT_TIMEOUT, memc->connect_timeout);
 645     test_compare(0, memc->poll_timeout);
 646
 647     memcached_return_t rc;
 648     size_t value_len;
 649     char *value= memcached_get(memc, memcached_literal_param("test"), &value_len, NULL, &rc);
 650     test_false(value);
 651     test_compare(0, value_len);
 652     test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_strerror(NULL, rc));
 653
 654     memcached_free(memc);
 655   }
 656
 657   return TEST_SUCCESS;
 658 }