git.m6w6.name Git - awesomized/libmemcached/blob - tests/plus.cpp

   1 /*
   2   C++ interface test
   3 */
   4 #include "libmemcached/memcached.hpp"
   5
   6 #include <assert.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10 #include <sys/time.h>
  11 #include <sys/types.h>
  12 #include <sys/stat.h>
  13 #include <unistd.h>
  14 #include <time.h>
  15 #include "server.h"
  16
  17 #include "test.h"
  18
  19 #include <string>
  20 #include <iostream>
  21
  22 using namespace std;
  23 using namespace memcache;
  24
  25 extern "C" {
  26    test_return_t basic_test(memcached_st *memc);
  27    test_return_t increment_test(memcached_st *memc);
  28    test_return_t basic_master_key_test(memcached_st *memc);
  29    test_return_t mget_result_function(memcached_st *memc);
  30    test_return_t basic_behavior(memcached_st *memc);
  31    test_return_t mget_test(memcached_st *memc);
  32    memcached_return_t callback_counter(memcached_st *,
  33                                        memcached_result_st *,
  34                                        void *context);
  35 }
  36
  37 static void populate_vector(vector<char> &vec, const string &str)
  38 {
  39   vec.reserve(str.length());
  40   vec.assign(str.begin(), str.end());
  41 }
  42
  43 static void copy_vec_to_string(vector<char> &vec, string &str)
  44 {
  45   str.clear();
  46   if (! vec.empty())
  47   {
  48     str.assign(vec.begin(), vec.end());
  49   }
  50 }
  51
  52 test_return_t basic_test(memcached_st *memc)
  53 {
  54   Memcache foo(memc);
  55   const string value_set("This is some data");
  56   std::vector<char> value;
  57   std::vector<char> test_value;
  58
  59   populate_vector(value, value_set);
  60
  61   foo.set("mine", value, 0, 0);
  62   foo.get("mine", test_value);
  63
  64   assert((memcmp(&test_value[0], &value[0], test_value.size()) == 0));
  65
  66   /*
  67    * Simple test of the exceptions here...this should throw an exception
  68    * saying that the key is empty.
  69    */
  70   try
  71   {
  72     foo.set("", value, 0, 0);
  73   }
  74   catch (Error &err)
  75   {
  76     return TEST_SUCCESS;
  77   }
  78
  79   return TEST_FAILURE;
  80 }
  81
  82 test_return_t increment_test(memcached_st *memc)
  83 {
  84   Memcache mcach(memc);
  85   bool rc;
  86   const string key("blah");
  87   const string inc_value("1");
  88   std::vector<char> inc_val;
  89   vector<char> ret_value;
  90   string ret_string;
  91   uint64_t int_inc_value;
  92   uint64_t int_ret_value;
  93
  94   populate_vector(inc_val, inc_value);
  95
  96   rc= mcach.set(key, inc_val, 0, 0);
  97   if (rc == false)
  98   {
  99     return TEST_FAILURE;
 100   }
 101   mcach.get(key, ret_value);
 102   if (ret_value.empty())
 103   {
 104     return TEST_FAILURE;
 105   }
 106   copy_vec_to_string(ret_value, ret_string);
 107
 108   int_inc_value= uint64_t(atol(inc_value.c_str()));
 109   int_ret_value= uint64_t(atol(ret_string.c_str()));
 110   assert(int_ret_value == int_inc_value);
 111
 112   rc= mcach.increment(key, 1, &int_ret_value);
 113   assert(rc == true);
 114   assert(int_ret_value == 2);
 115
 116   rc= mcach.increment(key, 1, &int_ret_value);
 117   assert(rc == true);
 118   assert(int_ret_value == 3);
 119
 120   rc= mcach.increment(key, 5, &int_ret_value);
 121   assert(rc == true);
 122   assert(int_ret_value == 8);
 123
 124   return TEST_SUCCESS;
 125 }
 126
 127 test_return_t basic_master_key_test(memcached_st *memc)
 128 {
 129   Memcache foo(memc);
 130   const string value_set("Data for server A");
 131   vector<char> value;
 132   vector<char> test_value;
 133   const string master_key_a("server-a");
 134   const string master_key_b("server-b");
 135   const string key("xyz");
 136
 137   populate_vector(value, value_set);
 138
 139   foo.setByKey(master_key_a, key, value, 0, 0);
 140   foo.getByKey(master_key_a, key, test_value);
 141
 142   assert((memcmp(&value[0], &test_value[0], value.size()) == 0));
 143
 144   test_value.clear();
 145
 146   foo.getByKey(master_key_b, key, test_value);
 147   assert((memcmp(&value[0], &test_value[0], value.size()) == 0));
 148
 149   return TEST_SUCCESS;
 150 }
 151
 152 /* Count the results */
 153 memcached_return_t callback_counter(memcached_st *,
 154                                   memcached_result_st *,
 155                                   void *context)
 156 {
 157   unsigned int *counter= static_cast<unsigned int *>(context);
 158
 159   *counter= *counter + 1;
 160
 161   return MEMCACHED_SUCCESS;
 162 }
 163
 164 test_return_t mget_result_function(memcached_st *memc)
 165 {
 166   Memcache mc(memc);
 167   bool rc;
 168   string key1("fudge");
 169   string key2("son");
 170   string key3("food");
 171   vector<string> keys;
 172   vector< vector<char> *> values;
 173   vector<char> val1;
 174   vector<char> val2;
 175   vector<char> val3;
 176   populate_vector(val1, key1);
 177   populate_vector(val2, key2);
 178   populate_vector(val3, key3);
 179   keys.reserve(3);
 180   keys.push_back(key1);
 181   keys.push_back(key2);
 182   keys.push_back(key3);
 183   values.reserve(3);
 184   values.push_back(&val1);
 185   values.push_back(&val2);
 186   values.push_back(&val3);
 187   unsigned int counter;
 188   memcached_execute_fn callbacks[1];
 189
 190   /* We need to empty the server before we continue the test */
 191   rc= mc.flush(0);
 192   rc= mc.setAll(keys, values, 50, 9);
 193   assert(rc == true);
 194
 195   rc= mc.mget(keys);
 196   assert(rc == true);
 197
 198   callbacks[0]= &callback_counter;
 199   counter= 0;
 200   rc= mc.fetchExecute(callbacks, static_cast<void *>(&counter), 1);
 201
 202   assert(counter == 3);
 203
 204   return TEST_SUCCESS;
 205 }
 206
 207 test_return_t mget_test(memcached_st *memc)
 208 {
 209   Memcache mc(memc);
 210   bool rc;
 211   memcached_return_t mc_rc;
 212   vector<string> keys;
 213   vector< vector<char> *> values;
 214   keys.reserve(3);
 215   keys.push_back("fudge");
 216   keys.push_back("son");
 217   keys.push_back("food");
 218   vector<char> val1;
 219   vector<char> val2;
 220   vector<char> val3;
 221   populate_vector(val1, "fudge");
 222   populate_vector(val2, "son");
 223   populate_vector(val3, "food");
 224   values.reserve(3);
 225   values.push_back(&val1);
 226   values.push_back(&val2);
 227   values.push_back(&val3);
 228
 229   string return_key;
 230   vector<char> return_value;
 231
 232   /* We need to empty the server before we continue the test */
 233   rc= mc.flush(0);
 234   assert(rc == true);
 235
 236   rc= mc.mget(keys);
 237   assert(rc == true);
 238
 239   while ((mc_rc= mc.fetch(return_key, return_value)) != MEMCACHED_END)
 240   {
 241     assert(return_value.size() != 0);
 242     return_value.clear();
 243   }
 244   assert(mc_rc == MEMCACHED_END);
 245
 246   rc= mc.setAll(keys, values, 50, 9);
 247   assert(rc == true);
 248
 249   rc= mc.mget(keys);
 250   assert(rc == true);
 251
 252   while ((mc_rc= mc.fetch(return_key, return_value)) != MEMCACHED_END)
 253   {
 254     assert(return_key.length() == return_value.size());
 255     assert(!memcmp(&return_value[0], return_key.c_str(), return_value.size()));
 256   }
 257
 258   return TEST_SUCCESS;
 259 }
 260
 261 test_return_t basic_behavior(memcached_st *memc)
 262 {
 263   Memcache mc(memc);
 264   bool rc;
 265   uint64_t value = 1;
 266   rc = mc.setBehavior(MEMCACHED_BEHAVIOR_VERIFY_KEY, value);
 267   assert(rc);
 268   uint64_t behavior = mc.getBehavior(MEMCACHED_BEHAVIOR_VERIFY_KEY);
 269   assert(behavior == value);
 270
 271   return TEST_SUCCESS;
 272 }
 273
 274 test_st tests[] ={
 275   { "basic", 0,
 276     reinterpret_cast<test_callback_fn>(basic_test) },
 277   { "basic_master_key", 0,
 278     reinterpret_cast<test_callback_fn>(basic_master_key_test) },
 279   { "increment_test", 0,
 280     reinterpret_cast<test_callback_fn>(increment_test) },
 281   { "mget", 1,
 282     reinterpret_cast<test_callback_fn>(mget_test) },
 283   { "mget_result_function", 1,
 284     reinterpret_cast<test_callback_fn>(mget_result_function) },
 285   { "basic_behavior", 0,
 286     reinterpret_cast<test_callback_fn>(basic_behavior) },
 287   {0, 0, 0}
 288 };
 289
 290 collection_st collection[] ={
 291   {"block", 0, 0, tests},
 292   {0, 0, 0, 0}
 293 };
 294
 295 #define SERVERS_TO_CREATE 5
 296
 297 #include "libmemcached_world.h"
 298
 299 void get_world(world_st *world)
 300 {
 301   world->collections= collection;
 302   world->collection_startup= reinterpret_cast<test_callback_fn>(world_collection_startup);
 303   world->flush= reinterpret_cast<test_callback_fn>(world_flush);
 304   world->pre_run= reinterpret_cast<test_callback_fn>(world_pre_run);
 305   world->create= reinterpret_cast<test_callback_create_fn>(world_create);
 306   world->post_run= reinterpret_cast<test_callback_fn>(world_post_run);
 307   world->on_error= reinterpret_cast<test_callback_error_fn>(world_on_error);
 308   world->destroy= reinterpret_cast<test_callback_fn>(world_destroy);
 309   world->runner= &defualt_libmemcached_runner;
 310 }