Index: chrome/browser/media/router/browser_presentation_connection_proxy_unittest.cc |
diff --git a/chrome/browser/media/router/browser_presentation_connection_proxy_unittest.cc b/chrome/browser/media/router/browser_presentation_connection_proxy_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8fe32c630daf17c486a6efbfaa618c4c8bf6ac08 |
--- /dev/null |
+++ b/chrome/browser/media/router/browser_presentation_connection_proxy_unittest.cc |
@@ -0,0 +1,122 @@ |
+// Copyright 2017 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 "chrome/browser/media/router/browser_presentation_connection_proxy.h" |
+ |
+#include "base/run_loop.h" |
+#include "chrome/browser/media/router/media_source.h" |
+#include "chrome/browser/media/router/media_source_helper.h" |
+#include "chrome/browser/media/router/mock_media_router.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
+ |
+using ::testing::_; |
+ |
+namespace media_router { |
+ |
+using OnMessageCallback = BrowserPresentationConnectionProxy::OnMessageCallback; |
imcheng
2017/01/31 01:53:25
or just:
using BrowserPresentationConnectionProxy
zhaobin
2017/01/31 18:44:15
Done.
|
+ |
+class MockOnMessageCallback { |
imcheng
2017/01/31 01:53:25
You can replace this using MockCallback: https://c
zhaobin
2017/01/31 18:44:15
Done.
|
+ public: |
+ MOCK_METHOD1(OnMessage, void(bool)); |
+}; |
+ |
+class MockPresentationConnectionProxy |
+ : public NON_EXPORTED_BASE(blink::mojom::PresentationConnection) { |
+ public: |
+ void OnMessage(blink::mojom::ConnectionMessagePtr message, |
+ const OnMessageCallback& on_message_callback) override { |
+ OnMessageRaw(message.get(), on_message_callback); |
+ } |
+ |
+ MOCK_METHOD2(OnMessageRaw, |
+ void(const blink::mojom::ConnectionMessage*, |
+ const OnMessageCallback&)); |
+ MOCK_METHOD1(DidChangeState, |
+ void(blink::mojom::PresentationConnectionState state)); |
+}; |
+ |
+class BrowserPresentationConnectionProxyTest : public ::testing::Test { |
+ public: |
+ BrowserPresentationConnectionProxyTest() { |
+ mock_route_ = base::MakeUnique<MediaRoute>( |
+ "MockRouteId", |
+ MediaSourceForPresentationUrl(GURL("MockPresentationUrl")), |
+ "mediaSinkId", "", true, "", true); |
+ } |
+ |
+ void SetUp() override { |
+ browser_connection_proxy_ = |
+ base::MakeUnique<BrowserPresentationConnectionProxy>(&mock_router_, |
+ mock_route_.get()); |
+ } |
+ |
+ void TearDown() override { browser_connection_proxy_.reset(); } |
+ |
+ std::unique_ptr<BrowserPresentationConnectionProxy> browser_connection_proxy_; |
+ MockMediaRouter mock_router_; |
+ std::unique_ptr<MediaRoute> mock_route_; |
+ |
+ private: |
+ content::TestBrowserThreadBundle thread_bundle_; |
+}; |
+ |
+TEST_F(BrowserPresentationConnectionProxyTest, TestBindControllerConnection) { |
+ MockPresentationConnectionProxy mock_controller_connection_proxy; |
+ blink::mojom::PresentationConnectionPtr controller_connection_ptr; |
+ mojo::Binding<blink::mojom::PresentationConnection> binding( |
+ &mock_controller_connection_proxy, |
+ mojo::MakeRequest(&controller_connection_ptr)); |
+ EXPECT_CALL( |
+ mock_controller_connection_proxy, |
+ DidChangeState(blink::mojom::PresentationConnectionState::CONNECTED)); |
+ |
+ base::RunLoop run_loop; |
+ browser_connection_proxy_->BindControllerConnection( |
+ std::move(controller_connection_ptr)); |
+ run_loop.RunUntilIdle(); |
+} |
+ |
+TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageTextMessage) { |
+ std::string message = "test message"; |
+ blink::mojom::ConnectionMessagePtr session_message = |
+ blink::mojom::ConnectionMessage::New(); |
+ session_message->type = blink::mojom::PresentationMessageType::TEXT; |
+ session_message->message = message; |
+ |
+ MockOnMessageCallback mock_on_message_callback; |
+ EXPECT_CALL(mock_router_, |
+ SendRouteMessage(mock_route_->media_route_id(), message, _)); |
+ |
+ browser_connection_proxy_->OnMessage( |
+ std::move(session_message), |
+ base::Bind(&MockOnMessageCallback::OnMessage, |
+ base::Unretained(&mock_on_message_callback))); |
+} |
+ |
+TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageBinaryMessage) { |
+ std::vector<uint8_t> expected_data; |
+ expected_data.push_back(42); |
+ expected_data.push_back(36); |
+ |
+ blink::mojom::ConnectionMessagePtr session_message = |
+ blink::mojom::ConnectionMessage::New(); |
+ session_message->type = blink::mojom::PresentationMessageType::BINARY; |
+ session_message->data = expected_data; |
+ |
+ MockOnMessageCallback mock_on_message_callback; |
+ EXPECT_CALL(mock_router_, SendRouteBinaryMessageInternal(_, _, _)) |
+ .WillOnce(::testing::Invoke([this, &expected_data]( |
+ const MediaRoute::Id& route_id, std::vector<uint8_t>* data, |
+ const OnMessageCallback& callback) { |
+ EXPECT_EQ(expected_data, *data); |
+ })); |
+ |
+ browser_connection_proxy_->OnMessage( |
+ std::move(session_message), |
+ base::Bind(&MockOnMessageCallback::OnMessage, |
+ base::Unretained(&mock_on_message_callback))); |
+} |
+ |
+} // namespace media_router |