ash/test/test_session_state_delegate.h

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef ASH_TEST_TEST_SESSION_STATE_DELEGATE_H_
   6 #define ASH_TEST_TEST_SESSION_STATE_DELEGATE_H_
   7
   8 #include <vector>
   9
  10 #include "ash/session/session_state_delegate.h"
  11 #include "base/basictypes.h"
  12 #include "base/compiler_specific.h"
  13 #include "ui/gfx/image/image_skia.h"
  14
  15 namespace ash {
  16 namespace test {
  17
  18 class MockUserInfo;
  19
  20 class TestSessionStateDelegate : public SessionStateDelegate {
  21  public:
  22   TestSessionStateDelegate();
  23   virtual ~TestSessionStateDelegate();
  24
  25   void set_logged_in_users(int users) { logged_in_users_ = users; }
  26   void AddUser(const std::string user_id);
  27   const user_manager::UserInfo* GetActiveUserInfo() const;
  28
  29   // SessionStateDelegate:
  30   virtual content::BrowserContext* GetBrowserContextByIndex(
  31       MultiProfileIndex index) OVERRIDE;
  32   virtual content::BrowserContext* GetBrowserContextForWindow(
  33       aura::Window* window) OVERRIDE;
  34   virtual int GetMaximumNumberOfLoggedInUsers() const OVERRIDE;
  35   virtual int NumberOfLoggedInUsers() const OVERRIDE;
  36   virtual bool IsActiveUserSessionStarted() const OVERRIDE;
  37   virtual bool CanLockScreen() const OVERRIDE;
  38   virtual bool IsScreenLocked() const OVERRIDE;
  39   virtual bool ShouldLockScreenBeforeSuspending() const OVERRIDE;
  40   virtual void LockScreen() OVERRIDE;
  41   virtual void UnlockScreen() OVERRIDE;
  42   virtual bool IsUserSessionBlocked() const OVERRIDE;
  43   virtual SessionState GetSessionState() const OVERRIDE;
  44   virtual const user_manager::UserInfo* GetUserInfo(
  45       ash::MultiProfileIndex index) const OVERRIDE;
  46   virtual const user_manager::UserInfo* GetUserInfo(
  47       content::BrowserContext* context) const OVERRIDE;
  48   virtual bool ShouldShowAvatar(aura::Window* window) const OVERRIDE;
  49   virtual void SwitchActiveUser(const std::string& user_id) OVERRIDE;
  50   virtual void CycleActiveUser(CycleUser cycle_user) OVERRIDE;
  51   virtual bool IsMultiProfileAllowedByPrimaryUserPolicy() const OVERRIDE;
  52   virtual void AddSessionStateObserver(
  53       ash::SessionStateObserver* observer) OVERRIDE;
  54   virtual void RemoveSessionStateObserver(
  55       ash::SessionStateObserver* observer) OVERRIDE;
  56
  57   // TODO(oshima): Use state machine instead of using boolean variables.
  58
  59   // Updates the internal state that indicates whether a session is in progress
  60   // and there is an active user. If |has_active_user| is |false|,
  61   // |active_user_session_started_| is reset to |false| as well (see below for
  62   // the difference between these two flags).
  63   void SetHasActiveUser(bool has_active_user);
  64
  65   // Updates the internal state that indicates whether the session has been
  66   // fully started for the active user. If |active_user_session_started| is
  67   // |true|, |has_active_user_| is set to |true| as well (see below for the
  68   // difference between these two flags).
  69   void SetActiveUserSessionStarted(bool active_user_session_started);
  70
  71   // Updates the internal state that indicates whether the screen can be locked.
  72   // Locking will only actually be allowed when this value is |true| and there
  73   // is an active user.
  74   void SetCanLockScreen(bool can_lock_screen);
  75
  76   // Updates |should_lock_screen_before_suspending_|.
  77   void SetShouldLockScreenBeforeSuspending(bool should_lock);
  78
  79   // Updates the internal state that indicates whether user adding screen is
  80   // running now.
  81   void SetUserAddingScreenRunning(bool user_adding_screen_running);
  82
  83   // Setting non NULL image enables avatar icon.
  84   void SetUserImage(const gfx::ImageSkia& user_image);
  85
  86  private:
  87   // Whether a session is in progress and there is an active user.
  88   bool has_active_user_;
  89
  90   // When a user becomes active, the profile and browser UI are not immediately
  91   // available. Only once this flag becomes |true| is the browser startup
  92   // complete and both profile and UI are fully available.
  93   bool active_user_session_started_;
  94
  95   // Whether the screen can be locked. Locking will only actually be allowed
  96   // when this is |true| and there is an active user.
  97   bool can_lock_screen_;
  98
  99   // Return value for ShouldLockScreenBeforeSuspending().
 100   bool should_lock_screen_before_suspending_;
 101
 102   // Whether the screen is currently locked.
 103   bool screen_locked_;
 104
 105   // Whether user addding screen is running now.
 106   bool user_adding_screen_running_;
 107
 108   // The number of users logged in.
 109   int logged_in_users_;
 110
 111   // The index for the activated user.
 112   int active_user_index_;
 113
 114   std::vector<MockUserInfo*> user_list_;
 115
 116   DISALLOW_COPY_AND_ASSIGN(TestSessionStateDelegate);
 117 };
 118
 119 }  // namespace test
 120 }  // namespace ash
 121
 122 #endif  // ASH_TEST_TEST_SESSION_STATE_DELEGATE_H_