Updates for version 0.37.
[m6w6/libmemcached] / tests / libmemcached_world.h
1 /* libMemcached Functions Test
2 * Copyright (C) 2006-2009 Brian Aker
3 * All rights reserved.
4 *
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
7 *
8 * Description: This is the startup bits for any libmemcached test.
9 *
10 */
11
12 #ifdef __cplusplus
13 extern "C" {
14 #endif
15
16 /* The structure we use for the test system */
17 typedef struct
18 {
19 server_startup_st construct;
20 memcached_st *parent;
21 memcached_st *memc;
22 } libmemcached_test_container_st;
23
24 /* Prototypes for functions we will pass to test framework */
25 libmemcached_test_container_st *world_create(test_return_t *error);
26 test_return_t world_test_startup(libmemcached_test_container_st *);
27 test_return_t world_flush(libmemcached_test_container_st *container);
28 test_return_t world_pre_run(libmemcached_test_container_st *);
29
30 test_return_t world_post_run(libmemcached_test_container_st *);
31 test_return_t world_on_error(test_return_t, libmemcached_test_container_st *);
32 test_return_t world_destroy(libmemcached_test_container_st *);
33
34 static libmemcached_test_container_st global_container;
35
36 /**
37 @note generic shutdown/startup for libmemcached tests.
38 */
39 test_return_t world_container_startup(libmemcached_test_container_st *container);
40 test_return_t world_container_shutdown(libmemcached_test_container_st *container);
41
42 libmemcached_test_container_st *world_create(test_return_t *error)
43 {
44 memset(&global_container, 0, sizeof(global_container));
45 global_container.construct.count= SERVERS_TO_CREATE;
46 global_container.construct.udp= 0;
47 server_startup(&global_container.construct);
48
49 if (! global_container.construct.servers)
50 {
51 *error= TEST_FAILURE;
52 server_shutdown(&global_container.construct);
53 return NULL;
54 }
55
56 *error= TEST_SUCCESS;
57
58 return &global_container;
59 }
60
61 test_return_t world_container_startup(libmemcached_test_container_st *container)
62 {
63 memcached_return_t rc;
64 container->parent= memcached_create(NULL);
65 test_truth((container->parent != NULL));
66
67 rc= memcached_server_push(container->parent, container->construct.servers);
68 test_truth(rc == MEMCACHED_SUCCESS);
69
70 return TEST_SUCCESS;
71 }
72
73 test_return_t world_container_shutdown(libmemcached_test_container_st *container)
74 {
75 memcached_free(container->parent);
76 container->parent= NULL;
77
78 return TEST_SUCCESS;
79 }
80
81 test_return_t world_test_startup(libmemcached_test_container_st *container)
82 {
83 container->memc= memcached_clone(NULL, container->parent);
84 test_truth((container->memc != NULL));
85
86 return TEST_SUCCESS;
87 }
88
89 test_return_t world_flush(libmemcached_test_container_st *container)
90 {
91 memcached_flush(container->memc, 0);
92 memcached_quit(container->memc);
93
94 return TEST_SUCCESS;
95 }
96
97 test_return_t world_pre_run(libmemcached_test_container_st *container)
98 {
99 uint32_t loop;
100
101 for (loop= 0; loop < memcached_server_list_count(container->construct.servers); loop++)
102 {
103 test_truth(container->memc->hosts[loop].fd == -1);
104 test_truth(container->memc->hosts[loop].cursor_active == 0);
105 }
106
107 return TEST_SUCCESS;
108 }
109
110
111 test_return_t world_post_run(libmemcached_test_container_st *container)
112 {
113 assert(container->memc);
114
115 return TEST_SUCCESS;
116 }
117
118 test_return_t world_on_error(test_return_t test_state, libmemcached_test_container_st *container)
119 {
120 (void)test_state;
121 memcached_free(container->memc);
122 container->memc= NULL;
123
124 return TEST_SUCCESS;
125 }
126
127 test_return_t world_destroy(libmemcached_test_container_st *container)
128 {
129 server_startup_st *construct= &container->construct;
130 memcached_server_st *servers= container->construct.servers;
131 memcached_server_list_free(servers);
132
133 server_shutdown(construct);
134
135 return TEST_SUCCESS;
136 }
137
138 typedef test_return_t (*libmemcached_test_callback_fn)(memcached_st *);
139 static test_return_t _runner_default(libmemcached_test_callback_fn func, libmemcached_test_container_st *container)
140 {
141 if (func)
142 {
143 return func(container->memc);
144 }
145 else
146 {
147 return TEST_SUCCESS;
148 }
149 }
150
151 #ifdef __cplusplus
152 }
153 #endif
154
155 #ifdef __cplusplus
156
157 static world_runner_st defualt_libmemcached_runner= {
158 reinterpret_cast<test_callback_runner_fn>(_runner_default),
159 reinterpret_cast<test_callback_runner_fn>(_runner_default),
160 reinterpret_cast<test_callback_runner_fn>(_runner_default)
161 };
162
163 #else
164
165 static world_runner_st defualt_libmemcached_runner= {
166 (test_callback_runner_fn)_runner_default,
167 (test_callback_runner_fn)_runner_default,
168 (test_callback_runner_fn)_runner_default
169 };
170
171 #endif