ipc/ipc_test_sink.cc

   1 // Copyright (c) 2011 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 "ipc/ipc_test_sink.h"
   6
   7 #include "ipc/ipc_listener.h"
   8 #include "ipc/ipc_message.h"
   9
  10 namespace IPC {
  11
  12 TestSink::TestSink() {
  13 }
  14
  15 TestSink::~TestSink() {
  16 }
  17
  18 bool TestSink::Send(Message* message) {
  19   OnMessageReceived(*message);
  20   delete message;
  21   return true;
  22 }
  23
  24 bool TestSink::OnMessageReceived(const Message& msg) {
  25   ObserverListBase<Listener>::Iterator it(filter_list_);
  26   Listener* observer;
  27   while ((observer = it.GetNext()) != NULL) {
  28     if (observer->OnMessageReceived(msg))
  29       return true;
  30   }
  31
  32   // No filter handled the message, so store it.
  33   messages_.push_back(Message(msg));
  34   return true;
  35 }
  36
  37 void TestSink::ClearMessages() {
  38   messages_.clear();
  39 }
  40
  41 const Message* TestSink::GetMessageAt(size_t index) const {
  42   if (index >= messages_.size())
  43     return NULL;
  44   return &messages_[index];
  45 }
  46
  47 const Message* TestSink::GetFirstMessageMatching(uint32 id) const {
  48   for (size_t i = 0; i < messages_.size(); i++) {
  49     if (messages_[i].type() == id)
  50       return &messages_[i];
  51   }
  52   return NULL;
  53 }
  54
  55 const Message* TestSink::GetUniqueMessageMatching(uint32 id) const {
  56   size_t found_index = 0;
  57   size_t found_count = 0;
  58   for (size_t i = 0; i < messages_.size(); i++) {
  59     if (messages_[i].type() == id) {
  60       found_count++;
  61       found_index = i;
  62     }
  63   }
  64   if (found_count != 1)
  65     return NULL;  // Didn't find a unique one.
  66   return &messages_[found_index];
  67 }
  68
  69 void TestSink::AddFilter(Listener* filter) {
  70   filter_list_.AddObserver(filter);
  71 }
  72
  73 void TestSink::RemoveFilter(Listener* filter) {
  74   filter_list_.RemoveObserver(filter);
  75 }
  76
  77 }  // namespace IPC