OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ | 5 #ifndef ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ |
6 #define ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ | 6 #define ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ |
7 | 7 |
8 #include "ash/system/tray/default_system_tray_delegate.h" | 8 #include "ash/system/tray/default_system_tray_delegate.h" |
9 #include "base/time/time.h" | 9 #include "base/time/time.h" |
10 | 10 |
11 namespace ash { | 11 namespace ash { |
12 namespace test { | 12 namespace test { |
13 | 13 |
14 class TestSystemTrayDelegate : public DefaultSystemTrayDelegate { | 14 class TestSystemTrayDelegate : public DefaultSystemTrayDelegate { |
15 public: | 15 public: |
16 TestSystemTrayDelegate(); | 16 TestSystemTrayDelegate(); |
17 virtual ~TestSystemTrayDelegate(); | 17 ~TestSystemTrayDelegate() override; |
18 | 18 |
19 // Changes the login status when initially the delegate is created. This will | 19 // Changes the login status when initially the delegate is created. This will |
20 // be called before AshTestBase::SetUp() to test the case when chrome is | 20 // be called before AshTestBase::SetUp() to test the case when chrome is |
21 // restarted right after the login (such like a flag is set). | 21 // restarted right after the login (such like a flag is set). |
22 // This value will be reset in AshTestHelper::TearDown, most test fixtures | 22 // This value will be reset in AshTestHelper::TearDown, most test fixtures |
23 // don't need to care its lifecycle. | 23 // don't need to care its lifecycle. |
24 static void SetInitialLoginStatus(user::LoginStatus login_status); | 24 static void SetInitialLoginStatus(user::LoginStatus login_status); |
25 | 25 |
26 // Changes the current login status in the test. This also invokes | 26 // Changes the current login status in the test. This also invokes |
27 // UpdateAfterLoginStatusChange(). Usually this is called in the test code to | 27 // UpdateAfterLoginStatusChange(). Usually this is called in the test code to |
28 // set up a login status. This will fit to most of the test cases, but this | 28 // set up a login status. This will fit to most of the test cases, but this |
29 // cannot be set during the initialization. To test the initialization, | 29 // cannot be set during the initialization. To test the initialization, |
30 // consider using SetInitialLoginStatus() instead. | 30 // consider using SetInitialLoginStatus() instead. |
31 void SetLoginStatus(user::LoginStatus login_status); | 31 void SetLoginStatus(user::LoginStatus login_status); |
32 | 32 |
33 void set_should_show_display_notification(bool should_show) { | 33 void set_should_show_display_notification(bool should_show) { |
34 should_show_display_notification_ = should_show; | 34 should_show_display_notification_ = should_show; |
35 } | 35 } |
36 | 36 |
37 // Updates the session length limit so that the limit will come from now in | 37 // Updates the session length limit so that the limit will come from now in |
38 // |new_limit|. | 38 // |new_limit|. |
39 void SetSessionLengthLimitForTest(const base::TimeDelta& new_limit); | 39 void SetSessionLengthLimitForTest(const base::TimeDelta& new_limit); |
40 | 40 |
41 // Clears the session length limit. | 41 // Clears the session length limit. |
42 void ClearSessionLengthLimit(); | 42 void ClearSessionLengthLimit(); |
43 | 43 |
44 // Overridden from SystemTrayDelegate: | 44 // Overridden from SystemTrayDelegate: |
45 virtual user::LoginStatus GetUserLoginStatus() const override; | 45 user::LoginStatus GetUserLoginStatus() const override; |
46 virtual bool IsUserSupervised() const override; | 46 bool IsUserSupervised() const override; |
47 virtual bool ShouldShowDisplayNotification() override; | 47 bool ShouldShowDisplayNotification() override; |
48 virtual bool GetSessionStartTime( | 48 bool GetSessionStartTime(base::TimeTicks* session_start_time) override; |
49 base::TimeTicks* session_start_time) override; | 49 bool GetSessionLengthLimit(base::TimeDelta* session_length_limit) override; |
50 virtual bool GetSessionLengthLimit( | 50 void ShutDown() override; |
51 base::TimeDelta* session_length_limit) override; | 51 void SignOut() override; |
52 virtual void ShutDown() override; | |
53 virtual void SignOut() override; | |
54 | 52 |
55 private: | 53 private: |
56 bool should_show_display_notification_; | 54 bool should_show_display_notification_; |
57 user::LoginStatus login_status_; | 55 user::LoginStatus login_status_; |
58 base::TimeDelta session_length_limit_; | 56 base::TimeDelta session_length_limit_; |
59 bool session_length_limit_set_; | 57 bool session_length_limit_set_; |
60 | 58 |
61 DISALLOW_COPY_AND_ASSIGN(TestSystemTrayDelegate); | 59 DISALLOW_COPY_AND_ASSIGN(TestSystemTrayDelegate); |
62 }; | 60 }; |
63 | 61 |
64 } // namespace test | 62 } // namespace test |
65 } // namespace ash | 63 } // namespace ash |
66 | 64 |
67 #endif // ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ | 65 #endif // ASH_TEST_TEST_SYSTEM_TRAY_DELEGATE_H_ |
OLD | NEW |