1 files changed, 74 insertions, 0 deletions
diff --git a/tests/backend-dbus/mock-login1-manager.h b/tests/backend-dbus/mock-login1-manager.h
new file mode 100644
index 0000000..f630329
--- /dev/null
+++ b/tests/backend-dbus/mock-login1-manager.h
@@ -0,0 +1,74 @@
+/*
+ * 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/>.
+ */
+
+#ifndef MOCK_LOGIN1_MANAGER_H
+#define MOCK_LOGIN1_MANAGER_H
+
+#include <set>
+#include <string>
+#include "mock-object.h"
+#include "backend-dbus/dbus-login1-manager.h"
+
+class MockLogin1Seat;
+class MockUser;
+
+class MockLogin1Manager: public MockObject
+{
+  public:
+
+    MockLogin1Manager (GMainLoop       * loop,
+                           GDBusConnection * bus_connection);
+    virtual ~MockLogin1Manager ();
+
+    int add_session (MockLogin1Seat * seat, MockUser * user);
+    void remove_session (MockLogin1Seat * seat, int session_tag);
+
+    void add_seat (MockLogin1Seat * seat);
+
+    const std::string& can_suspend () const;
+    const std::string& can_hibernate () const;
+
+    const std::string& last_action () const { return my_last_action; }
+    void clear_last_action () { my_last_action.clear(); }
+
+  private:
+
+    void emit_session_new (MockLogin1Seat * seat, int tag) const;
+    void emit_session_removed (MockLogin1Seat * seat, int tag) const;
+
+    GVariant * list_sessions () const;
+
+    static gboolean handle_list_sessions (Login1Manager *, GDBusMethodInvocation *, gpointer);
+    static gboolean handle_can_suspend   (Login1Manager *, GDBusMethodInvocation *, gpointer);
+    static gboolean handle_can_hibernate (Login1Manager *, GDBusMethodInvocation *, gpointer);
+    static gboolean handle_reboot        (Login1Manager *, GDBusMethodInvocation *, gboolean, gpointer);
+    static gboolean handle_power_off     (Login1Manager *, GDBusMethodInvocation *, gboolean, gpointer);
+    static gboolean handle_suspend       (Login1Manager *, GDBusMethodInvocation *, gboolean, gpointer);
+    static gboolean handle_hibernate     (Login1Manager *, GDBusMethodInvocation *, gboolean, gpointer);
+
+  private:
+
+    Login1Manager * my_skeleton;
+    std::set<MockLogin1Seat*> my_seats;
+    std::string my_can_suspend;
+    std::string my_can_hibernate;
+    std::string my_last_action;
+};
+
+#endif // #ifndef MOCK_LOGIN1_MANAGER_H