| // Copyright 2020 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "components/app_restore/app_restore_info.h" |
| |
| #include <map> |
| #include <set> |
| |
| #include "components/account_id/account_id.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace app_restore { |
| |
| class FakeAppRestoreInfoObserver : public AppRestoreInfo::Observer { |
| public: |
| void OnRestorePrefChanged(const AccountId& account_id, |
| bool could_restore) override { |
| if (could_restore) |
| restore_prefs_.insert(account_id); |
| else |
| restore_prefs_.erase(account_id); |
| |
| restore_pref_changed_count_[account_id]++; |
| } |
| |
| bool CanPerformRestore(const AccountId& account_id) const { |
| return restore_prefs_.find(account_id) != restore_prefs_.end(); |
| } |
| |
| int RestorePrefChangedCount(const AccountId& account_id) const { |
| auto it = restore_pref_changed_count_.find(account_id); |
| return it != restore_pref_changed_count_.end() ? it->second : 0; |
| } |
| |
| private: |
| std::set<AccountId> restore_prefs_; |
| std::map<AccountId, int> restore_pref_changed_count_; |
| }; |
| |
| using AppRestoreInfoTest = testing::Test; |
| |
| // Test the OnRestorePrefChanged callback when the restore pref is reset. |
| TEST_F(AppRestoreInfoTest, RestorePref) { |
| AccountId account_id1 = AccountId::FromUserEmail("aaa@gmail.com"); |
| AccountId account_id2 = AccountId::FromUserEmail("bbb@gmail.com"); |
| |
| AppRestoreInfo::GetInstance()->SetRestorePref(account_id1, true); |
| |
| FakeAppRestoreInfoObserver observer; |
| AppRestoreInfo::GetInstance()->AddObserver(&observer); |
| |
| // Not change the restore flag. |
| AppRestoreInfo::GetInstance()->SetRestorePref(account_id1, true); |
| AppRestoreInfo::GetInstance()->SetRestorePref(account_id2, false); |
| EXPECT_EQ(0, observer.RestorePrefChangedCount(account_id1)); |
| EXPECT_EQ(0, observer.RestorePrefChangedCount(account_id2)); |
| EXPECT_TRUE(AppRestoreInfo::GetInstance()->CanPerformRestore(account_id1)); |
| EXPECT_FALSE(AppRestoreInfo::GetInstance()->CanPerformRestore(account_id2)); |
| |
| // Change the restore flag. |
| AppRestoreInfo::GetInstance()->SetRestorePref(account_id1, false); |
| AppRestoreInfo::GetInstance()->SetRestorePref(account_id2, true); |
| EXPECT_EQ(1, observer.RestorePrefChangedCount(account_id1)); |
| EXPECT_EQ(1, observer.RestorePrefChangedCount(account_id2)); |
| EXPECT_FALSE(observer.CanPerformRestore(account_id1)); |
| EXPECT_TRUE(observer.CanPerformRestore(account_id2)); |
| EXPECT_FALSE(AppRestoreInfo::GetInstance()->CanPerformRestore(account_id1)); |
| EXPECT_TRUE(AppRestoreInfo::GetInstance()->CanPerformRestore(account_id2)); |
| |
| AppRestoreInfo::GetInstance()->RemoveObserver(&observer); |
| } |
| |
| } // namespace app_restore |