1 files changed, 4 insertions, 0 deletions
diff --git a/tests/backend-dbus/mock-unity-session.h b/tests/backend-dbus/mock-unity-session.h
index ded246a..9ca5fe9 100644
--- a/tests/backend-dbus/mock-unity-session.h
+++ b/tests/backend-dbus/mock-unity-session.h
@@ -35,6 +35,7 @@ class MockUnitySession: public MockObject
 
     enum Action { None, Lock, PromptLock, RequestLogout, RequestShutdown, RequestReboot };
     Action last_action () { return my_last_action; }
+    void clear_last_action () { my_last_action = None; }
 
   private:
 
@@ -47,6 +48,9 @@ class MockUnitySession: public MockObject
     static gboolean handle_prompt_lock (UnitySession *,
                                         GDBusMethodInvocation *,
                                         gpointer);
+    static gboolean handle_request_logout (UnitySession *,
+                                           GDBusMethodInvocation *,
+                                           gpointer);
 
 };