1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "sync/notifier/mock_ack_handler.h"
6
7 #include "sync/internal_api/public/base/ack_handle.h"
8 #include "sync/internal_api/public/base/invalidation.h"
9
10 namespace syncer {
11
12 namespace {
13
14 struct AckHandleMatcher {
15 AckHandleMatcher(const AckHandle& handle);
16 bool operator()(const syncer::Invalidation& invalidation) const;
17
18 syncer::AckHandle handle_;
19 };
20
AckHandleMatcher(const AckHandle & handle)21 AckHandleMatcher::AckHandleMatcher(const AckHandle& handle)
22 : handle_(handle) {}
23
operator ()(const syncer::Invalidation & invalidation) const24 bool AckHandleMatcher::operator()(
25 const syncer::Invalidation& invalidation) const {
26 return handle_.Equals(invalidation.ack_handle());
27 }
28
29 } // namespace
30
MockAckHandler()31 MockAckHandler::MockAckHandler() {}
32
~MockAckHandler()33 MockAckHandler::~MockAckHandler() {}
34
RegisterInvalidation(Invalidation * invalidation)35 void MockAckHandler::RegisterInvalidation(Invalidation* invalidation) {
36 unacked_invalidations_.push_back(*invalidation);
37 invalidation->set_ack_handler(WeakHandleThis());
38 }
39
RegisterUnsentInvalidation(Invalidation * invalidation)40 void MockAckHandler::RegisterUnsentInvalidation(Invalidation* invalidation) {
41 unsent_invalidations_.push_back(*invalidation);
42 }
43
IsUnacked(const Invalidation & invalidation) const44 bool MockAckHandler::IsUnacked(const Invalidation& invalidation) const {
45 AckHandleMatcher matcher(invalidation.ack_handle());
46 InvalidationVector::const_iterator it = std::find_if(
47 unacked_invalidations_.begin(),
48 unacked_invalidations_.end(),
49 matcher);
50 return it != unacked_invalidations_.end();
51 }
52
IsAcknowledged(const Invalidation & invalidation) const53 bool MockAckHandler::IsAcknowledged(const Invalidation& invalidation) const {
54 AckHandleMatcher matcher(invalidation.ack_handle());
55 InvalidationVector::const_iterator it = std::find_if(
56 acked_invalidations_.begin(),
57 acked_invalidations_.end(),
58 matcher);
59 return it != acked_invalidations_.end();
60 }
61
IsDropped(const Invalidation & invalidation) const62 bool MockAckHandler::IsDropped(const Invalidation& invalidation) const {
63 AckHandleMatcher matcher(invalidation.ack_handle());
64 InvalidationVector::const_iterator it = std::find_if(
65 dropped_invalidations_.begin(),
66 dropped_invalidations_.end(),
67 matcher);
68 return it != dropped_invalidations_.end();
69 }
70
IsUnsent(const Invalidation & invalidation) const71 bool MockAckHandler::IsUnsent(const Invalidation& invalidation) const {
72 AckHandleMatcher matcher(invalidation.ack_handle());
73 InvalidationVector::const_iterator it1 = std::find_if(
74 unsent_invalidations_.begin(),
75 unsent_invalidations_.end(),
76 matcher);
77 return it1 != unsent_invalidations_.end();
78 }
79
AllInvalidationsAccountedFor() const80 bool MockAckHandler::AllInvalidationsAccountedFor() const {
81 return unacked_invalidations_.empty() && unrecovered_drop_events_.empty();
82 }
83
Acknowledge(const invalidation::ObjectId & id,const AckHandle & handle)84 void MockAckHandler::Acknowledge(
85 const invalidation::ObjectId& id,
86 const AckHandle& handle) {
87 AckHandleMatcher matcher(handle);
88 InvalidationVector::iterator it = std::find_if(
89 unacked_invalidations_.begin(),
90 unacked_invalidations_.end(),
91 matcher);
92 if (it != unacked_invalidations_.end()) {
93 acked_invalidations_.push_back(*it);
94 unacked_invalidations_.erase(it);
95 }
96
97 IdHandleMap::iterator it2 = unrecovered_drop_events_.find(id);
98 if (it2 != unrecovered_drop_events_.end() && it2->second.Equals(handle)) {
99 unrecovered_drop_events_.erase(it2);
100 }
101 }
102
Drop(const invalidation::ObjectId & id,const AckHandle & handle)103 void MockAckHandler::Drop(
104 const invalidation::ObjectId& id,
105 const AckHandle& handle) {
106 AckHandleMatcher matcher(handle);
107 InvalidationVector::iterator it = std::find_if(
108 unacked_invalidations_.begin(),
109 unacked_invalidations_.end(),
110 matcher);
111 if (it != unacked_invalidations_.end()) {
112 dropped_invalidations_.push_back(*it);
113 unacked_invalidations_.erase(it);
114 }
115 unrecovered_drop_events_.erase(id);
116 unrecovered_drop_events_.insert(std::make_pair(id, handle));
117 }
118
WeakHandleThis()119 WeakHandle<AckHandler> MockAckHandler::WeakHandleThis() {
120 return WeakHandle<AckHandler>(AsWeakPtr());
121 }
122
123 } // namespace syncer
124