| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/signin/mutable_profile_oauth2_token_service_delegate.h" | 5 #include "chrome/browser/signin/mutable_profile_oauth2_token_service_delegate.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 #include <string> | 8 #include <string> |
| 9 #include <utility> | 9 #include <utility> |
| 10 #include <vector> | 10 #include <vector> |
| (...skipping 323 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 334 base::RunLoop().RunUntilIdle(); | 334 base::RunLoop().RunUntilIdle(); |
| 335 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_FINISHED_WITH_SUCCESS, | 335 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_FINISHED_WITH_SUCCESS, |
| 336 oauth2_service_delegate_->GetLoadCredentialsState()); | 336 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 337 EXPECT_EQ(2, token_available_count_); | 337 EXPECT_EQ(2, token_available_count_); |
| 338 EXPECT_EQ(0, token_revoked_count_); | 338 EXPECT_EQ(0, token_revoked_count_); |
| 339 EXPECT_EQ(1, tokens_loaded_count_); | 339 EXPECT_EQ(1, tokens_loaded_count_); |
| 340 EXPECT_EQ(1, start_batch_changes_); | 340 EXPECT_EQ(1, start_batch_changes_); |
| 341 EXPECT_EQ(1, end_batch_changes_); | 341 EXPECT_EQ(1, end_batch_changes_); |
| 342 ResetObserverCounts(); | 342 ResetObserverCounts(); |
| 343 | 343 |
| 344 // TODO(fgorski): Enable below when implemented: | 344 EXPECT_TRUE(oauth2_service_delegate_->RefreshTokenIsAvailable("account_id")); |
| 345 // EXPECT_TRUE(oauth2_servive_->RefreshTokenIsAvailable("account_id")); | |
| 346 EXPECT_TRUE(oauth2_service_delegate_->RefreshTokenIsAvailable("account_id2")); | 345 EXPECT_TRUE(oauth2_service_delegate_->RefreshTokenIsAvailable("account_id2")); |
| 347 | 346 |
| 348 oauth2_service_delegate_->RevokeAllCredentials(); | 347 oauth2_service_delegate_->RevokeAllCredentials(); |
| 348 EXPECT_EQ(0, token_available_count_); |
| 349 EXPECT_EQ(2, token_revoked_count_); |
| 350 EXPECT_EQ(0, tokens_loaded_count_); |
| 351 EXPECT_EQ(1, start_batch_changes_); |
| 352 EXPECT_EQ(1, end_batch_changes_); |
| 353 ResetObserverCounts(); |
| 354 } |
| 355 |
| 356 TEST_F(MutableProfileOAuth2TokenServiceDelegateTest, |
| 357 PersistenceLoadCredentialsEmptyPrimaryAccountId_DiceEnabled) { |
| 358 switches::EnableAccountConsistencyDiceForTesting( |
| 359 base::CommandLine::ForCurrentProcess()); |
| 360 |
| 361 // Ensure DB is clean. |
| 362 oauth2_service_delegate_->RevokeAllCredentials(); |
| 363 ResetObserverCounts(); |
| 364 // Perform a load from an empty DB. |
| 365 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_NOT_STARTED, |
| 366 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 367 oauth2_service_delegate_->LoadCredentials(""); |
| 368 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_IN_PROGRESS, |
| 369 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 370 base::RunLoop().RunUntilIdle(); |
| 371 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_FINISHED_WITH_SUCCESS, |
| 372 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 373 EXPECT_EQ(1, start_batch_changes_); |
| 374 EXPECT_EQ(1, end_batch_changes_); |
| 375 ExpectOneTokensLoadedNotification(); |
| 376 |
| 377 // No account should be present in the refresh token as no primary account |
| 378 // was passed to the token service. |
| 379 EXPECT_TRUE(oauth2_service_delegate_->refresh_tokens_.empty()); |
| 380 |
| 381 // Setup a DB with tokens that don't require upgrade and clear memory. |
| 382 oauth2_service_delegate_->UpdateCredentials("account_id", "refresh_token"); |
| 383 oauth2_service_delegate_->UpdateCredentials("account_id2", "refresh_token2"); |
| 384 oauth2_service_delegate_->refresh_tokens_.clear(); |
| 385 EXPECT_EQ(2, start_batch_changes_); |
| 386 EXPECT_EQ(2, end_batch_changes_); |
| 387 ResetObserverCounts(); |
| 388 |
| 389 oauth2_service_delegate_->LoadCredentials(""); |
| 390 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_IN_PROGRESS, |
| 391 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 392 base::RunLoop().RunUntilIdle(); |
| 393 EXPECT_EQ(OAuth2TokenServiceDelegate::LOAD_CREDENTIALS_FINISHED_WITH_SUCCESS, |
| 394 oauth2_service_delegate_->GetLoadCredentialsState()); |
| 395 EXPECT_EQ(2, token_available_count_); |
| 396 EXPECT_EQ(0, token_revoked_count_); |
| 397 EXPECT_EQ(1, tokens_loaded_count_); |
| 398 EXPECT_EQ(1, start_batch_changes_); |
| 399 EXPECT_EQ(1, end_batch_changes_); |
| 400 ResetObserverCounts(); |
| 401 |
| 402 EXPECT_TRUE(oauth2_service_delegate_->RefreshTokenIsAvailable("account_id")); |
| 403 EXPECT_TRUE(oauth2_service_delegate_->RefreshTokenIsAvailable("account_id2")); |
| 404 |
| 405 oauth2_service_delegate_->RevokeAllCredentials(); |
| 349 EXPECT_EQ(0, token_available_count_); | 406 EXPECT_EQ(0, token_available_count_); |
| 350 EXPECT_EQ(2, token_revoked_count_); | 407 EXPECT_EQ(2, token_revoked_count_); |
| 351 EXPECT_EQ(0, tokens_loaded_count_); | 408 EXPECT_EQ(0, tokens_loaded_count_); |
| 352 EXPECT_EQ(1, start_batch_changes_); | 409 EXPECT_EQ(1, start_batch_changes_); |
| 353 EXPECT_EQ(1, end_batch_changes_); | 410 EXPECT_EQ(1, end_batch_changes_); |
| 354 ResetObserverCounts(); | 411 ResetObserverCounts(); |
| 355 } | 412 } |
| 356 | 413 |
| 357 TEST_F(MutableProfileOAuth2TokenServiceDelegateTest, PersistanceNotifications) { | 414 TEST_F(MutableProfileOAuth2TokenServiceDelegateTest, PersistanceNotifications) { |
| 358 oauth2_service_delegate_->UpdateCredentials("account_id", "refresh_token"); | 415 oauth2_service_delegate_->UpdateCredentials("account_id", "refresh_token"); |
| (...skipping 353 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 712 EXPECT_EQ(1, tokens_loaded_count_); | 769 EXPECT_EQ(1, tokens_loaded_count_); |
| 713 EXPECT_EQ(2, token_available_count_); | 770 EXPECT_EQ(2, token_available_count_); |
| 714 EXPECT_EQ(0, token_revoked_count_); | 771 EXPECT_EQ(0, token_revoked_count_); |
| 715 EXPECT_EQ(1, start_batch_changes_); | 772 EXPECT_EQ(1, start_batch_changes_); |
| 716 EXPECT_EQ(1, end_batch_changes_); | 773 EXPECT_EQ(1, end_batch_changes_); |
| 717 EXPECT_TRUE( | 774 EXPECT_TRUE( |
| 718 oauth2_service_delegate_->RefreshTokenIsAvailable(primary_account)); | 775 oauth2_service_delegate_->RefreshTokenIsAvailable(primary_account)); |
| 719 EXPECT_TRUE( | 776 EXPECT_TRUE( |
| 720 oauth2_service_delegate_->RefreshTokenIsAvailable(secondary_account)); | 777 oauth2_service_delegate_->RefreshTokenIsAvailable(secondary_account)); |
| 721 } | 778 } |
| OLD | NEW |