bus/test-main.c

   1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
   2 /* test-main.c  main() for make check
   3  *
   4  * Copyright (C) 2003 Red Hat, Inc.
   5  *
   6  * Licensed under the Academic Free License version 2.1
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  21  *
  22  */
  23
  24 #include <config.h>
  25 #include "test.h"
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <dbus/dbus-string.h>
  29 #include <dbus/dbus-sysdeps.h>
  30 #include <dbus/dbus-internals.h>
  31 #include <dbus/dbus-message-private.h>
  32 #include "selinux.h"
  33
  34 #ifdef DBUS_BUILD_TESTS
  35 static void
  36 die (const char *failure)
  37 {
  38   fprintf (stderr, "Unit test failed: %s\n", failure);
  39   exit (1);
  40 }
  41
  42 static void
  43 check_memleaks (const char *name)
  44 {
  45   dbus_shutdown ();
  46
  47   printf ("%s: checking for memleaks\n", name);
  48   if (_dbus_get_malloc_blocks_outstanding () != 0)
  49     {
  50       _dbus_warn ("%d dbus_malloc blocks were not freed\n",
  51                   _dbus_get_malloc_blocks_outstanding ());
  52       die ("memleaks");
  53     }
  54 }
  55 #endif /* DBUS_BUILD_TESTS */
  56
  57 static void
  58 test_pre_hook (void)
  59 {
  60
  61   if (_dbus_getenv ("DBUS_TEST_SELINUX")
  62       && (!bus_selinux_pre_init ()
  63           || !bus_selinux_full_init ()))
  64     die ("could not init selinux support");
  65 }
  66
  67 static char *progname = "";
  68 static void
  69 test_post_hook (void)
  70 {
  71   if (_dbus_getenv ("DBUS_TEST_SELINUX"))
  72     bus_selinux_shutdown ();
  73   check_memleaks (progname);
  74   _dbus_check_fdleaks();
  75 }
  76
  77 int
  78 main (int argc, char **argv)
  79 {
  80 #ifdef DBUS_BUILD_TESTS
  81   const char *dir;
  82   const char *only;
  83   DBusString test_data_dir;
  84
  85   progname = argv[0];
  86
  87   if (argc > 1)
  88     dir = argv[1];
  89   else
  90     dir = _dbus_getenv ("DBUS_TEST_DATA");
  91
  92   if (argc > 2)
  93     only = argv[2];
  94   else
  95     only = NULL;
  96
  97   if (dir == NULL)
  98     {
  99       fprintf (stderr, "Must specify test data directory as argv[1] or in DBUS_TEST_DATA env variable\n");
 100       return 1;
 101     }
 102
 103   _dbus_string_init_const (&test_data_dir, dir);
 104
 105   if (!_dbus_threads_init_debug ())
 106     die ("initializing debug threads");
 107
 108   if (only == NULL || strcmp (only, "expire-list") == 0)
 109     {
 110       test_pre_hook ();
 111       printf ("%s: Running expire list test\n", argv[0]);
 112       if (!bus_expire_list_test (&test_data_dir))
 113         die ("expire list");
 114       test_post_hook ();
 115     }
 116
 117   if (only == NULL || strcmp (only, "config-parser") == 0)
 118     {
 119       test_pre_hook ();
 120       printf ("%s: Running config file parser test\n", argv[0]);
 121       if (!bus_config_parser_test (&test_data_dir))
 122         die ("parser");
 123       test_post_hook ();
 124     }
 125
 126   if (only == NULL || strcmp (only, "policy") == 0)
 127     {
 128       test_pre_hook ();
 129       printf ("%s: Running policy test\n", argv[0]);
 130       if (!bus_policy_test (&test_data_dir))
 131         die ("policy");
 132       test_post_hook ();
 133     }
 134
 135   if (only == NULL || strcmp (only, "signals") == 0)
 136     {
 137       test_pre_hook ();
 138       printf ("%s: Running signals test\n", argv[0]);
 139       if (!bus_signals_test (&test_data_dir))
 140         die ("signals");
 141       test_post_hook ();
 142     }
 143
 144   if (only == NULL || strcmp (only, "dispatch-sha1") == 0)
 145     {
 146       test_pre_hook ();
 147       printf ("%s: Running SHA1 connection test\n", argv[0]);
 148       if (!bus_dispatch_sha1_test (&test_data_dir))
 149         die ("sha1");
 150       test_post_hook ();
 151     }
 152
 153   if (only == NULL || strcmp (only, "dispatch") == 0)
 154     {
 155       test_pre_hook ();
 156       printf ("%s: Running message dispatch test\n", argv[0]);
 157       if (!bus_dispatch_test (&test_data_dir))
 158         die ("dispatch");
 159       test_post_hook ();
 160     }
 161
 162   if (only == NULL || strcmp (only, "activation-service-reload") == 0)
 163     {
 164       test_pre_hook ();
 165       printf ("%s: Running service files reloading test\n", argv[0]);
 166       if (!bus_activation_service_reload_test (&test_data_dir))
 167         die ("service reload");
 168       test_post_hook ();
 169     }
 170
 171 #ifdef HAVE_UNIX_FD_PASSING
 172   if (only == NULL || strcmp (only, "unix-fds-passing") == 0)
 173     {
 174       test_pre_hook ();
 175       printf ("%s: Running unix fd passing test\n", argv[0]);
 176       if (!bus_unix_fds_passing_test (&test_data_dir))
 177         die ("unix fd passing");
 178       test_post_hook ();
 179     }
 180 #endif
 181
 182   printf ("%s: Success\n", argv[0]);
 183
 184
 185   return 0;
 186 #else /* DBUS_BUILD_TESTS */
 187
 188   printf ("Not compiled with test support\n");
 189
 190   return 0;
 191 #endif
 192 }