test.c 4.91 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
/* -*- mode: C; c-file-style: "gnu" -*- */
/* test.c  unit test routines
 *
 * Copyright (C) 2003 Red Hat, Inc.
 *
 * Licensed under the Academic Free License version 1.2
 * 
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * 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
 *
 */

24 25 26 27 28
#include <config.h>

#ifdef DBUS_BUILD_TESTS
#include "test.h"
#include "loop.h"
29
#include <dbus/dbus-internals.h>
30 31 32 33 34 35 36

/* The "debug client" watch/timeout handlers don't dispatch messages,
 * as we manually pull them in order to verify them. This is why they
 * are different from the real handlers in connection.c
 */

static void
37
client_watch_callback (DBusWatch     *watch,
38 39 40 41 42 43 44 45 46 47 48 49 50
                           unsigned int   condition,
                           void          *data)
{
  DBusConnection *connection = data;

  dbus_connection_ref (connection);
  
  dbus_connection_handle_watch (connection, watch, condition);

  dbus_connection_unref (connection);
}

static dbus_bool_t
51
add_client_watch (DBusWatch      *watch,
52 53
                      DBusConnection *connection)
{
54
  return bus_loop_add_watch (watch, client_watch_callback, connection,
55 56 57 58
                             NULL);
}

static void
59
remove_client_watch (DBusWatch      *watch,
60 61
                         DBusConnection *connection)
{
62
  bus_loop_remove_watch (watch, client_watch_callback, connection);
63 64 65
}

static void
66
client_timeout_callback (DBusTimeout   *timeout,
67 68 69 70 71 72 73 74 75 76 77 78
                             void          *data)
{
  DBusConnection *connection = data;

  dbus_connection_ref (connection);
  
  dbus_timeout_handle (timeout);

  dbus_connection_unref (connection);
}

static dbus_bool_t
79
add_client_timeout (DBusTimeout    *timeout,
80 81
                        DBusConnection *connection)
{
82
  return bus_loop_add_timeout (timeout, client_timeout_callback, connection, NULL);
83 84 85
}

static void
86
remove_client_timeout (DBusTimeout    *timeout,
87 88
                           DBusConnection *connection)
{
89
  bus_loop_remove_timeout (timeout, client_timeout_callback, connection);
90 91
}

92 93 94 95 96 97 98 99 100 101
static DBusHandlerResult
client_disconnect_handler (DBusMessageHandler *handler,
                           DBusConnection     *connection,
                           DBusMessage        *message,
                           void               *user_data)
{
  dbus_connection_unref (connection);
  
  return DBUS_HANDLER_RESULT_ALLOW_MORE_HANDLERS;
}
102 103 104 105

dbus_bool_t
bus_setup_debug_client (DBusConnection *connection)
{
106 107 108
  DBusMessageHandler *disconnect_handler;
  const char *to_handle[] = { DBUS_MESSAGE_LOCAL_DISCONNECT };
  dbus_bool_t retval;
109
  
110 111 112 113 114 115 116 117 118 119
  disconnect_handler = dbus_message_handler_new (client_disconnect_handler,
                                                 NULL, NULL);

  if (disconnect_handler == NULL)
    return FALSE;

  if (!dbus_connection_register_handler (connection,
                                         disconnect_handler,
                                         to_handle,
                                         _DBUS_N_ELEMENTS (to_handle)))
120
    {
121
      dbus_message_handler_unref (disconnect_handler);
122 123
      return FALSE;
    }
124 125

  retval = FALSE;
126
  
127 128 129 130 131 132 133
  if (!dbus_connection_set_watch_functions (connection,
                                            (DBusAddWatchFunction) add_client_watch,
                                            (DBusRemoveWatchFunction) remove_client_watch,
                                            connection,
                                            NULL))
    goto out;
      
134
  if (!dbus_connection_set_timeout_functions (connection,
135 136
                                              (DBusAddTimeoutFunction) add_client_timeout,
                                              (DBusRemoveTimeoutFunction) remove_client_timeout,
137
                                              connection, NULL))
138 139 140 141 142 143
    goto out;

  retval = TRUE;
  
 out:
  if (!retval)
144
    {
145 146 147 148 149 150 151 152 153
      dbus_connection_unregister_handler (connection,
                                          disconnect_handler,
                                          to_handle,
                                          _DBUS_N_ELEMENTS (to_handle));
      
      dbus_connection_set_watch_functions (connection,
                                           NULL, NULL, NULL, NULL);
      dbus_connection_set_timeout_functions (connection,
                                             NULL, NULL, NULL, NULL);
154
    }
155 156 157 158
  
  dbus_message_handler_unref (disconnect_handler);
  
  return retval;
159 160
}
#endif