1 files changed, 156 insertions, 0 deletions
diff --git a/tests/backend-dbus/mock-consolekit-manager.cc b/tests/backend-dbus/mock-consolekit-manager.cc
new file mode 100644
index 0000000..40f9bf9
--- /dev/null
+++ b/tests/backend-dbus/mock-consolekit-manager.cc
@@ -0,0 +1,156 @@
+/*
+ * Copyright 2013 Canonical Ltd.
+ *
+ * Authors:
+ *   Charles Kerr <charles.kerr@canonical.com>
+ *
+ * This program is free software: you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 3, 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 warranties of
+ * MERCHANTABILITY, SATISFACTORY QUALITY, 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, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "mock-consolekit-manager.h"
+#include "mock-consolekit-seat.h"
+#include "mock-consolekit-session.h"
+
+namespace
+{
+  const char * CONSOLEKIT_MANAGER_NAME = "org.freedesktop.ConsoleKit";
+  
+  const char * CONSOLEKIT_MANAGER_PATH = "/org/freedesktop/ConsoleKit/Manager";
+
+  void on_active_session_changed (ConsoleKitSeat  * o          G_GNUC_UNUSED,
+                                  const gchar     * new_ssid,
+                                  gpointer          ssid)
+  {
+    *static_cast<std::string*>(ssid) = new_ssid;
+  }
+}
+
+/***
+****
+***/
+
+gboolean
+MockConsoleKitManager :: on_get_current_session (ConsoleKitManager     * m,
+                                                 GDBusMethodInvocation * inv,
+                                                 gpointer                gself)
+{
+  MockConsoleKitManager * self = static_cast<MockConsoleKitManager*>(gself);
+  const std::string& ssid = self->my_current_ssid;
+  console_kit_manager_complete_get_current_session (m, inv, ssid.c_str());
+  return true;
+}
+
+gboolean
+MockConsoleKitManager :: on_get_seats (ConsoleKitManager     * m,
+                                       GDBusMethodInvocation * inv,
+                                       gpointer                gself)
+{
+  int i;
+  char ** sids;
+  const seats_t& seats = static_cast<MockConsoleKitManager*>(gself)->my_seats;
+
+  i = 0;
+  sids = g_new0 (char*, seats.size()+1);
+  for (seats_t::const_iterator it(seats.begin()),
+                              end(seats.end()); it!=end; ++it)
+    sids[i++] = (char*) (*it)->path();
+  console_kit_manager_complete_get_seats (m, inv, sids);
+  g_strfreev (sids);
+
+  return true;
+}
+
+gboolean
+MockConsoleKitManager :: handle_restart (ConsoleKitManager     * ckm,
+                                         GDBusMethodInvocation * inv,
+                                         gpointer                gself)
+{
+  static_cast<MockConsoleKitManager*>(gself)->my_last_action = Restart;
+  console_kit_manager_complete_restart (ckm, inv);
+  return true;
+}
+
+gboolean
+MockConsoleKitManager :: handle_stop (ConsoleKitManager     * ckm,
+                                      GDBusMethodInvocation * inv,
+                                      gpointer                gself)
+{
+  static_cast<MockConsoleKitManager*>(gself)->my_last_action = Shutdown;
+  console_kit_manager_complete_stop (ckm, inv);
+  return true;
+}
+
+/***
+****
+***/
+
+MockConsoleKitSession *
+MockConsoleKitManager :: current_session ()
+{
+  MockConsoleKitSession * ret = 0;
+
+  for (seats_t::iterator it(my_seats.begin()),
+                        end(my_seats.end()); it!=end; ++it)
+    if ((ret = (*it)->find (my_current_ssid.c_str())))
+      break;
+
+  return ret;
+}
+
+void
+MockConsoleKitManager :: add_seat (MockConsoleKitSeat * seat)
+{
+  g_assert (my_seats.count(seat) == 0);
+
+  my_seats.insert (seat);
+
+  console_kit_manager_emit_seat_added (my_skeleton, seat->sid());
+
+  g_signal_connect (seat->skeleton(), "active-session-changed",
+                    G_CALLBACK(on_active_session_changed), &my_current_ssid);
+}
+
+/***
+****
+***/
+
+MockConsoleKitManager :: MockConsoleKitManager (GMainLoop       * loop,
+                                                GDBusConnection * conn):
+  MockObject (loop, conn, CONSOLEKIT_MANAGER_NAME, CONSOLEKIT_MANAGER_PATH),
+  my_skeleton (console_kit_manager_skeleton_new ()),
+  my_last_action (None)
+{
+  g_signal_connect (my_skeleton, "handle-get-current-session",
+                    G_CALLBACK(on_get_current_session), this);
+  g_signal_connect (my_skeleton, "handle-get-seats",
+                    G_CALLBACK(on_get_seats), this);
+  g_signal_connect (my_skeleton, "handle-stop",
+                    G_CALLBACK(handle_stop), this);
+  g_signal_connect (my_skeleton, "handle-restart",
+                    G_CALLBACK(handle_restart), this);
+
+  set_skeleton (G_DBUS_INTERFACE_SKELETON(my_skeleton));
+}
+
+MockConsoleKitManager :: ~MockConsoleKitManager ()
+{
+  for (seats_t::iterator it(my_seats.begin()); it!=my_seats.end(); ++it)
+    {
+      MockConsoleKitSeat * seat = *it;
+      g_signal_handlers_disconnect_by_data (seat->skeleton(), &my_current_ssid);
+      delete seat;
+    }
+
+  g_signal_handlers_disconnect_by_data (my_skeleton, this);
+  g_clear_object (&my_skeleton);
+}