Merge working tree with build tree.
[awesomized/libmemcached] / tests / failure.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
39 /*
40 C++ interface test
41 */
42 #include <libmemcached-1.0/memcached.hpp>
43 #include <libmemcached/server_instance.h>
44 #include <libtest/test.hpp>
45
46 #include <cstdio>
47 #include <cstdlib>
48 #include <cstring>
49 #include <sys/time.h>
50 #include <sys/types.h>
51 #include <sys/stat.h>
52 #include <unistd.h>
53 #include <ctime>
54
55 #include <string>
56 #include <iostream>
57
58 using namespace std;
59 using namespace memcache;
60 using namespace libtest;
61
62 Framework *global_framework= NULL;
63
64 static test_return_t shutdown_servers(memcached_st *memc)
65 {
66 test_compare(memcached_server_count(memc), 1U);
67
68 // Disable a single server, just the first
69 global_framework->servers().shutdown(0);
70
71 return TEST_SUCCESS;
72 }
73
74 static test_return_t add_shutdown_servers(memcached_st *memc)
75 {
76 while (memcached_server_count(memc) < 2)
77 {
78 const char *argv[1]= { "add_shutdown_server" };
79 test_true(global_framework->servers().start_socket_server("memcached", libtest::default_port(), 1, argv));
80 test_compare(MEMCACHED_SUCCESS, memcached_server_add(memc, "localhost", libtest::default_port()));
81 }
82
83 // Disable a single server, just the first
84 global_framework->servers().shutdown(0);
85
86 return TEST_SUCCESS;
87 }
88
89 static test_return_t restart_servers(memcached_st *)
90 {
91 // Restart the servers
92 global_framework->servers().restart();
93
94 return TEST_SUCCESS;
95 }
96
97 static test_return_t cull_TEST(memcached_st *memc)
98 {
99 uint32_t count= memcached_server_count(memc);
100
101 // Do not do this in your code, it is not supported.
102 memc->servers[0].options.is_dead= true;
103 memc->state.is_time_for_rebuild= true;
104
105 uint32_t new_count= memcached_server_count(memc);
106 test_compare(count, new_count);
107
108 return TEST_SUCCESS;
109 }
110
111 static test_return_t MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST(memcached_st *memc)
112 {
113 test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT, 30));
114 test_compare_got(MEMCACHED_CONNECTION_FAILURE,
115 memcached_set(memc,
116 test_literal_param("foo"),
117 NULL, 0, time_t(0), uint32_t(0)),
118 memcached_last_error_message(memc));
119
120 /*
121 Setting server_failure_counter==0 should not influence the timeout that we set above,
122 since we check the timeout that is created by the failure before we check how many times
123 a server has failed.
124 */
125 test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED,
126 memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0)));
127
128 return TEST_SUCCESS;
129 }
130
131 static test_return_t MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST(memcached_st *memc)
132 {
133 test_compare_got(MEMCACHED_CONNECTION_FAILURE,
134 memcached_set(memc,
135 test_literal_param("foo"),
136 NULL, 0, time_t(0), uint32_t(0)),
137 memcached_last_error_message(memc));
138
139 /*
140 Setting server_failure_counter==0 should not influence the timeout that we set above,
141 since we check the timeout that is created by the failure before we check how many times
142 a server has failed.
143 */
144 test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED,
145 memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0)));
146
147 global_framework->servers().restart();
148
149 memcached_return_t ret;
150 do {
151 sleep(3);
152 ret= memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0));
153 } while (ret == MEMCACHED_SERVER_TEMPORARILY_DISABLED);
154
155 test_compare_got(MEMCACHED_SUCCESS, ret, memcached_last_error_message(memc));
156
157 return TEST_SUCCESS;
158 }
159
160 static test_return_t MEMCACHED_SERVER_MARKED_DEAD_TEST(memcached_st *memc)
161 {
162 test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT, 30));
163 test_compare(MEMCACHED_SUCCESS, memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS, true));
164
165 memcached_return_t ret;
166 do {
167 ret= memcached_set(memc,
168 test_literal_param("foo"),
169 NULL, 0, time_t(0), uint32_t(0));
170 } while (ret == MEMCACHED_SUCCESS or ret == MEMCACHED_CONNECTION_FAILURE);
171 test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED, ret);
172
173 do {
174 sleep(3);
175 ret= memcached_set(memc, test_literal_param("foo"), NULL, 0, time_t(0), uint32_t(0));
176 } while (ret == MEMCACHED_SERVER_TEMPORARILY_DISABLED or ret == MEMCACHED_SUCCESS);
177
178 test_compare_got(MEMCACHED_SERVER_MARKED_DEAD, ret, memcached_last_error_message(memc));
179
180 return TEST_SUCCESS;
181 }
182
183 test_st cull_TESTS[] ={
184 { "cull servers", true, (test_callback_fn*)cull_TEST },
185 { 0, 0, 0 }
186 };
187
188 test_st server_temporarily_disabled_TESTS[] ={
189 { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED)", true, (test_callback_fn*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST },
190 { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED -> MEMCACHED_SUCCESS)", true, (test_callback_fn*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST },
191 { 0, 0, 0 }
192 };
193
194 test_st server_permanently_disabled_TESTS[] ={
195 { "memcached_set(MEMCACHED_SERVER_MARKED_DEAD)", true, (test_callback_fn*)MEMCACHED_SERVER_MARKED_DEAD_TEST },
196 { 0, 0, 0 }
197 };
198
199 collection_st collection[] ={
200 { "cull", (test_callback_fn*)shutdown_servers, (test_callback_fn*)restart_servers, cull_TESTS },
201 { "server failed", (test_callback_fn*)shutdown_servers, (test_callback_fn*)restart_servers, server_temporarily_disabled_TESTS },
202 { "server eject", (test_callback_fn*)add_shutdown_servers, (test_callback_fn*)restart_servers, server_permanently_disabled_TESTS },
203 { 0, 0, 0, 0 }
204 };
205
206 #include "libmemcached_world.h"
207
208 void get_world(Framework *world)
209 {
210 world->servers().set_count(1);
211
212 world->collections= collection;
213
214 world->_create= (test_callback_create_fn*)world_create;
215 world->_destroy= (test_callback_destroy_fn*)world_destroy;
216
217 world->item._startup= (test_callback_fn*)world_test_startup;
218 world->item.set_pre((test_callback_fn*)world_pre_run);
219 world->item.set_flush((test_callback_fn*)world_flush);
220 world->item.set_post((test_callback_fn*)world_post_run);
221 world->_on_error= (test_callback_error_fn*)world_on_error;
222
223 world->collection_startup= (test_callback_fn*)world_container_startup;
224 world->collection_shutdown= (test_callback_fn*)world_container_shutdown;
225
226 world->set_runner(&defualt_libmemcached_runner);
227
228 global_framework= world;
229 }