[Presentation API] Use connection proxy to change connection state to 'closed'

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/test/test_browser_thread_bundle.h"
 #include "content/renderer/presentation/presentation_connection_proxy.h"
@@ skipping 106 matching lines @@
   TestPresentationConnectionProxy(blink::WebPresentationConnection* connection)
       : PresentationConnectionProxy(connection) {}
 
   void SendConnectionMessage(blink::mojom::ConnectionMessagePtr session_message,
                              const OnMessageCallback& callback) const override {
     SendConnectionMessageInternal(session_message.get(), callback);
   }
   MOCK_CONST_METHOD2(SendConnectionMessageInternal,
                      void(blink::mojom::ConnectionMessage*,
                           const OnMessageCallback&));
+  MOCK_CONST_METHOD0(close, void());
 };
 
 class TestPresentationReceiver : public blink::WebPresentationReceiver {
  public:
   blink::WebPresentationConnection* onReceiverConnectionAvailable(
       const blink::WebPresentationSessionInfo&) override {
     return &connection_;
   }
 
   TestPresentationConnection connection_;
@@ skipping 362 matching lines @@
       std::move(session_info), std::move(controller_connection_ptr),
       mojo::MakeRequest(&receiver_connection_ptr));
 
   EXPECT_TRUE(receiver_connection_ptr);
   EXPECT_TRUE(receiver.connection_.proxy());
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationDispatcherTest, TestCloseSession) {
   base::RunLoop run_loop;
+  TestPresentationConnection connection;
+  TestPresentationConnectionProxy test_proxy(&connection);
+  EXPECT_CALL(test_proxy, close());
   EXPECT_CALL(presentation_service_,
               CloseConnection(gurl1_, presentation_id_.utf8()));
-  dispatcher_.closeSession(url1_, presentation_id_);
+  dispatcher_.closeSession(url1_, presentation_id_, &test_proxy);
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationDispatcherTest, TestTerminateSession) {
   base::RunLoop run_loop;
   EXPECT_CALL(presentation_service_,
               Terminate(gurl1_, presentation_id_.utf8()));
   dispatcher_.terminateSession(url1_, presentation_id_);
   run_loop.RunUntilIdle();
 }
@@ skipping 261 matching lines @@
 
   for (auto* mock_observer : mock_observers_)
     EXPECT_CALL(*mock_observer, availabilityChanged(false));
 
   base::RunLoop run_loop_2;
   ChangeURLState(gurl2_, URLState::Unavailable);
   run_loop_2.RunUntilIdle();
 }
 
 }  // namespace content