[Presentation API] (5th) (1-UA) integrate controller and receiver side for 1-UA messaging

chrome/browser/media/router/browser_presentation_connection_proxy_unittest.cc

+// 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::OnMessageCallback;

[zhaobin, 2017/01/31 18:44:15]

Done.

+
+class MockOnMessageCallback {

[imcheng, 2017/01/31 01:53:25]

You can replace this using MockCallback:
https://cs.chromium.org/chromium/src/base/test/mock_callback.h?q=mockcallback...

[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