Merge in all of the build tree.
[m6w6/libmemcached] / tests / parser.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Libmemcached Client and Server
4 *
5 * Copyright (C) 2011 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 #include <config.h>
39 #include <libtest/test.hpp>
40
41 using namespace libtest;
42
43 #include <vector>
44 #include <iostream>
45 #include <string>
46 #include <cerrno>
47 #include <cassert>
48
49 #include <libmemcached/memcached.h>
50 #include <libmemcached/util.h>
51
52 #include <tests/parser.h>
53 #include <tests/print.h>
54
55 enum scanner_type_t
56 {
57 NIL,
58 UNSIGNED,
59 SIGNED,
60 ARRAY
61 };
62
63
64 struct scanner_string_st {
65 const char *c_str;
66 size_t size;
67 };
68
69 static inline scanner_string_st scanner_string(const char *arg, size_t arg_size)
70 {
71 scanner_string_st local= { arg, arg_size };
72 return local;
73 }
74
75 #define make_scanner_string(X) scanner_string((X), static_cast<size_t>(sizeof(X) - 1))
76
77 static struct scanner_string_st scanner_string_null= { 0, 0};
78
79 struct scanner_variable_t {
80 enum scanner_type_t type;
81 struct scanner_string_st option;
82 struct scanner_string_st result;
83 test_return_t (*check_func)(memcached_st *memc, const scanner_string_st &hostname);
84 };
85
86 // Check and make sure the first host is what we expect it to be
87 static test_return_t __check_host(memcached_st *memc, const scanner_string_st &hostname)
88 {
89 memcached_server_instance_st instance=
90 memcached_server_instance_by_position(memc, 0);
91
92 test_true(instance);
93
94 const char *first_hostname = memcached_server_name(instance);
95 test_true(first_hostname);
96 test_strcmp(first_hostname, hostname.c_str);
97
98 return TEST_SUCCESS;
99 }
100
101 // Check and make sure the prefix_key is what we expect it to be
102 static test_return_t __check_namespace(memcached_st *memc, const scanner_string_st &arg)
103 {
104 const char *_namespace = memcached_get_namespace(memc);
105 test_true(_namespace);
106 test_strcmp(_namespace, arg.c_str);
107
108 return TEST_SUCCESS;
109 }
110
111 static test_return_t __check_IO_MSG_WATERMARK(memcached_st *memc, const scanner_string_st &value)
112 {
113 uint64_t value_number;
114
115 value_number= atoll(value.c_str);
116
117 test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK) == value_number);
118 return TEST_SUCCESS;
119 }
120
121 static test_return_t __check_REMOVE_FAILED_SERVERS(memcached_st *memc, const scanner_string_st &)
122 {
123 test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS));
124 return TEST_SUCCESS;
125 }
126
127 static test_return_t __check_NOREPLY(memcached_st *memc, const scanner_string_st &)
128 {
129 test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_NOREPLY));
130 return TEST_SUCCESS;
131 }
132
133 static test_return_t __check_VERIFY_KEY(memcached_st *memc, const scanner_string_st &)
134 {
135 test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_VERIFY_KEY));
136 return TEST_SUCCESS;
137 }
138
139 static test_return_t __check_distribution_RANDOM(memcached_st *memc, const scanner_string_st &)
140 {
141 test_true(memcached_behavior_get(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION) == MEMCACHED_DISTRIBUTION_RANDOM);
142 return TEST_SUCCESS;
143 }
144
145 scanner_variable_t test_server_strings[]= {
146 { ARRAY, make_scanner_string("--server=localhost"), make_scanner_string("localhost"), __check_host },
147 { ARRAY, make_scanner_string("--server=10.0.2.1"), make_scanner_string("10.0.2.1"), __check_host },
148 { ARRAY, make_scanner_string("--server=example.com"), make_scanner_string("example.com"), __check_host },
149 { ARRAY, make_scanner_string("--server=localhost:30"), make_scanner_string("localhost"), __check_host },
150 { ARRAY, make_scanner_string("--server=10.0.2.1:20"), make_scanner_string("10.0.2.1"), __check_host },
151 { ARRAY, make_scanner_string("--server=example.com:1024"), make_scanner_string("example.com"), __check_host },
152 { NIL, scanner_string_null, scanner_string_null, NULL }
153 };
154
155 scanner_variable_t test_server_strings_with_weights[]= {
156 { ARRAY, make_scanner_string("--server=10.0.2.1:30/?40"), make_scanner_string("10.0.2.1"), __check_host },
157 { ARRAY, make_scanner_string("--server=example.com:1024/?30"), make_scanner_string("example.com"), __check_host },
158 { ARRAY, make_scanner_string("--server=10.0.2.1/?20"), make_scanner_string("10.0.2.1"), __check_host },
159 { ARRAY, make_scanner_string("--server=example.com/?10"), make_scanner_string("example.com"), __check_host },
160 { NIL, scanner_string_null, scanner_string_null, NULL }
161 };
162
163 scanner_variable_t bad_test_strings[]= {
164 { ARRAY, make_scanner_string("-servers=localhost:11221,localhost:11222,localhost:11223,localhost:11224,localhost:11225"), scanner_string_null, NULL },
165 { ARRAY, make_scanner_string("-- servers=a.example.com:81,localhost:82,b.example.com"), scanner_string_null, NULL },
166 { ARRAY, make_scanner_string("--servers=localhost:+80"), scanner_string_null, NULL},
167 { ARRAY, make_scanner_string("--servers=localhost.com."), scanner_string_null, NULL},
168 { ARRAY, make_scanner_string("--server=localhost.com."), scanner_string_null, NULL},
169 { ARRAY, make_scanner_string("--server=localhost.com.:80"), scanner_string_null, NULL},
170 { NIL, scanner_string_null, scanner_string_null, NULL}
171 };
172
173 scanner_variable_t test_number_options[]= {
174 { ARRAY, make_scanner_string("--CONNECT-TIMEOUT=456"), scanner_string_null, NULL },
175 { ARRAY, make_scanner_string("--IO-BYTES-WATERMARK=456"), scanner_string_null, NULL },
176 { ARRAY, make_scanner_string("--IO-KEY-PREFETCH=456"), scanner_string_null, NULL },
177 { ARRAY, make_scanner_string("--IO-MSG-WATERMARK=456"), make_scanner_string("456"), __check_IO_MSG_WATERMARK },
178 { ARRAY, make_scanner_string("--NUMBER-OF-REPLICAS=456"), scanner_string_null, NULL },
179 { ARRAY, make_scanner_string("--POLL-TIMEOUT=456"), scanner_string_null, NULL },
180 { ARRAY, make_scanner_string("--RCV-TIMEOUT=456"), scanner_string_null, NULL },
181 { ARRAY, make_scanner_string("--REMOVE-FAILED-SERVERS=3"), scanner_string_null, __check_REMOVE_FAILED_SERVERS },
182 { ARRAY, make_scanner_string("--RETRY-TIMEOUT=456"), scanner_string_null, NULL },
183 { ARRAY, make_scanner_string("--SND-TIMEOUT=456"), scanner_string_null, NULL },
184 { ARRAY, make_scanner_string("--SOCKET-RECV-SIZE=456"), scanner_string_null, NULL },
185 { ARRAY, make_scanner_string("--SOCKET-SEND-SIZE=456"), scanner_string_null, NULL },
186 { NIL, scanner_string_null, scanner_string_null, NULL}
187 };
188
189 scanner_variable_t test_boolean_options[]= {
190 { ARRAY, make_scanner_string("--BINARY-PROTOCOL"), scanner_string_null, NULL },
191 { ARRAY, make_scanner_string("--BUFFER-REQUESTS"), scanner_string_null, NULL },
192 { ARRAY, make_scanner_string("--HASH-WITH-NAMESPACE"), scanner_string_null, NULL },
193 { ARRAY, make_scanner_string("--NOREPLY"), scanner_string_null, __check_NOREPLY },
194 { ARRAY, make_scanner_string("--RANDOMIZE-REPLICA-READ"), scanner_string_null, NULL },
195 { ARRAY, make_scanner_string("--SORT-HOSTS"), scanner_string_null, NULL },
196 { ARRAY, make_scanner_string("--SUPPORT-CAS"), scanner_string_null, NULL },
197 { ARRAY, make_scanner_string("--TCP-NODELAY"), scanner_string_null, NULL },
198 { ARRAY, make_scanner_string("--TCP-KEEPALIVE"), scanner_string_null, NULL },
199 { ARRAY, make_scanner_string("--TCP-KEEPIDLE"), scanner_string_null, NULL },
200 { ARRAY, make_scanner_string("--USE-UDP"), scanner_string_null, NULL },
201 { ARRAY, make_scanner_string("--VERIFY-KEY"), scanner_string_null, __check_VERIFY_KEY },
202 { NIL, scanner_string_null, scanner_string_null, NULL}
203 };
204
205 scanner_variable_t namespace_strings[]= {
206 { ARRAY, make_scanner_string("--NAMESPACE=foo"), make_scanner_string("foo"), __check_namespace },
207 { ARRAY, make_scanner_string("--NAMESPACE=\"foo\""), make_scanner_string("foo"), __check_namespace },
208 { ARRAY, make_scanner_string("--NAMESPACE=\"This_is_a_very_long_key\""), make_scanner_string("This_is_a_very_long_key"), __check_namespace },
209 { NIL, scanner_string_null, scanner_string_null, NULL}
210 };
211
212 scanner_variable_t distribution_strings[]= {
213 { ARRAY, make_scanner_string("--DISTRIBUTION=consistent"), scanner_string_null, NULL },
214 { ARRAY, make_scanner_string("--DISTRIBUTION=consistent,CRC"), scanner_string_null, NULL },
215 { ARRAY, make_scanner_string("--DISTRIBUTION=consistent,MD5"), scanner_string_null, NULL },
216 { ARRAY, make_scanner_string("--DISTRIBUTION=random"), scanner_string_null, __check_distribution_RANDOM },
217 { ARRAY, make_scanner_string("--DISTRIBUTION=modula"), scanner_string_null, NULL },
218 { NIL, scanner_string_null, scanner_string_null, NULL}
219 };
220
221 scanner_variable_t hash_strings[]= {
222 { ARRAY, make_scanner_string("--HASH=CRC"), scanner_string_null, NULL },
223 { ARRAY, make_scanner_string("--HASH=FNV1A_32"), scanner_string_null, NULL },
224 { ARRAY, make_scanner_string("--HASH=FNV1A_64"), scanner_string_null, NULL },
225 { ARRAY, make_scanner_string("--HASH=FNV1_32"), scanner_string_null, NULL },
226 { ARRAY, make_scanner_string("--HASH=FNV1_64"), scanner_string_null, NULL },
227 { ARRAY, make_scanner_string("--HASH=JENKINS"), scanner_string_null, NULL },
228 { ARRAY, make_scanner_string("--HASH=MD5"), scanner_string_null, NULL },
229 { ARRAY, make_scanner_string("--HASH=MURMUR"), scanner_string_null, NULL },
230 { NIL, scanner_string_null, scanner_string_null, NULL}
231 };
232
233
234 static test_return_t _test_option(scanner_variable_t *scanner, bool test_true_opt= true)
235 {
236 for (scanner_variable_t *ptr= scanner; ptr->type != NIL; ptr++)
237 {
238 memcached_st *memc= memcached(ptr->option.c_str, ptr->option.size);
239
240 // The case that it should have parsed, but it didn't. We will inspect
241 // for an error with libmemcached_check_configuration()
242 if (not memc and test_true_opt)
243 {
244 char buffer[2048];
245 bool success= libmemcached_check_configuration(ptr->option.c_str, ptr->option.size, buffer, sizeof(buffer));
246
247 std::string temp(buffer);
248 temp+= " with option string:";
249 temp+= ptr->option.c_str;
250 test_true_got(success, temp.c_str());
251
252 return TEST_FAILURE; // The line above should fail since memc should be null
253 }
254
255 if (test_true_opt)
256 {
257 if (ptr->check_func)
258 {
259 test_return_t test_rc= (*ptr->check_func)(memc, ptr->result);
260 if (test_rc != TEST_SUCCESS)
261 {
262 memcached_free(memc);
263 return test_rc;
264 }
265 }
266
267 memcached_free(memc);
268 }
269 else
270 {
271 test_false_with(memc, ptr->option.c_str);
272 }
273 }
274
275 return TEST_SUCCESS;
276 }
277
278 test_return_t server_test(memcached_st *)
279 {
280 return _test_option(test_server_strings);
281 }
282
283 test_return_t server_with_weight_test(memcached_st *)
284 {
285 return _test_option(test_server_strings_with_weights);
286 }
287
288 test_return_t servers_bad_test(memcached_st *)
289 {
290 test_return_t rc;
291 if ((rc= _test_option(bad_test_strings, false)) != TEST_SUCCESS)
292 {
293 return rc;
294 }
295
296 return TEST_SUCCESS;
297 }
298
299 test_return_t parser_number_options_test(memcached_st*)
300 {
301 return _test_option(test_number_options);
302 }
303
304 test_return_t parser_boolean_options_test(memcached_st*)
305 {
306 return _test_option(test_boolean_options);
307 }
308
309 test_return_t behavior_parser_test(memcached_st*)
310 {
311 return TEST_SUCCESS;
312 }
313
314 test_return_t parser_hash_test(memcached_st*)
315 {
316 return _test_option(hash_strings);
317 }
318
319 test_return_t parser_distribution_test(memcached_st*)
320 {
321 return _test_option(distribution_strings);
322 }
323
324 test_return_t parser_key_prefix_test(memcached_st*)
325 {
326 return _test_option(distribution_strings);
327 }
328
329 test_return_t test_namespace_keyword(memcached_st*)
330 {
331 return _test_option(namespace_strings);
332 }
333
334 #define SUPPORT_EXAMPLE_CNF "support/example.cnf"
335
336 test_return_t memcached_create_with_options_with_filename(memcached_st*)
337 {
338 if (access(SUPPORT_EXAMPLE_CNF, R_OK))
339 return TEST_SKIPPED;
340
341 memcached_st *memc_ptr;
342 memc_ptr= memcached(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""));
343 test_true_got(memc_ptr, "memcached() failed");
344 memcached_free(memc_ptr);
345
346 return TEST_SUCCESS;
347 }
348
349 test_return_t libmemcached_check_configuration_with_filename_test(memcached_st*)
350 {
351 if (access(SUPPORT_EXAMPLE_CNF, R_OK))
352 return TEST_SKIPPED;
353
354 memcached_return_t rc;
355 char buffer[BUFSIZ];
356
357 rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"support/example.cnf\""), buffer, sizeof(buffer));
358 test_true_got(rc == MEMCACHED_SUCCESS, (rc == MEMCACHED_ERRNO) ? strerror(errno) : memcached_strerror(NULL, rc));
359
360 rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=support/example.cnf"), buffer, sizeof(buffer));
361 test_false_with(rc == MEMCACHED_SUCCESS, memcached_strerror(NULL, rc));
362
363 rc= libmemcached_check_configuration(test_literal_param("--CONFIGURE-FILE=\"bad-path/example.cnf\""), buffer, sizeof(buffer));
364 test_true_got(rc == MEMCACHED_ERRNO, memcached_strerror(NULL, rc));
365
366 return TEST_SUCCESS;
367 }
368
369 test_return_t libmemcached_check_configuration_test(memcached_st*)
370 {
371 memcached_return_t rc;
372 char buffer[BUFSIZ];
373
374 test_compare(MEMCACHED_SUCCESS,
375 libmemcached_check_configuration(test_literal_param("--server=localhost"), buffer, sizeof(buffer)));
376
377 rc= libmemcached_check_configuration(test_literal_param("--dude=localhost"), buffer, sizeof(buffer));
378 test_false_with(rc == MEMCACHED_SUCCESS, buffer);
379 test_true(rc == MEMCACHED_PARSE_ERROR);
380
381 return TEST_SUCCESS;
382 }
383
384 test_return_t memcached_create_with_options_test(memcached_st*)
385 {
386 memcached_st *memc_ptr;
387 memc_ptr= memcached(test_literal_param("--server=localhost"));
388 test_true_got(memc_ptr, memcached_last_error_message(memc_ptr));
389 memcached_free(memc_ptr);
390
391 memc_ptr= memcached(test_literal_param("--dude=localhost"));
392 test_false_with(memc_ptr, memcached_last_error_message(memc_ptr));
393
394 return TEST_SUCCESS;
395 }
396
397 test_return_t test_include_keyword(memcached_st*)
398 {
399 if (access(SUPPORT_EXAMPLE_CNF, R_OK))
400 return TEST_SKIPPED;
401
402 char buffer[BUFSIZ];
403 test_compare(MEMCACHED_SUCCESS,
404 libmemcached_check_configuration(test_literal_param("INCLUDE \"support/example.cnf\""), buffer, sizeof(buffer)));
405
406 return TEST_SUCCESS;
407 }
408
409 test_return_t test_end_keyword(memcached_st*)
410 {
411 char buffer[BUFSIZ];
412 test_compare(MEMCACHED_SUCCESS,
413 libmemcached_check_configuration(test_literal_param("--server=localhost END bad keywords"), buffer, sizeof(buffer)));
414
415 return TEST_SUCCESS;
416 }
417
418 test_return_t test_reset_keyword(memcached_st*)
419 {
420 char buffer[BUFSIZ];
421 test_compare(MEMCACHED_SUCCESS,
422 libmemcached_check_configuration(test_literal_param("--server=localhost reset --server=bad.com"), buffer, sizeof(buffer)));
423
424 return TEST_SUCCESS;
425 }
426
427 test_return_t test_error_keyword(memcached_st*)
428 {
429 char buffer[BUFSIZ];
430 memcached_return_t rc;
431 rc= libmemcached_check_configuration(test_literal_param("--server=localhost ERROR --server=bad.com"), buffer, sizeof(buffer));
432 test_true_got(rc != MEMCACHED_SUCCESS, buffer);
433
434 return TEST_SUCCESS;
435 }
436
437 #define RANDOM_STRINGS 100
438 test_return_t random_statement_build_test(memcached_st*)
439 {
440 std::vector<scanner_string_st *> option_list;
441
442 for (scanner_variable_t *ptr= test_server_strings; ptr->type != NIL; ptr++)
443 option_list.push_back(&ptr->option);
444
445 for (scanner_variable_t *ptr= test_number_options; ptr->type != NIL; ptr++)
446 option_list.push_back(&ptr->option);
447
448 for (scanner_variable_t *ptr= test_boolean_options; ptr->type != NIL; ptr++)
449 option_list.push_back(&ptr->option);
450
451 for (scanner_variable_t *ptr= namespace_strings; ptr->type != NIL; ptr++)
452 option_list.push_back(&ptr->option);
453
454 for (scanner_variable_t *ptr= distribution_strings; ptr->type != NIL; ptr++)
455 option_list.push_back(&ptr->option);
456
457 for (scanner_variable_t *ptr= hash_strings; ptr->type != NIL; ptr++)
458 option_list.push_back(&ptr->option);
459
460 for (uint32_t x= 0; x < RANDOM_STRINGS; x++)
461 {
462 std::string random_options;
463
464 uint32_t number_of= random() % option_list.size();
465 for (uint32_t options= 0; options < number_of; options++)
466 {
467 random_options+= option_list[random() % option_list.size()]->c_str;
468 random_options+= " ";
469 }
470
471 memcached_st *memc_ptr= memcached(random_options.c_str(), random_options.size() -1);
472 if (not memc_ptr)
473 {
474 switch (errno)
475 {
476 case EINVAL:
477 #if 0 // Testing framework is not smart enough for this just yet.
478 {
479 // We will try to find the specific error
480 char buffer[2048];
481 memcached_return_t rc= libmemcached_check_configuration(random_options.c_str(), random_options.size(), buffer, sizeof(buffer));
482 test_true_got(rc != MEMCACHED_SUCCESS, "memcached_create_with_options() failed whiled libmemcached_check_configuration() was successful");
483 std::cerr << "Error occured on " << random_options.c_str() << " : " << buffer << std::endl;
484 return TEST_FAILURE;
485 }
486 #endif
487 break;
488 case ENOMEM:
489 std::cerr << "Failed to allocate memory for memcached_create_with_options()" << std::endl;
490 memcached_free(memc_ptr);
491 return TEST_FAILURE;
492 default:
493 std::cerr << "Unknown error from memcached_create_with_options?!!" << std::endl;
494 memcached_free(memc_ptr);
495 return TEST_FAILURE;
496 }
497 }
498 memcached_free(memc_ptr);
499 }
500
501 return TEST_SUCCESS;
502 }
503
504 static memcached_return_t dump_server_information(const memcached_st *,
505 const memcached_server_st *instance,
506 void *)
507 {
508 if (strcmp(memcached_server_name(instance), "localhost"))
509 {
510 assert(not memcached_server_name(instance));
511 return MEMCACHED_FAILURE;
512 }
513
514 if (memcached_server_port(instance) < 8888 or memcached_server_port(instance) > 8892)
515 {
516 assert(not memcached_server_port(instance));
517 return MEMCACHED_FAILURE;
518 }
519
520 if (instance->weight > 5 or instance->weight < 2)
521 {
522 assert(not instance->weight);
523 return MEMCACHED_FAILURE;
524 }
525
526 return MEMCACHED_SUCCESS;
527 }
528
529
530 test_return_t test_hostname_port_weight(memcached_st *)
531 {
532 const char *server_string= "--server=localhost:8888/?2 --server=localhost:8889/?3 --server=localhost:8890/?4 --server=localhost:8891/?5 --server=localhost:8892/?3";
533 char buffer[BUFSIZ];
534
535 test_compare_got(MEMCACHED_SUCCESS,
536 libmemcached_check_configuration(server_string, strlen(server_string), buffer, sizeof(buffer)), buffer);
537
538 memcached_st *memc= memcached(server_string, strlen(server_string));
539 test_true(memc);
540
541 memcached_server_fn callbacks[]= { dump_server_information };
542 test_true(memcached_success(memcached_server_cursor(memc, callbacks, NULL, 1)));
543
544 memcached_free(memc);
545
546 return TEST_SUCCESS;
547 }
548
549 struct socket_weight_t {
550 const char *socket;
551 size_t weight;
552 };
553
554 static memcached_return_t dump_socket_information(const memcached_st *,
555 const memcached_server_st *instance,
556 void *context)
557 {
558 socket_weight_t *check= (socket_weight_t *)context;
559
560 if (strcmp(memcached_server_name(instance), check->socket))
561 {
562 std::cerr << std::endl << __FILE__ << ":" << __LINE__ << " " << memcached_server_name(instance) << " != " << check->socket << std::endl;
563 return MEMCACHED_FAILURE;
564 }
565
566 if (instance->weight == check->weight)
567 {
568 return MEMCACHED_SUCCESS;
569 }
570
571 return MEMCACHED_FAILURE;
572 }
573
574 test_return_t test_parse_socket(memcached_st *)
575 {
576 char buffer[BUFSIZ];
577
578 memcached_server_fn callbacks[]= { dump_socket_information };
579 {
580 test_compare_got(MEMCACHED_SUCCESS,
581 libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\""), buffer, sizeof(buffer)),
582 buffer);
583
584 memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\""));
585 test_true(memc);
586 socket_weight_t check= { "/tmp/foo", 1 };
587 test_compare(MEMCACHED_SUCCESS,
588 memcached_server_cursor(memc, callbacks, &check, 1));
589 memcached_free(memc);
590 }
591
592 {
593 test_compare_got(MEMCACHED_SUCCESS,
594 libmemcached_check_configuration(test_literal_param("--socket=\"/tmp/foo\"/?23"), buffer, sizeof(buffer)),
595 buffer);
596
597 memcached_st *memc= memcached(test_literal_param("--socket=\"/tmp/foo\"/?23"));
598 test_true(memc);
599 socket_weight_t check= { "/tmp/foo", 23 };
600 test_compare(MEMCACHED_SUCCESS,
601 memcached_server_cursor(memc, callbacks, &check, 1));
602 memcached_free(memc);
603 }
604
605 return TEST_SUCCESS;
606 }
607
608 /*
609 By setting the timeout value to zero, we force poll() to return immediatly.
610 */
611 test_return_t regression_bug_71231153_connect(memcached_st *)
612 {
613 if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
614 return TEST_SKIPPED;
615
616 { // Test the connect-timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
617 memcached_st *memc= memcached(test_literal_param("--SERVER=10.0.2.252 --CONNECT-TIMEOUT=0"));
618 test_true(memc);
619 test_zero(memc->connect_timeout);
620 test_compare(MEMCACHED_DEFAULT_TIMEOUT, memc->poll_timeout);
621
622 memcached_return_t rc;
623 size_t value_len;
624 char *value= memcached_get(memc, test_literal_param("test"), &value_len, NULL, &rc);
625 test_false(value);
626 test_zero(value_len);
627 test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_last_error_message(memc));
628
629 memcached_free(memc);
630 }
631
632 return TEST_SUCCESS;
633 }
634
635 test_return_t regression_bug_71231153_poll(memcached_st *)
636 {
637 if (libmemcached_util_ping("10.0.2.252", 0, NULL)) // If for whatever reason someone has a host at this address, skip
638 return TEST_SKIPPED;
639
640 { // Test the poll timeout, on a bad host we should get MEMCACHED_CONNECTION_FAILURE
641 memcached_st *memc= memcached(test_literal_param("--SERVER=10.0.2.252 --POLL-TIMEOUT=0"));
642 test_true(memc);
643 test_compare(MEMCACHED_DEFAULT_CONNECT_TIMEOUT, memc->connect_timeout);
644 test_zero(memc->poll_timeout);
645
646 memcached_return_t rc;
647 size_t value_len;
648 char *value= memcached_get(memc, test_literal_param("test"), &value_len, NULL, &rc);
649 test_false(value);
650 test_zero(value_len);
651 #ifdef __APPLE__
652 test_compare_got(MEMCACHED_CONNECTION_FAILURE, rc, memcached_last_error_message(memc));
653 #else
654 test_compare_got(MEMCACHED_TIMEOUT, rc, memcached_last_error_message(memc));
655 #endif
656
657 memcached_free(memc);
658 }
659
660 return TEST_SUCCESS;
661 }