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

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

 // Copyright 2016 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 "modules/presentation/PresentationReceiver.h"
 
 #include "bindings/core/v8/ScriptPromiseResolver.h"
 #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>
 
 namespace blink {
 
 class MockEventListener : public EventListener {
  public:
   MockEventListener() : EventListener(CPPEventListenerType) {}
 
@@ skipping 48 matching lines @@
                void(const WebURL& availabilityUrl,
                     WebPresentationAvailabilityCallbacks*));
 
   MOCK_METHOD1(startListening, void(WebPresentationAvailabilityObserver*));
 
   MOCK_METHOD1(stopListening, void(WebPresentationAvailabilityObserver*));
 
   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:
   WebString getId() override { return WebString::fromUTF8("id"); }
   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 {
  public:
   void addConnectionavailableEventListener(EventListener*,
                                            const PresentationReceiver*);
   void verifyConnectionListPropertyState(ScriptPromisePropertyBase::State,
                                          const PresentationReceiver*);
   void verifyConnectionListSize(size_t expectedSize,
                                 const PresentationReceiver*);
@@ skipping 100 matching lines @@
 
 TEST_F(PresentationReceiverTest, CreateReceiver) {
   MockWebPresentationClient client;
   EXPECT_CALL(client, setReceiver(testing::_));
 
   V8TestingScope scope;
   new PresentationReceiver(&scope.frame(), &client);
 }
 
 }  // namespace blink