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 3 // found in the LICENSE file. 4 5 #include "chrome/browser/chromeos/login/login_manager_test.h" 6 7 #include "chrome/browser/browser_process.h" 8 #include "chrome/browser/chrome_notification_types.h" 9 #include "chrome/browser/chromeos/login/existing_user_controller.h" 10 #include "chrome/browser/chromeos/login/user_manager.h" 11 #include "chrome/browser/prefs/scoped_user_pref_update.h" 12 #include "chrome/common/chrome_switches.h" 13 #include "chromeos/chromeos_switches.h" 14 #include "content/public/browser/notification_service.h" 15 #include "content/public/test/test_utils.h" 16 #include "testing/gtest/include/gtest/gtest.h" 17 18 namespace chromeos { 19 20 LoginManagerTest::LoginManagerTest(bool should_launch_browser) 21 : should_launch_browser_(should_launch_browser) { 22 set_exit_when_last_browser_closes(false); 23 } 24 25 void LoginManagerTest::SetUpCommandLine(CommandLine* command_line) { 26 command_line->AppendSwitch(chromeos::switches::kLoginManager); 27 command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests); 28 command_line->AppendSwitch(::switches::kMultiProfiles); 29 } 30 31 void LoginManagerTest::SetUpInProcessBrowserTestFixture() { 32 mock_login_utils_ = new testing::NiceMock<MockLoginUtils>(); 33 mock_login_utils_->DelegateToFake(); 34 mock_login_utils_->GetFakeLoginUtils()->set_should_launch_browser( 35 should_launch_browser_); 36 LoginUtils::Set(mock_login_utils_); 37 } 38 39 void LoginManagerTest::RegisterUser(const std::string& username) { 40 ListPrefUpdate users_pref(g_browser_process->local_state(), "LoggedInUsers"); 41 users_pref->AppendIfNotPresent(new base::StringValue(username)); 42 } 43 44 void LoginManagerTest::SetExpectedCredentials(const std::string& username, 45 const std::string& password) { 46 login_utils().GetFakeLoginUtils()->SetExpectedCredentials(username, password); 47 } 48 49 bool LoginManagerTest::TryToLogin(const std::string& username, 50 const std::string& password) { 51 ExistingUserController* controller = 52 ExistingUserController::current_controller(); 53 EXPECT_TRUE(controller != NULL); 54 controller->Login(UserContext(username, password, std::string())); 55 content::WindowedNotificationObserver( 56 chrome::NOTIFICATION_SESSION_STARTED, 57 content::NotificationService::AllSources()).Wait(); 58 if (const User* active_user = UserManager::Get()->GetActiveUser()) 59 return active_user->email() == username; 60 return false; 61 } 62 63 void LoginManagerTest::LoginUser(const std::string& username) { 64 SetExpectedCredentials(username, "password"); 65 EXPECT_TRUE(TryToLogin(username, "password")); 66 } 67 68 } // namespace chromeos 69