git.m6w6.name Git - awesomized/libmemcached/blob - tests/libmemcached-1.0/sasl.cc

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached library
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *
   7  *  Redistribution and use in source and binary forms, with or without
   8  *  modification, are permitted provided that the following conditions are
   9  *  met:
  10  *
  11  *      * Redistributions of source code must retain the above copyright
  12  *  notice, this list of conditions and the following disclaimer.
  13  *
  14  *      * Redistributions in binary form must reproduce the above
  15  *  copyright notice, this list of conditions and the following disclaimer
  16  *  in the documentation and/or other materials provided with the
  17  *  distribution.
  18  *
  19  *      * The names of its contributors may not be used to endorse or
  20  *  promote products derived from this software without specific prior
  21  *  written permission.
  22  *
  23  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  25  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  26  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  27  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  28  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  29  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  30  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  31  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  32  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  33  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  34  *
  35  */
  36
  37 #include <config.h>
  38 #include <libtest/test.hpp>
  39
  40 using namespace libtest;
  41
  42 /*
  43   Test cases
  44 */
  45
  46 #define TEST_PORT_BASE MEMCACHED_DEFAULT_PORT +20
  47 #include <libmemcached/memcached.h>
  48
  49 static test_return_t pre_sasl(memcached_st *)
  50 {
  51   if (LIBMEMCACHED_WITH_SASL_SUPPORT == 0)
  52   {
  53     return TEST_SKIPPED;
  54   }
  55
  56   return TEST_SUCCESS;
  57 }
  58
  59 /*
  60  * Test that the sasl authentication works. We cannot use the default
  61  * pool of servers, because that would require that all servers we want
  62  * to test supports SASL authentication, and that they use the default
  63  * creds.
  64  */
  65 static test_return_t sasl_auth_test(memcached_st *memc)
  66 {
  67   if (LIBMEMCACHED_WITH_SASL_SUPPORT)
  68   {
  69     test_compare(MEMCACHED_SUCCESS, memcached_set(memc, "foo", 3, "bar", 3, (time_t)0, (uint32_t)0));
  70     test_compare(MEMCACHED_SUCCESS, memcached_delete(memc, "foo", 3, 0));
  71     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  72     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  73     test_compare(MEMCACHED_INVALID_ARGUMENTS, memcached_destroy_sasl_auth_data(NULL));
  74     memcached_quit(memc);
  75
  76     test_compare(MEMCACHED_AUTH_FAILURE,
  77                  memcached_set(memc, "foo", 3, "bar", 3, (time_t)0, (uint32_t)0));
  78     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  79
  80     memcached_quit(memc);
  81     return TEST_SUCCESS;
  82   }
  83
  84   return TEST_SKIPPED;
  85 }
  86
  87
  88 test_st sasl_auth_tests[]= {
  89   {"sasl_auth", true, (test_callback_fn*)sasl_auth_test },
  90   {0, 0, (test_callback_fn*)0}
  91 };
  92
  93 collection_st collection[] ={
  94   {"sasl_auth", (test_callback_fn*)pre_sasl, 0, sasl_auth_tests },
  95 #if 0
  96   {"sasl", (test_callback_fn*)pre_sasl, 0, tests },
  97 #endif
  98   {0, 0, 0, 0}
  99 };
 100
 101 #include "tests/libmemcached_world.h"
 102
 103 void get_world(Framework *world)
 104 {
 105   world->collections= collection;
 106
 107   world->_create= (test_callback_create_fn*)world_create;
 108   world->_destroy= (test_callback_destroy_fn*)world_destroy;
 109
 110   world->item._startup= (test_callback_fn*)world_test_startup;
 111   world->item.set_pre((test_callback_fn*)world_pre_run);
 112   world->item.set_flush((test_callback_fn*)world_flush);
 113   world->item.set_post((test_callback_fn*)world_post_run);
 114   world->_on_error= (test_callback_error_fn*)world_on_error;
 115
 116   world->collection_startup= (test_callback_fn*)world_container_startup;
 117   world->collection_shutdown= (test_callback_fn*)world_container_shutdown;
 118
 119   world->set_runner(&defualt_libmemcached_runner);
 120
 121   world->set_sasl("memcached", "memcached");
 122 }