git.m6w6.name Git - awesomized/libmemcached/blob - tests/test.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
  16 long int timedif(struct timeval a, struct timeval b)
  17 {
  18   register int us, s;
  19
  20   us = a.tv_usec - b.tv_usec;
  21   us /= 1000;
  22   s = a.tv_sec - b.tv_sec;
  23   s *= 1000;
  24   return s + us;
  25 }
  26
  27 void init_test(memcached_st *not_used)
  28 {
  29   memcached_st memc;
  30
  31   (void)memcached_create(&memc);
  32   memcached_free(&memc);
  33 }
  34
  35 void allocation_test(memcached_st *not_used)
  36 {
  37   memcached_st *memc;
  38   memc= memcached_create(NULL);
  39   assert(memc);
  40   memcached_free(memc);
  41 }
  42
  43 void clone_test(memcached_st *memc)
  44 {
  45   /* All null? */
  46   {
  47     memcached_st *clone;
  48     clone= memcached_clone(NULL, NULL);
  49     assert(clone);
  50     memcached_free(clone);
  51   }
  52
  53   /* Can we init from null? */
  54   {
  55     memcached_st *clone;
  56     clone= memcached_clone(NULL, memc);
  57     assert(clone);
  58     memcached_free(clone);
  59   }
  60
  61   /* Can we init from struct? */
  62   {
  63     memcached_st declared_clone;
  64     memcached_st *clone;
  65     clone= memcached_clone(&declared_clone, NULL);
  66     assert(clone);
  67     memcached_free(clone);
  68   }
  69
  70   /* Can we init from struct? */
  71   {
  72     memcached_st declared_clone;
  73     memcached_st *clone;
  74     clone= memcached_clone(&declared_clone, memc);
  75     assert(clone);
  76     memcached_free(clone);
  77   }
  78 }
  79
  80 void connection_test(memcached_st *memc)
  81 {
  82   memcached_return rc;
  83
  84   rc= memcached_server_add(memc, "localhost", 0);
  85   assert(rc == MEMCACHED_SUCCESS);
  86 }
  87
  88 void error_test(memcached_st *memc)
  89 {
  90   memcached_return rc;
  91
  92   for (rc= MEMCACHED_SUCCESS; rc < MEMCACHED_MAXIMUM_RETURN; rc++)
  93   {
  94     printf("Error %d -> %s\n", rc, memcached_strerror(memc, rc));
  95   }
  96 }
  97
  98 void set_test(memcached_st *memc)
  99 {
 100   memcached_return rc;
 101   char *key= "foo";
 102   char *value= "when we sanitize";
 103
 104   rc= memcached_set(memc, key, strlen(key),
 105                     value, strlen(value),
 106                     (time_t)0, (uint16_t)0);
 107   assert(rc == MEMCACHED_SUCCESS);
 108 }
 109
 110 void add_test(memcached_st *memc)
 111 {
 112   memcached_return rc;
 113   char *key= "foo";
 114   char *value= "when we sanitize";
 115
 116   rc= memcached_add(memc, key, strlen(key),
 117                     value, strlen(value),
 118                     (time_t)0, (uint16_t)0);
 119   assert(rc == MEMCACHED_NOTSTORED);
 120 }
 121
 122 void replace_test(memcached_st *memc)
 123 {
 124   memcached_return rc;
 125   char *key= "foo";
 126   char *value= "when we sanitize";
 127
 128   rc= memcached_replace(memc, key, strlen(key),
 129                     value, strlen(value),
 130                     (time_t)0, (uint16_t)0);
 131   assert(rc == MEMCACHED_SUCCESS);
 132 }
 133
 134 void delete_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, (uint16_t)0);
 143   assert(rc == MEMCACHED_SUCCESS);
 144
 145   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 146   assert(rc == MEMCACHED_SUCCESS);
 147 }
 148
 149 void flush_test(memcached_st *memc)
 150 {
 151   memcached_return rc;
 152
 153   rc= memcached_flush(memc, 0);
 154   assert(rc == MEMCACHED_SUCCESS);
 155 }
 156
 157 void get_test(memcached_st *memc)
 158 {
 159   memcached_return rc;
 160   char *key= "foo";
 161   char *string;
 162   size_t string_length;
 163   uint16_t flags;
 164
 165   rc= memcached_delete(memc, key, strlen(key), (time_t)0);
 166   assert(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_NOTFOUND);
 167
 168   string= memcached_get(memc, key, strlen(key),
 169                         &string_length, &flags, &rc);
 170
 171   assert(rc == MEMCACHED_NOTFOUND);
 172   assert(string_length ==  0);
 173   assert(!string);
 174 }
 175
 176 void get_test2(memcached_st *memc)
 177 {
 178   memcached_return rc;
 179   char *key= "foo";
 180   char *value= "when we sanitize";
 181   char *string;
 182   size_t string_length;
 183   uint16_t flags;
 184
 185   rc= memcached_set(memc, key, strlen(key),
 186                     value, strlen(value),
 187                     (time_t)0, (uint16_t)0);
 188   assert(rc == MEMCACHED_SUCCESS);
 189
 190   string= memcached_get(memc, key, strlen(key),
 191                         &string_length, &flags, &rc);
 192
 193   assert(string);
 194   assert(rc == MEMCACHED_SUCCESS);
 195   assert(string_length == strlen(value));
 196   assert(!memcmp(string, value, string_length));
 197
 198   free(string);
 199 }
 200
 201 void set_test2(memcached_st *memc)
 202 {
 203   memcached_return rc;
 204   char *key= "foo";
 205   char *value= "train in the brain";
 206   size_t value_length= strlen(value);
 207   unsigned int x;
 208
 209   for (x= 0; x < 10; x++)
 210   {
 211     rc= memcached_set(memc, key, strlen(key),
 212                       value, value_length,
 213                       (time_t)0, (uint16_t)0);
 214     assert(rc == MEMCACHED_SUCCESS);
 215   }
 216 }
 217
 218 void set_test3(memcached_st *memc)
 219 {
 220   memcached_return rc;
 221   char *key= "foo";
 222   char *value;
 223   size_t value_length= 8191;
 224   unsigned int x;
 225
 226   value = (char*)malloc(value_length);
 227   assert(value);
 228
 229   for (x= 0; x < value_length; x++)
 230     value[x] = (char) (x % 127);
 231
 232   for (x= 0; x < 1; x++)
 233   {
 234     rc= memcached_set(memc, key, strlen(key),
 235                       value, value_length,
 236                       (time_t)0, (uint16_t)0);
 237     assert(rc == MEMCACHED_SUCCESS);
 238   }
 239
 240   free(value);
 241 }
 242
 243 void get_test3(memcached_st *memc)
 244 {
 245   memcached_return rc;
 246   char *key= "foo";
 247   char *value;
 248   size_t value_length= 8191;
 249   char *string;
 250   size_t string_length;
 251   uint16_t flags;
 252   int x;
 253
 254   value = (char*)malloc(value_length);
 255   assert(value);
 256
 257   for (x= 0; x < value_length; x++)
 258     value[x] = (char) (x % 127);
 259
 260   rc= memcached_set(memc, key, strlen(key),
 261                     value, value_length,
 262                     (time_t)0, (uint16_t)0);
 263   assert(rc == MEMCACHED_SUCCESS);
 264
 265   string= memcached_get(memc, key, strlen(key),
 266                         &string_length, &flags, &rc);
 267
 268   assert(rc == MEMCACHED_SUCCESS);
 269   assert(string);
 270   assert(string_length == value_length);
 271   assert(!memcmp(string, value, string_length));
 272
 273   free(string);
 274   free(value);
 275 }
 276
 277 void get_test4(memcached_st *memc)
 278 {
 279   memcached_return rc;
 280   char *key= "foo";
 281   char *value;
 282   size_t value_length= 8191;
 283   char *string;
 284   size_t string_length;
 285   uint16_t flags;
 286   int x;
 287
 288   value = (char*)malloc(value_length);
 289   assert(value);
 290
 291   for (x= 0; x < value_length; x++)
 292     value[x] = (char) (x % 127);
 293
 294   rc= memcached_set(memc, key, strlen(key),
 295                     value, value_length,
 296                     (time_t)0, (uint16_t)0);
 297   assert(rc == MEMCACHED_SUCCESS);
 298
 299   for (x= 0; x < 10; x++)
 300   {
 301     string= memcached_get(memc, key, strlen(key),
 302                           &string_length, &flags, &rc);
 303
 304     assert(rc == MEMCACHED_SUCCESS);
 305     assert(string);
 306     assert(string_length == value_length);
 307     assert(!memcmp(string, value, string_length));
 308     free(string);
 309   }
 310
 311   free(value);
 312 }
 313
 314 void stats_servername_test(memcached_st *memc)
 315 {
 316   memcached_return rc;
 317   memcached_stat_st stat;
 318   rc= memcached_stat_servername(&stat, NULL,
 319                                  "localhost",
 320                                  MEMCACHED_DEFAULT_PORT);
 321 }
 322
 323 void increment_test(memcached_st *memc)
 324 {
 325   unsigned int new_number;
 326   memcached_return rc;
 327   char *key= "number";
 328   char *value= "0";
 329
 330   rc= memcached_set(memc, key, strlen(key),
 331                     value, strlen(value),
 332                     (time_t)0, (uint16_t)0);
 333   assert(rc == MEMCACHED_SUCCESS);
 334
 335   rc= memcached_increment(memc, key, strlen(key),
 336                           1, &new_number);
 337   assert(rc == MEMCACHED_SUCCESS);
 338   assert(new_number == 1);
 339
 340   rc= memcached_increment(memc, key, strlen(key),
 341                           1, &new_number);
 342   assert(rc == MEMCACHED_SUCCESS);
 343   assert(new_number == 2);
 344 }
 345
 346 void decrement_test(memcached_st *memc)
 347 {
 348   unsigned int new_number;
 349   memcached_return rc;
 350   char *key= "number";
 351   char *value= "3";
 352
 353   rc= memcached_set(memc, key, strlen(key),
 354                     value, strlen(value),
 355                     (time_t)0, (uint16_t)0);
 356   assert(rc == MEMCACHED_SUCCESS);
 357
 358   rc= memcached_decrement(memc, key, strlen(key),
 359                           1, &new_number);
 360   assert(rc == MEMCACHED_SUCCESS);
 361   assert(new_number == 2);
 362
 363   rc= memcached_decrement(memc, key, strlen(key),
 364                           1, &new_number);
 365   assert(rc == MEMCACHED_SUCCESS);
 366   assert(new_number == 1);
 367 }
 368
 369 void quit_test(memcached_st *memc)
 370 {
 371   memcached_return rc;
 372   char *key= "fudge";
 373   char *value= "sanford and sun";
 374
 375   rc= memcached_set(memc, key, strlen(key),
 376                     value, strlen(value),
 377                     (time_t)10, (uint16_t)3);
 378   assert(rc == MEMCACHED_SUCCESS);
 379   memcached_quit(memc);
 380
 381   rc= memcached_set(memc, key, strlen(key),
 382                     value, strlen(value),
 383                     (time_t)50, (uint16_t)9);
 384   assert(rc == MEMCACHED_SUCCESS);
 385 }
 386
 387 void mget_test(memcached_st *memc)
 388 {
 389   memcached_return rc;
 390   char *keys[]= {"fudge", "son", "food"};
 391   size_t key_length[]= {5, 3, 4};
 392   unsigned int x;
 393   uint16_t flags;
 394
 395   char return_key[MEMCACHED_MAX_KEY];
 396   size_t return_key_length;
 397   char *return_value;
 398   size_t return_value_length;
 399
 400   /* We need to empty the server before continueing test */
 401   rc= memcached_flush(memc, 0);
 402   assert(rc == MEMCACHED_SUCCESS);
 403
 404   rc= memcached_mget(memc, keys, key_length, 3);
 405   assert(rc == MEMCACHED_SUCCESS);
 406
 407   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 408                       &return_value_length, &flags, &rc)) != NULL)
 409   {
 410     assert(return_value);
 411   }
 412   assert(!return_value);
 413   assert(return_value_length == 0);
 414   assert(rc == MEMCACHED_NOTFOUND);
 415
 416   for (x= 0; x < 3; x++)
 417   {
 418     rc= memcached_set(memc, keys[x], key_length[x],
 419                       keys[x], key_length[x],
 420                       (time_t)50, (uint16_t)9);
 421     assert(rc == MEMCACHED_SUCCESS);
 422   }
 423
 424   rc= memcached_mget(memc, keys, key_length, 3);
 425   assert(rc == MEMCACHED_SUCCESS);
 426
 427   x= 0;
 428   while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 429                                         &return_value_length, &flags, &rc)))
 430   {
 431     assert(return_value);
 432     assert(rc == MEMCACHED_SUCCESS);
 433     assert(return_key_length == return_value_length);
 434     assert(!memcmp(return_value, return_key, return_value_length));
 435     free(return_value);
 436     x++;
 437   }
 438 }
 439
 440 void get_stats_keys(memcached_st *memc)
 441 {
 442  char **list;
 443  char **ptr;
 444  memcached_stat_st stat;
 445  memcached_return rc;
 446
 447  list= memcached_stat_get_keys(memc, &stat, &rc);
 448  assert(rc == MEMCACHED_SUCCESS);
 449  for (ptr= list; *ptr; ptr++)
 450    printf("Found key %s\n", *ptr);
 451  fflush(stdout);
 452
 453  free(list);
 454 }
 455
 456 void get_stats(memcached_st *memc)
 457 {
 458  unsigned int x;
 459  char **list;
 460  char **ptr;
 461  memcached_return rc;
 462  memcached_stat_st *stat;
 463
 464  stat= memcached_stat(memc, NULL, &rc);
 465  assert(rc == MEMCACHED_SUCCESS);
 466
 467  assert(rc == MEMCACHED_SUCCESS);
 468  assert(stat);
 469
 470  for (x= 0; x < memcached_server_count(memc); x++)
 471  {
 472    list= memcached_stat_get_keys(memc, &stat[x], &rc);
 473    assert(rc == MEMCACHED_SUCCESS);
 474    for (ptr= list; *ptr; ptr++)
 475      printf("Found key %s\n", *ptr);
 476
 477    free(list);
 478  }
 479
 480  memcached_stat_free(NULL, stat);
 481 }
 482
 483 void add_host_test(memcached_st *memc)
 484 {
 485   unsigned int x;
 486   memcached_server_st *servers;
 487   memcached_return rc;
 488   char servername[]= "0.example.com";
 489
 490   servers= memcached_server_list_append(NULL, servername, 400, &rc);
 491   assert(servers);
 492   assert(1 == memcached_server_list_count(servers));
 493
 494   for (x= 2; x < 20; x++)
 495   {
 496     char buffer[SMALL_STRING_LEN];
 497
 498     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
 499     servers= memcached_server_list_append(servers, buffer, 401,
 500                                      &rc);
 501     assert(rc == MEMCACHED_SUCCESS);
 502     assert(x == memcached_server_list_count(servers));
 503   }
 504
 505   rc= memcached_server_push(memc, servers);
 506   assert(rc == MEMCACHED_SUCCESS);
 507   rc= memcached_server_push(memc, servers);
 508   assert(rc == MEMCACHED_SUCCESS);
 509
 510   memcached_server_list_free(servers);
 511 }
 512
 513 /* We don't test the behavior itself, we test the switches */
 514 void behavior_test(memcached_st *memc)
 515 {
 516   unsigned long long value;
 517   unsigned int set= 1;
 518
 519   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
 520   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
 521   assert(value == 1);
 522
 523   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
 524   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
 525   assert(value == 1);
 526
 527   set= MEMCACHED_HASH_MD5;
 528   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
 529   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
 530   assert(value == MEMCACHED_HASH_MD5);
 531
 532   set= 0;
 533
 534   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
 535   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
 536   assert(value == 0);
 537
 538   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
 539   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
 540   assert(value == 0);
 541
 542   set= MEMCACHED_HASH_DEFAULT;
 543   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
 544   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
 545   assert(value == MEMCACHED_HASH_DEFAULT);
 546
 547   set= MEMCACHED_HASH_CRC;
 548   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &set);
 549   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_HASH);
 550   assert(value == MEMCACHED_HASH_CRC);
 551
 552   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 553   assert(value > 0);
 554
 555   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 556   assert(value > 0);
 557 }
 558
 559 /* Test case provided by Cal Haldenbrand */
 560 void user_supplied_bug1(memcached_st *memc)
 561 {
 562   unsigned int setter= 1;
 563   unsigned int x;
 564
 565   unsigned long long total= 0;
 566   int size= 0;
 567   char key[10];
 568   char randomstuff[6 * 1024];
 569   memcached_return rc;
 570
 571   memset(randomstuff, 0, 6 * 1024);
 572
 573   /* We just keep looking at the same values over and over */
 574   srandom(10);
 575
 576   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 577   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 578
 579
 580   /* add key */
 581   for (x= 0 ; total < 20 * 1024576 ; x++ )
 582   {
 583     unsigned int j= 0;
 584
 585     size= (rand() % ( 5 * 1024 ) ) + 400;
 586     memset(randomstuff, 0, 6 * 1024);
 587     assert(size < 6 * 1024); /* Being safe here */
 588
 589     for (j= 0 ; j < size ;j++)
 590       randomstuff[j] = (char) (rand() % 26) + 97;
 591
 592     total += size;
 593     sprintf(key, "%d", x);
 594     rc = memcached_set(memc, key, strlen(key),
 595                        randomstuff, strlen(randomstuff), 10, 0);
 596     /* If we fail, lets try again */
 597     if (rc != MEMCACHED_SUCCESS)
 598       rc = memcached_set(memc, key, strlen(key),
 599                          randomstuff, strlen(randomstuff), 10, 0);
 600     assert(rc == MEMCACHED_SUCCESS);
 601   }
 602 }
 603
 604 /* Test case provided by Cal Haldenbrand */
 605 void user_supplied_bug2(memcached_st *memc)
 606 {
 607   int errors;
 608   unsigned int setter;
 609   unsigned int x;
 610   unsigned long long total;
 611
 612   setter= 1;
 613   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 614   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 615 #ifdef NOT_YET
 616   setter = 20 * 1024576;
 617   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
 618   setter = 20 * 1024576;
 619   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
 620   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 621   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 622
 623   for (x= 0, errors= 0, total= 0 ; total < 20 * 1024576 ; x++)
 624 #endif
 625
 626   for (x= 0, errors= 0, total= 0 ; total < 24576 ; x++)
 627   {
 628     memcached_return rc= MEMCACHED_SUCCESS;
 629     char buffer[SMALL_STRING_LEN];
 630     uint16_t flags= 0;
 631     size_t val_len= 0;
 632     char *getval;
 633
 634     memset(buffer, 0, SMALL_STRING_LEN);
 635
 636     snprintf(buffer, SMALL_STRING_LEN, "%u", x);
 637     getval= memcached_get(memc, buffer, strlen(buffer),
 638                            &val_len, &flags, &rc);
 639     if (rc != MEMCACHED_SUCCESS)
 640     {
 641       if (rc == MEMCACHED_NOTFOUND)
 642         errors++;
 643       else
 644         assert(0);
 645
 646       continue;
 647     }
 648     total+= val_len;
 649     errors= 0;
 650     free(getval);
 651   }
 652 }
 653
 654 /* Do a large mget() over all the keys we think exist */
 655 #define KEY_COUNT 3000 // * 1024576
 656 void user_supplied_bug3(memcached_st *memc)
 657 {
 658   memcached_return rc;
 659   unsigned int setter;
 660   unsigned int x;
 661   char **keys;
 662   size_t key_lengths[KEY_COUNT];
 663
 664   setter= 1;
 665   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 666   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 667 #ifdef NOT_YET
 668   setter = 20 * 1024576;
 669   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
 670   setter = 20 * 1024576;
 671   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
 672   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 673   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 674 #endif
 675
 676   keys= (char **)malloc(sizeof(char *) * KEY_COUNT);
 677   assert(keys);
 678   memset(keys, 0, (sizeof(char *) * KEY_COUNT));
 679   for (x= 0; x < KEY_COUNT; x++)
 680   {
 681     char buffer[30];
 682
 683     snprintf(buffer, 30, "%u", x);
 684     keys[x]= strdup(buffer);
 685     key_lengths[x]= strlen(keys[x]);
 686   }
 687
 688   rc= memcached_mget(memc, keys, key_lengths, KEY_COUNT);
 689   assert(rc == MEMCACHED_SUCCESS);
 690
 691   /* Turn this into a help function */
 692   {
 693     char return_key[MEMCACHED_MAX_KEY];
 694     size_t return_key_length;
 695     char *return_value;
 696     size_t return_value_length;
 697     uint16_t flags;
 698
 699     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 700                                           &return_value_length, &flags, &rc)))
 701     {
 702       assert(return_value);
 703       assert(rc == MEMCACHED_SUCCESS);
 704       free(return_value);
 705     }
 706   }
 707
 708   for (x= 0; x < KEY_COUNT; x++)
 709     free(keys[x]);
 710   free(keys);
 711 }
 712
 713 void result_static(memcached_st *memc)
 714 {
 715   memcached_result_st result;
 716   memcached_result_st *result_ptr;
 717
 718   result_ptr= memcached_result_create(memc, &result);
 719   assert(result.is_allocated == MEMCACHED_NOT_ALLOCATED);
 720   assert(result_ptr);
 721   memcached_result_free(&result);
 722 }
 723
 724 void result_alloc(memcached_st *memc)
 725 {
 726   memcached_result_st *result;
 727
 728   result= memcached_result_create(memc, NULL);
 729   assert(result);
 730   memcached_result_free(result);
 731 }
 732
 733 void string_static_null(memcached_st *memc)
 734 {
 735   memcached_string_st string;
 736   memcached_string_st *string_ptr;
 737
 738   string_ptr= memcached_string_create(memc, &string, 0);
 739   assert(string.is_allocated == MEMCACHED_NOT_ALLOCATED);
 740   assert(string_ptr);
 741   memcached_string_free(&string);
 742 }
 743
 744 void string_alloc_null(memcached_st *memc)
 745 {
 746   memcached_string_st *string;
 747
 748   string= memcached_string_create(memc, NULL, 0);
 749   assert(string);
 750   memcached_string_free(string);
 751 }
 752
 753 void string_alloc_with_size(memcached_st *memc)
 754 {
 755   memcached_string_st *string;
 756
 757   string= memcached_string_create(memc, NULL, 1024);
 758   assert(string);
 759   memcached_string_free(string);
 760 }
 761
 762 void string_alloc_with_size_toobig(memcached_st *memc)
 763 {
 764   memcached_string_st *string;
 765
 766   string= memcached_string_create(memc, NULL, INT64_MAX);
 767   assert(string == NULL);
 768 }
 769
 770 void string_alloc_append(memcached_st *memc)
 771 {
 772   unsigned int x;
 773   char buffer[SMALL_STRING_LEN];
 774   memcached_string_st *string;
 775
 776   /* Ring the bell! */
 777   memset(buffer, 6, SMALL_STRING_LEN);
 778
 779   string= memcached_string_create(memc, NULL, 100);
 780   assert(string);
 781
 782   for (x= 0; x < 1024; x++)
 783   {
 784     memcached_return rc;
 785     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
 786     assert(rc == MEMCACHED_SUCCESS);
 787   }
 788   memcached_string_free(string);
 789 }
 790
 791 void string_alloc_append_toobig(memcached_st *memc)
 792 {
 793   memcached_return rc;
 794   unsigned int x;
 795   char buffer[SMALL_STRING_LEN];
 796   memcached_string_st *string;
 797
 798   /* Ring the bell! */
 799   memset(buffer, 6, SMALL_STRING_LEN);
 800
 801   string= memcached_string_create(memc, NULL, 100);
 802   assert(string);
 803
 804   for (x= 0; x < 1024; x++)
 805   {
 806     rc= memcached_string_append(string, buffer, SMALL_STRING_LEN);
 807     assert(rc == MEMCACHED_SUCCESS);
 808   }
 809   rc= memcached_string_append(string, buffer, INT64_MAX);
 810   assert(rc == MEMCACHED_MEMORY_ALLOCATION_FAILURE);
 811   memcached_string_free(string);
 812 }
 813
 814 void add_host_test1(memcached_st *memc)
 815 {
 816   unsigned int x;
 817   memcached_return rc;
 818   char servername[]= "0.example.com";
 819   memcached_server_st *servers;
 820
 821   servers= memcached_server_list_append(NULL, servername, 400, &rc);
 822   assert(servers);
 823   assert(1 == memcached_server_list_count(servers));
 824
 825   for (x= 2; x < 20; x++)
 826   {
 827     char buffer[SMALL_STRING_LEN];
 828
 829     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
 830     servers= memcached_server_list_append(servers, buffer, 401,
 831                                      &rc);
 832     assert(rc == MEMCACHED_SUCCESS);
 833     assert(x == memcached_server_list_count(servers));
 834   }
 835
 836   rc= memcached_server_push(memc, servers);
 837   assert(rc == MEMCACHED_SUCCESS);
 838   rc= memcached_server_push(memc, servers);
 839   assert(rc == MEMCACHED_SUCCESS);
 840
 841   memcached_server_list_free(servers);
 842 }
 843
 844 memcached_return pre_nonblock(memcached_st *memc)
 845 {
 846   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
 847
 848   return MEMCACHED_SUCCESS;
 849 }
 850
 851 memcached_return pre_md5(memcached_st *memc)
 852 {
 853   memcached_hash value= MEMCACHED_HASH_MD5;
 854   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 855
 856   return MEMCACHED_SUCCESS;
 857 }
 858
 859 memcached_return pre_crc(memcached_st *memc)
 860 {
 861   memcached_hash value= MEMCACHED_HASH_CRC;
 862   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 863
 864   return MEMCACHED_SUCCESS;
 865 }
 866
 867 memcached_return pre_hash_fnv1_64(memcached_st *memc)
 868 {
 869   memcached_hash value= MEMCACHED_HASH_FNV1_64;
 870   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 871
 872   return MEMCACHED_SUCCESS;
 873 }
 874
 875 memcached_return pre_hash_fnv1a_64(memcached_st *memc)
 876 {
 877   memcached_hash value= MEMCACHED_HASH_FNV1A_64;
 878   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 879
 880   return MEMCACHED_SUCCESS;
 881 }
 882
 883 memcached_return pre_hash_fnv1_32(memcached_st *memc)
 884 {
 885   memcached_hash value= MEMCACHED_HASH_FNV1_32;
 886   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 887
 888   return MEMCACHED_SUCCESS;
 889 }
 890
 891 memcached_return pre_hash_fnv1a_32(memcached_st *memc)
 892 {
 893   memcached_hash value= MEMCACHED_HASH_FNV1A_32;
 894   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 895
 896   return MEMCACHED_SUCCESS;
 897 }
 898
 899 memcached_return pre_hash_ketama(memcached_st *memc)
 900 {
 901   memcached_hash value= MEMCACHED_HASH_KETAMA;
 902   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, &value);
 903
 904   return MEMCACHED_SUCCESS;
 905 }
 906
 907 memcached_return pre_unix_socket(memcached_st *memc)
 908 {
 909   memcached_return rc;
 910   struct stat buf;
 911
 912   memcached_server_list_free(memc->hosts);
 913   memc->hosts= NULL;
 914   memc->number_of_hosts= 0;
 915
 916   if (stat("/tmp/memcached.socket", &buf))
 917     return MEMCACHED_FAILURE;
 918
 919   rc= memcached_server_add_unix_socket(memc, "/tmp/memcached.socket");
 920
 921   return rc;
 922 }
 923
 924 memcached_return pre_nodelay(memcached_st *memc)
 925 {
 926   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
 927   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, NULL);
 928
 929   return MEMCACHED_SUCCESS;
 930 }
 931
 932 typedef struct collection_st collection_st;
 933 typedef struct test_st test_st;
 934
 935 struct test_st {
 936   char *name;
 937   unsigned int requires_flush;
 938   void (*function)(memcached_st *memc);
 939 };
 940
 941 struct collection_st {
 942   char *name;
 943   memcached_return (*pre)(memcached_st *memc);
 944   memcached_return (*post)(memcached_st *memc);
 945   test_st *tests;
 946 };
 947
 948 int main(int argc, char *argv[])
 949 {
 950   unsigned int x;
 951   char *server_list;
 952   char *collection_to_run= NULL;
 953   char *wildcard= NULL;
 954   memcached_server_st *servers;
 955
 956
 957   if (argc > 1)
 958     collection_to_run= argv[1];
 959
 960   if (argc == 3)
 961     wildcard= argv[2];
 962
 963   if (!(server_list= getenv("MEMCACHED_SERVERS")))
 964     server_list= "localhost";
 965
 966   printf("servers %s\n", server_list);
 967   srandom(time(NULL));
 968
 969   servers= memcached_servers_parse(server_list);
 970   assert(servers);
 971
 972   for (x= 0; x < memcached_server_list_count(servers); x++)
 973   {
 974     printf("\t%s : %u\n", servers[x].hostname, servers[x].port);
 975     assert(servers[x].stack_responses == 0);
 976     assert(servers[x].fd == -1);
 977     assert(servers[x].cursor_active == 0);
 978   }
 979
 980   printf("\n");
 981
 982   /* Clean the server before beginning testing */
 983   test_st tests[] ={
 984     {"flush", 0, flush_test },
 985     {"init", 0, init_test },
 986     {"allocation", 0, allocation_test },
 987     {"clone_test", 0, clone_test },
 988     {"error", 0, error_test },
 989     {"set", 0, set_test },
 990     {"set2", 0, set_test2 },
 991     {"set3", 0, set_test3 },
 992     {"add", 0, add_test },
 993     {"replace", 0, replace_test },
 994     {"delete", 1, delete_test },
 995     {"get", 1, get_test },
 996     {"get2", 0, get_test2 },
 997     {"get3", 0, get_test3 },
 998     {"get4", 0, get_test4 },
 999     {"stats_servername", 0, stats_servername_test },
1000     {"increment", 0, increment_test },
1001     {"decrement", 0, decrement_test },
1002     {"quit", 0, quit_test },
1003     {"mget", 0, mget_test },
1004     {"get_stats", 0, get_stats },
1005     {"add_host_test", 0, add_host_test },
1006     {"get_stats_keys", 0, get_stats_keys },
1007     {"behavior_test", 0, get_stats_keys },
1008     {0, 0, 0}
1009   };
1010
1011   test_st string_tests[] ={
1012     {"string static with null", 0, string_static_null },
1013     {"string alloc with null", 0, string_alloc_null },
1014     {"string alloc with 1K", 0, string_alloc_with_size },
1015     {"string alloc with malloc failure", 0, string_alloc_with_size_toobig },
1016     {"string append", 0, string_alloc_append },
1017     {"string append failure (too big)", 0, string_alloc_append_toobig },
1018     {0, 0, 0}
1019   };
1020
1021   test_st result_tests[] ={
1022     {"result static", 0, result_static},
1023     {"result alloc", 0, result_alloc},
1024     {0, 0, 0}
1025   };
1026
1027   test_st user_tests[] ={
1028     {"user_supplied_bug1", 0, user_supplied_bug1 },
1029     {"user_supplied_bug2", 0, user_supplied_bug2 },
1030     {"user_supplied_bug3", 0, user_supplied_bug3 },
1031     {0, 0, 0}
1032   };
1033
1034
1035   collection_st collection[] ={
1036     {"block", 0, 0, tests},
1037     {"nonblock", pre_nonblock, 0, tests},
1038     {"nodelay", pre_nodelay, 0, tests},
1039     {"md5", pre_md5, 0, tests},
1040     {"crc", pre_crc, 0, tests},
1041     {"fnv1_64", pre_hash_fnv1_64, 0, tests},
1042     {"fnv1a_64", pre_hash_fnv1a_64, 0, tests},
1043     {"fnv1_32", pre_hash_fnv1_32, 0, tests},
1044     {"fnv1a_32", pre_hash_fnv1a_32, 0, tests},
1045     {"ketama", pre_hash_ketama, 0, tests},
1046     {"unix_socket", pre_unix_socket, 0, tests},
1047     {"unix_socket_nodelay", pre_nodelay, 0, tests},
1048     {"string", 0, 0, string_tests},
1049     {"result", 0, 0, result_tests},
1050     {"user", 0, 0, user_tests},
1051     {0, 0, 0, 0}
1052   };
1053
1054   collection_st *next;
1055   for (next= collection; next->name; next++)
1056   {
1057     test_st *run;
1058
1059     run= next->tests;
1060
1061     if (collection_to_run && strcmp(collection_to_run, next->name))
1062       continue;
1063
1064     fprintf(stderr, "\n%s\n\n", next->name);
1065
1066     for (x= 0; run->name; run++)
1067     {
1068       if (wildcard && strcmp(wildcard, run->name))
1069         continue;
1070
1071       fprintf(stderr, "Testing %s", run->name);
1072
1073       memcached_st *memc;
1074       memcached_return rc;
1075       struct timeval start_time, end_time;
1076
1077       memc= memcached_create(NULL);
1078       assert(memc);
1079
1080       if (run->requires_flush)
1081         memcached_flush(memc, 0);
1082
1083       rc= memcached_server_push(memc, servers);
1084       assert(rc == MEMCACHED_SUCCESS);
1085
1086       unsigned int loop;
1087       for (loop= 0; loop < memcached_server_list_count(servers); loop++)
1088       {
1089         assert(memc->hosts[loop].stack_responses == 0);
1090         assert(memc->hosts[loop].fd == -1);
1091         assert(memc->hosts[loop].cursor_active == 0);
1092       }
1093
1094       if (next->pre)
1095       {
1096         memcached_return rc;
1097         rc= next->pre(memc);
1098
1099         if (rc != MEMCACHED_SUCCESS)
1100         {
1101           fprintf(stderr, "\t\t\t\t\t [ skipping ]\n");
1102           goto error;
1103         }
1104       }
1105
1106       gettimeofday(&start_time, NULL);
1107       run->function(memc);
1108       gettimeofday(&end_time, NULL);
1109       long int load_time= timedif(end_time, start_time);
1110       fprintf(stderr, "\t\t\t\t\t %ld.%03ld [ ok ]\n", load_time / 1000,
1111               load_time % 1000);
1112
1113       if (next->post)
1114         (void)next->post(memc);
1115
1116       assert(memc);
1117 error:
1118       memcached_free(memc);
1119     }
1120   }
1121
1122   /* Clean up whatever we might have left */
1123   {
1124     memcached_st *memc;
1125     memc= memcached_create(NULL);
1126     assert(memc);
1127     flush_test(memc);
1128     memcached_free(memc);
1129   }
1130
1131   fprintf(stderr, "All tests completed successfully\n\n");
1132
1133   memcached_server_list_free(servers);
1134
1135   return 0;
1136 }