Index: blimp/client/render_widget_message_processor_unittest.cc |
diff --git a/blimp/client/render_widget_message_processor_unittest.cc b/blimp/client/render_widget_message_processor_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..09bf1f732158705e000c4e3922b1319a3ccd3493 |
--- /dev/null |
+++ b/blimp/client/render_widget_message_processor_unittest.cc |
@@ -0,0 +1,147 @@ |
+// Copyright 2015 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 "blimp/client/render_widget_message_processor.h" |
+ |
+#include "base/logging.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "blimp/common/proto/blimp_message.pb.h" |
+#include "blimp/common/proto/compositor.pb.h" |
+#include "blimp/common/proto/render_widget.pb.h" |
+#include "cc/proto/compositor_message.pb.h" |
+#include "net/base/net_errors.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using testing::_; |
+using testing::InvokeArgument; |
+using testing::Ref; |
+using testing::Return; |
+using testing::SaveArg; |
+ |
+namespace blimp { |
+ |
+namespace { |
+class MockBlimpMessageProcessor : public BlimpMessageProcessor { |
+ public: |
+ MockBlimpMessageProcessor() {} |
+ |
+ ~MockBlimpMessageProcessor() override {} |
+ |
+ // Adapts calls from ProcessMessage to MockableProcessMessage by |
+ // unboxing the |message| scoped_ptr for GMock compatibility. |
+ void ProcessMessage(scoped_ptr<BlimpMessage> message, |
+ const net::CompletionCallback& callback) { |
+ MockableProcessMessage(*message); |
+ if (!callback.is_null()) |
+ callback.Run(net::OK); |
+ } |
+ |
+ MOCK_METHOD1(MockableProcessMessage, |
+ void(const BlimpMessage& message)); |
+}; |
+ |
+class MockRenderWidgetMessageDelegate |
+ : public RenderWidgetMessageProcessor::RenderWidgetMessageDelegate { |
+ public: |
+ // RenderWidgetMessageDelegate implementation. |
+ void OnRenderWidgetInitialized() override { |
+ MockableOnRenderWidgetInitialized(); |
+ } |
+ void OnCompositorMessageReceived( |
+ scoped_ptr<cc::proto::CompositorMessage> message) override { |
+ MockableOnCompositorMessageReceived(*message); |
+ } |
+ |
+ MOCK_METHOD0(MockableOnRenderWidgetInitialized, void()); |
+ MOCK_METHOD1(MockableOnCompositorMessageReceived, |
+ void(const cc::proto::CompositorMessage& message)); |
+}; |
+ |
+MATCHER_P2(CompositorMessageEquals, tab_id, rw_id, "") { |
+ return arg.compositor().render_widget_id() == rw_id && |
+ arg.target_tab_id() == tab_id; |
+} |
+ |
+void SendRenderWidgetMessage(BlimpMessageProcessor* processor, |
+ int tab_id, |
+ uint32_t rw_id) { |
+ scoped_ptr<BlimpMessage> message(new BlimpMessage); |
+ message->set_type(BlimpMessage::RENDER_WIDGET); |
+ message->set_target_tab_id(tab_id); |
+ |
+ RenderWidgetMessage* details = message->mutable_render_widget(); |
+ details->set_type(RenderWidgetMessage::INITIALIZE); |
+ details->set_render_widget_id(rw_id); |
+ processor->ProcessMessage(message.Pass(), |
+ net::CompletionCallback()); |
+} |
+ |
+void SendCompositorMessage(BlimpMessageProcessor* processor, |
+ int tab_id, |
+ uint32_t rw_id) { |
+ scoped_ptr<BlimpMessage> message(new BlimpMessage); |
+ message->set_type(BlimpMessage::COMPOSITOR); |
+ message->set_target_tab_id(tab_id); |
+ |
+ CompositorMessage* details = message->mutable_compositor(); |
+ details->set_render_widget_id(rw_id); |
+ processor->ProcessMessage(message.Pass(), |
+ net::CompletionCallback()); |
+} |
+ |
+} // namespace |
+ |
+class RenderWidgetMessageProcessorTest : public testing::Test { |
+ public: |
+ RenderWidgetMessageProcessorTest() |
+ : processor_(&out_processor_) {} |
+ |
+ void SetUp() override { |
+ processor_.SetDelegate(1, &delegate1_); |
+ processor_.SetDelegate(2, &delegate2_); |
+ } |
+ |
+ protected: |
+ MockBlimpMessageProcessor out_processor_; |
+ MockRenderWidgetMessageDelegate delegate1_; |
+ MockRenderWidgetMessageDelegate delegate2_; |
+ |
+ RenderWidgetMessageProcessor processor_; |
+}; |
+ |
+TEST_F(RenderWidgetMessageProcessorTest, DelegateCallsOK) { |
+ EXPECT_CALL(delegate1_, MockableOnRenderWidgetInitialized()).Times(1); |
+ SendRenderWidgetMessage(&processor_, 1, 1U); |
+ |
+ EXPECT_CALL(delegate1_, MockableOnCompositorMessageReceived(_)).Times(1); |
+ SendCompositorMessage(&processor_, 1, 1U); |
+ |
+ EXPECT_CALL(delegate2_, MockableOnRenderWidgetInitialized()).Times(1); |
+ SendRenderWidgetMessage(&processor_, 2, 2U); |
+ |
+ EXPECT_CALL(delegate2_, MockableOnCompositorMessageReceived(_)).Times(1); |
+ SendCompositorMessage(&processor_, 2, 2U); |
+} |
+ |
+TEST_F(RenderWidgetMessageProcessorTest, RepliesHaveCorrectRenderWidgetId) { |
+ SendRenderWidgetMessage(&processor_, 1, 2U); |
+ SendRenderWidgetMessage(&processor_, 2, 1U); |
+ |
+ EXPECT_CALL(out_processor_, |
+ MockableProcessMessage(CompositorMessageEquals(1, 2U))).Times(1); |
+ processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); |
+ |
+ SendRenderWidgetMessage(&processor_, 1, 3U); |
+ |
+ EXPECT_CALL(out_processor_, |
+ MockableProcessMessage(CompositorMessageEquals(1, 3U))).Times(1); |
+ processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); |
+ |
+ EXPECT_CALL(out_processor_, |
+ MockableProcessMessage(CompositorMessageEquals(2, 1U))).Times(1); |
+ processor_.SendCompositorMessage(2, cc::proto::CompositorMessage()); |
+} |
+ |
+} // namespace blimp |