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 3// found in the LICENSE file. 4 5#include "chrome/browser/chromeos/login/lock/screen_locker.h" 6 7#include "ash/wm/window_state.h" 8#include "base/command_line.h" 9#include "base/memory/scoped_ptr.h" 10#include "base/message_loop/message_loop.h" 11#include "chrome/browser/chrome_notification_types.h" 12#include "chrome/browser/chromeos/login/lock/screen_locker_tester.h" 13#include "chrome/browser/profiles/profile_manager.h" 14#include "chrome/browser/ui/browser.h" 15#include "chrome/browser/ui/browser_window.h" 16#include "chrome/browser/ui/fullscreen/fullscreen_controller.h" 17#include "chrome/browser/ui/tabs/tab_strip_model.h" 18#include "chrome/common/chrome_switches.h" 19#include "chrome/test/base/in_process_browser_test.h" 20#include "chrome/test/base/ui_test_utils.h" 21#include "chromeos/chromeos_switches.h" 22#include "chromeos/dbus/dbus_thread_manager.h" 23#include "chromeos/dbus/fake_session_manager_client.h" 24#include "chromeos/login/auth/key.h" 25#include "chromeos/login/auth/mock_authenticator.h" 26#include "chromeos/login/auth/user_context.h" 27#include "chromeos/login/user_names.h" 28#include "content/public/browser/notification_service.h" 29#include "testing/gmock/include/gmock/gmock.h" 30#include "testing/gtest/include/gtest/gtest.h" 31#include "ui/base/test/ui_controls.h" 32#include "ui/compositor/layer_animator.h" 33#include "ui/compositor/scoped_animation_duration_scale_mode.h" 34#include "ui/views/widget/widget.h" 35 36using testing::_; 37using testing::AnyNumber; 38using testing::Return; 39 40namespace { 41 42// An object that wait for lock state and fullscreen state. 43class Waiter : public content::NotificationObserver { 44 public: 45 explicit Waiter(Browser* browser) 46 : browser_(browser), 47 running_(false) { 48 registrar_.Add(this, 49 chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED, 50 content::NotificationService::AllSources()); 51 registrar_.Add(this, 52 chrome::NOTIFICATION_FULLSCREEN_CHANGED, 53 content::NotificationService::AllSources()); 54 } 55 56 virtual ~Waiter() { 57 } 58 59 virtual void Observe(int type, 60 const content::NotificationSource& source, 61 const content::NotificationDetails& details) OVERRIDE { 62 DCHECK(type == chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED || 63 type == chrome::NOTIFICATION_FULLSCREEN_CHANGED); 64 if (running_) 65 base::MessageLoop::current()->Quit(); 66 } 67 68 // Wait until the two conditions are met. 69 void Wait(bool locker_state, bool fullscreen) { 70 running_ = true; 71 scoped_ptr<chromeos::test::ScreenLockerTester> 72 tester(chromeos::ScreenLocker::GetTester()); 73 while (tester->IsLocked() != locker_state || 74 browser_->window()->IsFullscreen() != fullscreen) { 75 content::RunMessageLoop(); 76 } 77 // Make sure all pending tasks are executed. 78 content::RunAllPendingInMessageLoop(); 79 running_ = false; 80 } 81 82 private: 83 Browser* browser_; 84 content::NotificationRegistrar registrar_; 85 86 // Are we currently running the message loop? 87 bool running_; 88 89 DISALLOW_COPY_AND_ASSIGN(Waiter); 90}; 91 92} // namespace 93 94namespace chromeos { 95 96class ScreenLockerTest : public InProcessBrowserTest { 97 public: 98 ScreenLockerTest() : fake_session_manager_client_(NULL) { 99 } 100 101 protected: 102 FakeSessionManagerClient* fake_session_manager_client_; 103 104 void LockScreen(test::ScreenLockerTester* tester) { 105 ScreenLocker::Show(); 106 tester->EmulateWindowManagerReady(); 107 content::WindowedNotificationObserver lock_state_observer( 108 chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED, 109 content::NotificationService::AllSources()); 110 if (!tester->IsLocked()) 111 lock_state_observer.Wait(); 112 EXPECT_TRUE(tester->IsLocked()); 113 } 114 115 // Verifies if LockScreenDismissed() was called once. 116 bool VerifyLockScreenDismissed() { 117 return 1 == fake_session_manager_client_-> 118 notify_lock_screen_dismissed_call_count(); 119 } 120 121 private: 122 virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { 123 fake_session_manager_client_ = new FakeSessionManagerClient; 124 DBusThreadManager::GetSetterForTesting()->SetSessionManagerClient( 125 scoped_ptr<SessionManagerClient>(fake_session_manager_client_)); 126 127 InProcessBrowserTest::SetUpInProcessBrowserTestFixture(); 128 zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode( 129 ui::ScopedAnimationDurationScaleMode::ZERO_DURATION)); 130 } 131 132 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { 133 command_line->AppendSwitchASCII(switches::kLoginProfile, "user"); 134 } 135 136 scoped_ptr<ui::ScopedAnimationDurationScaleMode> zero_duration_mode_; 137 138 DISALLOW_COPY_AND_ASSIGN(ScreenLockerTest); 139}; 140 141IN_PROC_BROWSER_TEST_F(ScreenLockerTest, TestBasic) { 142 ScreenLocker::Show(); 143 scoped_ptr<test::ScreenLockerTester> tester(ScreenLocker::GetTester()); 144 tester->EmulateWindowManagerReady(); 145 content::WindowedNotificationObserver lock_state_observer( 146 chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED, 147 content::NotificationService::AllSources()); 148 if (!tester->IsLocked()) 149 lock_state_observer.Wait(); 150 151 // Test to make sure that the widget is actually appearing and is of 152 // reasonable size, preventing a regression of 153 // http://code.google.com/p/chromium-os/issues/detail?id=5987 154 gfx::Rect lock_bounds = tester->GetChildWidget()->GetWindowBoundsInScreen(); 155 EXPECT_GT(lock_bounds.width(), 10); 156 EXPECT_GT(lock_bounds.height(), 10); 157 158 UserContext user_context(chromeos::login::kStubUser); 159 user_context.SetKey(Key("pass")); 160 tester->InjectMockAuthenticator(user_context); 161 EXPECT_TRUE(tester->IsLocked()); 162 tester->EnterPassword("fail"); 163 content::RunAllPendingInMessageLoop(); 164 EXPECT_TRUE(tester->IsLocked()); 165 tester->EnterPassword("pass"); 166 content::RunAllPendingInMessageLoop(); 167 // Successful authentication clears the lock screen and tells the 168 // SessionManager to announce this over DBus. 169 EXPECT_FALSE(tester->IsLocked()); 170 EXPECT_EQ( 171 1, 172 fake_session_manager_client_->notify_lock_screen_shown_call_count()); 173 174 EXPECT_TRUE(VerifyLockScreenDismissed()); 175} 176 177// Test how locking the screen affects an active fullscreen window. 178IN_PROC_BROWSER_TEST_F(ScreenLockerTest, TestFullscreenExit) { 179 // 1) If the active browser window is in fullscreen and the fullscreen window 180 // does not have all the pixels (e.g. the shelf is auto hidden instead of 181 // hidden), locking the screen should not exit fullscreen. The shelf is 182 // auto hidden when in immersive fullscreen. 183 scoped_ptr<test::ScreenLockerTester> tester(ScreenLocker::GetTester()); 184 BrowserWindow* browser_window = browser()->window(); 185 ash::wm::WindowState* window_state = ash::wm::GetWindowState( 186 browser_window->GetNativeWindow()); 187 { 188 Waiter waiter(browser()); 189 browser()->fullscreen_controller()->ToggleBrowserFullscreenMode(); 190 waiter.Wait(false /* not locked */, true /* full screen */); 191 EXPECT_TRUE(browser_window->IsFullscreen()); 192 EXPECT_FALSE(window_state->hide_shelf_when_fullscreen()); 193 EXPECT_FALSE(tester->IsLocked()); 194 } 195 { 196 Waiter waiter(browser()); 197 ScreenLocker::Show(); 198 tester->EmulateWindowManagerReady(); 199 waiter.Wait(true /* locked */, true /* full screen */); 200 EXPECT_TRUE(browser_window->IsFullscreen()); 201 EXPECT_FALSE(window_state->hide_shelf_when_fullscreen()); 202 EXPECT_TRUE(tester->IsLocked()); 203 } 204 UserContext user_context(chromeos::login::kStubUser); 205 user_context.SetKey(Key("pass")); 206 tester->InjectMockAuthenticator(user_context); 207 tester->EnterPassword("pass"); 208 content::RunAllPendingInMessageLoop(); 209 EXPECT_FALSE(tester->IsLocked()); 210 { 211 Waiter waiter(browser()); 212 browser()->fullscreen_controller()->ToggleBrowserFullscreenMode(); 213 waiter.Wait(false /* not locked */, false /* fullscreen */); 214 EXPECT_FALSE(browser_window->IsFullscreen()); 215 } 216 217 // 2) If the active browser window is in fullscreen and the fullscreen window 218 // has all of the pixels, locking the screen should exit fullscreen. The 219 // fullscreen window has all of the pixels when in tab fullscreen. 220 { 221 Waiter waiter(browser()); 222 content::WebContents* web_contents = 223 browser()->tab_strip_model()->GetActiveWebContents(); 224 browser()->fullscreen_controller()->ToggleFullscreenModeForTab( 225 web_contents, true); 226 waiter.Wait(false /* not locked */, true /* fullscreen */); 227 EXPECT_TRUE(browser_window->IsFullscreen()); 228 EXPECT_TRUE(window_state->hide_shelf_when_fullscreen()); 229 EXPECT_FALSE(tester->IsLocked()); 230 } 231 { 232 Waiter waiter(browser()); 233 ScreenLocker::Show(); 234 tester->EmulateWindowManagerReady(); 235 waiter.Wait(true /* locked */, false /* full screen */); 236 EXPECT_FALSE(browser_window->IsFullscreen()); 237 EXPECT_TRUE(tester->IsLocked()); 238 } 239 240 tester->EnterPassword("pass"); 241 content::RunAllPendingInMessageLoop(); 242 EXPECT_FALSE(tester->IsLocked()); 243 244 EXPECT_EQ( 245 2, 246 fake_session_manager_client_->notify_lock_screen_shown_call_count()); 247 EXPECT_EQ( 248 2, 249 fake_session_manager_client_->notify_lock_screen_dismissed_call_count()); 250} 251 252void SimulateKeyPress(views::Widget* widget, ui::KeyboardCode key_code) { 253 ui_controls::SendKeyPress(widget->GetNativeWindow(), 254 key_code, false, false, false, false); 255} 256 257void UnlockKeyPress(views::Widget* widget) { 258 SimulateKeyPress(widget, ui::VKEY_SPACE); 259} 260 261IN_PROC_BROWSER_TEST_F(ScreenLockerTest, TestShowTwice) { 262 scoped_ptr<test::ScreenLockerTester> tester(ScreenLocker::GetTester()); 263 LockScreen(tester.get()); 264 265 // Calling Show again simply send LockCompleted signal. 266 ScreenLocker::Show(); 267 EXPECT_TRUE(tester->IsLocked()); 268 EXPECT_EQ( 269 2, 270 fake_session_manager_client_->notify_lock_screen_shown_call_count()); 271 272 273 // Close the locker to match expectations. 274 ScreenLocker::Hide(); 275 content::RunAllPendingInMessageLoop(); 276 EXPECT_FALSE(tester->IsLocked()); 277 EXPECT_TRUE(VerifyLockScreenDismissed()); 278} 279 280// TODO(flackr): Find out why the RenderView isn't getting the escape press 281// and re-enable this test (currently this test is flaky). 282IN_PROC_BROWSER_TEST_F(ScreenLockerTest, DISABLED_TestEscape) { 283 scoped_ptr<test::ScreenLockerTester> tester(ScreenLocker::GetTester()); 284 LockScreen(tester.get()); 285 286 EXPECT_EQ( 287 1, 288 fake_session_manager_client_->notify_lock_screen_shown_call_count()); 289 290 tester->SetPassword("password"); 291 EXPECT_EQ("password", tester->GetPassword()); 292 // Escape clears the password. 293 SimulateKeyPress(tester->GetWidget(), ui::VKEY_ESCAPE); 294 content::RunAllPendingInMessageLoop(); 295 EXPECT_EQ("", tester->GetPassword()); 296 297 // Close the locker to match expectations. 298 ScreenLocker::Hide(); 299 content::RunAllPendingInMessageLoop(); 300 EXPECT_FALSE(tester->IsLocked()); 301 EXPECT_TRUE(VerifyLockScreenDismissed()); 302} 303 304} // namespace chromeos 305