[PresentationSevice] Use PresentationConnection to send messages from

content/renderer/presentation/presentation_dispatcher_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/common/presentation_connection_message.h"
 #include "content/public/test/test_browser_thread_bundle.h"
@@ skipping 88 matching lines @@
   MOCK_METHOD2(SetPresentationConnection,
                void(const PresentationInfo& presentation_info,
                     PresentationConnection* connection));
 
   MOCK_METHOD2(CloseConnection,
                void(const GURL& presentation_url,
                     const std::string& presentation_id));
   MOCK_METHOD2(Terminate,
                void(const GURL& presentation_url,
                     const std::string& presentation_id));
-  MOCK_METHOD1(ListenForConnectionMessages,
-               void(const PresentationInfo& presentation_info));
 };
 
 class TestPresentationConnectionProxy : public PresentationConnectionProxy {
  public:
   TestPresentationConnectionProxy(blink::WebPresentationConnection* connection)
       : PresentationConnectionProxy(connection) {}
 
   // PresentationConnectionMessage is move-only.
   void SendConnectionMessage(PresentationConnectionMessage message,
                              OnMessageCallback cb) const {
@@ skipping 210 matching lines @@
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
 };
 
 TEST_F(PresentationDispatcherTest, TestStartPresentation) {
   TestPresentationConnection connection;
   EXPECT_FALSE(connection.proxy());
   {
     base::RunLoop run_loop;
-    EXPECT_CALL(presentation_service_, ListenForConnectionMessages(_));
     EXPECT_CALL(presentation_service_, SetPresentationConnection(_, _));
     EXPECT_CALL(presentation_service_, StartPresentationInternal(gurls_, _))
         .WillOnce(Invoke(
             [this](const std::vector<GURL>& presentation_urls,
                    PresentationService::StartPresentationCallback& callback) {
               PresentationInfo presentation_info(gurl1_,
                                                  presentation_id_.Utf8());
               std::move(callback).Run(presentation_info, base::nullopt);
             }));
 
@@ skipping 51 matching lines @@
       base::MakeUnique<TestWebPresentationConnectionErrorCallback>(
           WebPresentationError::kErrorTypeNoAvailableScreens, error_message));
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationDispatcherTest, TestReconnectPresentation) {
   TestPresentationConnection connection;
   EXPECT_FALSE(connection.proxy());
   {
     base::RunLoop run_loop;
-    EXPECT_CALL(presentation_service_, ListenForConnectionMessages(_));
     EXPECT_CALL(presentation_service_, SetPresentationConnection(_, _));
     EXPECT_CALL(presentation_service_,
                 ReconnectPresentationInternal(gurls_, _, _))
         .WillOnce(Invoke(
             [this](
                 const std::vector<GURL>& presentation_urls,
                 const base::Optional<std::string>& presentation_id,
                 PresentationService::ReconnectPresentationCallback& callback) {
               EXPECT_TRUE(presentation_id.has_value());
               EXPECT_EQ(presentation_id_.Utf8(), presentation_id.value());
@@ skipping 445 matching lines @@
     EXPECT_CALL(*mock_observer,
                 AvailabilityChanged(ScreenAvailability::SOURCE_NOT_SUPPORTED));
   }
 
   base::RunLoop run_loop_3;
   ChangeURLState(gurl2_, ScreenAvailability::SOURCE_NOT_SUPPORTED);
   run_loop_3.RunUntilIdle();
 }
 
 }  // namespace content