Imported Upstream version 2.6.1

[debian/amanda] / common-src / testutils.c

diff --git a/common-src/testutils.c b/common-src/testutils.c
new file mode 100644 (file)
index 0000000..a614b7c
--- /dev/null
+++ b/common-src/testutils.c
@@ -0,0 +1,153 @@
+/*
+ * Copyright (c) 2005-2008 Zmanda Inc.  All Rights Reserved.
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 as published
+ * by the Free Software Foundation.
+ *
+ * This program 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 General Public License
+ * for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+ *
+ * Contact information: Zmanda Inc., 465 S Mathlida Ave, Suite 300
+ * Sunnyvale, CA 94086, USA, or: http://www.zmanda.com
+ */
+
+#include "amanda.h"
+#include "testutils.h"
+
+int tu_debugging_enabled = FALSE;
+
+static void
+alarm_hdlr(int sig G_GNUC_UNUSED)
+{
+    fprintf(stderr, "-- TEST TIMED OUT --\n");
+    exit(1);
+}
+
+/* Call testfn in a forked process, such that any failures will trigger a
+ * test failure, but allow the other tests to proceed.
+ */
+static int
+callinfork(TestUtilsTest *test, int ignore_timeouts)
+{
+    pid_t pid;
+    int success;
+    amwait_t status;
+
+    switch (pid = fork()) {
+       case 0: /* child */
+           /* kill the test after a bit */
+           signal(SIGALRM, alarm_hdlr);
+           if (!ignore_timeouts) alarm(test->timeout);
+
+           success = test->fn();
+           exit(success? 0:1);
+
+       case -1:
+           perror("fork");
+           exit(1);
+
+       default: /* parent */
+           waitpid(pid, &status, 0);
+           if (status == 0) {
+               fprintf(stderr, " PASS %s\n", test->name);
+           } else {
+               fprintf(stderr, " FAIL %s\n", test->name);
+           }
+           return status == 0;
+    }
+}
+
+static void
+usage(
+    TestUtilsTest *tests)
+{
+    printf("USAGE: <test-script> [-d] [-h] [testname [testname [..]]]\n"
+       "\n"
+       "\t-h: this message\n"
+       "\t-d: print debugging messages\n"
+       "\t-t: ignore timeouts\n"
+       "\n"
+       "If no test names are specified, all tests are run.  Available tests:\n"
+       "\n");
+    while (tests->fn) {
+       printf("\t%s\n", tests->name);
+       tests++;
+    }
+}
+
+static void
+ignore_debug_messages(
+           const gchar *log_domain G_GNUC_UNUSED,
+           GLogLevelFlags log_level G_GNUC_UNUSED,
+           const gchar *message G_GNUC_UNUSED,
+           gpointer user_data G_GNUC_UNUSED)
+{
+}
+
+int
+testutils_run_tests(
+    int argc,
+    char **argv,
+    TestUtilsTest *tests)
+{
+    TestUtilsTest *t;
+    int run_all = 1;
+    int success;
+    int ignore_timeouts = 0;
+
+    /* first_parse the command line */
+    while (argc > 1) {
+       if (strcmp(argv[1], "-d") == 0) {
+           tu_debugging_enabled = TRUE;
+       } else if (strcmp(argv[1], "-t") == 0) {
+           ignore_timeouts = TRUE;
+       } else if (strcmp(argv[1], "-h") == 0) {
+           usage(tests);
+           return 1;
+       } else {
+           int found = 0;
+
+           for (t = tests; t->fn; t++) {
+               if (strcmp(argv[1], t->name) == 0) {
+                   found = 1;
+                   t->selected = 1;
+                   break;
+               }
+           }
+
+           if (!found) {
+               fprintf(stderr, "Test '%s' not found\n", argv[1]);
+               return 1;
+           }
+
+           run_all = 0;
+       }
+
+       argc--; argv++;
+    }
+
+    /* Make sure g_critical and g_error will exit */
+    g_log_set_always_fatal(G_LOG_LEVEL_ERROR |  G_LOG_LEVEL_CRITICAL);
+
+    /* and silently drop debug messages unless we're debugging */
+    if (!tu_debugging_enabled) {
+       g_log_set_handler(NULL, G_LOG_LEVEL_DEBUG, ignore_debug_messages, NULL);
+    }
+
+    /* Now actually run the tests */
+    success = 1;
+    for (t = tests; t->fn; t++) {
+       if (run_all || t->selected) {
+           success = callinfork(t, ignore_timeouts) && success;
+       }
+    }
+
+    return success? 0:1;
+}