[Sync] Prevent uploading throttled datatypes.

chrome/browser/sync/sessions/sync_session_context_unittest.cc

Index: chrome/browser/sync/sessions/sync_session_context_unittest.cc
diff --git a/chrome/browser/sync/sessions/sync_session_context_unittest.cc b/chrome/browser/sync/sessions/sync_session_context_unittest.cc
new file mode 100755
index 0000000000000000000000000000000000000000..1560458c95a50798c996cbb75e99b393a5268b7a
--- /dev/null
+++ b/chrome/browser/sync/sessions/sync_session_context_unittest.cc
@@ -0,0 +1,47 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/sync/sessions/sync_session_context.h"
+
+#include "chrome/browser/sync/syncable/model_type.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace browser_sync {
+namespace sessions {
+TEST(SyncSessionContextTest, AddUnthrottleTimeTest) {
+  syncable::ModelTypeSet types;
+  types.insert(syncable::BOOKMARKS);
+  types.insert(syncable::PASSWORDS);
+
+  SyncSessionContext context;
+  base::TimeTicks now = base::TimeTicks::Now();
+  context.SetUnthrottleTime(types, base::TimeTicks::Now());
+
+  EXPECT_EQ(context.unthrottle_times_.size(), 2);
+  EXPECT_EQ(context.unthrottle_times_[syncable::BOOKMARKS], now);
+  EXPECT_EQ(context.unthrottle_times_[syncable::PASSWORDS], now);
+}
+
+TEST(SyncSessionContextTest, GetCurrentlyThrottledTypesTest) {
+  syncable::ModelTypeSet types;
+  types.insert(syncable::BOOKMARKS);
+  types.insert(syncable::PASSWORDS);
+
+  SyncSessionContext context;
+  base::TimeTicks now = base::TimeTicks::Now();
+
+  // Now update the throttled types with time set to 10 seconds earlier from
+  // now.
+  context.SetUnthrottleTime(types, now - base::TimeDelta::FromSeconds(10));
+  context.UpdateThrottledTypes(base::TimeTicks::Now());
+  EXPECT_EQ(context.GetThrottledTypes(), syncable::ModelTypeSet());
+
+  // Now update the throttled types with time set to 2 hours from now.
+  context.SetUnthrottleTime(types, now + base::TimeDelta::FromSeconds(1200));
+  context.UpdateThrottledTypes(base::TimeTicks::Now());
+  EXPECT_EQ(context.GetThrottledTypes(), types);
+}
+}  // namespace sessions.
+}  // namespace browser_sync
+