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

chrome/browser/media/router/browser_presentation_connection_proxy_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 "chrome/browser/media/router/browser_presentation_connection_proxy.h"
 
 #include "base/run_loop.h"
 #include "base/test/mock_callback.h"
 #include "chrome/browser/media/router/media_source.h"
 #include "chrome/browser/media/router/media_source_helper.h"
@@ skipping 13 matching lines @@
   void OnMessage(blink::mojom::ConnectionMessagePtr message,
                  const OnMessageCallback& on_message_callback) override {
     OnMessageRaw(message.get(), on_message_callback);
   }
 
   MOCK_METHOD2(OnMessageRaw,
                void(const blink::mojom::ConnectionMessage*,
                     const OnMessageCallback&));
   MOCK_METHOD1(DidChangeState,
                void(content::PresentationConnectionState state));
+  MOCK_METHOD0(OnClose, void());
 };
 
 class BrowserPresentationConnectionProxyTest : public ::testing::Test {
  public:
   BrowserPresentationConnectionProxyTest() = default;
 
   void SetUp() override {
     mock_controller_connection_proxy_ =
         base::MakeUnique<MockPresentationConnectionProxy>();
     blink::mojom::PresentationConnectionPtr controller_connection_ptr;
@@ skipping 63 matching lines @@
       .WillOnce(::testing::Invoke([&expected_data](
           const MediaRoute::Id& route_id, std::vector<uint8_t>* data,
           const BrowserPresentationConnectionProxy::OnMessageCallback&
               callback) { EXPECT_EQ(expected_data, *data); }));
 
   browser_connection_proxy()->OnMessage(std::move(session_message),
                                         mock_on_message_callback.Get());
 }
 
 }  // namespace media_router