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

   1 /* LibMemcached
   2  * Copyright (C) 2006-2009 Brian Aker
   3  * All rights reserved.
   4  *
   5  * Use and distribution licensed under the BSD license.  See
   6  * the COPYING file in the parent directory for full text.
   7  *
   8  * Summary:
   9  *
  10  */
  11
  12 /*
  13   Sample test application.
  14 */
  15 #include "libmemcached/common.h"
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include <sys/time.h>
  21 #include <sys/types.h>
  22 #include <sys/stat.h>
  23 #include <unistd.h>
  24 #include <time.h>
  25 #include "server.h"
  26 #include "../clients/generator.h"
  27 #include "../clients/execute.h"
  28
  29 #ifndef INT64_MAX
  30 #define INT64_MAX LONG_MAX
  31 #endif
  32 #ifndef INT32_MAX
  33 #define INT32_MAX INT_MAX
  34 #endif
  35
  36
  37 #include "test.h"
  38
  39 /* Number of items generated for tests */
  40 #define GLOBAL_COUNT 100000
  41
  42 /* Number of times to run the test loop */
  43 #define TEST_COUNTER 500000
  44 static uint32_t global_count;
  45
  46 static pairs_st *global_pairs;
  47 static char *global_keys[GLOBAL_COUNT];
  48 static size_t global_keys_length[GLOBAL_COUNT];
  49
  50 static test_return_t cleanup_pairs(memcached_st *memc __attribute__((unused)))
  51 {
  52   pairs_free(global_pairs);
  53
  54   return 0;
  55 }
  56
  57 static test_return_t generate_pairs(memcached_st *memc __attribute__((unused)))
  58 {
  59   global_pairs= pairs_generate(GLOBAL_COUNT, 400);
  60   global_count= GLOBAL_COUNT;
  61
  62   for (size_t x= 0; x < global_count; x++)
  63   {
  64     global_keys[x]= global_pairs[x].key;
  65     global_keys_length[x]=  global_pairs[x].key_length;
  66   }
  67
  68   return 0;
  69 }
  70
  71 static test_return_t drizzle(memcached_st *memc)
  72 {
  73   memcached_return_t rc;
  74   char *return_value;
  75   size_t return_value_length;
  76   uint32_t flags;
  77
  78 infinite:
  79   for (size_t x= 0; x < TEST_COUNTER; x++)
  80   {
  81     uint32_t test_bit;
  82     uint8_t which;
  83
  84     test_bit= (uint32_t)(random() % GLOBAL_COUNT);
  85     which= (uint8_t)(random() % 2);
  86
  87     if (which == 0)
  88     {
  89       return_value= memcached_get(memc, global_keys[test_bit], global_keys_length[test_bit],
  90                                   &return_value_length, &flags, &rc);
  91       if (rc == MEMCACHED_SUCCESS && return_value)
  92       {
  93         free(return_value);
  94       }
  95       else if (rc == MEMCACHED_NOTFOUND)
  96       {
  97         continue;
  98       }
  99       else
 100       {
 101         WATCHPOINT_ERROR(rc);
 102         WATCHPOINT_ASSERT(rc);
 103       }
 104     }
 105     else
 106     {
 107       rc= memcached_set(memc, global_pairs[test_bit].key,
 108                         global_pairs[test_bit].key_length,
 109                         global_pairs[test_bit].value,
 110                         global_pairs[test_bit].value_length,
 111                         0, 0);
 112       if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_BUFFERED)
 113       {
 114         WATCHPOINT_ERROR(rc);
 115         WATCHPOINT_ASSERT(0);
 116       }
 117     }
 118   }
 119
 120   if (getenv("MEMCACHED_ATOM_BURIN_IN"))
 121     goto infinite;
 122
 123   return TEST_SUCCESS;
 124 }
 125
 126 static test_return_t pre_nonblock(memcached_st *memc)
 127 {
 128   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_NO_BLOCK, 0);
 129
 130   return TEST_SUCCESS;
 131 }
 132
 133 /*
 134   Set the value, then quit to make sure it is flushed.
 135   Come back in and test that add fails.
 136 */
 137 static test_return_t add_test(memcached_st *memc)
 138 {
 139   memcached_return_t rc;
 140   const char *key= "foo";
 141   const char *value= "when we sanitize";
 142   unsigned long long setting_value;
 143
 144   setting_value= memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NO_BLOCK);
 145
 146   rc= memcached_set(memc, key, strlen(key),
 147                     value, strlen(value),
 148                     (time_t)0, (uint32_t)0);
 149   test_true(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_BUFFERED);
 150   memcached_quit(memc);
 151   rc= memcached_add(memc, key, strlen(key),
 152                     value, strlen(value),
 153                     (time_t)0, (uint32_t)0);
 154
 155   /* Too many broken OS'es have broken loopback in async, so we can't be sure of the result */
 156   if (setting_value)
 157   {
 158     test_true(rc == MEMCACHED_NOTSTORED || rc == MEMCACHED_STORED);
 159   }
 160   else
 161   {
 162     test_true(rc == MEMCACHED_NOTSTORED);
 163   }
 164
 165   return 0;
 166 }
 167
 168 /*
 169  * repeating add_tests many times
 170  * may show a problem in timing
 171  */
 172 static test_return_t many_adds(memcached_st *memc)
 173 {
 174   for (size_t x= 0; x < TEST_COUNTER; x++)
 175   {
 176     add_test(memc);
 177   }
 178   return 0;
 179 }
 180
 181 test_st smash_tests[] ={
 182   {"generate_pairs", 1, (test_callback_fn)generate_pairs },
 183   {"drizzle", 1, (test_callback_fn)drizzle },
 184   {"cleanup", 1, (test_callback_fn)cleanup_pairs },
 185   {"many_adds", 1, (test_callback_fn)many_adds },
 186   {0, 0, 0}
 187 };
 188
 189
 190 collection_st collection[] ={
 191   {"smash", 0, 0, smash_tests},
 192   {"smash_nonblock", (test_callback_fn)pre_nonblock, 0, smash_tests},
 193   {0, 0, 0, 0}
 194 };
 195
 196
 197 #define SERVERS_TO_CREATE 5
 198
 199 #include "libmemcached_world.h"
 200
 201 void get_world(world_st *world)
 202 {
 203   world->collections= collection;
 204
 205   world->create= (test_callback_create_fn)world_create;
 206   world->destroy= (test_callback_fn)world_destroy;
 207
 208   world->test.startup= (test_callback_fn)world_test_startup;
 209   world->test.flush= (test_callback_fn)world_flush;
 210   world->test.pre_run= (test_callback_fn)world_pre_run;
 211   world->test.post_run= (test_callback_fn)world_post_run;
 212   world->test.on_error= (test_callback_error_fn)world_on_error;
 213
 214   world->collection.startup= (test_callback_fn)world_container_startup;
 215   world->collection.shutdown= (test_callback_fn)world_container_shutdown;
 216
 217   world->runner= &defualt_libmemcached_runner;
 218 }