Adding custom IOSTREAM for libtest.
[awesomized/libmemcached] / tests / cycle.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
38 /*
39 Test that we are cycling the servers we are creating during testing.
40 */
41
42 #include <libtest/common.h>
43
44 #include <libmemcached/common.h>
45 #include <libmemcached/is.h>
46 #include <libmemcached/util.h>
47
48 #include <iostream>
49
50
51 #include <libtest/server.h>
52
53 #define SERVERS_TO_CREATE 5
54
55 #ifndef __INTEL_COMPILER
56 #pragma GCC diagnostic ignored "-Wstrict-aliasing"
57 #endif
58
59 static test_return_t alive(memcached_st *memc)
60 {
61 test_true(memc);
62 test_true(memcached_is_allocated(memc));
63 for (uint32_t x= 0; x < memcached_server_count(memc); ++x)
64 {
65 memcached_server_instance_st instance= memcached_server_instance_by_position(memc, x);
66 test_true(instance);
67
68 test_true(libmemcached_util_ping(memcached_server_name(instance),
69 memcached_server_port(instance), NULL));
70 }
71
72 return TEST_SUCCESS;
73 }
74
75 static test_return_t valid(memcached_st *memc)
76 {
77 test_true(memc);
78 test_true(memcached_is_allocated(memc));
79
80 for (uint32_t x= 0; x < memcached_server_count(memc); ++x)
81 {
82 memcached_server_instance_st instance= memcached_server_instance_by_position(memc, x);
83 test_true(instance);
84
85 pid_t pid= libmemcached_util_getpid(memcached_server_name(instance),
86 memcached_server_port(instance), NULL);
87 test_true(pid != -1);
88 }
89
90 return TEST_SUCCESS;
91 }
92
93 static test_return_t kill_test(memcached_st *)
94 {
95 static struct timespec global_sleep_value= { 2, 0 };
96
97 #ifdef WIN32
98 sleep(1);
99 #else
100 nanosleep(&global_sleep_value, NULL);
101 #endif
102
103 return TEST_SUCCESS;
104 }
105
106 test_st ping_tests[] ={
107 {"alive", true, (test_callback_fn*)alive },
108 {0, 0, 0}
109 };
110
111 test_st getpid_tests[] ={
112 {"valid", true, (test_callback_fn*)valid },
113 {0, 0, 0}
114 };
115
116 test_st kill_tests[] ={
117 {"kill", true, (test_callback_fn*)kill_test },
118 {0, 0, 0}
119 };
120
121 collection_st collection[] ={
122 {"libmemcached_util_ping()", 0, 0, ping_tests},
123 {"libmemcached_util_getpid()", 0, 0, getpid_tests},
124 {"kill", 0, 0, kill_tests},
125 {0, 0, 0, 0}
126 };
127
128 #if 0
129 static server_startup_st *world_create(test_return_t *error)
130 {
131 server_startup_st *servers= new server_startup_st();
132
133 server_startup(servers);
134
135 *error= TEST_SUCCESS;
136
137 return servers;
138 }
139
140 static test_return_t world_destroy(server_startup_st *servers)
141 {
142 server_shutdown(servers);
143 delete servers;
144
145 return TEST_SUCCESS;
146 }
147 #endif
148
149
150 #include "tests/libmemcached_world.h"
151
152 void get_world(Framework *world)
153 {
154 world->collections= collection;
155
156 world->_create= (test_callback_create_fn*)world_create;
157 world->_destroy= (test_callback_fn*)world_destroy;
158
159 world->item._startup= (test_callback_fn*)world_test_startup;
160 world->item.set_pre((test_callback_fn*)world_pre_run);
161 world->item.set_post((test_callback_fn*)world_post_run);
162 world->_on_error= (test_callback_error_fn*)world_on_error;
163
164 world->collection_startup= (test_callback_fn*)world_container_startup;
165 world->collection_shutdown= (test_callback_fn*)world_container_shutdown;
166
167 world->runner= &defualt_libmemcached_runner;
168 }
169