Chromium Code Reviews| 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..72b80d2a2461c3dbb3edff58b1ae91054696588e |
| --- /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 { |
|
haibinlu
2015/11/19 19:21:17
blimp/net/test_common.h has MockBlimpMessageProces
David Trainor- moved to gerrit
2015/11/20 17:58:53
Do we need a component that exposes test_common?
haibinlu
2015/11/20 18:20:15
Ah, I see. probably not worth doing that right now
|
| + 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_(new RenderWidgetMessageProcessor(&out_processor_)) {} |
| + |
| + void SetUp() override { |
| + processor_->SetDelegate(1, &delegate1_); |
| + processor_->SetDelegate(2, &delegate2_); |
| + } |
| + |
| + protected: |
| + MockBlimpMessageProcessor out_processor_; |
| + MockRenderWidgetMessageDelegate delegate1_; |
| + MockRenderWidgetMessageDelegate delegate2_; |
| + |
| + scoped_ptr<RenderWidgetMessageProcessor> processor_; |
| +}; |
| + |
| +TEST_F(RenderWidgetMessageProcessorTest, DelegateCallsOK) { |
| + EXPECT_CALL(delegate1_, MockableOnRenderWidgetInitialized()).Times(1); |
| + SendRenderWidgetMessage(processor_.get(), 1, 1U); |
| + |
| + EXPECT_CALL(delegate1_, MockableOnCompositorMessageReceived(_)).Times(1); |
| + SendCompositorMessage(processor_.get(), 1, 1U); |
| + |
| + EXPECT_CALL(delegate2_, MockableOnRenderWidgetInitialized()).Times(1); |
| + SendRenderWidgetMessage(processor_.get(), 2, 2U); |
| + |
| + EXPECT_CALL(delegate2_, MockableOnCompositorMessageReceived(_)).Times(1); |
| + SendCompositorMessage(processor_.get(), 2, 2U); |
| +} |
| + |
| +TEST_F(RenderWidgetMessageProcessorTest, RepliesHaveCorrectRenderWidgetId) { |
| + SendRenderWidgetMessage(processor_.get(), 1, 2U); |
| + SendRenderWidgetMessage(processor_.get(), 2, 1U); |
| + |
| + EXPECT_CALL(out_processor_, |
| + MockableProcessMessage(CompositorMessageEquals(1, 2U))).Times(1); |
| + processor_->SendCompositorMessage(1, cc::proto::CompositorMessage()); |
| + |
| + SendRenderWidgetMessage(processor_.get(), 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 |