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.
5 #include "ipc/ipc_test_sink.h"
7 #include "ipc/ipc_message.h"
11 TestSink::TestSink() {
14 TestSink::~TestSink() {
17 bool TestSink::Send(Message
* message
) {
18 OnMessageReceived(*message
);
23 bool TestSink::OnMessageReceived(const Message
& msg
) {
24 ObserverListBase
<Listener
>::Iterator
it(filter_list_
);
26 while ((observer
= it
.GetNext()) != NULL
) {
27 if (observer
->OnMessageReceived(msg
))
31 // No filter handled the message, so store it.
32 messages_
.push_back(Message(msg
));
36 void TestSink::ClearMessages() {
40 const Message
* TestSink::GetMessageAt(size_t index
) const {
41 if (index
>= messages_
.size())
43 return &messages_
[index
];
46 const Message
* TestSink::GetFirstMessageMatching(uint32 id
) const {
47 for (size_t i
= 0; i
< messages_
.size(); i
++) {
48 if (messages_
[i
].type() == id
)
54 const Message
* TestSink::GetUniqueMessageMatching(uint32 id
) const {
55 size_t found_index
= 0;
56 size_t found_count
= 0;
57 for (size_t i
= 0; i
< messages_
.size(); i
++) {
58 if (messages_
[i
].type() == id
) {
64 return NULL
; // Didn't find a unique one.
65 return &messages_
[found_index
];
68 void TestSink::AddFilter(Listener
* filter
) {
69 filter_list_
.AddObserver(filter
);
72 void TestSink::RemoveFilter(Listener
* filter
) {
73 filter_list_
.RemoveObserver(filter
);