git.m6w6.name Git - m6w6/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 #include <libmemcached/memcached.h>
  47
  48 static test_return_t pre_sasl(memcached_st *)
  49 {
  50   if (LIBMEMCACHED_WITH_SASL_SUPPORT == 0)
  51   {
  52     return TEST_SKIPPED;
  53   }
  54
  55   return TEST_SUCCESS;
  56 }
  57
  58 /*
  59  * Test that the sasl authentication works. We cannot use the default
  60  * pool of servers, because that would require that all servers we want
  61  * to test supports SASL authentication, and that they use the default
  62  * creds.
  63  */
  64 static test_return_t sasl_auth_test(memcached_st *memc)
  65 {
  66   if (LIBMEMCACHED_WITH_SASL_SUPPORT)
  67   {
  68     test_compare(MEMCACHED_SUCCESS, memcached_set(memc, "foo", 3, "bar", 3, (time_t)0, (uint32_t)0));
  69     test_compare(MEMCACHED_SUCCESS, memcached_delete(memc, "foo", 3, 0));
  70     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  71     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  72     test_compare(MEMCACHED_INVALID_ARGUMENTS, memcached_destroy_sasl_auth_data(NULL));
  73     memcached_quit(memc);
  74
  75     test_compare(MEMCACHED_AUTH_FAILURE,
  76                  memcached_set(memc, "foo", 3, "bar", 3, (time_t)0, (uint32_t)0));
  77     test_compare(MEMCACHED_SUCCESS, memcached_destroy_sasl_auth_data(memc));
  78
  79     memcached_quit(memc);
  80     return TEST_SUCCESS;
  81   }
  82
  83   return TEST_SKIPPED;
  84 }
  85
  86
  87 test_st sasl_auth_tests[]= {
  88   {"sasl_auth", true, (test_callback_fn*)sasl_auth_test },
  89   {0, 0, (test_callback_fn*)0}
  90 };
  91
  92 collection_st collection[] ={
  93   {"sasl_auth", (test_callback_fn*)pre_sasl, 0, sasl_auth_tests },
  94 #if 0
  95   {"sasl", (test_callback_fn*)pre_sasl, 0, tests },
  96 #endif
  97   {0, 0, 0, 0}
  98 };
  99
 100 #include "tests/libmemcached_world.h"
 101
 102 void get_world(Framework *world)
 103 {
 104   world->collections= collection;
 105
 106   world->_create= (test_callback_create_fn*)world_create;
 107   world->_destroy= (test_callback_destroy_fn*)world_destroy;
 108
 109   world->item._startup= (test_callback_fn*)world_test_startup;
 110   world->item.set_flush((test_callback_fn*)world_flush);
 111   world->_on_error= (test_callback_error_fn*)world_on_error;
 112
 113   world->collection_startup= (test_callback_fn*)world_container_startup;
 114   world->collection_shutdown= (test_callback_fn*)world_container_shutdown;
 115
 116   world->set_runner(new LibmemcachedRunner);
 117
 118   world->set_sasl("memcached", "memcached");
 119 }