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  free(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   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, &set);
 528   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_MD5_HASHING);
 529   assert(value == 1);
 530
 531   set= 0;
 532
 533   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &set);
 534   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
 535   assert(value == 0);
 536
 537   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &set);
 538   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY);
 539   assert(value == 0);
 540
 541   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, &set);
 542   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_MD5_HASHING);
 543   assert(value == 0);
 544
 545   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 546   assert(value > 0);
 547
 548   value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 549   assert(value > 0);
 550 }
 551
 552 /* Test case provided by Cal Haldenbrand */
 553 void user_supplied_bug1(memcached_st *memc)
 554 {
 555   unsigned int setter= 1;
 556   unsigned int x;
 557
 558   unsigned long long total= 0;
 559   int size= 0;
 560   char key[10];
 561   char randomstuff[6 * 1024];
 562   memcached_return rc;
 563
 564   memset(randomstuff, 0, 6 * 1024);
 565
 566   /* We just keep looking at the same values over and over */
 567   srandom(10);
 568
 569   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 570   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 571
 572
 573   /* add key */
 574   for (x= 0 ; total < 20 * 1024576 ; x++ )
 575   {
 576     unsigned int j= 0;
 577
 578     size= (rand() % ( 5 * 1024 ) ) + 400;
 579     memset(randomstuff, 0, 6 * 1024);
 580     assert(size < 6 * 1024); /* Being safe here */
 581
 582     for (j= 0 ; j < size ;j++)
 583       randomstuff[j] = (char) (rand() % 26) + 97;
 584
 585     total += size;
 586     sprintf(key, "%d", x);
 587     rc = memcached_set(memc, key, strlen(key),
 588                        randomstuff, strlen(randomstuff), 10, 0);
 589     /* If we fail, lets try again */
 590     if (rc != MEMCACHED_SUCCESS)
 591       rc = memcached_set(memc, key, strlen(key),
 592                          randomstuff, strlen(randomstuff), 10, 0);
 593     assert(rc == MEMCACHED_SUCCESS);
 594   }
 595 }
 596
 597 /* Test case provided by Cal Haldenbrand */
 598 void user_supplied_bug2(memcached_st *memc)
 599 {
 600   int errors;
 601   unsigned int setter;
 602   unsigned int x;
 603   unsigned long long total;
 604
 605   setter= 1;
 606   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 607   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 608 #ifdef NOT_YET
 609   setter = 20 * 1024576;
 610   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
 611   setter = 20 * 1024576;
 612   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
 613   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 614   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 615
 616   for (x= 0, errors= 0, total= 0 ; total < 20 * 1024576 ; x++)
 617 #endif
 618
 619   for (x= 0, errors= 0, total= 0 ; total < 24576 ; x++)
 620   {
 621     memcached_return rc= MEMCACHED_SUCCESS;
 622     char buffer[SMALL_STRING_LEN];
 623     uint16_t flags= 0;
 624     size_t val_len= 0;
 625     char *getval;
 626
 627     memset(buffer, 0, SMALL_STRING_LEN);
 628
 629     snprintf(buffer, SMALL_STRING_LEN, "%u", x);
 630     getval= memcached_get(memc, buffer, strlen(buffer),
 631                            &val_len, &flags, &rc);
 632     if (rc != MEMCACHED_SUCCESS)
 633     {
 634       if (rc == MEMCACHED_NOTFOUND)
 635         errors++;
 636       else
 637         assert(0);
 638
 639       continue;
 640     }
 641     total+= val_len;
 642     errors= 0;
 643     free(getval);
 644   }
 645 }
 646
 647 /* Do a large mget() over all the keys we think exist */
 648 #define KEY_COUNT 3000 // * 1024576
 649 void user_supplied_bug3(memcached_st *memc)
 650 {
 651   memcached_return rc;
 652   unsigned int setter;
 653   unsigned int x;
 654   char **keys;
 655   size_t key_lengths[KEY_COUNT];
 656
 657   setter= 1;
 658   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, &setter);
 659   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, &setter);
 660 #ifdef NOT_YET
 661   setter = 20 * 1024576;
 662   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE, &setter);
 663   setter = 20 * 1024576;
 664   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE, &setter);
 665   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE);
 666   getter = memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE);
 667 #endif
 668
 669   keys= (char **)malloc(sizeof(char *) * KEY_COUNT);
 670   assert(keys);
 671   memset(keys, 0, (sizeof(char *) * KEY_COUNT));
 672   for (x= 0; x < KEY_COUNT; x++)
 673   {
 674     char buffer[30];
 675
 676     snprintf(buffer, 30, "%u", x);
 677     keys[x]= strdup(buffer);
 678     key_lengths[x]= strlen(keys[x]);
 679   }
 680
 681   rc= memcached_mget(memc, keys, key_lengths, KEY_COUNT);
 682   assert(rc == MEMCACHED_SUCCESS);
 683
 684   /* Turn this into a help function */
 685   {
 686     char return_key[MEMCACHED_MAX_KEY];
 687     size_t return_key_length;
 688     char *return_value;
 689     size_t return_value_length;
 690     uint16_t flags;
 691
 692     while ((return_value= memcached_fetch(memc, return_key, &return_key_length,
 693                                           &return_value_length, &flags, &rc)))
 694     {
 695       assert(return_value);
 696       assert(rc == MEMCACHED_SUCCESS);
 697       free(return_value);
 698     }
 699   }
 700
 701   for (x= 0; x < KEY_COUNT; x++)
 702     free(keys[x]);
 703   free(keys);
 704 }
 705
 706 void string_alloc_null(memcached_st *memc)
 707 {
 708   memcached_string_st *string;
 709
 710   string= memcached_string_create(memc, 0);
 711   assert(string);
 712   memcached_string_free(memc, string);
 713 }
 714
 715 void string_alloc_with_size(memcached_st *memc)
 716 {
 717   memcached_string_st *string;
 718
 719   string= memcached_string_create(memc, 1024);
 720   assert(string);
 721   memcached_string_free(memc, string);
 722 }
 723
 724 void string_alloc_with_size_toobig(memcached_st *memc)
 725 {
 726   memcached_string_st *string;
 727
 728   string= memcached_string_create(memc, INT64_MAX);
 729   assert(string == NULL);
 730 }
 731
 732 void string_alloc_append(memcached_st *memc)
 733 {
 734   unsigned int x;
 735   char buffer[SMALL_STRING_LEN];
 736   memcached_string_st *string;
 737
 738   /* Ring the bell! */
 739   memset(buffer, 6, SMALL_STRING_LEN);
 740
 741   string= memcached_string_create(memc, 100);
 742   assert(string);
 743
 744   for (x= 0; x < 1024; x++)
 745   {
 746     memcached_return rc;
 747     rc= memcached_string_append(memc, string, buffer, SMALL_STRING_LEN);
 748     assert(rc == MEMCACHED_SUCCESS);
 749   }
 750   memcached_string_free(memc, string);
 751 }
 752
 753 void string_alloc_append_toobig(memcached_st *memc)
 754 {
 755   memcached_return rc;
 756   unsigned int x;
 757   char buffer[SMALL_STRING_LEN];
 758   memcached_string_st *string;
 759
 760   /* Ring the bell! */
 761   memset(buffer, 6, SMALL_STRING_LEN);
 762
 763   string= memcached_string_create(memc, 100);
 764   assert(string);
 765
 766   for (x= 0; x < 1024; x++)
 767   {
 768     rc= memcached_string_append(memc, string, buffer, SMALL_STRING_LEN);
 769     assert(rc == MEMCACHED_SUCCESS);
 770   }
 771   rc= memcached_string_append(memc, string, buffer, INT64_MAX);
 772   assert(rc == MEMCACHED_MEMORY_ALLOCATION_FAILURE);
 773   memcached_string_free(memc, string);
 774 }
 775
 776 void add_host_test1(memcached_st *memc)
 777 {
 778   unsigned int x;
 779   memcached_return rc;
 780   char servername[]= "0.example.com";
 781   memcached_server_st *servers;
 782
 783   servers= memcached_server_list_append(NULL, servername, 400, &rc);
 784   assert(servers);
 785   assert(1 == memcached_server_list_count(servers));
 786
 787   for (x= 2; x < 20; x++)
 788   {
 789     char buffer[SMALL_STRING_LEN];
 790
 791     snprintf(buffer, SMALL_STRING_LEN, "%u.example.com", 400+x);
 792     servers= memcached_server_list_append(servers, buffer, 401,
 793                                      &rc);
 794     assert(rc == MEMCACHED_SUCCESS);
 795     assert(x == memcached_server_list_count(servers));
 796   }
 797
 798   rc= memcached_server_push(memc, servers);
 799   assert(rc == MEMCACHED_SUCCESS);
 800   rc= memcached_server_push(memc, servers);
 801   assert(rc == MEMCACHED_SUCCESS);
 802
 803   memcached_server_list_free(servers);
 804 }
 805
 806 memcached_return pre_nonblock(memcached_st *memc)
 807 {
 808   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
 809
 810   return MEMCACHED_SUCCESS;
 811 }
 812
 813 memcached_return pre_md5(memcached_st *memc)
 814 {
 815   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_MD5_HASHING, NULL);
 816
 817   return MEMCACHED_SUCCESS;
 818 }
 819
 820 memcached_return pre_crc(memcached_st *memc)
 821 {
 822   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_CRC_HASHING, NULL);
 823
 824   return MEMCACHED_SUCCESS;
 825 }
 826
 827 memcached_return pre_unix_socket(memcached_st *memc)
 828 {
 829   memcached_return rc;
 830   struct stat buf;
 831
 832   memcached_server_list_free(memc->hosts);
 833   memc->hosts= NULL;
 834   memc->number_of_hosts= 0;
 835
 836   if (stat("/tmp/memcached.socket", &buf))
 837     return MEMCACHED_FAILURE;
 838
 839   rc= memcached_server_add_unix_socket(memc, "/tmp/memcached.socket");
 840
 841   return rc;
 842 }
 843
 844 memcached_return pre_nodelay(memcached_st *memc)
 845 {
 846   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, NULL);
 847   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_TCP_NODELAY, NULL);
 848
 849   return MEMCACHED_SUCCESS;
 850 }
 851
 852 typedef struct collection_st collection_st;
 853 typedef struct test_st test_st;
 854
 855 struct test_st {
 856   char *name;
 857   unsigned int requires_flush;
 858   void (*function)(memcached_st *memc);
 859 };
 860
 861 struct collection_st {
 862   char *name;
 863   memcached_return (*pre)(memcached_st *memc);
 864   memcached_return (*post)(memcached_st *memc);
 865   test_st *tests;
 866 };
 867
 868 int main(int argc, char *argv[])
 869 {
 870   unsigned int x;
 871   char *server_list;
 872   char *collection_to_run= NULL;
 873   char *wildcard= NULL;
 874   memcached_server_st *servers;
 875
 876
 877   if (argc > 1)
 878     collection_to_run= argv[1];
 879
 880   if (argc == 3)
 881     wildcard= argv[2];
 882
 883   if (!(server_list= getenv("MEMCACHED_SERVERS")))
 884     server_list= "localhost";
 885
 886   printf("servers %s\n", server_list);
 887   srandom(time(NULL));
 888
 889   servers= memcached_servers_parse(server_list);
 890   assert(servers);
 891
 892   for (x= 0; x < memcached_server_list_count(servers); x++)
 893   {
 894     printf("\t%s : %u\n", servers[x].hostname, servers[x].port);
 895     assert(servers[x].stack_responses == 0);
 896     assert(servers[x].fd == -1);
 897     assert(servers[x].cursor_active == 0);
 898   }
 899
 900   printf("\n");
 901
 902   /* Clean the server before beginning testing */
 903   test_st tests[] ={
 904     {"flush", 0, flush_test },
 905     {"init", 0, init_test },
 906     {"allocation", 0, allocation_test },
 907     {"clone_test", 0, clone_test },
 908     {"error", 0, error_test },
 909     {"set", 0, set_test },
 910     {"set2", 0, set_test2 },
 911     {"set3", 0, set_test3 },
 912     {"add", 0, add_test },
 913     {"replace", 0, replace_test },
 914     {"delete", 1, delete_test },
 915     {"get", 1, get_test },
 916     {"get2", 0, get_test2 },
 917     {"get3", 0, get_test3 },
 918     {"get4", 0, get_test4 },
 919     {"stats_servername", 0, stats_servername_test },
 920     {"increment", 0, increment_test },
 921     {"decrement", 0, decrement_test },
 922     {"quit", 0, quit_test },
 923     {"mget", 0, mget_test },
 924     {"get_stats", 0, get_stats },
 925     {"add_host_test", 0, add_host_test },
 926     {"get_stats_keys", 0, get_stats_keys },
 927     {"behavior_test", 0, get_stats_keys },
 928     {0, 0, 0}
 929   };
 930
 931   test_st string_tests[] ={
 932     {"string alloc with null", 0, string_alloc_null },
 933     {"string alloc with 1K", 0, string_alloc_with_size },
 934     {"string alloc with malloc failure", 0, string_alloc_with_size_toobig },
 935     {"string append", 0, string_alloc_append },
 936     {"string append failure (too big)", 0, string_alloc_append_toobig },
 937     {0, 0, 0}
 938   };
 939
 940   test_st user_tests[] ={
 941     {"user_supplied_bug1", 0, user_supplied_bug1 },
 942     {"user_supplied_bug2", 0, user_supplied_bug2 },
 943     {"user_supplied_bug3", 0, user_supplied_bug3 },
 944     {0, 0, 0}
 945   };
 946
 947
 948   collection_st collection[] ={
 949     {"block", 0, 0, tests},
 950     {"nonblock", pre_nonblock, 0, tests},
 951     {"nodelay", pre_nodelay, 0, tests},
 952     {"md5", pre_md5, 0, tests},
 953     {"crc", pre_crc, 0, tests},
 954     {"unix_socket", pre_unix_socket, 0, tests},
 955     {"string", 0, 0, string_tests},
 956     {"user", 0, 0, user_tests},
 957     {0, 0, 0, 0}
 958   };
 959
 960   collection_st *next;
 961   for (next= collection; next->name; next++)
 962   {
 963     test_st *run;
 964
 965     run= next->tests;
 966
 967     if (collection_to_run && strcmp(collection_to_run, next->name))
 968       continue;
 969
 970     fprintf(stderr, "\n%s\n\n", next->name);
 971
 972     for (x= 0; run->name; run++)
 973     {
 974       if (wildcard && strcmp(wildcard, run->name))
 975         continue;
 976
 977       fprintf(stderr, "Testing %s", run->name);
 978
 979       memcached_st *memc;
 980       memcached_return rc;
 981       struct timeval start_time, end_time;
 982
 983       memc= memcached_create(NULL);
 984       assert(memc);
 985
 986       if (run->requires_flush)
 987         memcached_flush(memc, 0);
 988
 989       rc= memcached_server_push(memc, servers);
 990       assert(rc == MEMCACHED_SUCCESS);
 991
 992       unsigned int loop;
 993       for (loop= 0; loop < memcached_server_list_count(servers); loop++)
 994       {
 995         assert(memc->hosts[loop].stack_responses == 0);
 996         assert(memc->hosts[loop].fd == -1);
 997         assert(memc->hosts[loop].cursor_active == 0);
 998       }
 999
1000       if (next->pre)
1001       {
1002         memcached_return rc;
1003         rc= next->pre(memc);
1004
1005         if (rc != MEMCACHED_SUCCESS)
1006         {
1007           fprintf(stderr, "\t\t\t\t\t [ skipping ]\n");
1008           goto error;
1009         }
1010       }
1011
1012       gettimeofday(&start_time, NULL);
1013       run->function(memc);
1014       gettimeofday(&end_time, NULL);
1015       long int load_time= timedif(end_time, start_time);
1016       fprintf(stderr, "\t\t\t\t\t %ld.%03ld [ ok ]\n", load_time / 1000,
1017               load_time % 1000);
1018
1019       if (next->post)
1020         (void)next->post(memc);
1021
1022       assert(memc);
1023 error:
1024       memcached_free(memc);
1025     }
1026   }
1027
1028   /* Clean up whatever we might have left */
1029   {
1030     memcached_st *memc;
1031     memc= memcached_create(NULL);
1032     assert(memc);
1033     flush_test(memc);
1034     memcached_free(memc);
1035   }
1036
1037   fprintf(stderr, "All tests completed successfully\n\n");
1038
1039   memcached_server_list_free(servers);
1040
1041   return 0;
1042 }