Refactor out portion of Item
[m6w6/libmemcached] / 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