diff options
author | brettw@google.com <brettw@google.com@0039d316-1c4b-4281-b951-d872f2087c98> | 2011-01-20 23:50:27 +0000 |
---|---|---|
committer | brettw@google.com <brettw@google.com@0039d316-1c4b-4281-b951-d872f2087c98> | 2011-01-20 23:50:27 +0000 |
commit | 3ff2a10362f73dfaf2b6509f6553f963cdbc7b7d (patch) | |
tree | 2b2e824e76b9b6c473ab91933a1759e10c9e685d /ipc/ipc_test_sink.h | |
parent | f63ad079d7c5b1bcd6a8b42aed298de328be0811 (diff) | |
download | chromium_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.h | 91 |
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_ |