git.m6w6.name Git - awesomized/libmemcached/blob - tests/runner.h

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached Client and Server
   4  *
   5  *  Copyright (C) 2012 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38
  39 #pragma once
  40
  41 #include "tests/libmemcached-1.0/generate.h"
  42
  43 class LibmemcachedRunner : public libtest::Runner {
  44 public:
  45   test_return_t run(test_callback_fn* func, void *object)
  46   {
  47     return _runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
  48   }
  49
  50   test_return_t flush(void* arg)
  51   {
  52     return flush((libmemcached_test_container_st*)arg);
  53   }
  54
  55   test_return_t flush(libmemcached_test_container_st *container)
  56   {
  57     test_true(container->memc);
  58     memcached_flush(container->memc, 0);
  59     memcached_quit(container->memc);
  60
  61     return TEST_SUCCESS;
  62   }
  63
  64   test_return_t pre(test_callback_fn* func, void *object)
  65   {
  66     return _pre_runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
  67   }
  68
  69   test_return_t post(test_callback_fn* func, void *object)
  70   {
  71     return _post_runner_default(libmemcached_test_callback_fn(func), (libmemcached_test_container_st*)object);
  72   }
  73
  74 private:
  75   test_return_t _runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
  76   {
  77     test_compare(true, check());
  78
  79     if (func)
  80     {
  81       test_true(container);
  82       test_true(container->memc);
  83       test_return_t ret;
  84       try {
  85         ret= func(container->memc);
  86       }
  87       catch (std::exception& e)
  88       {
  89         libtest::Error << e.what();
  90         return TEST_FAILURE;
  91       }
  92
  93       return ret;
  94     }
  95
  96     return TEST_SUCCESS;
  97   }
  98
  99   test_return_t _pre_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
 100   {
 101     test_compare(true, check());
 102
 103     if (func)
 104     {
 105       return func(container->parent);
 106     }
 107
 108     return TEST_SUCCESS;
 109   }
 110
 111   test_return_t _post_runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
 112   {
 113     test_compare(true, check());
 114     cleanup_pairs(NULL);
 115
 116     if (func)
 117     {
 118       return func(container->parent);
 119     }
 120
 121     return TEST_SUCCESS;
 122   }
 123 };
 124