Implement function to determine how many cores are available.
[m6w6/libmemcached] / libtest / unittest.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * libtest
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 3 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22
23 #include <config.h>
24
25 #include <libtest/test.hpp>
26
27 #if defined(LIBTEST_WITH_LIBMEMCACHED_SUPPORT) && LIBTEST_WITH_LIBMEMCACHED_SUPPORT
28 #include <libmemcached-1.0/memcached.h>
29 #endif
30
31 #if defined(LIBTEST_WITH_LIBGEARMAN_SUPPORT) && LIBTEST_WITH_LIBGEARMAN_SUPPORT
32 #include <libgearman/gearman.h>
33 #endif
34
35 #include <cstdlib>
36 #include <unistd.h>
37
38 using namespace libtest;
39
40 static test_return_t LIBTOOL_COMMAND_test(void *)
41 {
42 test_true(getenv("LIBTOOL_COMMAND"));
43 return TEST_SUCCESS;
44 }
45
46 static test_return_t VALGRIND_COMMAND_test(void *)
47 {
48 test_true(getenv("VALGRIND_COMMAND"));
49 return TEST_SUCCESS;
50 }
51
52 static test_return_t HELGRIND_COMMAND_test(void *)
53 {
54 test_true(getenv("HELGRIND_COMMAND"));
55 return TEST_SUCCESS;
56 }
57
58 static test_return_t GDB_COMMAND_test(void *)
59 {
60 test_true(getenv("GDB_COMMAND"));
61 return TEST_SUCCESS;
62 }
63
64 static test_return_t test_success_equals_one_test(void *)
65 {
66 test_skip(HAVE_LIBMEMCACHED, true);
67 #if defined(HAVE_LIBMEMCACHED) && HAVE_LIBMEMCACHED
68 test_zero(MEMCACHED_SUCCESS);
69 #endif
70 return TEST_SUCCESS;
71 }
72
73 static test_return_t test_success_test(void *)
74 {
75 return TEST_SUCCESS;
76 }
77
78 static test_return_t test_failure_test(void *)
79 {
80 return TEST_SKIPPED; // Only run this when debugging
81
82 test_compare(1, 2);
83 return TEST_SUCCESS;
84 }
85
86 static test_return_t local_test(void *)
87 {
88 if (getenv("LIBTEST_LOCAL"))
89 {
90 test_true(test_is_local());
91 }
92 else
93 {
94 test_false(test_is_local());
95 }
96
97 return TEST_SUCCESS;
98 }
99
100 static test_return_t local_not_test(void *)
101 {
102 return TEST_SKIPPED;
103
104 std::string temp;
105
106 const char *ptr;
107 if ((ptr= getenv("LIBTEST_LOCAL")) == NULL)
108 {
109 temp.append(ptr);
110 }
111
112 // unsetenv() will cause issues with valgrind
113 _compare(__FILE__, __LINE__, __func__, 0, unsetenv("LIBTEST_LOCAL"));
114 test_compare(0, unsetenv("LIBTEST_LOCAL"));
115 test_false(test_is_local());
116
117 test_compare(0, setenv("LIBTEST_LOCAL", "1", 1));
118 test_true(test_is_local());
119
120 if (temp.empty())
121 {
122 test_compare(0, unsetenv("LIBTEST_LOCAL"));
123 }
124 else
125 {
126 char *old_string= strdup(temp.c_str());
127 test_compare(0, setenv("LIBTEST_LOCAL", old_string, 1));
128 }
129
130 return TEST_SUCCESS;
131 }
132
133 static test_return_t var_exists_test(void *)
134 {
135 test_compare(0, access("var", R_OK | W_OK | X_OK));
136 return TEST_SUCCESS;
137 }
138
139 static test_return_t var_tmp_exists_test(void *)
140 {
141 test_compare(0, access("var/tmp", R_OK | W_OK | X_OK));
142 return TEST_SUCCESS;
143 }
144
145 static test_return_t var_run_exists_test(void *)
146 {
147 test_compare(0, access("var/run", R_OK | W_OK | X_OK));
148 return TEST_SUCCESS;
149 }
150
151 static test_return_t var_log_exists_test(void *)
152 {
153 test_compare(0, access("var/log", R_OK | W_OK | X_OK));
154 return TEST_SUCCESS;
155 }
156
157 static test_return_t var_tmp_test(void *)
158 {
159 FILE *file= fopen("var/tmp/junk", "w+");
160 char buffer[1024];
161 const char *dir= getcwd(buffer, sizeof(buffer));
162 test_true_got(file, dir);
163 fclose(file);
164 return TEST_SUCCESS;
165 }
166
167 static test_return_t var_run_test(void *)
168 {
169 FILE *file= fopen("var/run/junk", "w+");
170 test_true(file);
171 fclose(file);
172 return TEST_SUCCESS;
173 }
174
175 static test_return_t var_log_test(void *)
176 {
177 FILE *file= fopen("var/log/junk", "w+");
178 test_true(file);
179 fclose(file);
180 return TEST_SUCCESS;
181 }
182
183 static test_return_t var_tmp_rm_test(void *)
184 {
185 test_true(unlink("var/tmp/junk") == 0);
186 return TEST_SUCCESS;
187 }
188
189 static test_return_t var_run_rm_test(void *)
190 {
191 test_true(unlink("var/run/junk") == 0);
192 return TEST_SUCCESS;
193 }
194
195 static test_return_t var_log_rm_test(void *)
196 {
197 test_true(unlink("var/log/junk") == 0);
198 return TEST_SUCCESS;
199 }
200
201 static test_return_t _compare_test_return_t_test(void *)
202 {
203 test_compare(TEST_SUCCESS, TEST_SUCCESS);
204
205 return TEST_SUCCESS;
206 }
207
208 static test_return_t _compare_memcached_return_t_test(void *)
209 {
210 test_skip(HAVE_LIBMEMCACHED, true);
211 #if defined(HAVE_LIBMEMCACHED) && HAVE_LIBMEMCACHED
212 test_compare(MEMCACHED_SUCCESS, MEMCACHED_SUCCESS);
213 #endif
214
215 return TEST_SUCCESS;
216 }
217
218 static test_return_t _compare_gearman_return_t_test(void *)
219 {
220 test_skip(HAVE_LIBGEARMAN, true);
221 #if defined(HAVE_LIBGEARMAN) && HAVE_LIBGEARMAN
222 test_compare(GEARMAN_SUCCESS, GEARMAN_SUCCESS);
223 #endif
224
225 return TEST_SUCCESS;
226 }
227
228 static test_return_t gearmand_cycle_test(void *object)
229 {
230 server_startup_st *servers= (server_startup_st*)object;
231 test_true(servers);
232
233 #if defined(HAVE_GEARMAND_BINARY) && HAVE_GEARMAND_BINARY
234 test_true(has_gearmand_binary());
235 #else
236 test_skip(true, has_gearmand_binary());
237 #endif
238
239 test_true(server_startup(*servers, "gearmand", get_free_port(), 0, NULL));
240
241 return TEST_SUCCESS;
242 }
243
244 static test_return_t memcached_light_cycle_TEST(void *object)
245 {
246 server_startup_st *servers= (server_startup_st*)object;
247 test_true(servers);
248
249 test_skip(true, bool(HAVE_MEMCACHED_LIGHT_BINARY));
250
251 test_true(server_startup(*servers, "memcached-light", get_free_port(), 0, NULL));
252
253 return TEST_SUCCESS;
254 }
255
256 static test_return_t memcached_cycle_test(void *object)
257 {
258 server_startup_st *servers= (server_startup_st*)object;
259 test_true(servers);
260
261 if (MEMCACHED_BINARY and HAVE_LIBMEMCACHED)
262 {
263 test_true(has_memcached_binary());
264 test_true(server_startup(*servers, "memcached", get_free_port(), 0, NULL));
265
266 return TEST_SUCCESS;
267 }
268
269 return TEST_SKIPPED;
270 }
271
272 static test_return_t memcached_socket_cycle_test(void *object)
273 {
274 server_startup_st *servers= (server_startup_st*)object;
275 test_true(servers);
276
277 if (MEMCACHED_BINARY)
278 {
279 if (HAVE_LIBMEMCACHED)
280 {
281 test_true(has_memcached_binary());
282 test_true(servers->start_socket_server("memcached", get_free_port(), 0, NULL));
283
284 return TEST_SUCCESS;
285 }
286 }
287
288 return TEST_SKIPPED;
289 }
290
291 static test_return_t memcached_sasl_test(void *object)
292 {
293 server_startup_st *servers= (server_startup_st*)object;
294 test_true(servers);
295
296 if (getenv("TESTS_ENVIRONMENT"))
297 {
298 return TEST_SKIPPED;
299 }
300
301 if (MEMCACHED_SASL_BINARY)
302 {
303 if (HAVE_LIBMEMCACHED)
304 {
305 test_true(has_memcached_sasl_binary());
306 test_true(server_startup(*servers, "memcached-sasl", get_free_port(), 0, NULL));
307
308 return TEST_SUCCESS;
309 }
310 }
311
312 return TEST_SKIPPED;
313 }
314
315 static test_return_t application_true_BINARY(void *)
316 {
317 Application true_app("true");
318
319 test_compare(Application::SUCCESS, true_app.run());
320 test_compare(Application::SUCCESS, true_app.wait());
321
322 return TEST_SUCCESS;
323 }
324
325 static test_return_t application_true_fubar_BINARY(void *)
326 {
327 Application true_app("true");
328
329 const char *args[]= { "--fubar", 0 };
330 test_compare(Application::SUCCESS, true_app.run(args));
331 test_compare(Application::SUCCESS, true_app.wait());
332 test_compare(0, true_app.stdout_result().size());
333
334 return TEST_SUCCESS;
335 }
336
337 static test_return_t application_true_fubar_eq_doh_BINARY(void *)
338 {
339 Application true_app("true");
340
341 const char *args[]= { "--fubar=doh", 0 };
342 test_compare(Application::SUCCESS, true_app.run(args));
343 test_compare(Application::SUCCESS, true_app.wait());
344 test_compare(0, true_app.stdout_result().size());
345
346 return TEST_SUCCESS;
347 }
348
349 static test_return_t application_true_fubar_eq_doh_option_BINARY(void *)
350 {
351 Application true_app("true");
352
353 true_app.add_option("--fubar=", "doh");
354
355 test_compare(Application::SUCCESS, true_app.run());
356 test_compare(Application::SUCCESS, true_app.wait());
357 test_compare(0, true_app.stdout_result().size());
358
359 return TEST_SUCCESS;
360 }
361
362
363 static test_return_t GET_TEST(void *)
364 {
365 libtest::http::GET get("http://foo.example.com/");
366
367 test_compare(false, get.execute());
368
369 return TEST_SUCCESS;
370 }
371
372 static test_return_t POST_TEST(void *)
373 {
374 libtest::vchar_t body;
375 libtest::http::POST post("http://foo.example.com/", body);
376
377 test_compare(false, post.execute());
378
379 return TEST_SUCCESS;
380 }
381
382 static test_return_t TRACE_TEST(void *)
383 {
384 libtest::vchar_t body;
385 libtest::http::TRACE trace("http://foo.example.com/", body);
386
387 test_compare(false, trace.execute());
388
389 return TEST_SUCCESS;
390 }
391
392
393 static test_return_t vchar_t_TEST(void *)
394 {
395 libtest::vchar_t response;
396 libtest::make_vector(response, test_literal_param("fubar\n"));
397 test_compare(response, response);
398
399 return TEST_SUCCESS;
400 }
401
402 static test_return_t application_echo_fubar_BINARY(void *)
403 {
404 Application true_app("echo");
405
406 const char *args[]= { "fubar", 0 };
407 test_compare(Application::SUCCESS, true_app.run(args));
408 test_compare(Application::SUCCESS, true_app.wait());
409
410 libtest::vchar_t response;
411 make_vector(response, test_literal_param("fubar\n"));
412 test_compare(response, true_app.stdout_result());
413
414 return TEST_SUCCESS;
415 }
416
417 static test_return_t application_echo_fubar_BINARY2(void *)
418 {
419 Application true_app("echo");
420
421 true_app.add_option("fubar");
422
423 test_compare(Application::SUCCESS, true_app.run());
424 test_compare(Application::SUCCESS, true_app.wait());
425 libtest::vchar_t response;
426 make_vector(response, test_literal_param("fubar\n"));
427 test_compare(response, true_app.stdout_result());
428
429 return TEST_SUCCESS;
430 }
431
432 static test_return_t true_BINARY(void *)
433 {
434 const char *args[]= { 0 };
435 test_compare(EXIT_SUCCESS, exec_cmdline("true", args));
436
437 return TEST_SUCCESS;
438 }
439
440 static test_return_t true_fubar_BINARY(void *)
441 {
442 const char *args[]= { "--fubar", 0 };
443 test_compare(EXIT_SUCCESS, exec_cmdline("true", args));
444
445 return TEST_SUCCESS;
446 }
447
448 static test_return_t echo_fubar_BINARY(void *)
449 {
450 const char *args[]= { "fubar", 0 };
451 test_compare(EXIT_SUCCESS, exec_cmdline("echo", args));
452
453 return TEST_SUCCESS;
454 }
455
456 static test_return_t wait_BINARY(void *)
457 {
458 const char *args[]= { "--quiet", 0 };
459
460 test_compare(EXIT_FAILURE, exec_cmdline("libtest/wait", args, true));
461
462 return TEST_SUCCESS;
463 }
464
465 static test_return_t wait_help_BINARY(void *)
466 {
467 const char *args[]= { "--quiet", "--help", 0 };
468
469 test_compare(EXIT_SUCCESS, exec_cmdline("libtest/wait", args, true));
470
471 return TEST_SUCCESS;
472 }
473
474 static test_return_t wait_version_BINARY(void *)
475 {
476 const char *args[]= { "--quiet", "--version", 0 };
477
478 test_compare(EXIT_SUCCESS, exec_cmdline("libtest/wait", args, true));
479
480 return TEST_SUCCESS;
481 }
482
483 static test_return_t wait_services_BINARY(void *)
484 {
485 test_skip(0, access("/etc/services", R_OK ));
486
487 const char *args[]= { "--quiet", "/etc/services", 0 };
488
489 test_compare(EXIT_SUCCESS, exec_cmdline("libtest/wait", args, true));
490
491 return TEST_SUCCESS;
492 }
493
494 static test_return_t wait_services_BINARY2(void *)
495 {
496 test_skip(0, access("/etc/services", R_OK ));
497
498 const char *args[]= { "/etc/services", 0 };
499
500 test_compare(EXIT_SUCCESS, exec_cmdline("libtest/wait", args, true));
501
502 return TEST_SUCCESS;
503 }
504
505 static test_return_t application_wait_services_BINARY2(void *)
506 {
507 test_skip(0, access("/etc/services", R_OK ));
508
509 libtest::Application("libtest/wait", true);
510 const char *args[]= { "/etc/services", 0 };
511
512 test_compare(EXIT_SUCCESS, exec_cmdline("libtest/wait", args, true));
513
514 return TEST_SUCCESS;
515 }
516
517 static test_return_t get_free_port_TEST(void *)
518 {
519 in_port_t ret_port;
520 test_true_hint((ret_port= get_free_port()), ret_port);
521 test_true(get_free_port() != default_port());
522 test_true(get_free_port() != get_free_port());
523
524 return TEST_SUCCESS;
525 }
526
527 static uint32_t fatal_calls= 0;
528
529 static test_return_t fatal_TEST(void *)
530 {
531 test_compare(fatal_calls++, fatal::disabled_counter());
532 throw libtest::fatal(LIBYATL_DEFAULT_PARAM, "Testing va_args based fatal(): %d", 10);
533
534 return TEST_SUCCESS;
535 }
536
537 static test_return_t number_of_cpus_TEST(void *)
538 {
539 test_true(number_of_cpus() >= 1);
540
541 return TEST_SUCCESS;
542 }
543
544 static test_return_t fatal_message_TEST(void *)
545 {
546 test_compare(fatal_calls++, fatal::disabled_counter());
547 throw fatal_message("Fatal test");
548
549 return TEST_SUCCESS;
550 }
551
552 static test_return_t default_port_TEST(void *)
553 {
554 in_port_t ret_port= default_port();
555 test_compare(ret_port, libtest::default_port());
556 test_compare(ret_port, libtest::default_port());
557
558 return TEST_SUCCESS;
559 }
560
561 static test_return_t check_for_gearman(void *)
562 {
563 test_skip(true, HAVE_LIBGEARMAN);
564 test_skip(true, has_gearmand_binary());
565 return TEST_SUCCESS;
566 }
567
568
569 test_st gearmand_tests[] ={
570 #if 0
571 {"pause", 0, pause_test },
572 #endif
573 {"gearmand startup-shutdown", 0, gearmand_cycle_test },
574 {"_compare(gearman_return_t)", 0, _compare_gearman_return_t_test },
575 {0, 0, 0}
576 };
577
578 static test_return_t check_for_libmemcached(void *)
579 {
580 test_skip(true, HAVE_LIBMEMCACHED);
581 test_skip(true, has_memcached_binary());
582 return TEST_SUCCESS;
583 }
584
585 test_st memcached_tests[] ={
586 {"memcached startup-shutdown", 0, memcached_cycle_test },
587 {"memcached-light startup-shutdown", 0, memcached_light_cycle_TEST },
588 {"memcached(socket file) startup-shutdown", 0, memcached_socket_cycle_test },
589 {"memcached_sasl() startup-shutdown", 0, memcached_sasl_test },
590 {"_compare(memcached_return_t)", 0, _compare_memcached_return_t_test },
591 {0, 0, 0}
592 };
593
594 test_st environment_tests[] ={
595 {"LIBTOOL_COMMAND", 0, LIBTOOL_COMMAND_test },
596 {"VALGRIND_COMMAND", 0, VALGRIND_COMMAND_test },
597 {"HELGRIND_COMMAND", 0, HELGRIND_COMMAND_test },
598 {"GDB_COMMAND", 0, GDB_COMMAND_test },
599 {0, 0, 0}
600 };
601
602 test_st tests_log[] ={
603 {"TEST_SUCCESS", false, test_success_test },
604 {"TEST_FAILURE", false, test_failure_test },
605 {"TEST_SUCCESS == 0", false, test_success_equals_one_test },
606 {0, 0, 0}
607 };
608
609 test_st local_log[] ={
610 {"test_is_local()", 0, local_test },
611 {"test_is_local(NOT)", 0, local_not_test },
612 {0, 0, 0}
613 };
614
615 test_st directories_tests[] ={
616 {"var exists", 0, var_exists_test },
617 {"var/tmp exists", 0, var_tmp_exists_test },
618 {"var/run exists", 0, var_run_exists_test },
619 {"var/log exists", 0, var_log_exists_test },
620 {"var/tmp", 0, var_tmp_test },
621 {"var/run", 0, var_run_test },
622 {"var/log", 0, var_log_test },
623 {"var/tmp rm", 0, var_tmp_rm_test },
624 {"var/run rm", 0, var_run_rm_test },
625 {"var/log rm", 0, var_log_rm_test },
626 {0, 0, 0}
627 };
628
629 test_st comparison_tests[] ={
630 {"_compare(test_return_t)", 0, _compare_test_return_t_test },
631 {0, 0, 0}
632 };
633
634 test_st cmdline_tests[] ={
635 {"true", 0, true_BINARY },
636 {"true --fubar", 0, true_fubar_BINARY },
637 {"echo fubar", 0, echo_fubar_BINARY },
638 {"wait --quiet", 0, wait_BINARY },
639 {"wait --quiet --help", 0, wait_help_BINARY },
640 {"wait --quiet --version", 0, wait_version_BINARY },
641 {"wait --quiet /etc/services", 0, wait_services_BINARY },
642 {"wait /etc/services", 0, wait_services_BINARY2 },
643 {0, 0, 0}
644 };
645
646 test_st get_free_port_TESTS[] ={
647 {"get_free_port()", 0, get_free_port_TEST },
648 {"default_port()", 0, default_port_TEST },
649 {0, 0, 0}
650 };
651
652 test_st fatal_message_TESTS[] ={
653 {"libtest::fatal", 0, fatal_TEST },
654 {"fatal_message()", 0, fatal_message_TEST },
655 {0, 0, 0}
656 };
657
658 test_st number_of_cpus_TESTS[] ={
659 {"libtest::number_of_cpus()", 0, number_of_cpus_TEST },
660 {0, 0, 0}
661 };
662
663 test_st application_tests[] ={
664 {"vchar_t", 0, vchar_t_TEST },
665 {"true", 0, application_true_BINARY },
666 {"true --fubar", 0, application_true_fubar_BINARY },
667 {"true --fubar=doh", 0, application_true_fubar_eq_doh_BINARY },
668 {"true --fubar=doh add_option()", 0, application_true_fubar_eq_doh_option_BINARY },
669 {"echo fubar", 0, application_echo_fubar_BINARY },
670 {"echo fubar (as option)", 0, application_echo_fubar_BINARY2 },
671 {0, 0, 0}
672 };
673
674 static test_return_t check_for_curl(void *)
675 {
676 test_skip(true, HAVE_LIBCURL);
677 return TEST_SUCCESS;
678 }
679
680 static test_return_t disable_fatal_exception(void *)
681 {
682 fatal::disable();
683 return TEST_SUCCESS;
684 }
685
686 static test_return_t enable_fatal_exception(void *)
687 {
688 fatal::disable();
689 return TEST_SUCCESS;
690 }
691
692 test_st http_tests[] ={
693 {"GET", 0, GET_TEST },
694 {"POST", 0, POST_TEST },
695 {"TRACE", 0, TRACE_TEST },
696 {0, 0, 0}
697 };
698
699 collection_st collection[] ={
700 {"environment", 0, 0, environment_tests},
701 {"return values", 0, 0, tests_log},
702 {"local", 0, 0, local_log},
703 {"directories", 0, 0, directories_tests},
704 {"comparison", 0, 0, comparison_tests},
705 {"gearmand", check_for_gearman, 0, gearmand_tests},
706 {"memcached", check_for_libmemcached, 0, memcached_tests},
707 {"cmdline", 0, 0, cmdline_tests},
708 {"application", 0, 0, application_tests},
709 {"http", check_for_curl, 0, http_tests},
710 {"get_free_port()", 0, 0, get_free_port_TESTS },
711 {"fatal", disable_fatal_exception, enable_fatal_exception, fatal_message_TESTS },
712 {"number_of_cpus()", 0, 0, number_of_cpus_TESTS },
713 {0, 0, 0, 0}
714 };
715
716 static void *world_create(server_startup_st& servers, test_return_t&)
717 {
718 return &servers;
719 }
720
721 void get_world(Framework *world)
722 {
723 world->collections= collection;
724 world->_create= world_create;
725 }