git.m6w6.name Git - m6w6/libmemcached/blob - libtest/unittest.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  libtest
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *
   7  *  This library is free software; you can redistribute it and/or
   8  *  modify it under the terms of the GNU Lesser General Public
   9  *  License as published by the Free Software Foundation; either
  10  *  version 3 of the License, or (at your option) any later version.
  11  *
  12  *  This library is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  *  Lesser General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Lesser General Public
  18  *  License along with this library; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22
  23 #include <config.h>
  24
  25 #include <libtest/test.hpp>
  26
  27 #if defined(LIBTEST_WITH_LIBMEMCACHED_SUPPORT) && LIBTEST_WITH_LIBMEMCACHED_SUPPORT
  28 #include <libmemcached/memcached.h>
  29 #endif
  30
  31 #if defined(LIBTEST_WITH_LIBGEARMAN_SUPPORT) && LIBTEST_WITH_LIBGEARMAN_SUPPORT
  32 #include <libgearman/gearman.h>
  33 #endif
  34
  35 #include <cstdlib>
  36 #include <unistd.h>
  37
  38 using namespace libtest;
  39
  40 static test_return_t LIBTOOL_COMMAND_test(void *)
  41 {
  42   test_true(getenv("LIBTOOL_COMMAND"));
  43   return TEST_SUCCESS;
  44 }
  45
  46 static test_return_t VALGRIND_COMMAND_test(void *)
  47 {
  48   test_true(getenv("VALGRIND_COMMAND"));
  49   return TEST_SUCCESS;
  50 }
  51
  52 static test_return_t HELGRIND_COMMAND_test(void *)
  53 {
  54   test_true(getenv("HELGRIND_COMMAND"));
  55   return TEST_SUCCESS;
  56 }
  57
  58 static test_return_t GDB_COMMAND_test(void *)
  59 {
  60   test_true(getenv("GDB_COMMAND"));
  61   return TEST_SUCCESS;
  62 }
  63
  64 static test_return_t test_success_equals_one_test(void *)
  65 {
  66   test_skip(HAVE_LIBMEMCACHED, true);
  67 #if defined(HAVE_LIBMEMCACHED) && HAVE_LIBMEMCACHED
  68   test_zero(MEMCACHED_SUCCESS);
  69 #endif
  70   return TEST_SUCCESS;
  71 }
  72
  73 static test_return_t test_success_test(void *)
  74 {
  75   return TEST_SUCCESS;
  76 }
  77
  78 static test_return_t test_failure_test(void *)
  79 {
  80   return TEST_SKIPPED; // Only run this when debugging
  81
  82   test_compare(1, 2);
  83   return TEST_SUCCESS;
  84 }
  85
  86 static test_return_t local_test(void *)
  87 {
  88   if (getenv("LIBTEST_LOCAL"))
  89   {
  90     test_true(test_is_local());
  91   }
  92   else
  93   {
  94     test_false(test_is_local());
  95   }
  96
  97   return TEST_SUCCESS;
  98 }
  99
 100 static test_return_t local_not_test(void *)
 101 {
 102   return TEST_SKIPPED;
 103
 104   std::string temp;
 105
 106   const char *ptr;
 107   if ((ptr= getenv("LIBTEST_LOCAL")) == NULL)
 108   {
 109     temp.append(ptr);
 110   }
 111
 112   // unsetenv() will cause issues with valgrind
 113   _compare(__FILE__, __LINE__, __func__, 0, unsetenv("LIBTEST_LOCAL"));
 114   test_compare(0, unsetenv("LIBTEST_LOCAL"));
 115   test_false(test_is_local());
 116
 117   test_compare(0, setenv("LIBTEST_LOCAL", "1", 1));
 118   test_true(test_is_local());
 119
 120   if (temp.empty())
 121   {
 122     test_compare(0, unsetenv("LIBTEST_LOCAL"));
 123   }
 124   else
 125   {
 126     char *old_string= strdup(temp.c_str());
 127     test_compare(0, setenv("LIBTEST_LOCAL", old_string, 1));
 128   }
 129
 130   return TEST_SUCCESS;
 131 }
 132
 133 static test_return_t var_exists_test(void *)
 134 {
 135   test_compare(0, access("var", R_OK | W_OK | X_OK));
 136   return TEST_SUCCESS;
 137 }
 138
 139 static test_return_t var_tmp_exists_test(void *)
 140 {
 141   test_compare(0, access("var/tmp", R_OK | W_OK | X_OK));
 142   return TEST_SUCCESS;
 143 }
 144
 145 static test_return_t var_run_exists_test(void *)
 146 {
 147   test_compare(0, access("var/run", R_OK | W_OK | X_OK));
 148   return TEST_SUCCESS;
 149 }
 150
 151 static test_return_t var_log_exists_test(void *)
 152 {
 153   test_compare(0, access("var/log", R_OK | W_OK | X_OK));
 154   return TEST_SUCCESS;
 155 }
 156
 157 static test_return_t var_tmp_test(void *)
 158 {
 159   FILE *file= fopen("var/tmp/junk", "w+");
 160   char buffer[1024];
 161   const char *dir= getcwd(buffer, sizeof(buffer));
 162   test_true_got(file, dir);
 163   fclose(file);
 164   return TEST_SUCCESS;
 165 }
 166
 167 static test_return_t var_run_test(void *)
 168 {
 169   FILE *file= fopen("var/run/junk", "w+");
 170   test_true(file);
 171   fclose(file);
 172   return TEST_SUCCESS;
 173 }
 174
 175 static test_return_t var_log_test(void *)
 176 {
 177   FILE *file= fopen("var/log/junk", "w+");
 178   test_true(file);
 179   fclose(file);
 180   return TEST_SUCCESS;
 181 }
 182
 183 static test_return_t var_tmp_rm_test(void *)
 184 {
 185   test_true(unlink("var/tmp/junk") == 0);
 186   return TEST_SUCCESS;
 187 }
 188
 189 static test_return_t var_run_rm_test(void *)
 190 {
 191   test_true(unlink("var/run/junk") == 0);
 192   return TEST_SUCCESS;
 193 }
 194
 195 static test_return_t var_log_rm_test(void *)
 196 {
 197   test_true(unlink("var/log/junk") == 0);
 198   return TEST_SUCCESS;
 199 }
 200
 201 static test_return_t _compare_test_return_t_test(void *)
 202 {
 203   test_compare(TEST_SUCCESS, TEST_SUCCESS);
 204
 205   return TEST_SUCCESS;
 206 }
 207
 208 static test_return_t _compare_memcached_return_t_test(void *)
 209 {
 210   test_skip(HAVE_LIBMEMCACHED, true);
 211 #if defined(HAVE_LIBMEMCACHED) && HAVE_LIBMEMCACHED
 212   test_compare(MEMCACHED_SUCCESS, MEMCACHED_SUCCESS);
 213 #endif
 214
 215   return TEST_SUCCESS;
 216 }
 217
 218 static test_return_t _compare_gearman_return_t_test(void *)
 219 {
 220   test_skip(HAVE_LIBGEARMAN, true);
 221 #if defined(HAVE_LIBGEARMAN) && HAVE_LIBGEARMAN
 222   test_compare(GEARMAN_SUCCESS, GEARMAN_SUCCESS);
 223 #endif
 224
 225   return TEST_SUCCESS;
 226 }
 227
 228 static test_return_t gearmand_cycle_test(void *object)
 229 {
 230   server_startup_st *servers= (server_startup_st*)object;
 231   test_true(servers);
 232
 233   if (GEARMAND_BINARY)
 234   {
 235     if (HAVE_LIBGEARMAN)
 236     {
 237       test_true(has_gearmand_binary());
 238       const char *argv[1]= { "cycle_gearmand" };
 239       test_true(server_startup(*servers, "gearmand", 9999, 1, argv));
 240
 241       return TEST_SUCCESS;
 242     }
 243   }
 244
 245   return TEST_SKIPPED;
 246 }
 247
 248 static test_return_t memcached_cycle_test(void *object)
 249 {
 250   server_startup_st *servers= (server_startup_st*)object;
 251   test_true(servers);
 252
 253   if (MEMCACHED_BINARY and HAVE_LIBMEMCACHED)
 254   {
 255     test_true(has_memcached_binary());
 256     const char *argv[1]= { "cycle_memcached" };
 257     test_true(server_startup(*servers, "memcached", 9998, 1, argv));
 258
 259     return TEST_SUCCESS;
 260   }
 261
 262   return TEST_SKIPPED;
 263 }
 264
 265 static test_return_t memcached_socket_cycle_test(void *object)
 266 {
 267   server_startup_st *servers= (server_startup_st*)object;
 268   test_true(servers);
 269
 270   if (MEMCACHED_BINARY)
 271   {
 272     if (HAVE_LIBMEMCACHED)
 273     {
 274       test_true(has_memcached_binary());
 275       const char *argv[1]= { "cycle_memcached" };
 276       test_true(servers->start_socket_server("memcached", 9997, 1, argv));
 277
 278       return TEST_SUCCESS;
 279     }
 280   }
 281
 282   return TEST_SKIPPED;
 283 }
 284
 285 static test_return_t memcached_sasl_test(void *object)
 286 {
 287   server_startup_st *servers= (server_startup_st*)object;
 288   test_true(servers);
 289
 290   if (getenv("TESTS_ENVIRONMENT"))
 291   {
 292     return TEST_SKIPPED;
 293   }
 294
 295   if (MEMCACHED_SASL_BINARY)
 296   {
 297     if (HAVE_LIBMEMCACHED)
 298     {
 299       test_true(has_memcached_sasl_binary());
 300       const char *argv[1]= { "cycle_memcached_sasl" };
 301       test_true(server_startup(*servers, "memcached-sasl", 9996, 1, argv));
 302
 303       return TEST_SUCCESS;
 304     }
 305   }
 306
 307   return TEST_SKIPPED;
 308 }
 309
 310 static test_return_t wait_BINARY(void *)
 311 {
 312   const char *args[]= { "--quiet", 0 };
 313
 314   test_true(exec_cmdline("libtest/wait", args));
 315
 316   return TEST_SUCCESS;
 317 }
 318
 319 static test_return_t wait_help_BINARY(void *)
 320 {
 321   const char *args[]= { "--quiet", "--help", 0 };
 322
 323   test_true(exec_cmdline("libtest/wait", args));
 324
 325   return TEST_SUCCESS;
 326 }
 327
 328 static test_return_t wait_version_BINARY(void *)
 329 {
 330   const char *args[]= { "--quiet", "--version", 0 };
 331
 332   test_true(exec_cmdline("libtest/wait", args));
 333
 334   return TEST_SUCCESS;
 335 }
 336
 337 test_st gearmand_tests[] ={
 338 #if 0
 339   {"pause", 0, pause_test },
 340 #endif
 341   {"gearmand startup-shutdown", 0, gearmand_cycle_test },
 342   {0, 0, 0}
 343 };
 344
 345 test_st memcached_tests[] ={
 346   {"memcached startup-shutdown", 0, memcached_cycle_test },
 347   {"memcached(socket file) startup-shutdown", 0, memcached_socket_cycle_test },
 348   {"memcached_sasl() startup-shutdown", 0, memcached_sasl_test },
 349   {0, 0, 0}
 350 };
 351
 352 test_st environment_tests[] ={
 353   {"LIBTOOL_COMMAND", 0, LIBTOOL_COMMAND_test },
 354   {"VALGRIND_COMMAND", 0, VALGRIND_COMMAND_test },
 355   {"HELGRIND_COMMAND", 0, HELGRIND_COMMAND_test },
 356   {"GDB_COMMAND", 0, GDB_COMMAND_test },
 357   {0, 0, 0}
 358 };
 359
 360 test_st tests_log[] ={
 361   {"TEST_SUCCESS", false, test_success_test },
 362   {"TEST_FAILURE", false, test_failure_test },
 363   {"TEST_SUCCESS == 0", false, test_success_equals_one_test },
 364   {0, 0, 0}
 365 };
 366
 367 test_st local_log[] ={
 368   {"test_is_local()", 0, local_test },
 369   {"test_is_local(NOT)", 0, local_not_test },
 370   {0, 0, 0}
 371 };
 372
 373 test_st directories_tests[] ={
 374   {"var exists", 0, var_exists_test },
 375   {"var/tmp exists", 0, var_tmp_exists_test },
 376   {"var/run exists", 0, var_run_exists_test },
 377   {"var/log exists", 0, var_log_exists_test },
 378   {"var/tmp", 0, var_tmp_test },
 379   {"var/run", 0, var_run_test },
 380   {"var/log", 0, var_log_test },
 381   {"var/tmp rm", 0, var_tmp_rm_test },
 382   {"var/run rm", 0, var_run_rm_test },
 383   {"var/log rm", 0, var_log_rm_test },
 384   {0, 0, 0}
 385 };
 386
 387 test_st comparison_tests[] ={
 388   {"_compare(test_return_t)", 0, _compare_test_return_t_test },
 389   {"_compare(memcached_return_t)", 0, _compare_memcached_return_t_test },
 390   {"_compare(gearman_return_t)", 0, _compare_gearman_return_t_test },
 391   {0, 0, 0}
 392 };
 393
 394 test_st cmdline_tests[] ={
 395   {"wait --quiet", 0, wait_BINARY },
 396   {"wait --quiet --help", 0, wait_help_BINARY },
 397   {"wait --quiet --version", 0, wait_version_BINARY },
 398   {0, 0, 0}
 399 };
 400
 401 collection_st collection[] ={
 402   {"environment", 0, 0, environment_tests},
 403   {"return values", 0, 0, tests_log},
 404   {"local", 0, 0, local_log},
 405   {"directories", 0, 0, directories_tests},
 406   {"comparison", 0, 0, comparison_tests},
 407   {"gearmand", 0, 0, gearmand_tests},
 408   {"memcached", 0, 0, memcached_tests},
 409   {"cmdline", 0, 0, cmdline_tests},
 410   {0, 0, 0, 0}
 411 };
 412
 413 static void *world_create(server_startup_st& servers, test_return_t&)
 414 {
 415   return &servers;
 416 }
 417
 418 void get_world(Framework *world)
 419 {
 420   world->collections= collection;
 421   world->_create= world_create;
 422 }