Home | History | Annotate | Download | only in notifier 
      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 
     21 AckHandleMatcher::AckHandleMatcher(const AckHandle& handle)
     22   : handle_(handle) {}
     23 
     24 bool AckHandleMatcher::operator()(
     25     const syncer::Invalidation& invalidation) const {
     26   return handle_.Equals(invalidation.ack_handle());
     27 }
     28 
     29 }  // namespace
     30 
     31 MockAckHandler::MockAckHandler() {}
     32 
     33 MockAckHandler::~MockAckHandler() {}
     34 
     35 void MockAckHandler::RegisterInvalidation(Invalidation* invalidation) {
     36   unacked_invalidations_.push_back(*invalidation);
     37   invalidation->set_ack_handler(WeakHandleThis());
     38 }
     39 
     40 void MockAckHandler::RegisterUnsentInvalidation(Invalidation* invalidation) {
     41   unsent_invalidations_.push_back(*invalidation);
     42 }
     43 
     44 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 
     53 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 
     62 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 
     71 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 
     80 bool MockAckHandler::AllInvalidationsAccountedFor() const {
     81   return unacked_invalidations_.empty() && unrecovered_drop_events_.empty();
     82 }
     83 
     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 
    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 
    119 WeakHandle<AckHandler> MockAckHandler::WeakHandleThis() {
    120   return WeakHandle<AckHandler>(AsWeakPtr());
    121 }
    122 
    123 }  // namespace syncer
    124