| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "ash/session/session_state_delegate.h" |
| #include "ash/shell.h" |
| #include "base/prefs/pref_service.h" |
| #include "chrome/browser/chrome_notification_types.h" |
| #include "chrome/browser/chromeos/login/lock/screen_locker.h" |
| #include "chrome/browser/chromeos/login/lock/screen_locker_tester.h" |
| #include "chrome/browser/chromeos/login/login_manager_test.h" |
| #include "chrome/browser/chromeos/login/startup_utils.h" |
| #include "chrome/browser/chromeos/login/ui/login_display_host_impl.h" |
| #include "chrome/browser/chromeos/login/ui/user_adding_screen.h" |
| #include "chrome/browser/chromeos/login/ui/webui_login_view.h" |
| #include "chrome/browser/chromeos/login/users/multi_profile_user_controller.h" |
| #include "chrome/browser/chromeos/profiles/profile_helper.h" |
| #include "chrome/common/pref_names.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "components/user_manager/user_manager.h" |
| #include "content/public/test/test_utils.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using namespace testing; |
| |
| namespace { |
| |
| const char* const kTestUsers[] = {"test-user1@gmail.com", |
| "test-user2@gmail.com", |
| "test-user3@gmail.com"}; |
| |
| } // anonymous namespace |
| |
| namespace chromeos { |
| |
| class UserAddingScreenTest : public LoginManagerTest, |
| public UserAddingScreen::Observer { |
| public: |
| UserAddingScreenTest() : LoginManagerTest(false), |
| user_adding_started_(0), |
| user_adding_finished_(0) { |
| } |
| |
| virtual void SetUpInProcessBrowserTestFixture() override { |
| LoginManagerTest::SetUpInProcessBrowserTestFixture(); |
| UserAddingScreen::Get()->AddObserver(this); |
| } |
| |
| virtual void OnUserAddingFinished() override { ++user_adding_finished_; } |
| |
| virtual void OnUserAddingStarted() override { ++user_adding_started_; } |
| |
| void SetUserCanLock(user_manager::User* user, bool can_lock) { |
| user->set_can_lock(can_lock); |
| } |
| |
| void CheckScreenIsVisible() { |
| views::View* web_view = |
| LoginDisplayHostImpl::default_host()->GetWebUILoginView()->child_at(0); |
| for (views::View* current_view = web_view; |
| current_view; |
| current_view = current_view->parent()) { |
| EXPECT_TRUE(current_view->visible()); |
| if (current_view->layer()) |
| EXPECT_EQ(current_view->layer()->GetCombinedOpacity(), 1.f); |
| } |
| for (aura::Window* window = web_view->GetWidget()->GetNativeWindow(); |
| window; |
| window = window->parent()) { |
| EXPECT_TRUE(window->IsVisible()); |
| EXPECT_EQ(window->layer()->GetCombinedOpacity(), 1.f); |
| } |
| } |
| |
| int user_adding_started() { return user_adding_started_; } |
| |
| int user_adding_finished() { return user_adding_finished_; } |
| |
| private: |
| int user_adding_started_; |
| int user_adding_finished_; |
| |
| DISALLOW_COPY_AND_ASSIGN(UserAddingScreenTest); |
| }; |
| |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, PRE_CancelAdding) { |
| RegisterUser(kTestUsers[0]); |
| RegisterUser(kTestUsers[1]); |
| RegisterUser(kTestUsers[2]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, CancelAdding) { |
| EXPECT_CALL(login_utils(), DoBrowserLaunch(_, _)).Times(1); |
| EXPECT_EQ(3u, user_manager::UserManager::Get()->GetUsers().size()); |
| EXPECT_EQ(0u, user_manager::UserManager::Get()->GetLoggedInUsers().size()); |
| |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_PRIMARY, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| LoginUser(kTestUsers[0]); |
| EXPECT_EQ(1u, user_manager::UserManager::Get()->GetLoggedInUsers().size()); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| UserAddingScreen::Get()->Start(); |
| content::RunAllPendingInMessageLoop(); |
| EXPECT_EQ(1, user_adding_started()); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_SECONDARY, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| UserAddingScreen::Get()->Cancel(); |
| content::RunAllPendingInMessageLoop(); |
| EXPECT_EQ(1, user_adding_finished()); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| EXPECT_TRUE(LoginDisplayHostImpl::default_host() == NULL); |
| EXPECT_EQ(1u, user_manager::UserManager::Get()->GetLoggedInUsers().size()); |
| EXPECT_EQ(kTestUsers[0], |
| user_manager::UserManager::Get()->GetActiveUser()->email()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, PRE_AddingSeveralUsers) { |
| RegisterUser(kTestUsers[0]); |
| RegisterUser(kTestUsers[1]); |
| RegisterUser(kTestUsers[2]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, AddingSeveralUsers) { |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_PRIMARY, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| EXPECT_CALL(login_utils(), DoBrowserLaunch(_, _)).Times(3); |
| LoginUser(kTestUsers[0]); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| user_manager::UserManager* user_manager = user_manager::UserManager::Get(); |
| |
| for (int i = 1; i < 3; ++i) { |
| UserAddingScreen::Get()->Start(); |
| content::RunAllPendingInMessageLoop(); |
| EXPECT_EQ(i, user_adding_started()); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_LOGIN_SECONDARY, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| AddUser(kTestUsers[i]); |
| EXPECT_EQ(i, user_adding_finished()); |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| EXPECT_TRUE(LoginDisplayHostImpl::default_host() == NULL); |
| ASSERT_EQ(unsigned(i + 1), user_manager->GetLoggedInUsers().size()); |
| } |
| |
| EXPECT_EQ(ash::SessionStateDelegate::SESSION_STATE_ACTIVE, |
| ash::Shell::GetInstance()->session_state_delegate()-> |
| GetSessionState()); |
| |
| // Now check how unlock policy works for these users. |
| PrefService* prefs1 = |
| ProfileHelper::Get() |
| ->GetProfileByUserUnsafe(user_manager->GetLoggedInUsers()[0]) |
| ->GetPrefs(); |
| PrefService* prefs2 = |
| ProfileHelper::Get() |
| ->GetProfileByUserUnsafe(user_manager->GetLoggedInUsers()[1]) |
| ->GetPrefs(); |
| PrefService* prefs3 = |
| ProfileHelper::Get() |
| ->GetProfileByUserUnsafe(user_manager->GetLoggedInUsers()[2]) |
| ->GetPrefs(); |
| ASSERT_TRUE(prefs1 != NULL); |
| ASSERT_TRUE(prefs2 != NULL); |
| ASSERT_TRUE(prefs3 != NULL); |
| prefs1->SetBoolean(prefs::kEnableAutoScreenLock, false); |
| prefs2->SetBoolean(prefs::kEnableAutoScreenLock, false); |
| prefs3->SetBoolean(prefs::kEnableAutoScreenLock, false); |
| |
| // One of the users has the primary-only policy. |
| // List of unlock users doesn't depend on kEnableLockScreen preference. |
| prefs1->SetBoolean(prefs::kEnableAutoScreenLock, true); |
| prefs1->SetString(prefs::kMultiProfileUserBehavior, |
| MultiProfileUserController::kBehaviorPrimaryOnly); |
| prefs2->SetString(prefs::kMultiProfileUserBehavior, |
| MultiProfileUserController::kBehaviorUnrestricted); |
| prefs3->SetString(prefs::kMultiProfileUserBehavior, |
| MultiProfileUserController::kBehaviorUnrestricted); |
| user_manager::UserList unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(1UL, unlock_users.size()); |
| EXPECT_EQ(kTestUsers[0], unlock_users[0]->email()); |
| |
| prefs1->SetBoolean(prefs::kEnableAutoScreenLock, false); |
| unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(1UL, unlock_users.size()); |
| EXPECT_EQ(kTestUsers[0], unlock_users[0]->email()); |
| |
| // If all users have unrestricted policy then anyone can perform unlock. |
| prefs1->SetString(prefs::kMultiProfileUserBehavior, |
| MultiProfileUserController::kBehaviorUnrestricted); |
| unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(3UL, unlock_users.size()); |
| for (int i = 0; i < 3; ++i) |
| EXPECT_EQ(kTestUsers[i], unlock_users[i]->email()); |
| |
| // This preference doesn't affect list of unlock users. |
| prefs2->SetBoolean(prefs::kEnableAutoScreenLock, true); |
| unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(3UL, unlock_users.size()); |
| for (int i = 0; i < 3; ++i) |
| EXPECT_EQ(kTestUsers[i], unlock_users[i]->email()); |
| |
| // Now one of the users is unable to unlock. |
| SetUserCanLock(user_manager->GetLoggedInUsers()[2], false); |
| unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(2UL, unlock_users.size()); |
| for (int i = 0; i < 2; ++i) |
| EXPECT_EQ(kTestUsers[i], unlock_users[i]->email()); |
| SetUserCanLock(user_manager->GetLoggedInUsers()[2], true); |
| |
| // Now one of the users has not-allowed policy. |
| // In this scenario this user is not allowed in multi-profile session but |
| // if that user happened to still be part of multi-profile session it should |
| // not be listed on screen lock. |
| prefs3->SetString(prefs::kMultiProfileUserBehavior, |
| MultiProfileUserController::kBehaviorNotAllowed); |
| unlock_users = user_manager->GetUnlockUsers(); |
| ASSERT_EQ(2UL, unlock_users.size()); |
| for (int i = 0; i < 2; ++i) |
| EXPECT_EQ(kTestUsers[i], unlock_users[i]->email()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, PRE_ScreenVisibility) { |
| RegisterUser(kTestUsers[0]); |
| RegisterUser(kTestUsers[1]); |
| StartupUtils::MarkOobeCompleted(); |
| } |
| |
| // Trying to catch http://crbug.com/362153. |
| IN_PROC_BROWSER_TEST_F(UserAddingScreenTest, ScreenVisibility) { |
| LoginUser(kTestUsers[0]); |
| |
| UserAddingScreen::Get()->Start(); |
| content::RunAllPendingInMessageLoop(); |
| CheckScreenIsVisible(); |
| UserAddingScreen::Get()->Cancel(); |
| content::RunAllPendingInMessageLoop(); |
| |
| ScreenLocker::Show(); |
| content::WindowedNotificationObserver( |
| chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED, |
| content::NotificationService::AllSources()).Wait(); |
| |
| ScreenLocker::Hide(); |
| content::WindowedNotificationObserver( |
| chrome::NOTIFICATION_SCREEN_LOCK_STATE_CHANGED, |
| content::NotificationService::AllSources()).Wait(); |
| |
| UserAddingScreen::Get()->Start(); |
| content::RunAllPendingInMessageLoop(); |
| CheckScreenIsVisible(); |
| UserAddingScreen::Get()->Cancel(); |
| content::RunAllPendingInMessageLoop(); |
| } |
| |
| } // namespace chromeos |