TMP: add a ctdb snapshot of current ctdb master (git://git.samba.org/ctdb.git) to...

[obnox/samba/samba-obnox.git] / ctdb / lib / tevent / testsuite.c

diff --git a/ctdb/lib/tevent/testsuite.c b/ctdb/lib/tevent/testsuite.c
new file mode 100644 (file)
index 0000000..3a763de
--- /dev/null
+++ b/ctdb/lib/tevent/testsuite.c
@@ -0,0 +1,160 @@
+/*
+   Unix SMB/CIFS implementation.
+
+   testing of the events subsystem
+
+   Copyright (C) Stefan Metzmacher 2006-2009
+
+     ** NOTE! The following LGPL license applies to the tevent
+     ** library. This does NOT imply that all of Samba is released
+     ** under the LGPL
+
+   This library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Lesser General Public
+   License as published by the Free Software Foundation; either
+   version 3 of the License, or (at your option) any later version.
+
+   This library is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   Lesser General Public License for more details.
+
+   You should have received a copy of the GNU Lesser General Public
+   License along with this library; if not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "lib/events/events.h"
+#include "system/filesys.h"
+#include "torture/torture.h"
+
+static int fde_count;
+
+static void fde_handler(struct tevent_context *ev_ctx, struct tevent_fd *f,
+                       uint16_t flags, void *private_data)
+{
+       int *fd = (int *)private_data;
+       char c;
+#ifdef SA_SIGINFO
+       kill(getpid(), SIGUSR1);
+#endif
+       kill(getpid(), SIGALRM);
+       read(fd[0], &c, 1);
+       write(fd[1], &c, 1);
+       fde_count++;
+}
+
+static void finished_handler(struct tevent_context *ev_ctx, struct tevent_timer *te,
+                            struct timeval tval, void *private_data)
+{
+       int *finished = (int *)private_data;
+       (*finished) = 1;
+}
+
+static void count_handler(struct tevent_context *ev_ctx, struct signal_event *te,
+                         int signum, int count, void *info, void *private_data)
+{
+       int *countp = (int *)private_data;
+       (*countp) += count;
+}
+
+static bool test_event_context(struct torture_context *test,
+                              const void *test_data)
+{
+       struct tevent_context *ev_ctx;
+       int fd[2] = { -1, -1 };
+       const char *backend = (const char *)test_data;
+       int alarm_count=0, info_count=0;
+       struct tevent_fd *fde;
+#ifdef SA_RESTART
+       struct tevent_signal *se1 = NULL;
+#endif
+       struct tevent_signal *se2 = NULL;
+#ifdef SA_SIGINFO
+       struct tevent_signal *se3 = NULL;
+#endif
+       int finished=0;
+       struct timeval t;
+       char c = 0;
+
+       ev_ctx = event_context_init_byname(test, backend);
+       if (ev_ctx == NULL) {
+               torture_comment(test, "event backend '%s' not supported\n", backend);
+               return true;
+       }
+
+       torture_comment(test, "Testing event backend '%s'\n", backend);
+
+       /* reset globals */
+       fde_count = 0;
+
+       /* create a pipe */
+       pipe(fd);
+
+       fde = event_add_fd(ev_ctx, ev_ctx, fd[0], EVENT_FD_READ,
+                          fde_handler, fd);
+       tevent_fd_set_auto_close(fde);
+
+       event_add_timed(ev_ctx, ev_ctx, timeval_current_ofs(2,0),
+                       finished_handler, &finished);
+
+#ifdef SA_RESTART
+       se1 = event_add_signal(ev_ctx, ev_ctx, SIGALRM, SA_RESTART, count_handler, &alarm_count);
+#endif
+       se2 = event_add_signal(ev_ctx, ev_ctx, SIGALRM, SA_RESETHAND, count_handler, &alarm_count);
+#ifdef SA_SIGINFO
+       se3 = event_add_signal(ev_ctx, ev_ctx, SIGUSR1, SA_SIGINFO, count_handler, &info_count);
+#endif
+
+       write(fd[1], &c, 1);
+
+       t = timeval_current();
+       while (!finished) {
+               errno = 0;
+               if (event_loop_once(ev_ctx) == -1) {
+                       talloc_free(ev_ctx);
+                       torture_fail(test, talloc_asprintf(test, "Failed event loop %s\n", strerror(errno)));
+               }
+       }
+
+       talloc_free(fde);
+       close(fd[1]);
+
+       while (alarm_count < fde_count+1) {
+               if (event_loop_once(ev_ctx) == -1) {
+                       break;
+               }
+       }
+
+       torture_comment(test, "Got %.2f pipe events/sec\n", fde_count/timeval_elapsed(&t));
+
+#ifdef SA_RESTART
+       talloc_free(se1);
+#endif
+
+       torture_assert_int_equal(test, alarm_count, 1+fde_count, "alarm count mismatch");
+
+#ifdef SA_SIGINFO
+       talloc_free(se3);
+       torture_assert_int_equal(test, info_count, fde_count, "info count mismatch");
+#endif
+
+       talloc_free(ev_ctx);
+
+       return true;
+}
+
+struct torture_suite *torture_local_event(TALLOC_CTX *mem_ctx)
+{
+       struct torture_suite *suite = torture_suite_create(mem_ctx, "EVENT");
+       const char **list = event_backend_list(suite);
+       int i;
+
+       for (i=0;list && list[i];i++) {
+               torture_suite_add_simple_tcase_const(suite, list[i],
+                                              test_event_context,
+                                              (const void *)list[i]);
+       }
+
+       return suite;
+}