[Presentation API] (alternative) 1-UA: send message between controller and receiver page

third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp

Index: third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp
diff --git a/third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp b/third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp
index db0cb6e95dfff5d2cd71fed30bebb5004c1015d9..4ef162f815be7d8d2ecc42a0d0e93d183ee5e1d9 100644
--- a/third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp
+++ b/third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp
@@ -8,10 +8,12 @@
 #include "bindings/core/v8/V8BindingForTesting.h"
 #include "core/frame/LocalFrame.h"
 #include "core/testing/DummyPageHolder.h"
+#include "modules/presentation/PresentationConnection.h"
 #include "modules/presentation/PresentationConnectionList.h"
 #include "platform/testing/URLTestHelpers.h"
 #include "public/platform/modules/presentation/WebPresentationClient.h"
 #include "public/platform/modules/presentation/WebPresentationConnectionClient.h"
+#include "public/platform/modules/presentation/WebPresentationConnectionProxy.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include <v8.h>
@@ -80,6 +82,16 @@ class MockWebPresentationClient : public WebPresentationClient {
   MOCK_METHOD1(setDefaultPresentationUrls, void(const WebVector<WebURL>&));
 };
 
+class MockWebPresentationConnectionProxy
+    : public WebPresentationConnectionProxy {
+ public:
+  MOCK_METHOD1(SetSourceConnection, void(blink::WebPresentationConnection*));
+
+  MOCK_METHOD1(SendString, void(const blink::WebString& message));
+
+  MOCK_METHOD2(SendArrayBuffer, void(const uint8_t* data, size_t length));
+};
+
 class TestWebPresentationConnectionClient
     : public WebPresentationConnectionClient {
  public:
@@ -87,6 +99,10 @@ class TestWebPresentationConnectionClient
   WebURL getUrl() override {
     return URLTestHelpers::toKURL("http://www.example.com");
   }
+  std::unique_ptr<WebPresentationConnectionProxy> takeProxy() override {
+    return std::unique_ptr<WebPresentationConnectionProxy>(
+        new MockWebPresentationConnectionProxy());
+  }
 };
 
 class PresentationReceiverTest : public ::testing::Test {