git.m6w6.name Git - awesomized/libmemcached/blob - tests/function.c

   1 /*
   2   Sample test application.
   3 */
   4 #include <assert.h>
   5 #include <memcached.h>
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <sys/time.h>
  10 #include <sys/types.h>
  11 #include <sys/stat.h>
  12 #include <unistd.h>
  13 #include <time.h>
  14 #include "../lib/common.h"
  15 #include "../src/generator.h"
  16 #include "../src/execute.h"
  17
  18 #ifndef INT64_MAX
  19 #define INT64_MAX LONG_MAX
  20 #endif
  21 #ifndef INT32_MAX
  22 #define INT32_MAX INT_MAX
  23 #endif
  24
  25
  26 #include "test.h"
  27
  28 #define GLOBAL_COUNT 100000
  29 #define GLOBAL2_COUNT 1000
  30 static uint32_t global_count;
  31
  32 static pairs_st *global_pairs;
  33 static char *global_keys[GLOBAL_COUNT];
  34 static size_t global_keys_length[GLOBAL_COUNT];
  35
  36 uint8_t init_test(memcached_st *not_used)
  37 {
  38   memcached_st memc;
  39
  40   (void)memcached_create(&memc);
  41   memcached_free(&memc);
  42
  43   return 0;
  44 }
  45
  46 uint8_t server_list_null_test(memcached_st *ptr)
  47 {
  48   memcached_server_st *server_list;
  49   memcached_return rc;
  50
  51   server_list= memcached_server_list_append(NULL, NULL, 0, NULL);
  52   assert(server_list == NULL);
  53
  54   server_list= memcached_server_list_append(NULL, "localhost", 0, NULL);
  55   assert(server_list == NULL);
  56
  57   server_list= memcached_server_list_append(NULL, NULL, 0, &rc);
  58   assert(server_list == NULL);
  59
  60   return 0;
  61 }
  62
  63 uint8_t allocation_test(memcached_st *not_used)
  64 {
  65   memcached_st *memc;
  66   memc= memcached_create(NULL);
  67   assert(memc);
  68   memcached_free(memc);
  69
  70   return 0;
  71 }
  72
  73 uint8_t clone_test(memcached_st *memc)
  74 {
  75   /* All null? */
  76   {
  77     memcached_st *clone;
  78     clone= memcached_clone(NULL, NULL);
  79     assert(clone);
  80     memcached_free(clone);
  81   }
  82
  83   /* Can we init from null? */
  84   {
  85     memcached_st *clone;
  86     clone= memcached_clone(NULL, memc);
  87     assert(clone);
  88     memcached_free(clone);
  89   }
  90
  91   /* Can we init from struct? */
  92   {
  93     memcached_st declared_clone;
  94     memcached_st *clone;
  95     clone= memcached_clone(&declared_clone, NULL);
  96     assert(clone);
  97     memcached_free(clone);
  98   }
  99
 100   /* Can we init from struct? */
 101   {
 102     memcached_st declared_clone;
 103     memcached_st *clone;
 104     clone= memcached_clone(&declared_clone, memc);
 105     assert(clone);
 106     memcached_free(clone);
 107   }
 108
 109   return 0;
 110 }
 111
 112 uint8_t connection_test(memcached_st *memc)
 113 {
 114   memcached_return rc;
 115
 116   rc= memcached_server_add(memc, "localhost", 0);
 117   assert(rc == MEMCACHED_SUCCESS);
 118
 119   return 0;
 120 }
 121
 122 uint8_t error_test(memcached_st *memc)
 123 {
 124   memcached_return rc;
 125
 126   for (rc= MEMCACHED_SUCCESS; rc < MEMCACHED_MAXIMUM_RETURN; rc++)
 127   {
 128     printf("Error %d -> %s\n", rc, memcached_strerror(memc, rc));
 129   }
 130
 131   return 0;
 132 }
 133
 134 uint8_t set_test(memcached_st *memc)
 135 {
 136   memcached_return rc;
 137   char *key= "foo";
 138   char *value= "when we sanitize";
 139
 140   rc= memcached_set(memc, key, strlen(key),
 141                     value, strlen(value),
 142                     (time_t)0, (uint32_t)0);
 143   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 144
 145   return 0;
 146 }
 147
 148 uint8_t append_test(memcached_st *memc)
 149 {
 150   memcached_return rc;
 151   char *key= "fig";
 152   char *value= "we";
 153   size_t value_length;
 154   uint32_t flags;
 155
 156   rc= memcached_flush(memc, 0);
 157   assert(rc == MEMCACHED_SUCCESS);
 158
 159   rc= memcached_set(memc, key, strlen(key),
 160                     value, strlen(value),
 161                     (time_t)0, (uint32_t)0);
 162   assert(rc == MEMCACHED_SUCCESS);
 163
 164   rc= memcached_append(memc, key, strlen(key),
 165                        " the", strlen(" the"),
 166                        (time_t)0, (uint32_t)0);
 167   assert(rc == MEMCACHED_SUCCESS);
 168
 169   rc= memcached_append(memc, key, strlen(key),
 170                        " people", strlen(" people"),
 171                        (time_t)0, (uint32_t)0);
 172   assert(rc == MEMCACHED_SUCCESS);
 173
 174   value= memcached_get(memc, key, strlen(key),
 175                        &value_length, &flags, &rc);
 176   assert(!memcmp(value, "we the people", strlen("we the people")));
 177   assert(strlen("we the people") == value_length);
 178   assert(rc == MEMCACHED_SUCCESS);
 179   free(value);
 180
 181   return 0;
 182 }
 183
 184 uint8_t append_binary_test(memcached_st *memc)
 185 {
 186   memcached_return rc;
 187   char *key= "numbers";
 188   unsigned int *store_ptr;
 189   unsigned int store_list[] = { 23, 56, 499, 98, 32847, 0 };
 190   char *value;
 191   size_t value_length;
 192   uint32_t flags;
 193   unsigned int x;
 194
 195   rc= memcached_flush(memc, 0);
 196   assert(rc == MEMCACHED_SUCCESS);
 197
 198   rc= memcached_set(memc,
 199                     key, strlen(key),
 200                     NULL, 0,
 201                     (time_t)0, (uint32_t)0);
 202   assert(rc == MEMCACHED_SUCCESS);
 203
 204   for (x= 0; store_list[x] ; x++)
 205   {
 206     rc= memcached_append(memc,
 207                          key, strlen(key),
 208                          (char *)&store_list[x], sizeof(unsigned int),
 209                          (time_t)0, (uint32_t)0);
 210     assert(rc == MEMCACHED_SUCCESS);
 211   }
 212
 213   value= memcached_get(memc, key, strlen(key),
 214                        &value_length, &flags, &rc);
 215   assert((value_length == (sizeof(unsigned int) * x)));
 216   assert(rc == MEMCACHED_SUCCESS);
 217
 218   store_ptr= (unsigned int *)value;
 219   x= 0;
 220   while ((size_t)store_ptr < (size_t)(value + value_length))
 221   {
 222     assert(*store_ptr == store_list[x++]);
 223     store_ptr++;
 224   }
 225   free(value);
 226
 227   return 0;
 228 }
 229
 230 uint8_t cas2_test(memcached_st *memc)
 231 {
 232   memcached_return rc;
 233   char *keys[]= {"fudge", "son", "food"};
 234   size_t key_length[]= {5, 3, 4};
 235   char *value= "we the people";
 236   size_t value_length= strlen("we the people");
 237   unsigned int x;
 238   memcached_result_st results_obj;
 239   memcached_result_st *results;
 240   unsigned int set= 1;
 241
 242   rc= memcached_flush(memc, 0);
 243   assert(rc == MEMCACHED_SUCCESS);
 244
 245   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, &set);
 246
 247   for (x= 0; x < 3; x++)
 248   {
 249     rc= memcached_set(memc, keys[x], key_length[x],
 250                       keys[x], key_length[x],
 251                       (time_t)50, (uint32_t)9);
 252     assert(rc == MEMCACHED_SUCCESS);
 253   }
 254
 255   rc= memcached_mget(memc, keys, key_length, 3);
 256
 257   results= memcached_result_create(memc, &results_obj);
 258
 259   results= memcached_fetch_result(memc, &results_obj, &rc);
 260   assert(results);
 261   assert(results->cas);
 262   assert(rc == MEMCACHED_SUCCESS);
 263   WATCHPOINT_ASSERT(memcached_result_cas(results));
 264
 265   assert(!memcmp(value, "we the people", strlen("we the people")));
 266   assert(strlen("we the people") == value_length);
 267   assert(rc == MEMCACHED_SUCCESS);
 268
 269   memcached_result_free(&results_obj);
 270
 271   return 0;
 272 }
 273
 274 uint8_t cas_test(memcached_st *memc)
 275 {
 276   memcached_return rc;
 277   char *key= "fun";
 278   size_t key_length= strlen("fun");
 279   char *value= "we the people";
 280   size_t value_length= strlen("we the people");
 281   memcached_result_st results_obj;
 282   memcached_result_st *results;
 283   unsigned int set= 1;
 284
 285   rc= memcached_flush(memc, 0);
 286   assert(rc == MEMCACHED_SUCCESS);
 287
 288   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, &set);
 289
 290   rc= memcached_set(memc, key, strlen(key),
 291                     value, strlen(value),
 292                     (time_t)0, (uint32_t)0);
 293   assert(rc == MEMCACHED_SUCCESS);
 294
 295   rc= memcached_mget(memc, &key, &key_length, 1);
 296
 297   results= memcached_result_create(memc, &results_obj);
 298
 299   results= memcached_fetch_result(memc, &results_obj, &rc);
 300   assert(results);
 301   assert(rc == MEMCACHED_SUCCESS);
 302   WATCHPOINT_ASSERT(memcached_result_cas(results));
 303
 304   assert(!memcmp(value, "we the people", strlen("we the people")));
 305   assert(strlen("we the people") == value_length);
 306   assert(rc == MEMCACHED_SUCCESS);
 307
 308   memcached_result_free(&results_obj);
 309
 310   return 0;
 311 }
 312
 313 uint8_t prepend_test(memcached_st *memc)
 314 {
 315   memcached_return rc;
 316   char *key= "fig";
 317   char *value= "people";
 318   size_t value_length;
 319   uint32_t flags;
 320
 321   rc= memcached_flush(memc, 0);
 322   assert(rc == MEMCACHED_SUCCESS);
 323
 324   rc= memcached_set(memc, key, strlen(key),
 325                     value, strlen(value),
 326                     (time_t)0, (uint32_t)0);
 327   assert(rc == MEMCACHED_SUCCESS);
 328
 329   rc= memcached_prepend(memc, key, strlen(key),
 330                        "the ", strlen("the "),
 331                        (time_t)0, (uint32_t)0);
 332   assert(rc == MEMCACHED_SUCCESS);
 333
 334   rc= memcached_prepend(memc, key, strlen(key),
 335                        "we ", strlen("we "),
 336                        (time_t)0, (uint32_t)0);
 337   assert(rc == MEMCACHED_SUCCESS);
 338
 339   value= memcached_get(memc, key, strlen(key),
 340                        &value_length, &flags, &rc);
 341   assert(!memcmp(value, "we the people", strlen("we the people")));
 342   assert(strlen("we the people") == value_length);
 343   assert(rc == MEMCACHED_SUCCESS);
 344   free(value);
 345
 346   return 0;
 347 }
 348
 349 /*
 350   Set the value, then quit to make sure it is flushed.
 351   Come back in and test that add fails.
 352 */
 353 uint8_t add_test(memcached_st *memc)
 354 {
 355   memcached_return rc;
 356   char *key= "foo";
 357   char *value= "when we sanitize";
 358
 359   rc= memcached_set(memc, key, strlen(key),
 360                     value, strlen(value),
 361                     (time_t)0, (uint32_t)0);
 362   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 363   memcached_quit(memc);
 364   rc= memcached_add(memc, key, strlen(key),
 365                     value, strlen(value),
 366                     (time_t)0, (uint32_t)0);
 367   assert(rc == MEMCACHED_NOTSTORED);
 368
 369   return 0;
 370 }
 371
 372 uint8_t add_wrapper(memcached_st *memc)
 373 {
 374   unsigned int x;
 375
 376   for (x= 0; x < 10000; x++)
 377     add_test(memc);
 378
 379   return 0;
 380 }
 381
 382 uint8_t replace_test(memcached_st *memc)
 383 {
 384   memcached_return rc;
 385   char *key= "foo";
 386   char *value= "when we sanitize";
 387
 388   rc= memcached_replace(memc, key, strlen(key),
 389                     value, strlen(value),
 390                     (time_t)0, (uint32_t)0);
 391   assert(rc == MEMCACHED_SUCCESS);
 392
 393   return 0;
 394 }
 395
 396 uint8_t delete_test(memcached_st *memc)
 397 {
 398   memcached_return rc;
 399   char *key= "foo";
 400   char *value= "when we sanitize";
 401
 402   rc= memcached_set(memc, key, strlen(key),
 403                     value, strlen(value),
 404                     (time_t)0, (uint32_t)0);
 405   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 406
 407   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 408   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 409
 410   return 0;
 411 }
 412
 413 uint8_t flush_test(memcached_st *memc)
 414 {
 415   memcached_return rc;
 416
 417   rc= memcached_flush(memc, 0);
 418   assert(rc == MEMCACHED_SUCCESS);
 419
 420   return 0;
 421 }
 422
 423 uint8_t get_test(memcached_st *memc)
 424 {
 425   memcached_return rc;
 426   char *key= "foo";
 427   char *string;
 428   size_t string_length;
 429   uint32_t flags;
 430
 431   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 432   assert(rc == MEMCACHED_BUFFERED || rc == MEMCACHED_NOTFOUND);
 433
 434   string= memcached_get(memc, key, strlen(key),
 435                         &string_length, &flags, &rc);
 436
 437   assert(rc == MEMCACHED_NOTFOUND);
 438   assert(string_length ==  0);
 439   assert(!string);
 440
 441   return 0;
 442 }
 443
 444 uint8_t get_test2(memcached_st *memc)
 445 {
 446   memcached_return rc;
 447   char *key= "foo";
 448   char *value= "when we sanitize";
 449   char *string;
 450   size_t string_length;
 451   uint32_t flags;
 452
 453   rc= memcached_set(memc, key, strlen(key),
 454                     value, strlen(value),
 455                     (time_t)0, (uint32_t)0);
 456   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 457
 458   string= memcached_get(memc, key, strlen(key),
 459                         &string_length, &flags, &rc);
 460
 461   assert(string);
 462   assert(rc == MEMCACHED_SUCCESS);
 463   assert(string_length == strlen(value));
 464   assert(!memcmp(string, value, string_length));
 465
 466   free(string);
 467
 468   return 0;
 469 }
 470
 471 uint8_t set_test2(memcached_st *memc)
 472 {
 473   memcached_return rc;
 474   char *key= "foo";
 475   char *value= "train in the brain";
 476   size_t value_length= strlen(value);
 477   unsigned int x;
 478
 479   for (x= 0; x < 10; x++)
 480   {
 481     rc= memcached_set(memc, key, strlen(key),
 482                       value, value_length,
 483                       (time_t)0, (uint32_t)0);
 484     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 485   }
 486
 487   return 0;
 488 }
 489
 490 uint8_t set_test3(memcached_st *memc)
 491 {
 492   memcached_return rc;
 493   char *key= "foo";
 494   char *value;
 495   size_t value_length= 8191;
 496   unsigned int x;
 497
 498   value = (char*)malloc(value_length);
 499   assert(value);
 500
 501   for (x= 0; x < value_length; x++)
 502     value[x] = (char) (x % 127);
 503
 504   for (x= 0; x < 1; x++)
 505   {
 506     rc= memcached_set(memc, key, strlen(key),
 507                       value, value_length,
 508                       (time_t)0, (uint32_t)0);
 509     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 510   }
 511
 512   free(value);
 513
 514   return 0;
 515 }
 516
 517 uint8_t get_test3(memcached_st *memc)
 518 {
 519   memcached_return rc;
 520   char *key= "foo";
 521   char *value;
 522   size_t value_length= 8191;
 523   char *string;
 524   size_t string_length;
 525   uint32_t flags;
 526   int x;
 527
 528   value = (char*)malloc(value_length);
 529   assert(value);
 530
 531   for (x= 0; x < value_length; x++)
 532     value[x] = (char) (x % 127);
 533
 534   rc= memcached_set(memc, key, strlen(key),
 535                     value, value_length,
 536                     (time_t)0, (uint32_t)0);
 537   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 538
 539   string= memcached_get(memc, key, strlen(key),
 540                         &string_length, &flags, &rc);
 541
 542   assert(rc == MEMCACHED_SUCCESS);
 543   assert(string);
 544   assert(string_length == value_length);
 545   assert(!memcmp(string, value, string_length));
 546
 547   free(string);
 548   free(value);
 549
 550   return 0;
 551 }
 552
 553 uint8_t get_test4(memcached_st *memc)
 554 {
 555   memcached_return rc;
 556   char *key= "foo";
 557   char *value;
 558   size_t value_length= 8191;
 559   char *string;
 560   size_t string_length;
 561   uint32_t flags;
 562   int x;
 563
 564   value = (char*)malloc(value_length);
 565   assert(value);
 566
 567   for (x= 0; x < value_length; x++)
 568     value[x] = (char) (x % 127);
 569
 570   rc= memcached_set(memc, key, strlen(key),
 571                     value, value_length,
 572                     (time_t)0, (uint32_t)0);
 573   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 574
 575   for (x= 0; x < 10; x++)
 576   {
 577     string= memcached_get(memc, key, strlen(key),
 578                           &string_length, &flags, &rc);
 579
 580     assert(rc == MEMCACHED_SUCCESS);
 581     assert(string);
 582     assert(string_length == value_length);
 583     assert(!memcmp(string, value, string_length));
 584     free(string);
 585   }
 586
 587   free(value);
 588
 589   return 0;
 590 }
 591
 592 /* Do not copy the style of this code, I just access hosts to testthis function */
 593 uint8_t stats_servername_test(memcached_st *memc)
 594 {
 595   memcached_return rc;
 596   memcached_stat_st stat;
 597   rc= memcached_stat_servername(&stat, NULL,
 598                                  memc->hosts[0].hostname,
 599                                  memc->hosts[0].port);
 600
 601   return 0;
 602 }
 603
 604 uint8_t increment_test(memcached_st *memc)
 605 {
 606   uint64_t new_number;
 607   memcached_return rc;
 608   char *key= "number";
 609   char *value= "0";
 610
 611   rc= memcached_set(memc, key, strlen(key),
 612                     value, strlen(value),
 613                     (time_t)0, (uint32_t)0);
 614   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 615
 616   rc= memcached_increment(memc, key, strlen(key),
 617                           1, &new_number);
 618   assert(rc == MEMCACHED_SUCCESS);
 619   assert(new_number == 1);
 620
 621   rc= memcached_increment(memc, key, strlen(key),
 622                           1, &new_number);
 623   assert(rc == MEMCACHED_SUCCESS);
 624   assert(new_number == 2);
 625
 626   return 0;
 627 }
 628
 629 uint8_t decrement_test(memcached_st *memc)
 630 {
 631   uint64_t new_number;
 632   memcached_return rc;
 633   char *key= "number";
 634   char *value= "3";
 635
 636   rc= memcached_set(memc, key, strlen(key),
 637                     value, strlen(value),
 638                     (time_t)0, (uint32_t)0);
 639   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 640
 641   rc= memcached_decrement(memc, key, strlen(key),
 642                           1, &new_number);
 643   assert(rc == MEMCACHED_SUCCESS);
 644   assert(new_number == 2);
 645
 646   rc= memcached_decrement(memc, key, strlen(key),
 647                           1, &new_number);
 648   assert(rc == MEMCACHED_SUCCESS);
 649   assert(new_number == 1);
 650
 651   return 0;
 652 }
 653
 654 uint8_t quit_test(memcached_st *memc)
 655 {
 656   memcached_return rc;
 657   char *key= "fudge";
 658   char *value= "sanford and sun";
 659
 660   rc= memcached_set(memc, key, strlen(key),
 661                     value, strlen(value),
 662                     (time_t)10, (uint32_t)3);
 663   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 664   memcached_quit(memc);
 665
 666   rc= memcached_set(memc, key, strlen(key),
 667                     value, strlen(value),
 668                     (time_t)50, (uint32_t)9);
 669   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 670
 671   return 0;
 672 }
 673
 674 uint8_t mget_result_test(memcached_st *memc)
 675 {
 676   memcached_return rc;
 677   char *keys[]= {"fudge", "son", "food"};
 678   size_t key_length[]= {5, 3, 4};
 679   unsigned int x;
 680
 681   memcached_result_st results_obj;
 682   memcached_result_st *results;
 683
 684   results= memcached_result_create(memc, &results_obj);
 685   assert(results);
 686   assert(&results_obj == results);
 687
 688   /* We need to empty the server before continueing test */
 689   rc= memcached_flush(memc, 0);
 690   assert(rc == MEMCACHED_SUCCESS);
 691
 692   rc= memcached_mget(memc, keys, key_length, 3);
 693   assert(rc == MEMCACHED_SUCCESS);
 694
 695   while ((results= memcached_fetch_result(memc, &results_obj, &rc)) != NULL)
 696   {
 697     assert(results);
 698   }
 699
 700   while ((results= memcached_fetch_result(memc, &results_obj, &rc)) != NULL)
 701   assert(!results);
 702   assert(rc == MEMCACHED_END);
 703
 704   for (x= 0; x < 3; x++)
 705   {
 706     rc= memcached_set(memc, keys[x], key_length[x],
 707                       keys[x], key_length[x],
 708                       (time_t)50, (uint32_t)9);
 709     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 710   }
 711
 712   rc= memcached_mget(memc, keys, key_length, 3);
 713   assert(rc == MEMCACHED_SUCCESS);
 714
 715   while ((results= memcached_fetch_result(memc, &results_obj, &rc)))
 716   {
 717     assert(results);
 718     assert(&results_obj == results);
 719     assert(rc == MEMCACHED_SUCCESS);
 720     assert(memcached_result_key_length(results) == memcached_result_length(results));
 721     assert(!memcmp(memcached_result_key_value(results),
 722                    memcached_result_value(results),
 723                    memcached_result_length(results)));
 724   }
 725
 726   memcached_result_free(&results_obj);
 727
 728   return 0;
 729 }
 730
 731 uint8_t mget_result_alloc_test(memcached_st *memc)
 732 {
 733   memcached_return rc;
 734   char *keys[]= {"fudge", "son", "food"};
 735   size_t key_length[]= {5, 3, 4};
 736   unsigned int x;
 737
 738   memcached_result_st *results;
 739
 740   /* We need to empty the server before continueing test */
 741   rc= memcached_flush(memc, 0);
 742   assert(rc == MEMCACHED_SUCCESS);
 743
 744   rc= memcached_mget(memc, keys, key_length, 3);
 745   assert(rc == MEMCACHED_SUCCESS);
 746
 747   while ((results= memcached_fetch_result(memc, NULL, &rc)) != NULL)
 748   {
 749     assert(results);
 750   }
 751   assert(!results);
 752   assert(rc == MEMCACHED_END);
 753
 754   for (x= 0; x < 3; x++)
 755   {
 756     rc= memcached_set(memc, keys[x], key_length[x],
 757                       keys[x], key_length[x],
 758                       (time_t)50, (uint32_t)9);
 759     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 760   }
 761
 762   rc= memcached_mget(memc, keys, key_length, 3);
 763   assert(rc == MEMCACHED_SUCCESS);
 764
 765   x= 0;
 766   while ((results= memcached_fetch_result(memc, NULL, &rc)))
 767   {
 768     assert(results);
 769     assert(rc == MEMCACHED_SUCCESS);
 770     assert(memcached_result_key_length(results) == memcached_result_length(results));
 771     assert(!memcmp(memcached_result_key_value(results),
 772                    memcached_result_value(results),
 773                    memcached_result_length(results)));
 774     memcached_result_free(results);
 775     x++;
 776   }
 777
 778   return 0;
 779 }
 780
 781 /* Count the results */
 782 unsigned int callback_counter(memcached_st *ptr, memcached_result_st *result, void *context)
 783 {
 784   unsigned int *counter= (unsigned int *)context;
 785
 786   *counter= *counter + 1;
 787
 788   return 0;
 789 }
 790
 791 uint8_t mget_result_function(memcached_st *memc)
 792 {
 793   memcached_return rc;
 794   char *keys[]= {"fudge", "son", "food"};
 795   size_t key_length[]= {5, 3, 4};
 796   unsigned int x;
 797   unsigned int counter;
 798   unsigned int (*callbacks[1])(memcached_st *, memcached_result_st *, void *);
 799
 800   /* We need to empty the server before continueing test */
 801   rc= memcached_flush(memc, 0);
 802   for (x= 0; x < 3; x++)
 803   {
 804     rc= memcached_set(memc, keys[x], key_length[x],
 805                       keys[x], key_length[x],
 806                       (time_t)50, (uint32_t)9);
 807     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 808   }
 809
 810   rc= memcached_mget(memc, keys, key_length, 3);
 811   assert(rc == MEMCACHED_SUCCESS);
 812
 813   callbacks[0]= &callback_counter;
 814   counter= 0;
 815   rc= memcached_fetch_execute(memc, callbacks, (void *)&counter, 1);
 816
 817   assert(counter == 3);
 818
 819   return 0;
 820 }
 821
 822 uint8_t mget_test(memcached_st *memc)
 823 {
 824   memcached_return rc;
 825   char *keys[]= {"fudge", "son", "food"};
 826   size_t key_length[]= {5, 3, 4};
 827   unsigned int x;
 828   uint32_t flags;
 829
 830   char return_key[MEMCACHED_MAX_KEY];
 831   size_t return_key_length;
 832   char *return_value;
 833   size_t return_value_length;
 834
 835   /* We need to empty the server before continueing test */
 836   rc= memcached_flush(memc, 0);
 837   assert(rc == MEMCACHED_SUCCESS);
 838
 839   rc= memcached_mget(memc, keys, key_length, 3);
 840   assert(rc == MEMCACHED_SUCCESS);
 841
 842   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 843                       &return_value_length, &flags, &rc)) != NULL)
 844   {
 845     assert(return_value);
 846   }
 847   assert(!return_value);
 848   assert(return_value_length == 0);
 849   assert(rc == MEMCACHED_END);
 850
 851   for (x= 0; x < 3; x++)
 852   {
 853     rc= memcached_set(memc, keys[x], key_length[x],
 854                       keys[x], key_length[x],
 855                       (time_t)50, (uint32_t)9);
 856     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 857   }
 858
 859   rc= memcached_mget(memc, keys, key_length, 3);
 860   assert(rc == MEMCACHED_SUCCESS);
 861
 862   x= 0;
 863   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 864                                         &return_value_length, &flags, &rc)))
 865   {
 866     assert(return_value);
 867     assert(rc == MEMCACHED_SUCCESS);
 868     assert(return_key_length == return_value_length);
 869     assert(!memcmp(return_value, return_key, return_value_length));
 870     free(return_value);
 871     x++;
 872   }
 873
 874   return 0;
 875 }
 876
 877 uint8_t get_stats_keys(memcached_st *memc)
 878 {
 879  char **list;
 880  char **ptr;
 881  memcached_stat_st stat;
 882  memcached_return rc;
 883
 884  list= memcached_stat_get_keys(memc, &stat, &rc);
 885  assert(rc == MEMCACHED_SUCCESS);
 886  for (ptr= list; *ptr; ptr++)
 887    assert(*ptr);
 888  fflush(stdout);
 889
 890  free(list);
 891
 892  return 0;
 893 }
 894
 895 uint8_t version_string_test(memcached_st *memc)
 896 {
 897   const char *version_string;
 898
 899   version_string= memcached_lib_version();
 900
 901   assert(!strcmp(version_string, LIBMEMCACHED_VERSION_STRING));
 902
 903   return 0;
 904 }
 905
 906 uint8_t get_stats(memcached_st *memc)
 907 {
 908  unsigned int x;
 909  char **list;
 910  char **ptr;
 911  memcached_return rc;
 912  memcached_stat_st *stat;
 913
 914  stat= memcached_stat(memc, NULL, &rc);
 915  assert(rc == MEMCACHED_SUCCESS);
 916
 917  assert(rc == MEMCACHED_SUCCESS);
 918  assert(stat);
 919
 920  for (x= 0; x < memcached_server_count(memc); x++)
 921  {
 922    list= memcached_stat_get_keys(memc, stat+x, &rc);
 923    assert(rc == MEMCACHED_SUCCESS);
 924    for (ptr= list; *ptr; ptr++);
 925
 926    free(list);
 927  }
 928
 929  memcached_stat_free(NULL, stat);
 930
 931   return 0;
 932 }
 933
 934 uint8_t add_host_test(memcached_st *memc)
 935 {
 936   unsigned int x;
 937   memcached_server_st *servers;
 938   memcached_return rc;
 939   char servername[]= "0.example.com";
 940
 941   servers= memcached_server_list_append(NULL, servername, 400, &rc);
 942   assert(servers);
 943   assert(1 == memcached_server_list_count(servers));
 944
 945   for (x= 2; x < 20; x++)
 946   {
 947     char buffer[SMALL_STRING_LEN];
 948
 949     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
 950     servers= memcached_server_list_append(servers, buffer, 401,
 951                                      &rc);
 952     assert(rc == MEMCACHED_SUCCESS);
 953     assert(x == memcached_server_list_count(servers));
 954   }
 955
 956   rc= memcached_server_push(memc, servers);
 957   assert(rc == MEMCACHED_SUCCESS);
 958   rc= memcached_server_push(memc, servers);
 959   assert(rc == MEMCACHED_SUCCESS);
 960
 961   memcached_server_list_free(servers);
 962
 963   return 0;
 964 }
 965
 966 memcached_return clone_test_callback(memcached_st *parent, memcached_st *clone)
 967 {
 968   return MEMCACHED_SUCCESS;
 969 }
 970
 971 memcached_return cleanup_test_callback(memcached_st *ptr)
 972 {
 973   return MEMCACHED_SUCCESS;
 974 }
 975
 976 uint8_t callback_test(memcached_st *memc)
 977 {
 978   /* Test User Data */
 979   {
 980     int x= 5;
 981     int *test_ptr;
 982     memcached_return rc;
 983
 984     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_USER_DATA, &x);
 985     assert(rc == MEMCACHED_SUCCESS);
 986     test_ptr= (int *)memcached_callback_get(memc, MEMCACHED_CALLBACK_USER_DATA, &rc);
 987     assert(*test_ptr == x);
 988   }
 989
 990   /* Test Clone Callback */
 991   {
 992     memcached_clone_func temp_function;
 993     memcached_return rc;
 994
 995     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, clone_test_callback);
 996     assert(rc == MEMCACHED_SUCCESS);
 997     temp_function= (memcached_clone_func)memcached_callback_get(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, &rc);
 998     assert(temp_function == clone_test_callback);
 999   }
1000
1001   /* Test Cleanup Callback */
1002   {
1003     memcached_cleanup_func temp_function;
1004     memcached_return rc;
1005
1006     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, cleanup_test_callback);
1007     assert(rc == MEMCACHED_SUCCESS);
1008     temp_function= (memcached_cleanup_func)memcached_callback_get(memc, MEMCACHED_CALLBACK_CLONE_FUNCTION, &rc);
1009     assert(temp_function == cleanup_test_callback);
1010   }
1011
1012   return 0;
1013 }
1014
1015 /* We don't test the behavior itself, we test the switches */
1016 uint8_t behavior_test(memcached_st *memc)
1017 {
1018   unsigned long long value;
1019   unsigned int set= 1;
1020
1021   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
1022   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
1023   assert(value == 1);
1024
1025   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
1026   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
1027   assert(value == 1);
1028
1029   set= MEMCACHED_HASH_MD5;
1030   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
1031   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1032   assert(value == MEMCACHED_HASH_MD5);
1033
1034   set= 0;
1035
1036   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
1037   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
1038   assert(value == 0);
1039
1040   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
1041   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
1042   assert(value == 0);
1043
1044   set= MEMCACHED_HASH_DEFAULT;
1045   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
1046   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1047   assert(value == MEMCACHED_HASH_DEFAULT);
1048
1049   set= MEMCACHED_HASH_CRC;
1050   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
1051   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
1052   assert(value == MEMCACHED_HASH_CRC);
1053
1054   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1055   assert(value > 0);
1056
1057   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1058   assert(value > 0);
1059
1060   return 0;
1061 }
1062
1063 /* Test case provided by Cal Haldenbrand */
1064 uint8_t user_supplied_bug1(memcached_st *memc)
1065 {
1066   unsigned int setter= 1;
1067   unsigned int x;
1068
1069   unsigned long long total= 0;
1070   int size= 0;
1071   char key[10];
1072   char randomstuff[6 * 1024];
1073   memcached_return rc;
1074
1075   memset(randomstuff, 0, 6 * 1024);
1076
1077   /* We just keep looking at the same values over and over */
1078   srandom(10);
1079
1080   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
1081   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
1082
1083
1084   /* add key */
1085   for (x= 0 ; total < 20 * 1024576 ; x++ )
1086   {
1087     unsigned int j= 0;
1088
1089     size= (rand() % ( 5 * 1024 ) ) + 400;
1090     memset(randomstuff, 0, 6 * 1024);
1091     assert(size < 6 * 1024); /* Being safe here */
1092
1093     for (j= 0 ; j < size ;j++)
1094       randomstuff[j] = (char) (rand() % 26) + 97;
1095
1096     total += size;
1097     sprintf(key, "%d", x);
1098     rc = memcached_set(memc, key, strlen(key),
1099                        randomstuff, strlen(randomstuff), 10, 0);
1100     /* If we fail, lets try again */
1101     if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_BUFFERED)
1102       rc = memcached_set(memc, key, strlen(key),
1103                          randomstuff, strlen(randomstuff), 10, 0);
1104     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
1105   }
1106
1107   return 0;
1108 }
1109
1110 /* Test case provided by Cal Haldenbrand */
1111 uint8_t user_supplied_bug2(memcached_st *memc)
1112 {
1113   int errors;
1114   unsigned int setter;
1115   unsigned int x;
1116   unsigned long long total;
1117
1118   setter= 1;
1119   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
1120   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
1121 #ifdef NOT_YET
1122   setter = 20 * 1024576;
1123   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
1124   setter = 20 * 1024576;
1125   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
1126   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1127   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1128
1129   for (x= 0, errors= 0, total= 0 ; total < 20 * 1024576 ; x++)
1130 #endif
1131
1132   for (x= 0, errors= 0, total= 0 ; total < 24576 ; x++)
1133   {
1134     memcached_return rc= MEMCACHED_SUCCESS;
1135     char buffer[SMALL_STRING_LEN];
1136     uint32_t flags= 0;
1137     size_t val_len= 0;
1138     char *getval;
1139
1140     memset(buffer, 0, SMALL_STRING_LEN);
1141
1142     snprintf(buffer, SMALL_STRING_LEN, "%u", x);
1143     getval= memcached_get(memc, buffer, strlen(buffer),
1144                            &val_len, &flags, &rc);
1145     if (rc != MEMCACHED_SUCCESS)
1146     {
1147       if (rc == MEMCACHED_NOTFOUND)
1148         errors++;
1149       else
1150         assert(0);
1151
1152       continue;
1153     }
1154     total+= val_len;
1155     errors= 0;
1156     free(getval);
1157   }
1158
1159   return 0;
1160 }
1161
1162 /* Do a large mget() over all the keys we think exist */
1163 #define KEY_COUNT 3000 // * 1024576
1164 uint8_t user_supplied_bug3(memcached_st *memc)
1165 {
1166   memcached_return rc;
1167   unsigned int setter;
1168   unsigned int x;
1169   char **keys;
1170   size_t key_lengths[KEY_COUNT];
1171
1172   setter= 1;
1173   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
1174   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
1175 #ifdef NOT_YET
1176   setter = 20 * 1024576;
1177   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
1178   setter = 20 * 1024576;
1179   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
1180   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
1181   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
1182 #endif
1183
1184   keys= (char **)malloc(sizeof(char *) * KEY_COUNT);
1185   assert(keys);
1186   memset(keys, 0, (sizeof(char *) * KEY_COUNT));
1187   for (x= 0; x < KEY_COUNT; x++)
1188   {
1189     char buffer[30];
1190
1191     snprintf(buffer, 30, "%u", x);
1192     keys[x]= strdup(buffer);
1193     key_lengths[x]= strlen(keys[x]);
1194   }
1195
1196   rc= memcached_mget(memc, keys, key_lengths, KEY_COUNT);
1197   assert(rc == MEMCACHED_SUCCESS);
1198
1199   /* Turn this into a help function */
1200   {
1201     char return_key[MEMCACHED_MAX_KEY];
1202     size_t return_key_length;
1203     char *return_value;
1204     size_t return_value_length;
1205     uint32_t flags;
1206
1207     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1208                                           &return_value_length, &flags, &rc)))
1209     {
1210       assert(return_value);
1211       assert(rc == MEMCACHED_SUCCESS);
1212       free(return_value);
1213     }
1214   }
1215
1216   for (x= 0; x < KEY_COUNT; x++)
1217     free(keys[x]);
1218   free(keys);
1219
1220   return 0;
1221 }
1222
1223 /* Make sure we behave properly if server list has no values */
1224 uint8_t user_supplied_bug4(memcached_st *memc)
1225 {
1226   memcached_return rc;
1227   char *keys[]= {"fudge", "son", "food"};
1228   size_t key_length[]= {5, 3, 4};
1229   unsigned int x;
1230   uint32_t flags;
1231   char return_key[MEMCACHED_MAX_KEY];
1232   size_t return_key_length;
1233   char *return_value;
1234   size_t return_value_length;
1235
1236   /* Here we free everything before running a bunch of mget tests */
1237   {
1238     memcached_server_list_free(memc->hosts);
1239     memc->hosts= NULL;
1240     memc->number_of_hosts= 0;
1241   }
1242
1243
1244   /* We need to empty the server before continueing test */
1245   rc= memcached_flush(memc, 0);
1246   assert(rc == MEMCACHED_NO_SERVERS);
1247
1248   rc= memcached_mget(memc, keys, key_length, 3);
1249   assert(rc == MEMCACHED_NO_SERVERS);
1250
1251   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1252                       &return_value_length, &flags, &rc)) != NULL)
1253   {
1254     assert(return_value);
1255   }
1256   assert(!return_value);
1257   assert(return_value_length == 0);
1258   assert(rc == MEMCACHED_NO_SERVERS);
1259
1260   for (x= 0; x < 3; x++)
1261   {
1262     rc= memcached_set(memc, keys[x], key_length[x],
1263                       keys[x], key_length[x],
1264                       (time_t)50, (uint32_t)9);
1265     assert(rc == MEMCACHED_NO_SERVERS);
1266   }
1267
1268   rc= memcached_mget(memc, keys, key_length, 3);
1269   assert(rc == MEMCACHED_NO_SERVERS);
1270
1271   x= 0;
1272   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1273                                         &return_value_length, &flags, &rc)))
1274   {
1275     assert(return_value);
1276     assert(rc == MEMCACHED_SUCCESS);
1277     assert(return_key_length == return_value_length);
1278     assert(!memcmp(return_value, return_key, return_value_length));
1279     free(return_value);
1280     x++;
1281   }
1282
1283   return 0;
1284 }
1285
1286 #define VALUE_SIZE_BUG5 1048064
1287 uint8_t user_supplied_bug5(memcached_st *memc)
1288 {
1289   memcached_return rc;
1290   char *keys[]= {"036790384900", "036790384902", "036790384904", "036790384906"};
1291   size_t key_length[]=  {strlen("036790384900"), strlen("036790384902"), strlen("036790384904"), strlen("036790384906")};
1292   char return_key[MEMCACHED_MAX_KEY];
1293   size_t return_key_length;
1294   char *value;
1295   size_t value_length;
1296   uint32_t flags;
1297   unsigned int count;
1298   unsigned int x;
1299   char insert_data[VALUE_SIZE_BUG5];
1300
1301   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1302     insert_data[x]= rand();
1303
1304   memcached_flush(memc, 0);
1305   value= memcached_get(memc, keys[0], key_length[0],
1306                         &value_length, &flags, &rc);
1307   assert(value == NULL);
1308   rc= memcached_mget(memc, keys, key_length, 4);
1309
1310   count= 0;
1311   while ((value= memcached_fetch(memc, return_key, &return_key_length,
1312                                         &value_length, &flags, &rc)))
1313     count++;
1314   assert(count == 0);
1315
1316   for (x= 0; x < 4; x++)
1317   {
1318     rc= memcached_set(memc, keys[x], key_length[x],
1319                       insert_data, VALUE_SIZE_BUG5,
1320                       (time_t)0, (uint32_t)0);
1321     assert(rc == MEMCACHED_SUCCESS);
1322   }
1323
1324   for (x= 0; x < 10; x++)
1325   {
1326     value= memcached_get(memc, keys[0], key_length[0],
1327                          &value_length, &flags, &rc);
1328     assert(value);
1329     free(value);
1330
1331     rc= memcached_mget(memc, keys, key_length, 4);
1332     count= 0;
1333     while ((value= memcached_fetch(memc, return_key, &return_key_length,
1334                                           &value_length, &flags, &rc)))
1335     {
1336       count++;
1337       free(value);
1338     }
1339     assert(count == 4);
1340   }
1341
1342   return 0;
1343 }
1344
1345 uint8_t user_supplied_bug6(memcached_st *memc)
1346 {
1347   memcached_return rc;
1348   char *keys[]= {"036790384900", "036790384902", "036790384904", "036790384906"};
1349   size_t key_length[]=  {strlen("036790384900"), strlen("036790384902"), strlen("036790384904"), strlen("036790384906")};
1350   char return_key[MEMCACHED_MAX_KEY];
1351   size_t return_key_length;
1352   char *value;
1353   size_t value_length;
1354   uint32_t flags;
1355   unsigned int count;
1356   unsigned int x;
1357   char insert_data[VALUE_SIZE_BUG5];
1358
1359   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1360     insert_data[x]= rand();
1361
1362   memcached_flush(memc, 0);
1363   value= memcached_get(memc, keys[0], key_length[0],
1364                         &value_length, &flags, &rc);
1365   assert(value == NULL);
1366   assert(rc == MEMCACHED_NOTFOUND);
1367   rc= memcached_mget(memc, keys, key_length, 4);
1368   assert(rc == MEMCACHED_SUCCESS);
1369
1370   count= 0;
1371   while ((value= memcached_fetch(memc, return_key, &return_key_length,
1372                                         &value_length, &flags, &rc)))
1373     count++;
1374   assert(count == 0);
1375   assert(rc == MEMCACHED_END);
1376
1377   for (x= 0; x < 4; x++)
1378   {
1379     rc= memcached_set(memc, keys[x], key_length[x],
1380                       insert_data, VALUE_SIZE_BUG5,
1381                       (time_t)0, (uint32_t)0);
1382     assert(rc == MEMCACHED_SUCCESS);
1383   }
1384
1385   for (x= 0; x < 2; x++)
1386   {
1387     value= memcached_get(memc, keys[0], key_length[0],
1388                          &value_length, &flags, &rc);
1389     assert(value);
1390     free(value);
1391
1392     rc= memcached_mget(memc, keys, key_length, 4);
1393     assert(rc == MEMCACHED_SUCCESS);
1394     count= 3;
1395     /* We test for purge of partial complete fetches */
1396     for (count= 3; count; count--)
1397     {
1398       value= memcached_fetch(memc, return_key, &return_key_length,
1399                              &value_length, &flags, &rc);
1400       assert(rc == MEMCACHED_SUCCESS);
1401       assert(!(memcmp(value, insert_data, value_length)));
1402       assert(value_length);
1403       free(value);
1404     }
1405   }
1406
1407   return 0;
1408 }
1409
1410 uint8_t user_supplied_bug8(memcached_st *memc)
1411 {
1412   memcached_return rc;
1413   memcached_st *mine;
1414   memcached_st *clone;
1415
1416   memcached_server_st *servers;
1417   char *server_list= "memcache1.memcache.bk.sapo.pt:11211, memcache1.memcache.bk.sapo.pt:11212, memcache1.memcache.bk.sapo.pt:11213, memcache1.memcache.bk.sapo.pt:11214, memcache2.memcache.bk.sapo.pt:11211, memcache2.memcache.bk.sapo.pt:11212, memcache2.memcache.bk.sapo.pt:11213, memcache2.memcache.bk.sapo.pt:11214";
1418
1419   servers= memcached_servers_parse(server_list);
1420   assert(servers);
1421
1422   mine= memcached_create(NULL);
1423   rc= memcached_server_push(mine, servers);
1424   assert(rc == MEMCACHED_SUCCESS);
1425   memcached_server_list_free(servers);
1426
1427   assert(mine);
1428   clone= memcached_clone(NULL, mine);
1429
1430   memcached_quit(mine);
1431   memcached_quit(clone);
1432
1433
1434   memcached_free(mine);
1435   memcached_free(clone);
1436
1437   return 0;
1438 }
1439
1440 /* Test flag store/retrieve */
1441 uint8_t user_supplied_bug7(memcached_st *memc)
1442 {
1443   memcached_return rc;
1444   char *keys= "036790384900";
1445   size_t key_length=  strlen("036790384900");
1446   char return_key[MEMCACHED_MAX_KEY];
1447   size_t return_key_length;
1448   char *value;
1449   size_t value_length;
1450   uint32_t flags;
1451   unsigned int x;
1452   char insert_data[VALUE_SIZE_BUG5];
1453
1454   for (x= 0; x < VALUE_SIZE_BUG5; x++)
1455     insert_data[x]= rand();
1456
1457   memcached_flush(memc, 0);
1458
1459   flags= 245;
1460   rc= memcached_set(memc, keys, key_length,
1461                     insert_data, VALUE_SIZE_BUG5,
1462                     (time_t)0, flags);
1463   assert(rc == MEMCACHED_SUCCESS);
1464
1465   flags= 0;
1466   value= memcached_get(memc, keys, key_length,
1467                         &value_length, &flags, &rc);
1468   assert(flags == 245);
1469   assert(value);
1470   free(value);
1471
1472   rc= memcached_mget(memc, &keys, &key_length, 1);
1473
1474   flags= 0;
1475   value= memcached_fetch(memc, return_key, &return_key_length,
1476                          &value_length, &flags, &rc);
1477   assert(flags == 245);
1478   assert(value);
1479   free(value);
1480
1481
1482   return 0;
1483 }
1484
1485 uint8_t user_supplied_bug9(memcached_st *memc)
1486 {
1487   memcached_return rc;
1488   char *keys[]= {"UDATA:edevil@sapo.pt", "fudge&*@#", "for^#@&$not"};
1489   size_t key_length[3];
1490   unsigned int x;
1491   uint32_t flags;
1492   unsigned count= 0;
1493
1494   char return_key[MEMCACHED_MAX_KEY];
1495   size_t return_key_length;
1496   char *return_value;
1497   size_t return_value_length;
1498
1499
1500   key_length[0]= strlen("UDATA:edevil@sapo.pt");
1501   key_length[1]= strlen("fudge&*@#");
1502   key_length[2]= strlen("for^#@&$not");
1503
1504
1505   for (x= 0; x < 3; x++)
1506   {
1507     rc= memcached_set(memc, keys[x], key_length[x],
1508                       keys[x], key_length[x],
1509                       (time_t)50, (uint32_t)9);
1510     assert(rc == MEMCACHED_SUCCESS);
1511   }
1512
1513   rc= memcached_mget(memc, keys, key_length, 3);
1514   assert(rc == MEMCACHED_SUCCESS);
1515
1516   /* We need to empty the server before continueing test */
1517   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1518                       &return_value_length, &flags, &rc)) != NULL)
1519   {
1520     assert(return_value);
1521     free(return_value);
1522     count++;
1523   }
1524   assert(count == 3);
1525
1526   return 0;
1527 }
1528
1529 /* We are testing with aggressive timeout to get failures */
1530 uint8_t user_supplied_bug10(memcached_st *memc)
1531 {
1532   char *key= "foo";
1533   char *value;
1534   size_t value_length= 512;
1535   unsigned int x;
1536   int key_len= 3;
1537   memcached_return rc;
1538   unsigned int set= 1;
1539   memcached_st *mclone= memcached_clone(NULL, memc);
1540   int32_t timeout;
1541
1542   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
1543   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
1544   timeout= 2;
1545   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, &timeout);
1546
1547   value = (char*)malloc(value_length * sizeof(char));
1548
1549   for (x= 0; x < value_length; x++)
1550     value[x]= (char) (x % 127);
1551
1552   for (x= 1; x <= 100000; ++x)
1553   {
1554     rc= memcached_set(mclone, key, key_len,value, value_length, 0, 0);
1555
1556     assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_WRITE_FAILURE || rc == MEMCACHED_BUFFERED);
1557
1558     if (rc == MEMCACHED_WRITE_FAILURE)
1559       x--;
1560   }
1561
1562   free(value);
1563   memcached_free(mclone);
1564
1565   return 0;
1566 }
1567
1568 /*
1569   We are looking failures in the async protocol
1570 */
1571 uint8_t user_supplied_bug11(memcached_st *memc)
1572 {
1573   char *key= "foo";
1574   char *value;
1575   size_t value_length= 512;
1576   unsigned int x;
1577   int key_len= 3;
1578   memcached_return rc;
1579   unsigned int set= 1;
1580   int32_t timeout;
1581   memcached_st *mclone= memcached_clone(NULL, memc);
1582
1583   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
1584   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
1585   timeout= -1;
1586   memcached_behavior_set(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, &timeout);
1587
1588   timeout= (int32_t)memcached_behavior_get(mclone, MEMCACHED_BEHAVIOR_POLL_TIMEOUT);
1589
1590   assert(timeout == -1);
1591
1592   value = (char*)malloc(value_length * sizeof(char));
1593
1594   for (x= 0; x < value_length; x++)
1595     value[x]= (char) (x % 127);
1596
1597   for (x= 1; x <= 100000; ++x)
1598   {
1599     rc= memcached_set(mclone, key, key_len,value, value_length, 0, 0);
1600   }
1601
1602   free(value);
1603   memcached_free(mclone);
1604
1605   return 0;
1606 }
1607
1608 /*
1609   Bug found where incr was not returning MEMCACHED_NOTFOUND when object did not exist.
1610 */
1611 uint8_t user_supplied_bug12(memcached_st *memc)
1612 {
1613   memcached_return rc;
1614   uint32_t flags;
1615   size_t value_length;
1616   char *value;
1617   uint64_t number_value;
1618
1619   value= memcached_get(memc, "autoincrement", strlen("autoincrement"),
1620                         &value_length, &flags, &rc);
1621   assert(value == NULL);
1622   assert(rc == MEMCACHED_NOTFOUND);
1623
1624   rc= memcached_increment(memc, "autoincrement", strlen("autoincrement"),
1625                           1, &number_value);
1626
1627   assert(value == NULL);
1628   assert(rc == MEMCACHED_NOTFOUND);
1629
1630   rc= memcached_set(memc, "autoincrement", strlen("autoincrement"), "1", 1, 0, 0);
1631
1632   value= memcached_get(memc, "autoincrement", strlen("autoincrement"),
1633                         &value_length, &flags, &rc);
1634   assert(value);
1635   assert(rc == MEMCACHED_SUCCESS);
1636   free(value);
1637
1638   rc= memcached_increment(memc, "autoincrement", strlen("autoincrement"),
1639                           1, &number_value);
1640   assert(number_value == 2);
1641   assert(rc == MEMCACHED_SUCCESS);
1642
1643   return 0;
1644 }
1645
1646 uint8_t result_static(memcached_st *memc)
1647 {
1648   memcached_result_st result;
1649   memcached_result_st *result_ptr;
1650
1651   result_ptr= memcached_result_create(memc, &result);
1652   assert(result.is_allocated == MEMCACHED_NOT_ALLOCATED);
1653   assert(result_ptr);
1654   memcached_result_free(&result);
1655
1656   return 0;
1657 }
1658
1659 uint8_t result_alloc(memcached_st *memc)
1660 {
1661   memcached_result_st *result;
1662
1663   result= memcached_result_create(memc, NULL);
1664   assert(result);
1665   memcached_result_free(result);
1666
1667   return 0;
1668 }
1669
1670 uint8_t string_static_null(memcached_st *memc)
1671 {
1672   memcached_string_st string;
1673   memcached_string_st *string_ptr;
1674
1675   string_ptr= memcached_string_create(memc, &string, 0);
1676   assert(string.is_allocated == MEMCACHED_NOT_ALLOCATED);
1677   assert(string_ptr);
1678   memcached_string_free(&string);
1679
1680   return 0;
1681 }
1682
1683 uint8_t string_alloc_null(memcached_st *memc)
1684 {
1685   memcached_string_st *string;
1686
1687   string= memcached_string_create(memc, NULL, 0);
1688   assert(string);
1689   memcached_string_free(string);
1690
1691   return 0;
1692 }
1693
1694 uint8_t string_alloc_with_size(memcached_st *memc)
1695 {
1696   memcached_string_st *string;
1697
1698   string= memcached_string_create(memc, NULL, 1024);
1699   assert(string);
1700   memcached_string_free(string);
1701
1702   return 0;
1703 }
1704
1705 uint8_t string_alloc_with_size_toobig(memcached_st *memc)
1706 {
1707   memcached_string_st *string;
1708
1709   string= memcached_string_create(memc, NULL, INT64_MAX);
1710   assert(string == NULL);
1711
1712   return 0;
1713 }
1714
1715 uint8_t string_alloc_append(memcached_st *memc)
1716 {
1717   unsigned int x;
1718   char buffer[SMALL_STRING_LEN];
1719   memcached_string_st *string;
1720
1721   /* Ring the bell! */
1722   memset(buffer, 6, SMALL_STRING_LEN);
1723
1724   string= memcached_string_create(memc, NULL, 100);
1725   assert(string);
1726
1727   for (x= 0; x < 1024; x++)
1728   {
1729     memcached_return rc;
1730     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
1731     assert(rc == MEMCACHED_SUCCESS);
1732   }
1733   memcached_string_free(string);
1734
1735   return 0;
1736 }
1737
1738 uint8_t string_alloc_append_toobig(memcached_st *memc)
1739 {
1740   memcached_return rc;
1741   unsigned int x;
1742   char buffer[SMALL_STRING_LEN];
1743   memcached_string_st *string;
1744
1745   /* Ring the bell! */
1746   memset(buffer, 6, SMALL_STRING_LEN);
1747
1748   string= memcached_string_create(memc, NULL, 100);
1749   assert(string);
1750
1751   for (x= 0; x < 1024; x++)
1752   {
1753     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
1754     assert(rc == MEMCACHED_SUCCESS);
1755   }
1756   rc= memcached_string_append(string, buffer, INT64_MAX);
1757   assert(rc == MEMCACHED_MEMORY_ALLOCATION_FAILURE);
1758   memcached_string_free(string);
1759
1760   return 0;
1761 }
1762
1763 uint8_t cleanup_pairs(memcached_st *memc)
1764 {
1765   pairs_free(global_pairs);
1766
1767   return 0;
1768 }
1769
1770 uint8_t generate_pairs(memcached_st *memc)
1771 {
1772   unsigned long long x;
1773   global_pairs= pairs_generate(GLOBAL_COUNT, 400);
1774   global_count= GLOBAL_COUNT;
1775
1776   for (x= 0; x < global_count; x++)
1777   {
1778     global_keys[x]= global_pairs[x].key;
1779     global_keys_length[x]=  global_pairs[x].key_length;
1780   }
1781
1782   return 0;
1783 }
1784
1785 uint8_t generate_large_pairs(memcached_st *memc)
1786 {
1787   unsigned long long x;
1788   global_pairs= pairs_generate(GLOBAL2_COUNT, MEMCACHED_MAX_BUFFER+10);
1789   global_count= GLOBAL2_COUNT;
1790
1791   for (x= 0; x < global_count; x++)
1792   {
1793     global_keys[x]= global_pairs[x].key;
1794     global_keys_length[x]=  global_pairs[x].key_length;
1795   }
1796
1797   return 0;
1798 }
1799
1800 uint8_t generate_data(memcached_st *memc)
1801 {
1802   execute_set(memc, global_pairs, global_count);
1803
1804   return 0;
1805 }
1806
1807 uint8_t generate_buffer_data(memcached_st *memc)
1808 {
1809   int latch= 0;
1810
1811   latch= 1;
1812   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, &latch);
1813   generate_data(memc);
1814
1815   return 0;
1816 }
1817
1818 uint8_t get_read(memcached_st *memc)
1819 {
1820   unsigned int x;
1821   memcached_return rc;
1822
1823   {
1824     char *return_value;
1825     size_t return_value_length;
1826     uint32_t flags;
1827
1828     for (x= 0; x < global_count; x++)
1829     {
1830       return_value= memcached_get(memc, global_keys[x], global_keys_length[x],
1831                                   &return_value_length, &flags, &rc);
1832       /*
1833       assert(return_value);
1834       assert(rc == MEMCACHED_SUCCESS);
1835     */
1836       if (rc == MEMCACHED_SUCCESS && return_value)
1837         free(return_value);
1838     }
1839   }
1840
1841   return 0;
1842 }
1843
1844 uint8_t mget_read(memcached_st *memc)
1845 {
1846   memcached_return rc;
1847
1848   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
1849   assert(rc == MEMCACHED_SUCCESS);
1850   /* Turn this into a help function */
1851   {
1852     char return_key[MEMCACHED_MAX_KEY];
1853     size_t return_key_length;
1854     char *return_value;
1855     size_t return_value_length;
1856     uint32_t flags;
1857
1858     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
1859                                           &return_value_length, &flags, &rc)))
1860     {
1861       assert(return_value);
1862       assert(rc == MEMCACHED_SUCCESS);
1863       free(return_value);
1864     }
1865   }
1866
1867   return 0;
1868 }
1869
1870 uint8_t mget_read_result(memcached_st *memc)
1871 {
1872   memcached_return rc;
1873
1874   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
1875   assert(rc == MEMCACHED_SUCCESS);
1876   /* Turn this into a help function */
1877   {
1878     memcached_result_st results_obj;
1879     memcached_result_st *results;
1880
1881     results= memcached_result_create(memc, &results_obj);
1882
1883     while ((results= memcached_fetch_result(memc, &results_obj, &rc)))
1884     {
1885       assert(results);
1886       assert(rc == MEMCACHED_SUCCESS);
1887     }
1888
1889     memcached_result_free(&results_obj);
1890   }
1891
1892   return 0;
1893 }
1894
1895 uint8_t mget_read_function(memcached_st *memc)
1896 {
1897   memcached_return rc;
1898   unsigned int counter;
1899   unsigned int (*callbacks[1])(memcached_st *, memcached_result_st *, void *);
1900
1901   rc= memcached_mget(memc, global_keys, global_keys_length, global_count);
1902   assert(rc == MEMCACHED_SUCCESS);
1903
1904   callbacks[0]= &callback_counter;
1905   counter= 0;
1906   rc= memcached_fetch_execute(memc, callbacks, (void *)&counter, 1);
1907
1908   return 0;
1909 }
1910
1911 uint8_t delete_generate(memcached_st *memc)
1912 {
1913   unsigned int x;
1914
1915   for (x= 0; x < global_count; x++)
1916   {
1917     (void)memcached_delete(memc, global_keys[x], global_keys_length[x], (time_t)0);
1918   }
1919
1920   return 0;
1921 }
1922
1923 uint8_t delete_buffer_generate(memcached_st *memc)
1924 {
1925   int latch= 0;
1926   unsigned int x;
1927
1928   latch= 1;
1929   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BUFFER_REQUESTS, &latch);
1930
1931   for (x= 0; x < global_count; x++)
1932   {
1933     (void)memcached_delete(memc, global_keys[x], global_keys_length[x], (time_t)0);
1934   }
1935
1936   return 0;
1937 }
1938
1939 uint8_t free_data(memcached_st *memc)
1940 {
1941   pairs_free(global_pairs);
1942
1943   return 0;
1944 }
1945
1946 uint8_t add_host_test1(memcached_st *memc)
1947 {
1948   unsigned int x;
1949   memcached_return rc;
1950   char servername[]= "0.example.com";
1951   memcached_server_st *servers;
1952
1953   servers= memcached_server_list_append(NULL, servername, 400, &rc);
1954   assert(servers);
1955   assert(1 == memcached_server_list_count(servers));
1956
1957   for (x= 2; x < 20; x++)
1958   {
1959     char buffer[SMALL_STRING_LEN];
1960
1961     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
1962     servers= memcached_server_list_append(servers, buffer, 401,
1963                                      &rc);
1964     assert(rc == MEMCACHED_SUCCESS);
1965     assert(x == memcached_server_list_count(servers));
1966   }
1967
1968   rc= memcached_server_push(memc, servers);
1969   assert(rc == MEMCACHED_SUCCESS);
1970   rc= memcached_server_push(memc, servers);
1971   assert(rc == MEMCACHED_SUCCESS);
1972
1973   memcached_server_list_free(servers);
1974
1975   return 0;
1976 }
1977
1978 memcached_return pre_nonblock(memcached_st *memc)
1979 {
1980   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
1981
1982   return MEMCACHED_SUCCESS;
1983 }
1984
1985 memcached_return pre_md5(memcached_st *memc)
1986 {
1987   memcached_hash value= MEMCACHED_HASH_MD5;
1988   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
1989
1990   return MEMCACHED_SUCCESS;
1991 }
1992
1993 memcached_return pre_crc(memcached_st *memc)
1994 {
1995   memcached_hash value= MEMCACHED_HASH_CRC;
1996   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
1997
1998   return MEMCACHED_SUCCESS;
1999 }
2000
2001 memcached_return pre_hsieh(memcached_st *memc)
2002 {
2003   memcached_hash value= MEMCACHED_HASH_HSIEH;
2004   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2005
2006   return MEMCACHED_SUCCESS;
2007 }
2008
2009 memcached_return pre_hash_fnv1_64(memcached_st *memc)
2010 {
2011   memcached_hash value= MEMCACHED_HASH_FNV1_64;
2012   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2013
2014   return MEMCACHED_SUCCESS;
2015 }
2016
2017 memcached_return pre_hash_fnv1a_64(memcached_st *memc)
2018 {
2019   memcached_hash value= MEMCACHED_HASH_FNV1A_64;
2020   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2021
2022   return MEMCACHED_SUCCESS;
2023 }
2024
2025 memcached_return pre_hash_fnv1_32(memcached_st *memc)
2026 {
2027   memcached_hash value= MEMCACHED_HASH_FNV1_32;
2028   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2029
2030   return MEMCACHED_SUCCESS;
2031 }
2032
2033 memcached_return pre_hash_fnv1a_32(memcached_st *memc)
2034 {
2035   memcached_hash value= MEMCACHED_HASH_FNV1A_32;
2036   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2037
2038   return MEMCACHED_SUCCESS;
2039 }
2040
2041 memcached_return pre_hash_ketama(memcached_st *memc)
2042 {
2043   memcached_hash value= MEMCACHED_HASH_KETAMA;
2044   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
2045
2046   return MEMCACHED_SUCCESS;
2047 }
2048
2049 void my_free(memcached_st *ptr, void *mem)
2050 {
2051   free(mem);
2052 }
2053
2054 void *my_malloc(memcached_st *ptr, const size_t size)
2055 {
2056   return malloc(size);
2057 }
2058
2059 void *my_realloc(memcached_st *ptr, void *mem, const size_t size)
2060 {
2061   return realloc(mem, size);
2062 }
2063
2064 memcached_return set_memory_alloc(memcached_st *memc)
2065 {
2066   {
2067     memcached_malloc_function test_ptr;
2068     memcached_return rc;
2069
2070     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_MALLOC_FUNCTION, &my_malloc);
2071     assert(rc == MEMCACHED_SUCCESS);
2072     test_ptr= (memcached_malloc_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_MALLOC_FUNCTION, &rc);
2073     assert(rc == MEMCACHED_SUCCESS);
2074     assert(test_ptr == my_malloc);
2075   }
2076
2077   {
2078     memcached_realloc_function test_ptr;
2079     memcached_return rc;
2080
2081     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_REALLOC_FUNCTION, &my_realloc);
2082     assert(rc == MEMCACHED_SUCCESS);
2083     test_ptr= (memcached_realloc_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_REALLOC_FUNCTION, &rc);
2084     assert(rc == MEMCACHED_SUCCESS);
2085     assert(test_ptr == my_realloc);
2086   }
2087
2088   {
2089     memcached_free_function test_ptr;
2090     memcached_return rc;
2091
2092     rc= memcached_callback_set(memc, MEMCACHED_CALLBACK_FREE_FUNCTION, my_free);
2093     assert(rc == MEMCACHED_SUCCESS);
2094     test_ptr= (memcached_free_function)memcached_callback_get(memc, MEMCACHED_CALLBACK_FREE_FUNCTION, &rc);
2095     assert(rc == MEMCACHED_SUCCESS);
2096     assert(test_ptr == my_free);
2097   }
2098
2099   return MEMCACHED_SUCCESS;
2100 }
2101
2102 memcached_return enable_consistent(memcached_st *memc)
2103 {
2104   memcached_server_distribution value= MEMCACHED_DISTRIBUTION_CONSISTENT;
2105   memcached_hash hash;
2106   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, &value);
2107   pre_hsieh(memc);
2108
2109   value= (memcached_server_distribution)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION);
2110   assert(value == MEMCACHED_DISTRIBUTION_CONSISTENT);
2111
2112   hash= (memcached_hash)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
2113   assert(hash == MEMCACHED_HASH_HSIEH);
2114
2115
2116   return MEMCACHED_SUCCESS;
2117 }
2118
2119 memcached_return enable_cas(memcached_st *memc)
2120 {
2121   unsigned int set= 1;
2122
2123   memcached_version(memc);
2124
2125   if (memc->hosts[0].major_version >= 1 &&
2126       memc->hosts[0].minor_version >= 2 &&
2127       memc->hosts[0].micro_version >= 4)
2128   {
2129     memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SUPPORT_CAS, &set);
2130
2131     return MEMCACHED_SUCCESS;
2132   }
2133
2134   return MEMCACHED_FAILURE;
2135 }
2136
2137 memcached_return check_for_1_2_3(memcached_st *memc)
2138 {
2139   memcached_version(memc);
2140
2141   if (memc->hosts[0].major_version >= 1 &&
2142       memc->hosts[0].minor_version >= 2 &&
2143       memc->hosts[0].micro_version >= 4)
2144     return MEMCACHED_SUCCESS;
2145
2146   return MEMCACHED_FAILURE;
2147 }
2148
2149 memcached_return pre_unix_socket(memcached_st *memc)
2150 {
2151   memcached_return rc;
2152   struct stat buf;
2153
2154   memcached_server_list_free(memc->hosts);
2155   memc->hosts= NULL;
2156   memc->number_of_hosts= 0;
2157
2158   if (stat("/tmp/memcached.socket", &buf))
2159     return MEMCACHED_FAILURE;
2160
2161   rc= memcached_server_add_unix_socket(memc, "/tmp/memcached.socket");
2162
2163   return rc;
2164 }
2165
2166 memcached_return pre_udp(memcached_st *memc)
2167 {
2168   memcached_return rc;
2169
2170   memcached_server_list_free(memc->hosts);
2171   memc->hosts= NULL;
2172   memc->number_of_hosts= 0;
2173
2174   if (0)
2175     return MEMCACHED_FAILURE;
2176
2177   rc= memcached_server_add_udp(memc, "localhost", MEMCACHED_DEFAULT_PORT);
2178
2179   return rc;
2180 }
2181
2182 memcached_return pre_nodelay(memcached_st *memc)
2183 {
2184   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
2185   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, NULL);
2186
2187   return MEMCACHED_SUCCESS;
2188 }
2189
2190 memcached_return poll_timeout(memcached_st *memc)
2191 {
2192   int32_t timeout;
2193
2194   timeout= 100;
2195
2196   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_POLL_TIMEOUT, &timeout);
2197
2198   timeout= (int32_t)memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_POLL_TIMEOUT);
2199
2200   assert(timeout == 100);
2201
2202   return MEMCACHED_SUCCESS;
2203 }
2204
2205
2206 /* Clean the server before beginning testing */
2207 test_st tests[] ={
2208   {"flush", 0, flush_test },
2209   {"init", 0, init_test },
2210   {"allocation", 0, allocation_test },
2211   {"server_list_null_test", 0, server_list_null_test},
2212   {"clone_test", 0, clone_test },
2213   {"error", 0, error_test },
2214   {"set", 0, set_test },
2215   {"set2", 0, set_test2 },
2216   {"set3", 0, set_test3 },
2217   {"add", 1, add_test },
2218   {"replace", 0, replace_test },
2219   {"delete", 1, delete_test },
2220   {"get", 1, get_test },
2221   {"get2", 0, get_test2 },
2222   {"get3", 0, get_test3 },
2223   {"get4", 0, get_test4 },
2224   {"stats_servername", 0, stats_servername_test },
2225   {"increment", 0, increment_test },
2226   {"decrement", 0, decrement_test },
2227   {"quit", 0, quit_test },
2228   {"mget", 1, mget_test },
2229   {"mget_result", 1, mget_result_test },
2230   {"mget_result_alloc", 1, mget_result_alloc_test },
2231   {"mget_result_function", 1, mget_result_function },
2232   {"get_stats", 0, get_stats },
2233   {"add_host_test", 0, add_host_test },
2234   {"get_stats_keys", 0, get_stats_keys },
2235   {"behavior_test", 0, get_stats_keys },
2236   {"callback_test", 0, get_stats_keys },
2237   {"version_string_test", 0, version_string_test},
2238   {0, 0, 0}
2239 };
2240
2241 test_st async_tests[] ={
2242   {"add", 1, add_wrapper },
2243   {0, 0, 0}
2244 };
2245
2246 test_st string_tests[] ={
2247   {"string static with null", 0, string_static_null },
2248   {"string alloc with null", 0, string_alloc_null },
2249   {"string alloc with 1K", 0, string_alloc_with_size },
2250   {"string alloc with malloc failure", 0, string_alloc_with_size_toobig },
2251   {"string append", 0, string_alloc_append },
2252   {"string append failure (too big)", 0, string_alloc_append_toobig },
2253   {0, 0, 0}
2254 };
2255
2256 test_st result_tests[] ={
2257   {"result static", 0, result_static},
2258   {"result alloc", 0, result_alloc},
2259   {0, 0, 0}
2260 };
2261
2262 test_st version_1_2_3[] ={
2263   {"append", 0, append_test },
2264   {"prepend", 0, prepend_test },
2265   {"cas", 0, cas_test },
2266   {"cas2", 0, cas2_test },
2267   {"append_binary", 0, append_binary_test },
2268   {0, 0, 0}
2269 };
2270
2271 test_st user_tests[] ={
2272   {"user_supplied_bug1", 0, user_supplied_bug1 },
2273   {"user_supplied_bug2", 0, user_supplied_bug2 },
2274   {"user_supplied_bug3", 0, user_supplied_bug3 },
2275   {"user_supplied_bug4", 0, user_supplied_bug4 },
2276   {"user_supplied_bug5", 1, user_supplied_bug5 },
2277   {"user_supplied_bug6", 1, user_supplied_bug6 },
2278   {"user_supplied_bug7", 1, user_supplied_bug7 },
2279   {"user_supplied_bug8", 1, user_supplied_bug8 },
2280   {"user_supplied_bug9", 1, user_supplied_bug9 },
2281   {"user_supplied_bug10", 1, user_supplied_bug10 },
2282   {"user_supplied_bug11", 1, user_supplied_bug11 },
2283   {"user_supplied_bug12", 1, user_supplied_bug12 },
2284   {0, 0, 0}
2285 };
2286
2287 test_st generate_tests[] ={
2288   {"generate_pairs", 1, generate_pairs },
2289   {"generate_data", 1, generate_data },
2290   {"get_read", 0, get_read },
2291   {"delete_generate", 0, delete_generate },
2292   {"generate_buffer_data", 1, generate_buffer_data },
2293   {"delete_buffer", 0, delete_buffer_generate},
2294   {"generate_data", 1, generate_data },
2295   {"mget_read", 0, mget_read },
2296   {"mget_read_result", 0, mget_read_result },
2297   {"mget_read_function", 0, mget_read_function },
2298   {"cleanup", 1, cleanup_pairs },
2299   {"generate_large_pairs", 1, generate_large_pairs },
2300   {"generate_data", 1, generate_data },
2301   {"generate_buffer_data", 1, generate_buffer_data },
2302   {"cleanup", 1, cleanup_pairs },
2303   {0, 0, 0}
2304 };
2305
2306
2307 collection_st collection[] ={
2308   {"block", 0, 0, tests},
2309   {"nonblock", pre_nonblock, 0, tests},
2310   {"nodelay", pre_nodelay, 0, tests},
2311   {"md5", pre_md5, 0, tests},
2312   {"crc", pre_crc, 0, tests},
2313   {"hsieh", pre_hsieh, 0, tests},
2314   {"fnv1_64", pre_hash_fnv1_64, 0, tests},
2315   {"fnv1a_64", pre_hash_fnv1a_64, 0, tests},
2316   {"fnv1_32", pre_hash_fnv1_32, 0, tests},
2317   {"fnv1a_32", pre_hash_fnv1a_32, 0, tests},
2318   {"ketama", pre_hash_ketama, 0, tests},
2319   {"unix_socket", pre_unix_socket, 0, tests},
2320   {"unix_socket_nodelay", pre_nodelay, 0, tests},
2321   {"poll_timeout", poll_timeout, 0, tests},
2322   {"gets", enable_cas, 0, tests},
2323   {"consistent", enable_consistent, 0, tests},
2324   {"memory_allocators", set_memory_alloc, 0, tests},
2325 //  {"udp", pre_udp, 0, tests},
2326   {"version_1_2_3", check_for_1_2_3, 0, version_1_2_3},
2327   {"string", 0, 0, string_tests},
2328   {"result", 0, 0, result_tests},
2329   {"async", pre_nonblock, 0, async_tests},
2330   {"user", 0, 0, user_tests},
2331   {"generate", 0, 0, generate_tests},
2332   {"generate_hsieh", pre_hsieh, 0, generate_tests},
2333   {"generate_hsieh_consistent", enable_consistent, 0, generate_tests},
2334   {"generate_md5", pre_md5, 0, generate_tests},
2335   {"generate_nonblock", pre_nonblock, 0, generate_tests},
2336   {0, 0, 0, 0}
2337 };
2338
2339 collection_st *gets_collections(void)
2340 {
2341   return collection;
2342 }