| Index: chrome/browser/policy/cloud/user_policy_signin_service_unittest.cc
|
| diff --git a/chrome/browser/policy/cloud/user_policy_signin_service_unittest.cc b/chrome/browser/policy/cloud/user_policy_signin_service_unittest.cc
|
| index 9bd390857d68ad98a1925e764dbeb9a7bf0c045f..381788832cf00eb8dffa18ceb0993bbec30776d9 100644
|
| --- a/chrome/browser/policy/cloud/user_policy_signin_service_unittest.cc
|
| +++ b/chrome/browser/policy/cloud/user_policy_signin_service_unittest.cc
|
| @@ -65,6 +65,7 @@ namespace policy {
|
|
|
| namespace {
|
|
|
| +const char kTestGaiaId[] = "gaia-id-testuser@test.com";
|
| const char kTestUser[] = "testuser@test.com";
|
|
|
| #if !defined(OS_ANDROID)
|
| @@ -134,8 +135,10 @@ class UserPolicySigninServiceTest : public testing::Test {
|
| // a valid login token, while on other platforms, the login refresh token
|
| // is specified directly.
|
| #if defined(OS_ANDROID)
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth2_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth2_login_refresh_token");
|
| #endif
|
| service->RegisterForPolicy(
|
| kTestUser,
|
| @@ -390,7 +393,7 @@ class UserPolicySigninServiceSignedInTest : public UserPolicySigninServiceTest {
|
|
|
| // Set the user as signed in.
|
| SigninManagerFactory::GetForProfile(profile_.get())->
|
| - SetAuthenticatedUsername(kTestUser);
|
| + SetAuthenticatedAccountInfo(kTestGaiaId, kTestUser);
|
|
|
| // Let the SigninService know that the profile has been created.
|
| content::NotificationService::current()->Notify(
|
| @@ -425,8 +428,10 @@ TEST_F(UserPolicySigninServiceSignedInTest, InitWhileSignedIn) {
|
| ASSERT_FALSE(IsRequestActive());
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // Client registration should be in progress since we now have an oauth token.
|
| EXPECT_EQ(mock_store_->signin_username_, kTestUser);
|
| @@ -444,8 +449,10 @@ TEST_F(UserPolicySigninServiceSignedInTest, InitWhileSignedInOAuthError) {
|
| ASSERT_FALSE(IsRequestActive());
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // Client registration should be in progress since we now have an oauth token.
|
| ASSERT_TRUE(IsRequestActive());
|
| @@ -463,15 +470,17 @@ TEST_F(UserPolicySigninServiceTest, SignInAfterInit) {
|
| ASSERT_FALSE(manager_->core()->service());
|
|
|
| // Now sign in the user.
|
| - SigninManagerFactory::GetForProfile(profile_.get())->SetAuthenticatedUsername(
|
| - kTestUser);
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->SetAuthenticatedAccountInfo(kTestGaiaId, kTestUser);
|
|
|
| // Complete initialization of the store.
|
| mock_store_->NotifyStoreLoaded();
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // UserCloudPolicyManager should be initialized.
|
| EXPECT_EQ(mock_store_->signin_username_, kTestUser);
|
| @@ -487,15 +496,18 @@ TEST_F(UserPolicySigninServiceTest, SignInWithNonEnterpriseUser) {
|
| ASSERT_FALSE(manager_->core()->service());
|
|
|
| // Now sign in a non-enterprise user (blacklisted gmail.com domain).
|
| - SigninManagerFactory::GetForProfile(profile_.get())->SetAuthenticatedUsername(
|
| - "non_enterprise_user@gmail.com");
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->SetAuthenticatedAccountInfo("gaia-id-non_enterprise_user@gmail.com",
|
| + "non_enterprise_user@gmail.com");
|
|
|
| // Complete initialization of the store.
|
| mock_store_->NotifyStoreLoaded();
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // UserCloudPolicyManager should not be initialized and there should be no
|
| // DMToken request active.
|
| @@ -509,12 +521,14 @@ TEST_F(UserPolicySigninServiceTest, UnregisteredClient) {
|
| ASSERT_FALSE(manager_->core()->service());
|
|
|
| // Now sign in the user.
|
| - SigninManagerFactory::GetForProfile(profile_.get())->SetAuthenticatedUsername(
|
| - kTestUser);
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->SetAuthenticatedAccountInfo(kTestGaiaId, kTestUser);
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // UserCloudPolicyManager should be initialized.
|
| EXPECT_EQ(mock_store_->signin_username_, kTestUser);
|
| @@ -537,12 +551,14 @@ TEST_F(UserPolicySigninServiceTest, RegisteredClient) {
|
| ASSERT_FALSE(manager_->core()->service());
|
|
|
| // Now sign in the user.
|
| - SigninManagerFactory::GetForProfile(profile_.get())->SetAuthenticatedUsername(
|
| - kTestUser);
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->SetAuthenticatedAccountInfo(kTestGaiaId, kTestUser);
|
|
|
| // Make oauth token available.
|
| - GetTokenService()->IssueRefreshTokenForUser(kTestUser,
|
| - "oauth_login_refresh_token");
|
| + GetTokenService()->IssueRefreshTokenForUser(
|
| + SigninManagerFactory::GetForProfile(profile_.get())
|
| + ->GetAuthenticatedAccountId(),
|
| + "oauth_login_refresh_token");
|
|
|
| // UserCloudPolicyManager should be initialized.
|
| EXPECT_EQ(mock_store_->signin_username_, kTestUser);
|
|
|