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

chrome/browser/media/router/offscreen_presentation_manager_unittest.cc

 // 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 <vector>
 
 #include "base/bind.h"
 #include "chrome/browser/media/router/offscreen_presentation_manager.h"
 #include "chrome/browser/media/router/test_helper.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 
 namespace media_router {
 
 namespace {
 const char kPresentationId[] = "presentationId";
 const char kPresentationId2[] = "presentationId2";
+const char kPresentationUrl[] = "presentationUrl";
+const char kPresentationUrl2[] = "presentationUrl2";
 }  // namespace
 
 class MockOffscreenPresentationClient
     : public content::OffscreenPresentationClient {
  public:
   MOCK_METHOD1(SetOffscreenPresentationClient,
                void(OffscreenPresentationClient*));
   MOCK_METHOD1(RemoveOffscreenPresentationClient,
                void(OffscreenPresentationClient*));
+  MOCK_METHOD3(OnSessionMessages,
+               void(const content::PresentationSessionInfo&,
+                    const ScopedVector<content::PresentationSessionMessage>&,
+                    bool));
 };
 
 class OffscreenPresentationManagerTest : public ::testing::Test {
  public:
   OffscreenPresentationManager* manager() { return &manager_; }
 
   void verifyPresentationsSize(size_t expected);
   void verifyControllerSize(size_t expected,
                             const OffscreenPresentation* presentation);
 
@@ skipping 15 matching lines @@
 
 void OffscreenPresentationManagerTest::verifyControllerSize(
     size_t expected,
     const OffscreenPresentation* presentation) {
   EXPECT_EQ(expected, presentation->controllers_.size());
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterControllerCreateNewPresentation) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
+
   verifyPresentationsSize(1);
 }
 
 TEST_F(OffscreenPresentationManagerTest, RegisterOneReceiverDoNothing) {
   MockReceiverConnectionAvailableCallback receiver_callback;
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerNoCrash) {
   MockOffscreenPresentationClient controller;
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverNoCrash) {
   manager()->UnregisterOffscreenPresentationReceiver(kPresentationId);
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterControllerCreatePresentationWithTwoControllers) {
   MockOffscreenPresentationClient controller1;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller1);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller1);
   MockOffscreenPresentationClient controller2;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 2,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 2, &controller2);
   verifyPresentationsSize(1);
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterControllerCreateTwoPresentations) {
   MockOffscreenPresentationClient controller1;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller1);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller1);
   MockOffscreenPresentationClient controller2;
-  manager()->RegisterOffscreenPresentationController(kPresentationId2, 1,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId2, kPresentationUrl2, 1, &controller2);
   verifyPresentationsSize(2);
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterControllerAndReceiverFireOnConnectionAvailable) {
   MockOffscreenPresentationClient controller;
   MockReceiverConnectionAvailableCallback receiver_callback;
 
   verifyPresentationsSize(0);
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
 
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 }
 
 TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverOnly) {
   MockOffscreenPresentationClient controller;
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
 
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
   manager()->UnregisterOffscreenPresentationReceiver(kPresentationId);
 
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerOnly) {
   MockOffscreenPresentationClient controller;
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
 
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
   verifyPresentationsSize(1);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverController) {
   MockOffscreenPresentationClient controller;
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
 
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 
   manager()->UnregisterOffscreenPresentationReceiver(kPresentationId);
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
 
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerReceiver) {
   MockOffscreenPresentationClient controller;
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
 
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
   manager()->UnregisterOffscreenPresentationReceiver(kPresentationId);
 
   verifyPresentationsSize(0);
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterTwoControllersThenReceiverFireOnConnectionAvailableTwice) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   MockOffscreenPresentationClient controller2;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 2,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 2, &controller2);
 
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
   EXPECT_CALL(receiver_callback,
               OnReceiverConnectionAvailable(_, &controller2));
 
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        RegisterControllerReceiverConontrollerFireOnConnectionAvailableTwice) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
 
   MockOffscreenPresentationClient controller2;
   EXPECT_CALL(receiver_callback,
               OnReceiverConnectionAvailable(_, &controller2));
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 2,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 2, &controller2);
 }
 
 TEST_F(OffscreenPresentationManagerTest,
        OnePresentationUnregisterFirstController) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   MockOffscreenPresentationClient controller2;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 2,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 2, &controller2);
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, _)).Times(2);
 
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 1);
   manager()->UnregisterOffscreenPresentationController(kPresentationId, 2);
 
   verifyPresentationsSize(1);
 }
 
 TEST_F(OffscreenPresentationManagerTest, TwoPresentations) {
   MockOffscreenPresentationClient controller;
-  manager()->RegisterOffscreenPresentationController(kPresentationId, 1,
-                                                     &controller);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId, kPresentationUrl, 1, &controller);
   MockReceiverConnectionAvailableCallback receiver_callback;
   EXPECT_CALL(receiver_callback, OnReceiverConnectionAvailable(_, &controller));
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId, base::Bind(&MockReceiverConnectionAvailableCallback::
                                       OnReceiverConnectionAvailable,
                                   base::Unretained(&receiver_callback)));
   MockOffscreenPresentationClient controller2;
-  manager()->RegisterOffscreenPresentationController(kPresentationId2, 2,
-                                                     &controller2);
+  manager()->RegisterOffscreenPresentationController(
+      kPresentationId2, kPresentationUrl, 2, &controller2);
   MockReceiverConnectionAvailableCallback receiver_callback2;
   EXPECT_CALL(receiver_callback2,
               OnReceiverConnectionAvailable(_, &controller2));
   manager()->RegisterOffscreenPresentationReceiver(
       kPresentationId2, base::Bind(&MockReceiverConnectionAvailableCallback::
                                        OnReceiverConnectionAvailable,
                                    base::Unretained(&receiver_callback2)));
 
   verifyPresentationsSize(2);
 
   manager()->UnregisterOffscreenPresentationReceiver(kPresentationId2);
   verifyPresentationsSize(1);
 }
 
 }  // namespace media_router