| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #include "ash/shell.h" | 5 #include "ash/shell.h" |
| 6 #include "ash/system/chromeos/multi_user/user_switch_util.h" | 6 #include "ash/system/chromeos/multi_user/user_switch_util.h" |
| 7 #include "ash/system/chromeos/screen_security/screen_tray_item.h" | 7 #include "ash/system/chromeos/screen_security/screen_tray_item.h" |
| 8 #include "ash/system/tray/system_tray.h" | 8 #include "ash/system/tray/system_tray.h" |
| 9 #include "ash/test/ash_test_base.h" | 9 #include "ash/test/ash_test_base.h" |
| 10 | 10 |
| 11 namespace ash { | 11 namespace ash { |
| 12 | 12 |
| 13 class TrySwitchingUserTest : public ash::test::AshTestBase { | 13 class TrySwitchingUserTest : public ash::test::AshTestBase { |
| 14 public: | 14 public: |
| 15 // The action type to perform / check for upon user switching. | 15 // The action type to perform / check for upon user switching. |
| 16 enum ActionType { | 16 enum ActionType { |
| 17 NO_DIALOG, // No dialog should be shown. | 17 NO_DIALOG, // No dialog should be shown. |
| 18 ACCEPT_DIALOG, // A dialog should be shown and we should accept it. | 18 ACCEPT_DIALOG, // A dialog should be shown and we should accept it. |
| 19 DECLINE_DIALOG, // A dialog should be shown and we do not accept it. | 19 DECLINE_DIALOG, // A dialog should be shown and we do not accept it. |
| 20 }; | 20 }; |
| 21 TrySwitchingUserTest() | 21 TrySwitchingUserTest() |
| 22 : capture_item_(NULL), | 22 : capture_item_(NULL), |
| 23 share_item_(NULL), | 23 share_item_(NULL), |
| 24 stop_capture_callback_hit_count_(0), | 24 stop_capture_callback_hit_count_(0), |
| 25 stop_share_callback_hit_count_(0), | 25 stop_share_callback_hit_count_(0), |
| 26 switch_callback_hit_count_(0) {} | 26 switch_callback_hit_count_(0) {} |
| 27 ~TrySwitchingUserTest() override {} | 27 ~TrySwitchingUserTest() override {} |
| 28 | 28 |
| 29 void SetUp() override { | 29 void SetUp() override { |
| 30 test::AshTestBase::SetUp(); | 30 test::AshTestBase::SetUp(); |
| 31 TrayItemView::DisableAnimationsForTest(); | 31 TrayItemView::DisableAnimationsForTest(); |
| 32 SystemTray* system_tray = Shell::GetInstance()->GetPrimarySystemTray(); | 32 SystemTray* system_tray = Shell::GetInstance()->GetPrimarySystemTray(); |
| 33 share_item_ = system_tray->GetScreenShareItem(); | 33 share_item_ = system_tray->GetScreenShareItem(); |
| 34 capture_item_ = system_tray->GetScreenCaptureItem(); | 34 capture_item_ = system_tray->GetScreenCaptureItem(); |
| 35 EXPECT_TRUE(share_item_); | 35 EXPECT_TRUE(share_item_); |
| 36 EXPECT_TRUE(capture_item_); | 36 EXPECT_TRUE(capture_item_); |
| 37 } | 37 } |
| 38 | 38 |
| 39 // Accessing the capture session functionality. | 39 // Accessing the capture session functionality. |
| 40 // Simulates a screen capture session start. | 40 // Simulates a screen capture session start. |
| 41 void StartCaptureSession() { | 41 void StartCaptureSession() { |
| 42 capture_item_->Start( | 42 capture_item_->Start(base::Bind(&TrySwitchingUserTest::StopCaptureCallback, |
| 43 base::Bind(&TrySwitchingUserTest::StopCaptureCallback, | 43 base::Unretained(this))); |
| 44 base::Unretained(this))); | |
| 45 } | 44 } |
| 46 | 45 |
| 47 // The callback which gets called when the screen capture gets stopped. | 46 // The callback which gets called when the screen capture gets stopped. |
| 48 void StopCaptureSession() { | 47 void StopCaptureSession() { capture_item_->Stop(); } |
| 49 capture_item_->Stop(); | |
| 50 } | |
| 51 | 48 |
| 52 // Simulates a screen capture session stop. | 49 // Simulates a screen capture session stop. |
| 53 void StopCaptureCallback() { | 50 void StopCaptureCallback() { stop_capture_callback_hit_count_++; } |
| 54 stop_capture_callback_hit_count_++; | |
| 55 } | |
| 56 | 51 |
| 57 // Accessing the share session functionality. | 52 // Accessing the share session functionality. |
| 58 // Simulate a Screen share session start. | 53 // Simulate a Screen share session start. |
| 59 void StartShareSession() { | 54 void StartShareSession() { |
| 60 share_item_->Start( | 55 share_item_->Start(base::Bind(&TrySwitchingUserTest::StopShareCallback, |
| 61 base::Bind(&TrySwitchingUserTest::StopShareCallback, | 56 base::Unretained(this))); |
| 62 base::Unretained(this))); | |
| 63 } | 57 } |
| 64 | 58 |
| 65 // Simulates a screen share session stop. | 59 // Simulates a screen share session stop. |
| 66 void StopShareSession() { | 60 void StopShareSession() { share_item_->Stop(); } |
| 67 share_item_->Stop(); | |
| 68 } | |
| 69 | 61 |
| 70 // The callback which gets called when the screen share gets stopped. | 62 // The callback which gets called when the screen share gets stopped. |
| 71 void StopShareCallback() { | 63 void StopShareCallback() { stop_share_callback_hit_count_++; } |
| 72 stop_share_callback_hit_count_++; | |
| 73 } | |
| 74 | 64 |
| 75 // Issuing a switch user call which might or might not create a dialog. | 65 // Issuing a switch user call which might or might not create a dialog. |
| 76 // The passed |action| type parameter defines the outcome (which will be | 66 // The passed |action| type parameter defines the outcome (which will be |
| 77 // checked) and the action the user will choose. | 67 // checked) and the action the user will choose. |
| 78 void SwitchUser(ActionType action) { | 68 void SwitchUser(ActionType action) { |
| 79 TrySwitchingActiveUser( | 69 TrySwitchingActiveUser(base::Bind(&TrySwitchingUserTest::SwitchCallback, |
| 80 base::Bind(&TrySwitchingUserTest::SwitchCallback, | 70 base::Unretained(this))); |
| 81 base::Unretained(this))); | 71 switch (action) { |
| 82 switch(action) { | |
| 83 case NO_DIALOG: | 72 case NO_DIALOG: |
| 84 EXPECT_TRUE(!TestAndTerminateDesktopCastingWarningForTest(true)); | 73 EXPECT_TRUE(!TestAndTerminateDesktopCastingWarningForTest(true)); |
| 85 return; | 74 return; |
| 86 case ACCEPT_DIALOG: | 75 case ACCEPT_DIALOG: |
| 87 EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(true)); | 76 EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(true)); |
| 88 return; | 77 return; |
| 89 case DECLINE_DIALOG: | 78 case DECLINE_DIALOG: |
| 90 EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(false)); | 79 EXPECT_TRUE(TestAndTerminateDesktopCastingWarningForTest(false)); |
| 91 return; | 80 return; |
| 92 } | 81 } |
| 93 } | 82 } |
| 94 | 83 |
| 95 // Called when the user will get actually switched. | 84 // Called when the user will get actually switched. |
| 96 void SwitchCallback() { | 85 void SwitchCallback() { switch_callback_hit_count_++; } |
| 97 switch_callback_hit_count_++; | |
| 98 } | |
| 99 | 86 |
| 100 // Various counter accessors. | 87 // Various counter accessors. |
| 101 int stop_capture_callback_hit_count() const { | 88 int stop_capture_callback_hit_count() const { |
| 102 return stop_capture_callback_hit_count_; | 89 return stop_capture_callback_hit_count_; |
| 103 } | 90 } |
| 104 int stop_share_callback_hit_count() const { | 91 int stop_share_callback_hit_count() const { |
| 105 return stop_share_callback_hit_count_; | 92 return stop_share_callback_hit_count_; |
| 106 } | 93 } |
| 107 int switch_callback_hit_count() const { return switch_callback_hit_count_; } | 94 int switch_callback_hit_count() const { return switch_callback_hit_count_; } |
| 108 | 95 |
| (...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 219 EXPECT_EQ(1, stop_share_callback_hit_count()); | 206 EXPECT_EQ(1, stop_share_callback_hit_count()); |
| 220 // Another stop should have no effect. | 207 // Another stop should have no effect. |
| 221 StopShareSession(); | 208 StopShareSession(); |
| 222 StopCaptureSession(); | 209 StopCaptureSession(); |
| 223 EXPECT_EQ(1, switch_callback_hit_count()); | 210 EXPECT_EQ(1, switch_callback_hit_count()); |
| 224 EXPECT_EQ(1, stop_capture_callback_hit_count()); | 211 EXPECT_EQ(1, stop_capture_callback_hit_count()); |
| 225 EXPECT_EQ(1, stop_share_callback_hit_count()); | 212 EXPECT_EQ(1, stop_share_callback_hit_count()); |
| 226 } | 213 } |
| 227 | 214 |
| 228 } // namespace ash | 215 } // namespace ash |
| OLD | NEW |