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 <config.h>
  39 #include <libtest/test.hpp>
  40
  41 using namespace libtest;
  42
  43 #include <vector>
  44 #include <iostream>
  45 #include <string>
  46 #include <cerrno>
  47 #include <cassert>
  48
  49 #include <libmemcached/memcached.h>
  50 #include <libmemcached/util.h>
  51
  52 #include <tests/parser.h>
  53 #include <tests/print.h>
  54
  55 enum scanner_type_t
  56 {
  57   NIL,
  58   UNSIGNED,
  59   SIGNED,
  60   ARRAY
  61 };
  62
  63
  64 struct scanner_string_st {
  65   const char *c_str;
  66   size_t size;
  67 };
  68
  69 static inline scanner_string_st scanner_string(const char *arg, size_t arg_size)
  70 {
  71   scanner_string_st local= { arg, arg_size };
  72   return local;
  73 }
  74
  75 #define make_scanner_string(X) scanner_string((X), static_cast<size_t>(sizeof(X) - 1))
  76
  77 static struct scanner_string_st scanner_string_null= { 0, 0};
  78
  79 struct scanner_variable_t {
  80   enum scanner_type_t type;
  81   struct scanner_string_st option;
  82   struct scanner_string_st result;
  83   test_return_t (*check_func)(memcached_st *memc, const scanner_string_st &hostname);
  84 };
  85
  86 // Check and make sure the first host is what we expect it to be
  87 static test_return_t __check_host(memcached_st *memc, const scanner_string_st &hostname)
  88 {
  89   memcached_server_instance_st instance=
  90     memcached_server_instance_by_position(memc, 0);
  91
  92   test_true(instance);
  93
  94   const char *first_hostname = memcached_server_name(instance);
  95   test_true(first_hostname);
  96   test_strcmp(first_hostname, hostname.c_str);
  97
  98   return TEST_SUCCESS;
  99 }
 100
 101 // Check and make sure the prefix_key is what we expect it to be
 102 static test_return_t __check_namespace(memcached_st *, const scanner_string_st &)
 103 {
 104 #if 0
 105   const char *_namespace = memcached_get_namespace(memc);
 106   test_true(_namespace);
 107   test_strcmp(_namespace, arg.c_str);
 108 #endif
 109
 110   return TEST_SUCCESS;
 111 }
 112
 113 static test_return_t __check_IO_MSG_WATERMARK(memcached_st *memc, const scanner_string_st &value)
 114 {
 115   uint64_t value_number;
 116
 117   value_number= atoll(value.c_str);
 118
 119   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK) == value_number);
 120   return TEST_SUCCESS;
 121 }
 122
 123 static test_return_t __check_REMOVE_FAILED_SERVERS(memcached_st *memc, const scanner_string_st &)
 124 {
 125   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS));
 126   return TEST_SUCCESS;
 127 }
 128
 129 static test_return_t __check_NOREPLY(memcached_st *memc, const scanner_string_st &)
 130 {
 131   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NOREPLY));
 132   return TEST_SUCCESS;
 133 }
 134
 135 static test_return_t __check_VERIFY_KEY(memcached_st *memc, const scanner_string_st &)
 136 {
 137   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_VERIFY_KEY));
 138   return TEST_SUCCESS;
 139 }
 140
 141 static test_return_t __check_distribution_RANDOM(memcached_st *memc, const scanner_string_st &)
 142 {
 143   test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION) == MEMCACHED_DISTRIBUTION_RANDOM);
 144   return TEST_SUCCESS;
 145 }
 146
 147 scanner_variable_t test_server_strings[]= {
 148   { ARRAY, make_scanner_string("--server=localhost"), make_scanner_string("localhost"), __check_host },
 149   { ARRAY, make_scanner_string("--server=10.0.2.1"), make_scanner_string("10.0.2.1"), __check_host },
 150   { ARRAY, make_scanner_string("--server=example.com"), make_scanner_string("example.com"), __check_host },
 151   { ARRAY, make_scanner_string("--server=localhost:30"), make_scanner_string("localhost"), __check_host },
 152   { ARRAY, make_scanner_string("--server=10.0.2.1:20"), make_scanner_string("10.0.2.1"), __check_host },
 153   { ARRAY, make_scanner_string("--server=example.com:1024"), make_scanner_string("example.com"), __check_host },
 154   { NIL, scanner_string_null, scanner_string_null, NULL }
 155 };
 156
 157 scanner_variable_t test_server_strings_with_weights[]= {
 158   { ARRAY, make_scanner_string("--server=10.0.2.1:30/?40"), make_scanner_string("10.0.2.1"), __check_host },
 159   { ARRAY, make_scanner_string("--server=example.com:1024/?30"), make_scanner_string("example.com"), __check_host },
 160   { ARRAY, make_scanner_string("--server=10.0.2.1/?20"), make_scanner_string("10.0.2.1"), __check_host },
 161   { ARRAY, make_scanner_string("--server=example.com/?10"), make_scanner_string("example.com"), __check_host },
 162   { NIL, scanner_string_null, scanner_string_null, NULL }
 163 };
 164
 165 scanner_variable_t bad_test_strings[]= {
 166   { ARRAY, make_scanner_string("-servers=localhost:11221,localhost:11222,localhost:11223,localhost:11224,localhost:11225"), scanner_string_null, NULL },
 167   { ARRAY, make_scanner_string("-- servers=a.example.com:81,localhost:82,b.example.com"), scanner_string_null, NULL },
 168   { ARRAY, make_scanner_string("--servers=localhost:+80"), scanner_string_null, NULL},
 169   { ARRAY, make_scanner_string("--servers=localhost.com."), scanner_string_null, NULL},
 170   { ARRAY, make_scanner_string("--server=localhost.com."), scanner_string_null, NULL},
 171   { ARRAY, make_scanner_string("--server=localhost.com.:80"), scanner_string_null, NULL},
 172   { NIL, scanner_string_null, scanner_string_null, NULL}
 173 };
 174
 175 scanner_variable_t test_number_options[]= {
 176   { ARRAY,  make_scanner_string("--CONNECT-TIMEOUT=456"), scanner_string_null, NULL },
 177   { ARRAY,  make_scanner_string("--IO-BYTES-WATERMARK=456"), scanner_string_null, NULL },
 178   { ARRAY,  make_scanner_string("--IO-KEY-PREFETCH=456"), scanner_string_null, NULL },
 179   { ARRAY,  make_scanner_string("--IO-MSG-WATERMARK=456"), make_scanner_string("456"), __check_IO_MSG_WATERMARK },
 180   { ARRAY,  make_scanner_string("--NUMBER-OF-REPLICAS=456"), scanner_string_null, NULL },
 181   { ARRAY,  make_scanner_string("--POLL-TIMEOUT=456"), scanner_string_null, NULL },
 182   { ARRAY,  make_scanner_string("--RCV-TIMEOUT=456"), scanner_string_null, NULL },
 183   { ARRAY,  make_scanner_string("--REMOVE-FAILED-SERVERS=3"), scanner_string_null, __check_REMOVE_FAILED_SERVERS },
 184   { ARRAY,  make_scanner_string("--RETRY-TIMEOUT=456"), scanner_string_null, NULL },
 185   { ARRAY,  make_scanner_string("--SND-TIMEOUT=456"), scanner_string_null, NULL },
 186   { ARRAY,  make_scanner_string("--SOCKET-RECV-SIZE=456"), scanner_string_null, NULL },
 187   { ARRAY,  make_scanner_string("--SOCKET-SEND-SIZE=456"), scanner_string_null, NULL },
 188   { NIL, scanner_string_null, scanner_string_null, NULL}
 189 };
 190
 191 scanner_variable_t test_boolean_options[]= {
 192   { ARRAY,  make_scanner_string("--BINARY-PROTOCOL"), scanner_string_null, NULL },
 193   { ARRAY,  make_scanner_string("--BUFFER-REQUESTS"), scanner_string_null, NULL },
 194   { ARRAY,  make_scanner_string("--HASH-WITH-NAMESPACE"), scanner_string_null, NULL },
 195   { ARRAY,  make_scanner_string("--NOREPLY"), scanner_string_null, __check_NOREPLY },
 196   { ARRAY,  make_scanner_string("--RANDOMIZE-REPLICA-READ"), scanner_string_null, NULL },
 197   { ARRAY,  make_scanner_string("--SORT-HOSTS"), scanner_string_null, NULL },
 198   { ARRAY,  make_scanner_string("--SUPPORT-CAS"), scanner_string_null, NULL },
 199   { ARRAY,  make_scanner_string("--TCP-NODELAY"), scanner_string_null, NULL },
 200   { ARRAY,  make_scanner_string("--TCP-KEEPALIVE"), scanner_string_null, NULL },
 201   { ARRAY,  make_scanner_string("--TCP-KEEPIDLE"), scanner_string_null, NULL },
 202   { ARRAY,  make_scanner_string("--USE-UDP"), scanner_string_null, NULL },
 203   { ARRAY,  make_scanner_string("--VERIFY-KEY"), scanner_string_null, __check_VERIFY_KEY },
 204   { NIL, scanner_string_null, scanner_string_null, NULL}
 205 };
 206
 207 scanner_variable_t namespace_strings[]= {
 208   { ARRAY, make_scanner_string("--NAMESPACE=foo"), make_scanner_string("foo"), __check_namespace },
 209   { ARRAY, make_scanner_string("--NAMESPACE=\"foo\""), make_scanner_string("foo"), __check_namespace },
 210   { ARRAY, make_scanner_string("--NAMESPACE=\"This_is_a_very_long_key\""), make_scanner_string("This_is_a_very_long_key"), __check_namespace },
 211   { NIL, scanner_string_null, scanner_string_null, NULL}
 212 };
 213
 214 scanner_variable_t distribution_strings[]= {
 215   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent"), scanner_string_null, NULL },
 216   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent,CRC"), scanner_string_null, NULL },
 217   { ARRAY,  make_scanner_string("--DISTRIBUTION=consistent,MD5"), scanner_string_null, NULL },
 218   { ARRAY,  make_scanner_string("--DISTRIBUTION=random"), scanner_string_null, __check_distribution_RANDOM },
 219   { ARRAY,  make_scanner_string("--DISTRIBUTION=modula"), scanner_string_null, NULL },
 220   { NIL, scanner_string_null, scanner_string_null, NULL}
 221 };
 222
 223 scanner_variable_t hash_strings[]= {
 224   { ARRAY,  make_scanner_string("--HASH=CRC"), scanner_string_null, NULL },
 225   { ARRAY,  make_scanner_string("--HASH=FNV1A_32"), scanner_string_null, NULL },
 226   { ARRAY,  make_scanner_string("--HASH=FNV1_32"), scanner_string_null, NULL },
 227   { ARRAY,  make_scanner_string("--HASH=JENKINS"), scanner_string_null, NULL },
 228   { ARRAY,  make_scanner_string("--HASH=MD5"), 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
 239     // The case that it should have parsed, but it didn't. We will inspect
 240     // for an error with libmemcached_check_configuration()
 241     if (not memc and test_true_opt)
 242     {
 243       char buffer[2048];
 244       bool success= libmemcached_check_configuration(ptr->option.c_str, ptr->option.size, buffer, sizeof(buffer));
 245
 246       std::string temp(buffer);
 247       temp+= " with option string:";
 248       temp+= ptr->option.c_str;
 249       test_true_got(success, temp.c_str());
 250
 251       return TEST_FAILURE; // The line above should fail since memc should be null
 252     }
 253
 254     if (test_true_opt)
 255     {
 256       if (ptr->check_func)
 257       {
 258         test_return_t test_rc= (*ptr->check_func)(memc, ptr->result);
 259         if (test_rc != TEST_SUCCESS)
 260         {
 261           memcached_free(memc);
 262           return test_rc;
 263         }
 264       }
 265
 266       memcached_free(memc);
 267     }
 268     else
 269     {
 270       test_false_with(memc, ptr->option.c_str);
 271     }
 272   }
 273
 274   return TEST_SUCCESS;
 275 }
 276
 277 test_return_t server_test(memcached_st *)
 278 {
 279   return _test_option(test_server_strings);
 280 }
 281
 282 test_return_t server_with_weight_test(memcached_st *)
 283 {
 284   return _test_option(test_server_strings_with_weights);
 285 }
 286
 287 test_return_t servers_bad_test(memcached_st *)
 288 {
 289   test_return_t rc;
 290   if ((rc= _test_option(bad_test_strings, false)) != TEST_SUCCESS)
 291   {
 292     return rc;
 293   }
 294
 295   return TEST_SUCCESS;
 296 }
 297
 298 test_return_t parser_number_options_test(memcached_st*)
 299 {
 300   return _test_option(test_number_options);
 301 }
 302
 303 test_return_t parser_boolean_options_test(memcached_st*)
 304 {
 305   return _test_option(test_boolean_options);
 306 }
 307
 308 test_return_t behavior_parser_test(memcached_st*)
 309 {
 310   return TEST_SUCCESS;
 311 }
 312
 313 test_return_t parser_hash_test(memcached_st*)
 314 {
 315   return _test_option(hash_strings);
 316 }
 317
 318 test_return_t parser_distribution_test(memcached_st*)
 319 {
 320   return _test_option(distribution_strings);
 321 }
 322
 323 test_return_t parser_key_prefix_test(memcached_st*)
 324 {
 325   return _test_option(distribution_strings);
 326 }
 327
 328 test_return_t test_namespace_keyword(memcached_st*)
 329 {
 330   return _test_option(namespace_strings);
 331 }
 332
 333 #define SUPPORT_EXAMPLE_CNF "support/example.cnf"
 334
 335 test_return_t memcached_create_with_options_with_filename(memcached_st*)
 336 {
 337   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 338     return TEST_SKIPPED;
 339
 340   memcached_st *memc_ptr;
 341   memc_ptr= memcached(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""));
 342   test_true_got(memc_ptr, "memcached() failed");
 343   memcached_free(memc_ptr);
 344
 345   return TEST_SUCCESS;
 346 }
 347
 348 test_return_t libmemcached_check_configuration_with_filename_test(memcached_st*)
 349 {
 350   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 351     return TEST_SKIPPED;
 352
 353   memcached_return_t rc;
 354   char buffer[BUFSIZ];
 355
 356   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""), buffer, sizeof(buffer));
 357   test_true_got(rc == MEMCACHED_SUCCESS, (rc == MEMCACHED_ERRNO) ? strerror(errno) : memcached_strerror(NULL, rc));
 358
 359   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=support/example.cnf"), buffer, sizeof(buffer));
 360   test_false_with(rc == MEMCACHED_SUCCESS, memcached_strerror(NULL, rc));
 361
 362   rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"bad-path/example.cnf\""), buffer, sizeof(buffer));
 363   test_true_got(rc == MEMCACHED_ERRNO, memcached_strerror(NULL, rc));
 364
 365   return TEST_SUCCESS;
 366 }
 367
 368 test_return_t libmemcached_check_configuration_test(memcached_st*)
 369 {
 370   memcached_return_t rc;
 371   char buffer[BUFSIZ];
 372
 373   test_compare(MEMCACHED_SUCCESS,
 374                libmemcached_check_configuration(test_literal_param("--server=localhost"), buffer, sizeof(buffer)));
 375
 376   rc= libmemcached_check_configuration(test_literal_param("--dude=localhost"), buffer, sizeof(buffer));
 377   test_false_with(rc == MEMCACHED_SUCCESS, buffer);
 378   test_true(rc == MEMCACHED_PARSE_ERROR);
 379
 380   return TEST_SUCCESS;
 381 }
 382
 383 test_return_t memcached_create_with_options_test(memcached_st*)
 384 {
 385   memcached_st *memc_ptr;
 386   memc_ptr= memcached(test_literal_param("--server=localhost"));
 387   test_true_got(memc_ptr, memcached_last_error_message(memc_ptr));
 388   memcached_free(memc_ptr);
 389
 390   memc_ptr= memcached(test_literal_param("--dude=localhost"));
 391   test_false_with(memc_ptr, memcached_last_error_message(memc_ptr));
 392
 393   return TEST_SUCCESS;
 394 }
 395
 396 test_return_t test_include_keyword(memcached_st*)
 397 {
 398   if (access(SUPPORT_EXAMPLE_CNF, R_OK))
 399     return TEST_SKIPPED;
 400
 401   char buffer[BUFSIZ];
 402   test_compare(MEMCACHED_SUCCESS,
 403                libmemcached_check_configuration(test_literal_param("INCLUDE \"support/example.cnf\""), buffer, sizeof(buffer)));
 404
 405   return TEST_SUCCESS;
 406 }
 407
 408 test_return_t test_end_keyword(memcached_st*)
 409 {
 410   char buffer[BUFSIZ];
 411   test_compare(MEMCACHED_SUCCESS,
 412                libmemcached_check_configuration(test_literal_param("--server=localhost END bad keywords"), buffer, sizeof(buffer)));
 413
 414   return TEST_SUCCESS;
 415 }
 416
 417 test_return_t test_reset_keyword(memcached_st*)
 418 {
 419   char buffer[BUFSIZ];
 420   test_compare(MEMCACHED_SUCCESS,
 421                libmemcached_check_configuration(test_literal_param("--server=localhost reset --server=bad.com"), buffer, sizeof(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   test_compare_got(MEMCACHED_SUCCESS,
 535                    libmemcached_check_configuration(server_string, strlen(server_string), buffer, sizeof(buffer)), buffer);
 536
 537   memcached_st *memc= memcached(server_string, strlen(server_string));
 538   test_true(memc);
 539
 540   memcached_server_fn callbacks[]= { dump_server_information };
 541   test_true(memcached_success(memcached_server_cursor(memc, callbacks, NULL, 1)));
 542
 543   memcached_free(memc);
 544
 545   return TEST_SUCCESS;
 546 }
 547
 548 struct socket_weight_t {
 549   const char *socket;
 550   size_t weight;
 551 };
 552
 553 static memcached_return_t dump_socket_information(const memcached_st *,
 554                                                   const memcached_server_st *instance,
 555                                                   void *context)
 556 {
 557   socket_weight_t *check= (socket_weight_t *)context;
 558
 559   if (strcmp(memcached_server_name(instance), check->socket))
 560   {
 561     std::cerr << std::endl << __FILE__ << ":" << __LINE__ << " " << memcached_server_name(instance) << " != " << check->socket << std::endl;
 562     return MEMCACHED_FAILURE;
 563   }
 564
 565   if (instance->weight == check->weight)
 566   {
 567     return MEMCACHED_SUCCESS;
 568   }
 569
 570   return MEMCACHED_FAILURE;
 571 }
 572
 573 test_return_t test_parse_socket(memcached_st *)
 574 {
 575   char buffer[BUFSIZ];
 576
 577   memcached_server_fn callbacks[]= { dump_socket_information };
 578   {
 579     test_compare_got(MEMCACHED_SUCCESS,
 580                      libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\""), buffer, sizeof(buffer)),
 581                      buffer);
 582
 583     memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\""));
 584     test_true(memc);
 585     socket_weight_t check= { "/tmp/foo", 1 };
 586     test_compare(MEMCACHED_SUCCESS,
 587                  memcached_server_cursor(memc, callbacks, &check, 1));
 588     memcached_free(memc);
 589   }
 590
 591   {
 592     test_compare_got(MEMCACHED_SUCCESS,
 593                      libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\"/?23"), buffer, sizeof(buffer)),
 594                      buffer);
 595
 596     memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\"/?23"));
 597     test_true(memc);
 598     socket_weight_t check= { "/tmp/foo", 23 };
 599     test_compare(MEMCACHED_SUCCESS,
 600                  memcached_server_cursor(memc, callbacks, &check, 1));
 601     memcached_free(memc);
 602   }
 603
 604   return TEST_SUCCESS;
 605 }
 606
 607 /*
 608   By setting the timeout value to zero, we force poll() to return immediatly.
 609 */
 610 test_return_t regression_bug_71231153_connect(memcached_st *)
 611 {
 612   if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
 613     return TEST_SKIPPED;
 614
 615   { // Test the connect-timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
 616     memcached_st *memc= memcached(test_literal_param("--SERVER=10.0.2.252 --CONNECT-TIMEOUT=0"));
 617     test_true(memc);
 618     test_zero(memc->connect_timeout);
 619     test_compare(MEMCACHED_DEFAULT_TIMEOUT, memc->poll_timeout);
 620
 621     memcached_return_t rc;
 622     size_t value_len;
 623     char *value= memcached_get(memc, test_literal_param("test"), &value_len, NULL, &rc);
 624     test_false(value);
 625     test_zero(value_len);
 626     test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_last_error_message(memc));
 627
 628     memcached_free(memc);
 629   }
 630
 631   return TEST_SUCCESS;
 632 }
 633
 634 test_return_t regression_bug_71231153_poll(memcached_st *)
 635 {
 636   if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
 637     return TEST_SKIPPED;
 638
 639   { // Test the poll timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
 640     memcached_st *memc= memcached(test_literal_param("--SERVER=10.0.2.252 --POLL-TIMEOUT=0"));
 641     test_true(memc);
 642     test_compare(MEMCACHED_DEFAULT_CONNECT_TIMEOUT, memc->connect_timeout);
 643     test_zero(memc->poll_timeout);
 644
 645     memcached_return_t rc;
 646     size_t value_len;
 647     char *value= memcached_get(memc, test_literal_param("test"), &value_len, NULL, &rc);
 648     test_false(value);
 649     test_zero(value_len);
 650     test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_last_error_message(memc));
 651
 652     memcached_free(memc);
 653   }
 654
 655   return TEST_SUCCESS;
 656 }