1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
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
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
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.
42 #include <libmemcached/memcached.hpp>
43 #include <libmemcached/server_instance.h>
44 #include <libtest/test.hpp>
50 #include <sys/types.h>
59 using namespace memcache
;
60 using namespace libtest
;
62 Framework
*global_framework
= NULL
;
64 static test_return_t
shutdown_servers(memcached_st
*memc
)
66 test_compare(memcached_server_count(memc
), 1U);
68 // Disable a single server, just the first
69 global_framework
->servers().shutdown(0);
74 static test_return_t
restart_servers(memcached_st
*)
76 // Restart the servers
77 global_framework
->servers().restart();
82 static test_return_t
cull_TEST(memcached_st
*memc
)
84 uint32_t count
= memcached_server_count(memc
);
86 // Do not do this in your code, it is not supported.
87 memc
->servers
[0].options
.is_dead
= true;
88 memc
->state
.is_time_for_rebuild
= true;
90 uint32_t new_count
= memcached_server_count(memc
);
91 test_compare(count
, new_count
);
96 static test_return_t
MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST(memcached_st
*memc
)
98 test_compare(MEMCACHED_SUCCESS
, memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
, 30));
99 test_compare_got(MEMCACHED_CONNECTION_FAILURE
,
101 test_literal_param("foo"),
102 NULL
, 0, time_t(0), uint32_t(0)),
103 memcached_last_error_message(memc
));
106 Setting server_failure_counter==0 should not influence the timeout that we set above,
107 since we check the timeout that is created by the failure before we check how many times
110 test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED
,
111 memcached_set(memc
, test_literal_param("foo"), NULL
, 0, time_t(0), uint32_t(0)));
116 static test_return_t
MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST(memcached_st
*memc
)
118 test_compare_got(MEMCACHED_CONNECTION_FAILURE
,
120 test_literal_param("foo"),
121 NULL
, 0, time_t(0), uint32_t(0)),
122 memcached_last_error_message(memc
));
125 Setting server_failure_counter==0 should not influence the timeout that we set above,
126 since we check the timeout that is created by the failure before we check how many times
129 test_compare(MEMCACHED_SERVER_TEMPORARILY_DISABLED
,
130 memcached_set(memc
, test_literal_param("foo"), NULL
, 0, time_t(0), uint32_t(0)));
132 global_framework
->servers().restart();
134 memcached_return_t ret
;
137 ret
= memcached_set(memc
, test_literal_param("foo"), NULL
, 0, time_t(0), uint32_t(0));
138 } while (ret
== MEMCACHED_SERVER_TEMPORARILY_DISABLED
);
140 test_compare_got(MEMCACHED_SUCCESS
, ret
, memcached_last_error_message(memc
));
145 test_st cull_TESTS
[] ={
146 { "cull servers", true, (test_callback_fn
*)cull_TEST
},
150 test_st server_temporarily_disabled_TESTS
[] ={
151 { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED)", true, (test_callback_fn
*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_TEST
},
152 { "memcached_set(MEMCACHED_SERVER_TEMPORARILY_DISABLED -> MEMCACHED_SUCCESS)", true, (test_callback_fn
*)MEMCACHED_SERVER_TEMPORARILY_DISABLED_to_success_TEST
},
156 collection_st collection
[] ={
157 { "cull", (test_callback_fn
*)shutdown_servers
, (test_callback_fn
*)restart_servers
, cull_TESTS
},
158 { "server failed", (test_callback_fn
*)shutdown_servers
, (test_callback_fn
*)restart_servers
, server_temporarily_disabled_TESTS
},
162 #define TEST_PORT_BASE MEMCACHED_DEFAULT_PORT +10
163 #include "libmemcached_world.h"
165 void get_world(Framework
*world
)
167 world
->servers().set_count(1);
169 world
->collections
= collection
;
171 world
->_create
= (test_callback_create_fn
*)world_create
;
172 world
->_destroy
= (test_callback_destroy_fn
*)world_destroy
;
174 world
->item
._startup
= (test_callback_fn
*)world_test_startup
;
175 world
->item
.set_pre((test_callback_fn
*)world_pre_run
);
176 world
->item
.set_flush((test_callback_fn
*)world_flush
);
177 world
->item
.set_post((test_callback_fn
*)world_post_run
);
178 world
->_on_error
= (test_callback_error_fn
*)world_on_error
;
180 world
->collection_startup
= (test_callback_fn
*)world_container_startup
;
181 world
->collection_shutdown
= (test_callback_fn
*)world_container_shutdown
;
183 world
->set_runner(&defualt_libmemcached_runner
);
185 global_framework
= world
;