cc: Make separate interface for BeginFrame ipc from OutputSurface

ipc/ipc_forwarding_message_filter_unittest.cc

Index: ipc/ipc_forwarding_message_filter_unittest.cc
diff --git a/ipc/ipc_forwarding_message_filter_unittest.cc b/ipc/ipc_forwarding_message_filter_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b611bad61e467d651f6743adec7b6a6678d57839
--- /dev/null
+++ b/ipc/ipc_forwarding_message_filter_unittest.cc
@@ -0,0 +1,87 @@
+// 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 "ipc/ipc_forwarding_message_filter.h"
+
+#include "base/bind.h"
+#include "base/test/test_simple_task_runner.h"
+#include "ipc/ipc_message.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+#define IPC_MESSAGE_IMPL
+#include "ipc/ipc_message_macros.h"
+
+#define IPC_MESSAGE_START TestMsgStart
+
+namespace {
+
+IPC_MESSAGE_ROUTED0(TestMessage);
+
+class ForwardingMessageFilterTestHandler
+    : public base::RefCounted<ForwardingMessageFilterTestHandler> {
+ public:
+  ForwardingMessageFilterTestHandler() : count_(0) {
+  }
+
+  void OnPlusMethod(const IPC::Message& msg) {
+    count_++;
+  }
+
+  void OnMinusMethod(const IPC::Message& msg) {
+    count_--;
+  }
+
+  int count() { return count_; }
+
+  void reset_count() { count_ = 0; }
+
+ private:
+  friend class base::RefCounted<ForwardingMessageFilterTestHandler>;
+  ~ForwardingMessageFilterTestHandler() {}
+
+  int count_;
+};
+
+TEST(ForwardingMessageFilterTest, BasicTest) {
+  scoped_refptr<ForwardingMessageFilterTestHandler> handler =
+      new ForwardingMessageFilterTestHandler;
+  base::TestSimpleTaskRunner* task_runner = new base::TestSimpleTaskRunner;
+  uint32 messages_to_filter[] = {
+    TestMessage::ID
+  };
+  int route_id = 0;
+
+  TestMessage msg(route_id);
+
+  IPC::ForwardingMessageFilter::Handler plus_handler =
+      base::Bind(&ForwardingMessageFilterTestHandler::OnPlusMethod, handler);
+  IPC::ForwardingMessageFilter::Handler minus_handler =
+      base::Bind(&ForwardingMessageFilterTestHandler::OnMinusMethod, handler);
+
+  scoped_refptr<IPC::ForwardingMessageFilter> filter =
+      new IPC::ForwardingMessageFilter(messages_to_filter, 1, task_runner);
+
+  filter->AddRoute(route_id, plus_handler);
+  filter->OnMessageReceived(msg);
+  task_runner->RunPendingTasks();
+  EXPECT_EQ(1, handler->count());
+
+  handler->reset_count();
+  EXPECT_EQ(0, handler->count());
+
+  filter->AddRoute(route_id, minus_handler);
+  filter->OnMessageReceived(msg);
+  task_runner->RunPendingTasks();
+  EXPECT_EQ(0, handler->count());
+
+  handler->reset_count();
+  EXPECT_EQ(0, handler->count());
+
+  filter->RemoveRoute(route_id, plus_handler);
+  filter->OnMessageReceived(msg);
+  task_runner->RunPendingTasks();
+  EXPECT_EQ(-1, handler->count());
+}
+
+}  // namespace