OLD | NEW |
(Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/media/router/browser_presentation_connection_proxy.h" |
| 6 |
| 7 #include "base/run_loop.h" |
| 8 #include "base/test/mock_callback.h" |
| 9 #include "chrome/browser/media/router/media_source.h" |
| 10 #include "chrome/browser/media/router/media_source_helper.h" |
| 11 #include "chrome/browser/media/router/mock_media_router.h" |
| 12 #include "content/public/test/test_browser_thread_bundle.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" |
| 14 |
| 15 using ::testing::_; |
| 16 |
| 17 namespace media_router { |
| 18 |
| 19 constexpr char kMediaRouteId[] = "MockRouteId"; |
| 20 |
| 21 class MockPresentationConnectionProxy |
| 22 : public NON_EXPORTED_BASE(blink::mojom::PresentationConnection) { |
| 23 public: |
| 24 void OnMessage(blink::mojom::ConnectionMessagePtr message, |
| 25 const OnMessageCallback& on_message_callback) override { |
| 26 OnMessageRaw(message.get(), on_message_callback); |
| 27 } |
| 28 |
| 29 MOCK_METHOD2(OnMessageRaw, |
| 30 void(const blink::mojom::ConnectionMessage*, |
| 31 const OnMessageCallback&)); |
| 32 MOCK_METHOD1(DidChangeState, |
| 33 void(content::PresentationConnectionState state)); |
| 34 }; |
| 35 |
| 36 class BrowserPresentationConnectionProxyTest : public ::testing::Test { |
| 37 public: |
| 38 BrowserPresentationConnectionProxyTest() = default; |
| 39 |
| 40 void SetUp() override { |
| 41 mock_controller_connection_proxy_ = |
| 42 base::MakeUnique<MockPresentationConnectionProxy>(); |
| 43 blink::mojom::PresentationConnectionPtr controller_connection_ptr; |
| 44 mojo::Binding<blink::mojom::PresentationConnection> binding( |
| 45 mock_controller_connection_proxy_.get(), |
| 46 mojo::MakeRequest(&controller_connection_ptr)); |
| 47 EXPECT_CALL( |
| 48 *mock_controller_connection_proxy_, |
| 49 DidChangeState(content::PRESENTATION_CONNECTION_STATE_CONNECTED)); |
| 50 |
| 51 blink::mojom::PresentationConnectionPtr receiver_connection_ptr; |
| 52 |
| 53 base::RunLoop run_loop; |
| 54 browser_connection_proxy_ = |
| 55 base::MakeUnique<BrowserPresentationConnectionProxy>( |
| 56 &mock_router_, "MockRouteId", |
| 57 mojo::MakeRequest(&receiver_connection_ptr), |
| 58 std::move(controller_connection_ptr)); |
| 59 run_loop.RunUntilIdle(); |
| 60 } |
| 61 |
| 62 void TearDown() override { |
| 63 browser_connection_proxy_.reset(); |
| 64 mock_controller_connection_proxy_.reset(); |
| 65 } |
| 66 |
| 67 BrowserPresentationConnectionProxy* browser_connection_proxy() { |
| 68 return browser_connection_proxy_.get(); |
| 69 } |
| 70 |
| 71 MockMediaRouter* mock_router() { return &mock_router_; } |
| 72 |
| 73 private: |
| 74 std::unique_ptr<MockPresentationConnectionProxy> |
| 75 mock_controller_connection_proxy_; |
| 76 std::unique_ptr<BrowserPresentationConnectionProxy> browser_connection_proxy_; |
| 77 MockMediaRouter mock_router_; |
| 78 content::TestBrowserThreadBundle thread_bundle_; |
| 79 }; |
| 80 |
| 81 TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageTextMessage) { |
| 82 std::string message = "test message"; |
| 83 blink::mojom::ConnectionMessagePtr session_message = |
| 84 blink::mojom::ConnectionMessage::New(); |
| 85 session_message->type = blink::mojom::PresentationMessageType::TEXT; |
| 86 session_message->message = message; |
| 87 |
| 88 base::MockCallback<base::Callback<void(bool)>> mock_on_message_callback; |
| 89 EXPECT_CALL(*mock_router(), SendRouteMessage(kMediaRouteId, message, _)); |
| 90 |
| 91 browser_connection_proxy()->OnMessage(std::move(session_message), |
| 92 mock_on_message_callback.Get()); |
| 93 } |
| 94 |
| 95 TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageBinaryMessage) { |
| 96 std::vector<uint8_t> expected_data; |
| 97 expected_data.push_back(42); |
| 98 expected_data.push_back(36); |
| 99 |
| 100 blink::mojom::ConnectionMessagePtr session_message = |
| 101 blink::mojom::ConnectionMessage::New(); |
| 102 session_message->type = blink::mojom::PresentationMessageType::BINARY; |
| 103 session_message->data = expected_data; |
| 104 |
| 105 base::MockCallback<base::Callback<void(bool)>> mock_on_message_callback; |
| 106 EXPECT_CALL(*mock_router(), SendRouteBinaryMessageInternal(_, _, _)) |
| 107 .WillOnce(::testing::Invoke([this, &expected_data]( |
| 108 const MediaRoute::Id& route_id, std::vector<uint8_t>* data, |
| 109 const BrowserPresentationConnectionProxy::OnMessageCallback& |
| 110 callback) { EXPECT_EQ(expected_data, *data); })); |
| 111 |
| 112 browser_connection_proxy()->OnMessage(std::move(session_message), |
| 113 mock_on_message_callback.Get()); |
| 114 } |
| 115 |
| 116 } // namespace media_router |
OLD | NEW |