test_session_state_delegate.h revision 5c02ac1a9c1b504631c0a3d2b6e737b5d738bae1 
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
15namespace ash {
16namespace test {
17
18class MockUserInfo;
19
20class 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 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 UserInfo* GetUserInfo(
45      ash::MultiProfileIndex index) const OVERRIDE;
46  virtual const 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 void AddSessionStateObserver(
52      ash::SessionStateObserver* observer) OVERRIDE;
53  virtual void RemoveSessionStateObserver(
54      ash::SessionStateObserver* observer) OVERRIDE;
55
56  // TODO(oshima): Use state machine instead of using boolean variables.
57
58  // Updates the internal state that indicates whether a session is in progress
59  // and there is an active user. If |has_active_user| is |false|,
60  // |active_user_session_started_| is reset to |false| as well (see below for
61  // the difference between these two flags).
62  void SetHasActiveUser(bool has_active_user);
63
64  // Updates the internal state that indicates whether the session has been
65  // fully started for the active user. If |active_user_session_started| is
66  // |true|, |has_active_user_| is set to |true| as well (see below for the
67  // difference between these two flags).
68  void SetActiveUserSessionStarted(bool active_user_session_started);
69
70  // Updates the internal state that indicates whether the screen can be locked.
71  // Locking will only actually be allowed when this value is |true| and there
72  // is an active user.
73  void SetCanLockScreen(bool can_lock_screen);
74
75  // Updates |should_lock_screen_before_suspending_|.
76  void SetShouldLockScreenBeforeSuspending(bool should_lock);
77
78  // Updates the internal state that indicates whether user adding screen is
79  // running now.
80  void SetUserAddingScreenRunning(bool user_adding_screen_running);
81
82  // Setting non NULL image enables avatar icon.
83  void SetUserImage(const gfx::ImageSkia& user_image);
84
85 private:
86  // Whether a session is in progress and there is an active user.
87  bool has_active_user_;
88
89  // When a user becomes active, the profile and browser UI are not immediately
90  // available. Only once this flag becomes |true| is the browser startup
91  // complete and both profile and UI are fully available.
92  bool active_user_session_started_;
93
94  // Whether the screen can be locked. Locking will only actually be allowed
95  // when this is |true| and there is an active user.
96  bool can_lock_screen_;
97
98  // Return value for ShouldLockScreenBeforeSuspending().
99  bool should_lock_screen_before_suspending_;
100
101  // Whether the screen is currently locked.
102  bool screen_locked_;
103
104  // Whether user addding screen is running now.
105  bool user_adding_screen_running_;
106
107  // The number of users logged in.
108  int logged_in_users_;
109
110  // The index for the activated user.
111  int active_user_index_;
112
113  std::vector<MockUserInfo*> user_list_;
114
115  DISALLOW_COPY_AND_ASSIGN(TestSessionStateDelegate);
116};
117
118}  // namespace test
119}  // namespace ash
120
121#endif  // ASH_TEST_TEST_SESSION_STATE_DELEGATE_H_
122