1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// Copyright 2014 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 "components/invalidation/impl/mock_ack_handler.h"
#include <algorithm>
#include "base/threading/thread_task_runner_handle.h"
#include "components/invalidation/public/ack_handle.h"
#include "components/invalidation/public/invalidation.h"
namespace syncer {
namespace {
struct AckHandleMatcher {
AckHandleMatcher(const AckHandle& handle);
bool operator()(const syncer::Invalidation& invalidation) const;
syncer::AckHandle handle_;
};
AckHandleMatcher::AckHandleMatcher(const AckHandle& handle)
: handle_(handle) {}
bool AckHandleMatcher::operator()(
const syncer::Invalidation& invalidation) const {
return handle_.Equals(invalidation.ack_handle());
}
} // namespace
MockAckHandler::MockAckHandler() {}
MockAckHandler::~MockAckHandler() {}
void MockAckHandler::RegisterInvalidation(Invalidation* invalidation) {
unacked_invalidations_.push_back(*invalidation);
invalidation->SetAckHandler(AsWeakPtr(), base::ThreadTaskRunnerHandle::Get());
}
void MockAckHandler::RegisterUnsentInvalidation(Invalidation* invalidation) {
unsent_invalidations_.push_back(*invalidation);
}
bool MockAckHandler::IsUnacked(const Invalidation& invalidation) const {
AckHandleMatcher matcher(invalidation.ack_handle());
InvalidationVector::const_iterator it = std::find_if(
unacked_invalidations_.begin(),
unacked_invalidations_.end(),
matcher);
return it != unacked_invalidations_.end();
}
bool MockAckHandler::IsAcknowledged(const Invalidation& invalidation) const {
AckHandleMatcher matcher(invalidation.ack_handle());
InvalidationVector::const_iterator it = std::find_if(
acked_invalidations_.begin(),
acked_invalidations_.end(),
matcher);
return it != acked_invalidations_.end();
}
bool MockAckHandler::IsDropped(const Invalidation& invalidation) const {
AckHandleMatcher matcher(invalidation.ack_handle());
InvalidationVector::const_iterator it = std::find_if(
dropped_invalidations_.begin(),
dropped_invalidations_.end(),
matcher);
return it != dropped_invalidations_.end();
}
bool MockAckHandler::IsUnsent(const Invalidation& invalidation) const {
AckHandleMatcher matcher(invalidation.ack_handle());
InvalidationVector::const_iterator it1 = std::find_if(
unsent_invalidations_.begin(),
unsent_invalidations_.end(),
matcher);
return it1 != unsent_invalidations_.end();
}
bool MockAckHandler::AllInvalidationsAccountedFor() const {
return unacked_invalidations_.empty() && unrecovered_drop_events_.empty();
}
void MockAckHandler::Acknowledge(
const invalidation::ObjectId& id,
const AckHandle& handle) {
AckHandleMatcher matcher(handle);
InvalidationVector::iterator it = std::find_if(
unacked_invalidations_.begin(),
unacked_invalidations_.end(),
matcher);
if (it != unacked_invalidations_.end()) {
acked_invalidations_.push_back(*it);
unacked_invalidations_.erase(it);
}
IdHandleMap::iterator it2 = unrecovered_drop_events_.find(id);
if (it2 != unrecovered_drop_events_.end() && it2->second.Equals(handle)) {
unrecovered_drop_events_.erase(it2);
}
}
void MockAckHandler::Drop(
const invalidation::ObjectId& id,
const AckHandle& handle) {
AckHandleMatcher matcher(handle);
InvalidationVector::iterator it = std::find_if(
unacked_invalidations_.begin(),
unacked_invalidations_.end(),
matcher);
if (it != unacked_invalidations_.end()) {
dropped_invalidations_.push_back(*it);
unacked_invalidations_.erase(it);
}
unrecovered_drop_events_.erase(id);
unrecovered_drop_events_.insert(std::make_pair(id, handle));
}
} // namespace syncer
|
