text: // 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
  