Index: chrome/browser/sync/engine/syncer_proto_util_unittest.cc |
diff --git a/chrome/browser/sync/engine/syncer_proto_util_unittest.cc b/chrome/browser/sync/engine/syncer_proto_util_unittest.cc |
index c05b1921c04c1c4da44462c9c914f8db912c848a..1fd0f670c3f1d5cc1a324fda35c406cbce2a7110 100644 |
--- a/chrome/browser/sync/engine/syncer_proto_util_unittest.cc |
+++ b/chrome/browser/sync/engine/syncer_proto_util_unittest.cc |
@@ -18,6 +18,7 @@ |
#include "chrome/browser/sync/protocol/sync.pb.h" |
#include "chrome/browser/sync/syncable/blob.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/test/engine/mock_connection_manager.h" |
#include "chrome/browser/sync/test/engine/test_directory_setter_upper.h" |
@@ -26,7 +27,6 @@ |
using syncable::Blob; |
using syncable::ScopedDirLookup; |
-using syncable::ModelTypeSet; |
using ::testing::_; |
namespace browser_sync { |
@@ -36,7 +36,7 @@ class MockSyncSessionContext : public SyncSessionContext { |
public: |
MockSyncSessionContext() {} |
~MockSyncSessionContext() {} |
- MOCK_METHOD2(SetUnthrottleTime, void(const ModelTypeSet&, |
+ MOCK_METHOD2(SetUnthrottleTime, void(syncable::ModelEnumSet, |
const base::TimeTicks&)); |
}; |
@@ -275,14 +275,14 @@ TEST_F(SyncerProtoUtilTest, HandleThrottlingWithDatatypes) { |
MockSyncSessionContext context; |
SyncProtocolError error; |
error.error_type = browser_sync::THROTTLED; |
- syncable::ModelTypeSet types; |
- types.insert(syncable::BOOKMARKS); |
- types.insert(syncable::PASSWORDS); |
+ syncable::ModelEnumSet types; |
+ types.Put(syncable::BOOKMARKS); |
+ types.Put(syncable::PASSWORDS); |
error.error_data_types = types; |
base::TimeTicks ticks = base::TimeTicks::Now(); |
- EXPECT_CALL(context, SetUnthrottleTime(types, ticks)); |
+ EXPECT_CALL(context, SetUnthrottleTime(HasModelTypes(types), ticks)); |
SyncerProtoUtil::HandleThrottleError(error, ticks, &context, NULL); |
} |