Index: components/invalidation/sync_invalidation_listener_unittest.cc |
diff --git a/components/invalidation/sync_invalidation_listener_unittest.cc b/components/invalidation/sync_invalidation_listener_unittest.cc |
index f7ff05a24ceab9f11789472e87945c7da6893a13..0399039e47ececa05246d9bda0952530810a115d 100644 |
--- a/components/invalidation/sync_invalidation_listener_unittest.cc |
+++ b/components/invalidation/sync_invalidation_listener_unittest.cc |
@@ -18,7 +18,6 @@ |
#include "google/cacheinvalidation/include/types.h" |
#include "jingle/notifier/listener/fake_push_client.h" |
#include "sync/internal_api/public/util/weak_handle.h" |
-#include "sync/notifier/dropped_invalidation_tracker.h" |
#include "sync/notifier/invalidation_util.h" |
#include "sync/notifier/object_id_invalidation_map.h" |
#include "sync/notifier/unacked_invalidation_set_test_util.h" |
@@ -138,8 +137,7 @@ class FakeInvalidationClient : public invalidation::InvalidationClient { |
class FakeDelegate : public SyncInvalidationListener::Delegate { |
public: |
explicit FakeDelegate(SyncInvalidationListener* listener) |
- : state_(TRANSIENT_INVALIDATION_ERROR), |
- drop_handlers_deleter_(&drop_handlers_) {} |
+ : state_(TRANSIENT_INVALIDATION_ERROR) {} |
virtual ~FakeDelegate() {} |
size_t GetInvalidationCount(const ObjectId& id) const { |
@@ -195,17 +193,6 @@ class FakeDelegate : public SyncInvalidationListener::Delegate { |
return state_; |
} |
- DroppedInvalidationTracker* GetDropTrackerForObject(const ObjectId& id) { |
- DropHandlers::iterator it = drop_handlers_.find(id); |
- if (it == drop_handlers_.end()) { |
- drop_handlers_.insert( |
- std::make_pair(id, new DroppedInvalidationTracker(id))); |
- return drop_handlers_.find(id)->second; |
- } else { |
- return it->second; |
- } |
- } |
- |
void AcknowledgeNthInvalidation(const ObjectId& id, size_t n) { |
List& list = invalidations_[id]; |
List::iterator it = list.begin() + n; |
@@ -220,15 +207,19 @@ class FakeDelegate : public SyncInvalidationListener::Delegate { |
} |
void DropNthInvalidation(const ObjectId& id, size_t n) { |
- DroppedInvalidationTracker* drop_tracker = GetDropTrackerForObject(id); |
List& list = invalidations_[id]; |
List::iterator it = list.begin() + n; |
- it->Drop(drop_tracker); |
+ it->Drop(); |
+ dropped_invalidations_map_.erase(id); |
+ dropped_invalidations_map_.insert(std::make_pair(id, *it)); |
} |
void RecoverFromDropEvent(const ObjectId& id) { |
- DroppedInvalidationTracker* drop_tracker = GetDropTrackerForObject(id); |
- drop_tracker->RecordRecoveryFromDropEvent(); |
+ DropMap::iterator it = dropped_invalidations_map_.find(id); |
+ if (it != dropped_invalidations_map_.end()) { |
+ it->second.Acknowledge(); |
+ dropped_invalidations_map_.erase(it); |
+ } |
} |
// SyncInvalidationListener::Delegate implementation. |
@@ -250,14 +241,11 @@ class FakeDelegate : public SyncInvalidationListener::Delegate { |
private: |
typedef std::vector<Invalidation> List; |
typedef std::map<ObjectId, List, ObjectIdLessThan> Map; |
- typedef std::map<ObjectId, |
- DroppedInvalidationTracker*, |
- ObjectIdLessThan> DropHandlers; |
+ typedef std::map<ObjectId, Invalidation, ObjectIdLessThan> DropMap; |
Map invalidations_; |
InvalidatorState state_; |
- DropHandlers drop_handlers_; |
- STLValueDeleter<DropHandlers> drop_handlers_deleter_; |
+ DropMap dropped_invalidations_map_; |
}; |
invalidation::InvalidationClient* CreateFakeInvalidationClient( |