[Presentation API] Mojo typemap for content::PresentationConnectionMessage

content/renderer/presentation/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 <memory>
 #include <utility>
 
 #include "base/run_loop.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/renderer/presentation/presentation_connection_proxy.h"
@@ skipping 53 matching lines @@
   std::unique_ptr<TestPresentationConnection> receiver_connection_;
   ControllerConnectionProxy* controller_connection_proxy_;
   ReceiverConnectionProxy* receiver_connection_proxy_;
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
 };
 
 TEST_F(PresentationConnectionProxyTest, TestSendString) {
   blink::WebString message = blink::WebString::fromUTF8("test message");
-  blink::mojom::ConnectionMessagePtr session_message =
-      blink::mojom::ConnectionMessage::New();
-  session_message->type = blink::mojom::PresentationMessageType::TEXT;
-  session_message->message = message.utf8();
-
   base::RunLoop run_loop;
   EXPECT_CALL(*receiver_connection_, didReceiveTextMessage(message));
   controller_connection_proxy_->SendConnectionMessage(
-      std::move(session_message),
+      PresentationConnectionMessage(message.utf8()),
       base::Bind(
           &PresentationConnectionProxyTest::ExpectSendConnectionMessageCallback,
           base::Unretained(this)));
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationConnectionProxyTest, TestSendArrayBuffer) {
   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;
-
   base::RunLoop run_loop;
   EXPECT_CALL(*receiver_connection_, didReceiveBinaryMessage(_, _))
       .WillOnce(::testing::Invoke(
           [&expected_data](const uint8_t* data, size_t length) {
             std::vector<uint8_t> message_data(data, data + length);
             EXPECT_EQ(expected_data, message_data);
           }));
 
   controller_connection_proxy_->SendConnectionMessage(
-      std::move(session_message),
+      PresentationConnectionMessage(expected_data),
       base::Bind(
           &PresentationConnectionProxyTest::ExpectSendConnectionMessageCallback,
           base::Unretained(this)));
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationConnectionProxyTest, TestControllerConnectionCallsClose) {
   base::RunLoop run_loop;
   EXPECT_CALL(*controller_connection_,
               didChangeState(blink::WebPresentationConnectionState::Closed));
   EXPECT_CALL(*receiver_connection_,
               didChangeState(blink::WebPresentationConnectionState::Closed));
   controller_connection_proxy_->close();
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationConnectionProxyTest, TestReceiverConnectionCallsClose) {
   base::RunLoop run_loop;
   EXPECT_CALL(*controller_connection_,
               didChangeState(blink::WebPresentationConnectionState::Closed));
   EXPECT_CALL(*receiver_connection_,
               didChangeState(blink::WebPresentationConnectionState::Closed));
   receiver_connection_proxy_->close();
   run_loop.RunUntilIdle();
 }
 
 }  // namespace content