| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/sync/sessions/sync_session_context.h" | 5 #include "chrome/browser/sync/sessions/sync_session_context.h" |
| 6 | 6 |
| 7 #include "chrome/browser/sync/syncable/model_type.h" | 7 #include "chrome/browser/sync/syncable/model_type.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 | 9 |
| 10 namespace browser_sync { | 10 namespace browser_sync { |
| 11 namespace sessions { | 11 namespace sessions { |
| 12 TEST(SyncSessionContextTest, AddUnthrottleTimeTest) { | 12 TEST(SyncSessionContextTest, AddUnthrottleTimeTest) { |
| 13 syncable::ModelTypeSet types; | 13 syncable::ModelEnumSet types; |
| 14 types.insert(syncable::BOOKMARKS); | 14 types.Put(syncable::BOOKMARKS); |
| 15 types.insert(syncable::PASSWORDS); | 15 types.Put(syncable::PASSWORDS); |
| 16 | 16 |
| 17 SyncSessionContext context; | 17 SyncSessionContext context; |
| 18 base::TimeTicks now = base::TimeTicks::Now(); | 18 base::TimeTicks now = base::TimeTicks::Now(); |
| 19 context.SetUnthrottleTime(types, now); | 19 context.SetUnthrottleTime(types, now); |
| 20 | 20 |
| 21 EXPECT_EQ(context.unthrottle_times_.size(), 2U); | 21 EXPECT_EQ(context.unthrottle_times_.size(), 2U); |
| 22 EXPECT_EQ(context.unthrottle_times_[syncable::BOOKMARKS], now); | 22 EXPECT_EQ(context.unthrottle_times_[syncable::BOOKMARKS], now); |
| 23 EXPECT_EQ(context.unthrottle_times_[syncable::PASSWORDS], now); | 23 EXPECT_EQ(context.unthrottle_times_[syncable::PASSWORDS], now); |
| 24 } | 24 } |
| 25 | 25 |
| 26 TEST(SyncSessionContextTest, GetCurrentlyThrottledTypesTest) { | 26 TEST(SyncSessionContextTest, GetCurrentlyThrottledTypesTest) { |
| 27 syncable::ModelTypeSet types; | 27 syncable::ModelEnumSet types; |
| 28 types.insert(syncable::BOOKMARKS); | 28 types.Put(syncable::BOOKMARKS); |
| 29 types.insert(syncable::PASSWORDS); | 29 types.Put(syncable::PASSWORDS); |
| 30 | 30 |
| 31 SyncSessionContext context; | 31 SyncSessionContext context; |
| 32 base::TimeTicks now = base::TimeTicks::Now(); | 32 base::TimeTicks now = base::TimeTicks::Now(); |
| 33 | 33 |
| 34 // Now update the throttled types with time set to 10 seconds earlier from | 34 // Now update the throttled types with time set to 10 seconds earlier from |
| 35 // now. | 35 // now. |
| 36 context.SetUnthrottleTime(types, now - base::TimeDelta::FromSeconds(10)); | 36 context.SetUnthrottleTime(types, now - base::TimeDelta::FromSeconds(10)); |
| 37 context.PruneUnthrottledTypes(base::TimeTicks::Now()); | 37 context.PruneUnthrottledTypes(base::TimeTicks::Now()); |
| 38 EXPECT_EQ(context.GetThrottledTypes(), syncable::ModelTypeSet()); | 38 EXPECT_TRUE(context.GetThrottledTypes().Empty()); |
| 39 | 39 |
| 40 // Now update the throttled types with time set to 2 hours from now. | 40 // Now update the throttled types with time set to 2 hours from now. |
| 41 context.SetUnthrottleTime(types, now + base::TimeDelta::FromSeconds(1200)); | 41 context.SetUnthrottleTime(types, now + base::TimeDelta::FromSeconds(1200)); |
| 42 context.PruneUnthrottledTypes(base::TimeTicks::Now()); | 42 context.PruneUnthrottledTypes(base::TimeTicks::Now()); |
| 43 EXPECT_EQ(context.GetThrottledTypes(), types); | 43 EXPECT_TRUE(context.GetThrottledTypes().Equals(types)); |
| 44 } | 44 } |
| 45 } // namespace sessions. | 45 } // namespace sessions. |
| 46 } // namespace browser_sync | 46 } // namespace browser_sync |
| 47 | 47 |
| OLD | NEW |