[Presentation API] 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 "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include <v8.h>
 
 namespace blink {
 
@@ skipping 57 matching lines @@
 class TestWebPresentationConnectionClient : public WebPresentationConnectionClient {
 public:
     WebString getId() override { return WebString::fromUTF8("id"); }
     WebURL getUrl() override { return URLTestHelpers::toKURL("http://www.example.com"); }
 };
 
 class PresentationReceiverTest : public ::testing::Test {
 public:
     void addConnectionavailableEventListener(
         EventListener*, const PresentationReceiver*);
+    void addOnMessageEventListener(
+        EventListener*, const PresentationReceiver*);
     void verifyConnectionListPropertyState(
         ScriptPromisePropertyBase::State, const PresentationReceiver*);
     void verifyConnectionListSize(
         size_t expectedSize, const PresentationReceiver*);
 };
 
 void PresentationReceiverTest::addConnectionavailableEventListener(
     EventListener* eventHandler,
     const PresentationReceiver* receiver)
 {
     receiver->m_connectionList->addEventListener(EventTypeNames::connectionavailable, eventHandler);
 }
 
+void PresentationReceiverTest::addOnMessageEventListener(
+    EventListener* eventHandler,
+    const PresentationReceiver* receiver)
+{
+    for (auto const& connection : receiver->m_connectionList->m_connections)
+        connection->addEventListener(EventTypeNames::connect, eventHandler);
+}
+
 void PresentationReceiverTest::verifyConnectionListPropertyState(
     ScriptPromisePropertyBase::State expectedState,
     const PresentationReceiver* receiver)
 {
     EXPECT_EQ(expectedState, receiver->m_connectionListProperty->getState());
 }
 
 void PresentationReceiverTest::verifyConnectionListSize(
     size_t expectedSize,
     const PresentationReceiver* receiver)
@@ skipping 82 matching lines @@
 
 TEST_F(PresentationReceiverTest, CreateReceiver)
 {
     MockWebPresentationClient client;
     EXPECT_CALL(client, setReceiver(testing::_));
 
     V8TestingScope scope;
     new PresentationReceiver(&scope.frame(), &client);
 }
 
+TEST_F(PresentationReceiverTest, didReceiveSessionTextMessage)
+{
+    V8TestingScope scope;
+    auto receiver = new PresentationReceiver(&scope.frame(), nullptr);
+
+    // Receive first connection.
+    auto connectionClient1 = new TestWebPresentationConnectionClient();
+    receiver->onReceiverConnectionAvailable(connectionClient1);
+
+    // verify onMessage event
+    StrictMock<MockEventListener>* eventHandler = new StrictMock<MockEventListener>();
+    addOnMessageEventListener(eventHandler, receiver);
+    EXPECT_CALL(*eventHandler, handleEvent(testing::_, testing::_))
+        .Times(1);
+
+    // Receive second connection.
+    auto connectionClient2 = new TestWebPresentationConnectionClient();
+    receiver->didReceiveSessionTextMessage(connectionClient2, "Hello World");
+}
+
 } // namespace blink