summaryrefslogtreecommitdiffstats
path: root/ipc/ipc_test_sink.h
diff options
context:
space:
mode:
authorbrettw@google.com <brettw@google.com@0039d316-1c4b-4281-b951-d872f2087c98>2011-01-20 23:50:27 +0000
committerbrettw@google.com <brettw@google.com@0039d316-1c4b-4281-b951-d872f2087c98>2011-01-20 23:50:27 +0000
commit3ff2a10362f73dfaf2b6509f6553f963cdbc7b7d (patch)
tree2b2e824e76b9b6c473ab91933a1759e10c9e685d /ipc/ipc_test_sink.h
parentf63ad079d7c5b1bcd6a8b42aed298de328be0811 (diff)
downloadchromium_src-3ff2a10362f73dfaf2b6509f6553f963cdbc7b7d.zip
chromium_src-3ff2a10362f73dfaf2b6509f6553f963cdbc7b7d.tar.gz
chromium_src-3ff2a10362f73dfaf2b6509f6553f963cdbc7b7d.tar.bz2
Move the TestSink for doing IPC tests from chrome/common into IPC and create a new IPC test_support project that references it.
This is necessary because I want to make a test that uses the sink outside of chrome. TEST=this is a test BUG=none Review URL: http://codereview.chromium.org/6290008 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@72044 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'ipc/ipc_test_sink.h')
-rw-r--r--ipc/ipc_test_sink.h91
1 files changed, 91 insertions, 0 deletions
diff --git a/ipc/ipc_test_sink.h b/ipc/ipc_test_sink.h
new file mode 100644
index 0000000..c6114f1
--- /dev/null
+++ b/ipc/ipc_test_sink.h
@@ -0,0 +1,91 @@
+// Copyright (c) 2011 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.
+
+#ifndef IPC_IPC_TEST_SINK_H_
+#define IPC_IPC_TEST_SINK_H_
+#pragma once
+
+#include <utility>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "ipc/ipc_channel.h"
+
+namespace IPC {
+
+class Message;
+
+// This test sink provides a "sink" for IPC messages that are sent. It allows
+// the caller to query messages received in various different ways. It is
+// designed for tests for objects that use the IPC system.
+//
+// Typical usage:
+//
+// test_sink.ClearMessages();
+// do_something();
+//
+// // We should have gotten exactly one update state message.
+// EXPECT_TRUE(test_sink.GetUniqeMessageMatching(ViewHostMsg_Update::ID));
+// // ...and no start load messages.
+// EXPECT_FALSE(test_sink.GetFirstMessageMatching(ViewHostMsg_Start::ID));
+//
+// // Now inspect a message. This assumes a message that was declared like
+// // this: IPC_MESSAGE_ROUTED2(ViewMsg_Foo, bool, int)
+// IPC::Message* msg = test_sink.GetFirstMessageMatching(ViewMsg_Foo::ID));
+// ASSERT_TRUE(msg);
+// bool first_param;
+// int second_param;
+// ViewMsg_Foo::Read(msg, &first_param, &second_param);
+//
+// // Go on to the next phase of the test.
+// test_sink.ClearMessages();
+//
+// To hook up the sink, all you need to do is call OnMessageReceived when a
+// message is received.
+class TestSink : public Channel {
+ public:
+ TestSink();
+ ~TestSink();
+
+ // Interface in IPC::Channel. This copies the message to the sink and then
+ // deletes it.
+ virtual bool Send(IPC::Message* message);
+
+ // Used by the source of the messages to send the message to the sink. This
+ // will make a copy of the message and store it in the list.
+ bool OnMessageReceived(const Message& msg);
+
+ // Returns the number of messages in the queue.
+ size_t message_count() const { return messages_.size(); }
+
+ // Clears the message queue of saved messages.
+ void ClearMessages();
+
+ // Returns the message at the given index in the queue. The index may be out
+ // of range, in which case the return value is NULL. The returned pointer will
+ // only be valid until another message is received or the list is cleared.
+ const Message* GetMessageAt(size_t index) const;
+
+ // Returns the first message with the given ID in the queue. If there is no
+ // message with the given ID, returns NULL. The returned pointer will only be
+ // valid until another message is received or the list is cleared.
+ const Message* GetFirstMessageMatching(uint32 id) const;
+
+ // Returns the message with the given ID in the queue. If there is no such
+ // message or there is more than one of that message, this will return NULL
+ // (with the expectation that you'll do an ASSERT_TRUE() on the result).
+ // The returned pointer will only be valid until another message is received
+ // or the list is cleared.
+ const Message* GetUniqueMessageMatching(uint32 id) const;
+
+ private:
+ // The actual list of received messages.
+ std::vector<Message> messages_;
+
+ DISALLOW_COPY_AND_ASSIGN(TestSink);
+};
+
+} // namespace IPC
+
+#endif // IPC_IPC_TEST_SINK_H_