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