| 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
|
|
|