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 |