Chromium Code Reviews| Index: chrome/browser/sync/internal_api/syncapi_unittest.cc |
| diff --git a/chrome/browser/sync/internal_api/syncapi_unittest.cc b/chrome/browser/sync/internal_api/syncapi_unittest.cc |
| index 0c3cd6f10874184c3738a5103502cb2769b863fb..33ef803cc6862006a4f1a1da24e2e10363f9b682 100644 |
| --- a/chrome/browser/sync/internal_api/syncapi_unittest.cc |
| +++ b/chrome/browser/sync/internal_api/syncapi_unittest.cc |
| @@ -46,6 +46,7 @@ |
| #include "chrome/browser/sync/protocol/sync.pb.h" |
| #include "chrome/browser/sync/sessions/sync_session.h" |
| #include "chrome/browser/sync/syncable/directory_manager.h" |
| +#include "chrome/browser/sync/syncable/model_type_test_util.h" |
| #include "chrome/browser/sync/syncable/syncable.h" |
| #include "chrome/browser/sync/syncable/syncable_id.h" |
| #include "chrome/browser/sync/test/engine/test_user_share.h" |
| @@ -72,10 +73,9 @@ using browser_sync::ModelSafeWorkerRegistrar; |
| using browser_sync::sessions::SyncSessionSnapshot; |
| using browser_sync::WeakHandle; |
| using content::BrowserThread; |
| -using syncable::GetAllRealModelTypes; |
| using syncable::kEncryptedString; |
| +using syncable::ModelEnumSet; |
| using syncable::ModelType; |
| -using syncable::ModelTypeSet; |
| using test::ExpectDictStringValue; |
| using testing::_; |
| using testing::AnyNumber; |
| @@ -632,7 +632,7 @@ class SyncManagerObserverMock : public SyncManager::Observer { |
| MOCK_METHOD0(OnClearServerDataFailed, void()); // NOLINT |
| MOCK_METHOD0(OnClearServerDataSucceeded, void()); // NOLINT |
| MOCK_METHOD2(OnEncryptedTypesChanged, |
| - void(const ModelTypeSet&, bool)); // NOLINT |
| + void(ModelEnumSet, bool)); // NOLINT |
| MOCK_METHOD0(OnEncryptionComplete, void()); // NOLINT |
| MOCK_METHOD1(OnActionableError, |
| void(const browser_sync::SyncProtocolError&)); // NOLINT |
| @@ -1197,9 +1197,9 @@ TEST_F(SyncManagerTest, RefreshEncryptionReady) { |
| EXPECT_TRUE(SetUpEncryption(true)); |
| EXPECT_CALL(observer_, OnEncryptionComplete()); |
| sync_manager_.RefreshEncryption(); |
| - syncable::ModelTypeSet encrypted_types = |
| + syncable::ModelEnumSet encrypted_types = |
| sync_manager_.GetEncryptedDataTypesForTest(); |
| - EXPECT_EQ(1U, encrypted_types.count(syncable::PASSWORDS)); |
| + EXPECT_TRUE(encrypted_types.Has(syncable::PASSWORDS)); |
| EXPECT_FALSE(sync_manager_.EncryptEverythingEnabledForTest()); |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| @@ -1217,9 +1217,9 @@ TEST_F(SyncManagerTest, RefreshEncryptionReady) { |
| TEST_F(SyncManagerTest, RefreshEncryptionNotReady) { |
| // Don't set up encryption (no nigori node created). |
| sync_manager_.RefreshEncryption(); // Should fail. |
| - syncable::ModelTypeSet encrypted_types = |
| + syncable::ModelEnumSet encrypted_types = |
| sync_manager_.GetEncryptedDataTypesForTest(); |
| - EXPECT_EQ(1U, encrypted_types.count(syncable::PASSWORDS)); // Hardcoded. |
| + EXPECT_TRUE(encrypted_types.Has(syncable::PASSWORDS)); // Hardcoded. |
| EXPECT_FALSE(sync_manager_.EncryptEverythingEnabledForTest()); |
| } |
| @@ -1228,9 +1228,9 @@ TEST_F(SyncManagerTest, RefreshEncryptionEmptyNigori) { |
| EXPECT_TRUE(SetUpEncryption(false)); |
| EXPECT_CALL(observer_, OnEncryptionComplete()); |
| sync_manager_.RefreshEncryption(); // Should write to nigori. |
| - syncable::ModelTypeSet encrypted_types = |
| + syncable::ModelEnumSet encrypted_types = |
| sync_manager_.GetEncryptedDataTypesForTest(); |
| - EXPECT_EQ(1U, encrypted_types.count(syncable::PASSWORDS)); // Hardcoded. |
| + EXPECT_TRUE(encrypted_types.Has(syncable::PASSWORDS)); // Hardcoded. |
| EXPECT_FALSE(sync_manager_.EncryptEverythingEnabledForTest()); |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| @@ -1247,7 +1247,8 @@ TEST_F(SyncManagerTest, RefreshEncryptionEmptyNigori) { |
| TEST_F(SyncManagerTest, EncryptDataTypesWithNoData) { |
| EXPECT_TRUE(SetUpEncryption(true)); |
| EXPECT_CALL(observer_, |
| - OnEncryptedTypesChanged(GetAllRealModelTypes(), true)); |
| + OnEncryptedTypesChanged( |
| + HasModelTypes(syncable::ModelEnumSet::All()), true)); |
| EXPECT_CALL(observer_, OnEncryptionComplete()); |
| sync_manager_.EnableEncryptEverything(); |
| EXPECT_TRUE(sync_manager_.EncryptEverythingEnabledForTest()); |
| @@ -1283,7 +1284,8 @@ TEST_F(SyncManagerTest, EncryptDataTypesWithData) { |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| - EXPECT_EQ(Cryptographer::SensitiveTypes(), GetEncryptedTypes(&trans)); |
| + EXPECT_TRUE(Cryptographer::SensitiveTypes().Equals( |
| + GetEncryptedTypes(&trans))); |
| EXPECT_TRUE(syncable::VerifyDataTypeEncryptionForTest( |
| trans.GetWrappedTrans(), |
| trans.GetCryptographer(), |
| @@ -1302,13 +1304,15 @@ TEST_F(SyncManagerTest, EncryptDataTypesWithData) { |
| } |
| EXPECT_CALL(observer_, |
| - OnEncryptedTypesChanged(GetAllRealModelTypes(), true)); |
| + OnEncryptedTypesChanged( |
| + HasModelTypes(syncable::ModelEnumSet::All()), true)); |
| EXPECT_CALL(observer_, OnEncryptionComplete()); |
| sync_manager_.EnableEncryptEverything(); |
| EXPECT_TRUE(sync_manager_.EncryptEverythingEnabledForTest()); |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| - EXPECT_EQ(GetAllRealModelTypes(), GetEncryptedTypes(&trans)); |
| + EXPECT_TRUE(GetEncryptedTypes(&trans) |
|
Nicolas Zea
2011/12/08 01:49:37
make consistent with below lines
akalin
2011/12/09 19:10:10
Done.
|
| + .Equals(syncable::ModelEnumSet::All())); |
| EXPECT_TRUE(syncable::VerifyDataTypeEncryptionForTest( |
| trans.GetWrappedTrans(), |
| trans.GetCryptographer(), |
| @@ -1334,7 +1338,8 @@ TEST_F(SyncManagerTest, EncryptDataTypesWithData) { |
| EXPECT_TRUE(sync_manager_.EncryptEverythingEnabledForTest()); |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| - EXPECT_EQ(GetAllRealModelTypes(), GetEncryptedTypes(&trans)); |
| + EXPECT_TRUE(syncable::ModelEnumSet::All().Equals( |
| + GetEncryptedTypes(&trans))); |
| EXPECT_TRUE(syncable::VerifyDataTypeEncryptionForTest( |
| trans.GetWrappedTrans(), |
| trans.GetCryptographer(), |
| @@ -1494,14 +1499,16 @@ TEST_F(SyncManagerTest, EncryptBookmarksWithLegacyData) { |
| } |
| EXPECT_CALL(observer_, |
| - OnEncryptedTypesChanged(GetAllRealModelTypes(), true)); |
| + OnEncryptedTypesChanged( |
| + HasModelTypes(syncable::ModelEnumSet::All()), true)); |
| EXPECT_CALL(observer_, OnEncryptionComplete()); |
| sync_manager_.EnableEncryptEverything(); |
| EXPECT_TRUE(sync_manager_.EncryptEverythingEnabledForTest()); |
| { |
| ReadTransaction trans(FROM_HERE, sync_manager_.GetUserShare()); |
| - EXPECT_EQ(GetAllRealModelTypes(), GetEncryptedTypes(&trans)); |
| + EXPECT_TRUE(syncable::ModelEnumSet::All().Equals( |
| + GetEncryptedTypes(&trans))); |
| EXPECT_TRUE(syncable::VerifyDataTypeEncryptionForTest( |
| trans.GetWrappedTrans(), |
| trans.GetCryptographer(), |