| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "blimp/client/compositor/render_widget_message_processor.h" | 5 #include "blimp/client/compositor/render_widget_message_processor.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "blimp/common/proto/blimp_message.pb.h" | 9 #include "blimp/common/proto/blimp_message.pb.h" |
| 10 #include "blimp/common/proto/compositor.pb.h" | 10 #include "blimp/common/proto/compositor.pb.h" |
| 11 #include "blimp/common/proto/render_widget.pb.h" | 11 #include "blimp/common/proto/render_widget.pb.h" |
| 12 #include "blimp/net/test_common.h" |
| 12 #include "cc/proto/compositor_message.pb.h" | 13 #include "cc/proto/compositor_message.pb.h" |
| 13 #include "net/base/net_errors.h" | 14 #include "net/base/net_errors.h" |
| 14 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 17 |
| 17 using testing::_; | 18 using testing::_; |
| 18 using testing::InvokeArgument; | 19 using testing::InvokeArgument; |
| 19 using testing::Ref; | 20 using testing::Ref; |
| 20 using testing::Return; | 21 using testing::Return; |
| 21 using testing::SaveArg; | 22 using testing::SaveArg; |
| 22 | 23 |
| 23 namespace blimp { | 24 namespace blimp { |
| 24 | 25 |
| 25 namespace { | 26 namespace { |
| 26 class MockBlimpMessageProcessor : public BlimpMessageProcessor { | |
| 27 public: | |
| 28 MockBlimpMessageProcessor() {} | |
| 29 | |
| 30 ~MockBlimpMessageProcessor() override {} | |
| 31 | |
| 32 // Adapts calls from ProcessMessage to MockableProcessMessage by | |
| 33 // unboxing the |message| scoped_ptr for GMock compatibility. | |
| 34 void ProcessMessage(scoped_ptr<BlimpMessage> message, | |
| 35 const net::CompletionCallback& callback) { | |
| 36 MockableProcessMessage(*message); | |
| 37 if (!callback.is_null()) | |
| 38 callback.Run(net::OK); | |
| 39 } | |
| 40 | |
| 41 MOCK_METHOD1(MockableProcessMessage, | |
| 42 void(const BlimpMessage& message)); | |
| 43 }; | |
| 44 | 27 |
| 45 class MockRenderWidgetMessageDelegate | 28 class MockRenderWidgetMessageDelegate |
| 46 : public RenderWidgetMessageProcessor::RenderWidgetMessageDelegate { | 29 : public RenderWidgetMessageProcessor::RenderWidgetMessageDelegate { |
| 47 public: | 30 public: |
| 48 // RenderWidgetMessageDelegate implementation. | 31 // RenderWidgetMessageDelegate implementation. |
| 49 void OnRenderWidgetInitialized() override { | 32 void OnRenderWidgetInitialized() override { |
| 50 MockableOnRenderWidgetInitialized(); | 33 MockableOnRenderWidgetInitialized(); |
| 51 } | 34 } |
| 52 void OnCompositorMessageReceived( | 35 void OnCompositorMessageReceived( |
| 53 scoped_ptr<cc::proto::CompositorMessage> message) override { | 36 scoped_ptr<cc::proto::CompositorMessage> message) override { |
| (...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 123 | 106 |
| 124 EXPECT_CALL(delegate2_, MockableOnCompositorMessageReceived(_)).Times(1); | 107 EXPECT_CALL(delegate2_, MockableOnCompositorMessageReceived(_)).Times(1); |
| 125 SendCompositorMessage(&processor_, 2, 2U); | 108 SendCompositorMessage(&processor_, 2, 2U); |
| 126 } | 109 } |
| 127 | 110 |
| 128 TEST_F(RenderWidgetMessageProcessorTest, RepliesHaveCorrectRenderWidgetId) { | 111 TEST_F(RenderWidgetMessageProcessorTest, RepliesHaveCorrectRenderWidgetId) { |
| 129 SendRenderWidgetMessage(&processor_, 1, 2U); | 112 SendRenderWidgetMessage(&processor_, 1, 2U); |
| 130 SendRenderWidgetMessage(&processor_, 2, 1U); | 113 SendRenderWidgetMessage(&processor_, 2, 1U); |
| 131 | 114 |
| 132 EXPECT_CALL(out_processor_, | 115 EXPECT_CALL(out_processor_, |
| 133 MockableProcessMessage(CompMsgEquals(1, 2U))).Times(1); | 116 MockableProcessMessage(CompMsgEquals(1, 2U), _)).Times(1); |
| 134 processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); | 117 processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); |
| 135 | 118 |
| 136 SendRenderWidgetMessage(&processor_, 1, 3U); | 119 SendRenderWidgetMessage(&processor_, 1, 3U); |
| 137 | 120 |
| 138 EXPECT_CALL(out_processor_, | 121 EXPECT_CALL(out_processor_, |
| 139 MockableProcessMessage(CompMsgEquals(1, 3U))).Times(1); | 122 MockableProcessMessage(CompMsgEquals(1, 3U), _)).Times(1); |
| 140 processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); | 123 processor_.SendCompositorMessage(1, cc::proto::CompositorMessage()); |
| 141 | 124 |
| 142 EXPECT_CALL(out_processor_, | 125 EXPECT_CALL(out_processor_, |
| 143 MockableProcessMessage(CompMsgEquals(2, 1U))).Times(1); | 126 MockableProcessMessage(CompMsgEquals(2, 1U), _)).Times(1); |
| 144 processor_.SendCompositorMessage(2, cc::proto::CompositorMessage()); | 127 processor_.SendCompositorMessage(2, cc::proto::CompositorMessage()); |
| 145 } | 128 } |
| 146 | 129 |
| 147 } // namespace blimp | 130 } // namespace blimp |
| OLD | NEW |