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