Index: chrome/browser/chromeos/login/existing_user_controller_browsertest.cc |
diff --git a/chrome/browser/chromeos/login/existing_user_controller_browsertest.cc b/chrome/browser/chromeos/login/existing_user_controller_browsertest.cc |
index 251d4084d90ad0610e08247713d1e12c717ed8ce..f29ba9502ae1f61f5c2c36ba7cc7010b3fbadffb 100644 |
--- a/chrome/browser/chromeos/login/existing_user_controller_browsertest.cc |
+++ b/chrome/browser/chromeos/login/existing_user_controller_browsertest.cc |
@@ -13,6 +13,7 @@ |
#include "base/run_loop.h" |
#include "chrome/browser/chrome_notification_types.h" |
#include "chrome/browser/chromeos/login/auth/authenticator.h" |
+#include "chrome/browser/chromeos/login/auth/key.h" |
#include "chrome/browser/chromeos/login/auth/mock_authenticator.h" |
#include "chrome/browser/chromeos/login/auth/mock_url_fetchers.h" |
#include "chrome/browser/chromeos/login/auth/user_context.h" |
@@ -74,9 +75,8 @@ const int kAutoLoginNoDelay = 0; |
const int kAutoLoginShortDelay = 1; |
const int kAutoLoginLongDelay = 10000; |
- |
-ACTION_P2(CreateAuthenticator, username, password) { |
- return new MockAuthenticator(arg0, username, password); |
+ACTION_P(CreateAuthenticator, user_context) { |
+ return new MockAuthenticator(arg0, user_context); |
} |
} // namespace |
@@ -236,12 +236,12 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerTest, ExistingUserLogin) { |
// auto-enrollment, and again after doing an ownership status check. |
EXPECT_CALL(*mock_login_display_, SetUIEnabled(false)) |
.Times(2); |
- EXPECT_CALL(*mock_login_utils_, CreateAuthenticator(_)) |
- .Times(1) |
- .WillOnce(WithArg<0>(CreateAuthenticator(kUsername, kPassword))); |
UserContext user_context(kUsername); |
- user_context.SetPassword(kPassword); |
+ user_context.SetKey(Key(kPassword)); |
user_context.SetUserIDHash(kUsername); |
+ EXPECT_CALL(*mock_login_utils_, CreateAuthenticator(_)) |
+ .Times(1) |
+ .WillOnce(WithArg<0>(CreateAuthenticator(user_context))); |
EXPECT_CALL(*mock_login_utils_, PrepareProfile(user_context, _, _, _, _)) |
.Times(1) |
.WillOnce(InvokeWithoutArgs(&profile_prepared_cb_, |
@@ -283,7 +283,7 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerTest, AutoEnrollAfterSignIn) { |
.InSequence(uiEnabledSequence); |
existing_user_controller()->DoAutoEnrollment(); |
UserContext user_context(kUsername); |
- user_context.SetPassword(kPassword); |
+ user_context.SetKey(Key(kPassword)); |
existing_user_controller()->CompleteLogin(user_context); |
content::RunAllPendingInMessageLoop(); |
} |
@@ -298,15 +298,15 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerTest, |
StartWizardPtr(WizardController::kTermsOfServiceScreenName, |
NULL)) |
.Times(1); |
+ UserContext user_context(kNewUsername); |
+ user_context.SetKey(Key(kPassword)); |
EXPECT_CALL(*mock_login_utils_, CreateAuthenticator(_)) |
.Times(1) |
- .WillOnce(WithArg<0>(CreateAuthenticator(kNewUsername, kPassword))); |
+ .WillOnce(WithArg<0>(CreateAuthenticator(user_context))); |
base::Callback<void(void)> add_user_cb = |
base::Bind(&MockUserManager::AddUser, |
base::Unretained(mock_user_manager_), |
kNewUsername); |
- UserContext user_context(kNewUsername); |
- user_context.SetPassword(kPassword); |
user_context.SetUserIDHash(kNewUsername); |
EXPECT_CALL(*mock_login_utils_, PrepareProfile(user_context, _, _, _, _)) |
.Times(1) |
@@ -426,16 +426,12 @@ class ExistingUserControllerPublicSessionTest |
virtual void SetUpUserManager() OVERRIDE { |
} |
- void ExpectSuccessfulLogin(const std::string& username, |
- const std::string& password) { |
+ void ExpectSuccessfulLogin(const UserContext& user_context) { |
EXPECT_CALL(*mock_login_display_, SetUIEnabled(false)) |
.Times(AnyNumber()); |
EXPECT_CALL(*mock_login_utils_, CreateAuthenticator(_)) |
.Times(1) |
- .WillOnce(WithArg<0>(CreateAuthenticator(username, password))); |
- UserContext user_context(username); |
- user_context.SetPassword(password); |
- user_context.SetUserIDHash(username); |
+ .WillOnce(WithArg<0>(CreateAuthenticator(user_context))); |
EXPECT_CALL(*mock_login_utils_, PrepareProfile(user_context, _, _, _, _)) |
.Times(1) |
.WillOnce(InvokeWithoutArgs(&profile_prepared_cb_, |
@@ -533,7 +529,9 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
AutoLoginNoDelay) { |
// Set up mocks to check login success. |
- ExpectSuccessfulLogin(public_session_user_id_, ""); |
+ UserContext user_context(public_session_user_id_); |
+ user_context.SetUserIDHash(user_context.GetUserID()); |
+ ExpectSuccessfulLogin(user_context); |
existing_user_controller()->OnSigninScreenReady(); |
// Start auto-login and wait for login tasks to complete. |
@@ -544,7 +542,9 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
AutoLoginShortDelay) { |
// Set up mocks to check login success. |
- ExpectSuccessfulLogin(public_session_user_id_, ""); |
+ UserContext user_context(public_session_user_id_); |
+ user_context.SetUserIDHash(user_context.GetUserID()); |
+ ExpectSuccessfulLogin(user_context); |
existing_user_controller()->OnSigninScreenReady(); |
SetAutoLoginPolicy(kPublicSessionAccountId, kAutoLoginShortDelay); |
ASSERT_TRUE(auto_login_timer()); |
@@ -567,15 +567,16 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
LoginStopsAutoLogin) { |
// Set up mocks to check login success. |
- ExpectSuccessfulLogin(kUsername, kPassword); |
+ UserContext user_context(kUsername); |
+ user_context.SetKey(Key(kPassword)); |
+ user_context.SetUserIDHash(user_context.GetUserID()); |
+ ExpectSuccessfulLogin(user_context); |
existing_user_controller()->OnSigninScreenReady(); |
SetAutoLoginPolicy(kPublicSessionAccountId, kAutoLoginLongDelay); |
ASSERT_TRUE(auto_login_timer()); |
- // Login and check that it stopped the timer. |
- UserContext user_context(kUsername); |
- user_context.SetPassword(kPassword); |
+ // Log in and check that it stopped the timer. |
existing_user_controller()->Login(user_context); |
EXPECT_TRUE(is_login_in_progress()); |
ASSERT_TRUE(auto_login_timer()); |
@@ -593,9 +594,11 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
GuestModeLoginStopsAutoLogin) { |
EXPECT_CALL(*mock_login_display_, SetUIEnabled(false)) |
.Times(1); |
+ UserContext user_context(kUsername); |
+ user_context.SetKey(Key(kPassword)); |
EXPECT_CALL(*mock_login_utils_, CreateAuthenticator(_)) |
.Times(1) |
- .WillOnce(WithArg<0>(CreateAuthenticator(kUsername, kPassword))); |
+ .WillOnce(WithArg<0>(CreateAuthenticator(user_context))); |
EXPECT_CALL(*mock_login_utils_, CompleteOffTheRecordLogin(_)) |
.Times(1); |
@@ -620,7 +623,10 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
CompleteLoginStopsAutoLogin) { |
// Set up mocks to check login success. |
- ExpectSuccessfulLogin(kUsername, kPassword); |
+ UserContext user_context(kUsername); |
+ user_context.SetKey(Key(kPassword)); |
+ user_context.SetUserIDHash(user_context.GetUserID()); |
+ ExpectSuccessfulLogin(user_context); |
EXPECT_CALL(*mock_login_display_host_, OnCompleteLogin()) |
.Times(1); |
@@ -629,8 +635,6 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
ASSERT_TRUE(auto_login_timer()); |
// Check that login completes and stops the timer. |
- UserContext user_context(kUsername); |
- user_context.SetPassword(kPassword); |
existing_user_controller()->CompleteLogin(user_context); |
ASSERT_TRUE(auto_login_timer()); |
EXPECT_FALSE(auto_login_timer()->IsRunning()); |
@@ -646,7 +650,9 @@ IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
IN_PROC_BROWSER_TEST_F(ExistingUserControllerPublicSessionTest, |
PublicSessionLoginStopsAutoLogin) { |
// Set up mocks to check login success. |
- ExpectSuccessfulLogin(public_session_user_id_, ""); |
+ UserContext user_context(public_session_user_id_); |
+ user_context.SetUserIDHash(user_context.GetUserID()); |
+ ExpectSuccessfulLogin(user_context); |
existing_user_controller()->OnSigninScreenReady(); |
SetAutoLoginPolicy(kPublicSessionAccountId, kAutoLoginLongDelay); |
ASSERT_TRUE(auto_login_timer()); |