[Presentation API] Remove closed connections from PresentationConnectionList

content/renderer/presentation/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 <memory>
 #include <utility>
 
 #include "base/run_loop.h"
+#include "base/test/mock_callback.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/renderer/presentation/presentation_connection_proxy.h"
 #include "content/renderer/presentation/test_presentation_connection.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/modules/presentation/WebPresentationConnection.h"
 #include "third_party/WebKit/public/platform/modules/presentation/WebPresentationController.h"
 
 using ::testing::_;
 
 namespace content {
 
 class PresentationConnectionProxyTest : public ::testing::Test {
  public:
   PresentationConnectionProxyTest() = default;
   ~PresentationConnectionProxyTest() override = default;
 
   void SetUp() override {
     // Set up test connections and test connection proxies.
     controller_connection_ = base::MakeUnique<TestPresentationConnection>();
     receiver_connection_ = base::MakeUnique<TestPresentationConnection>();
 
     controller_connection_proxy_ =
         new ControllerConnectionProxy(controller_connection_.get());
     controller_connection_->BindProxy(
         base::WrapUnique(controller_connection_proxy_));
-    receiver_connection_proxy_ =
-        new ReceiverConnectionProxy(receiver_connection_.get());
+    receiver_connection_proxy_ = new ReceiverConnectionProxy(
+        receiver_connection_.get(), mock_connection_closed_callback_.Get());
     receiver_connection_->BindProxy(
         base::WrapUnique(receiver_connection_proxy_));
 
     EXPECT_CALL(
         *controller_connection_,
         DidChangeState(blink::WebPresentationConnectionState::kConnected));
     EXPECT_CALL(
         *receiver_connection_,
         DidChangeState(blink::WebPresentationConnectionState::kConnected));
 
@@ skipping 10 matching lines @@
 
   void ExpectSendConnectionMessageCallback(bool success) {
     EXPECT_TRUE(success);
   }
 
  protected:
   std::unique_ptr<TestPresentationConnection> controller_connection_;
   std::unique_ptr<TestPresentationConnection> receiver_connection_;
   ControllerConnectionProxy* controller_connection_proxy_;
   ReceiverConnectionProxy* receiver_connection_proxy_;
+  base::MockCallback<ReceiverConnectionProxy::OnConnectionClosedCallback>
+      mock_connection_closed_callback_;
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
 };
 
 TEST_F(PresentationConnectionProxyTest, TestSendString) {
   blink::WebString message = blink::WebString::FromUTF8("test message");
   base::RunLoop run_loop;
   EXPECT_CALL(*receiver_connection_, DidReceiveTextMessage(message));
   controller_connection_proxy_->SendConnectionMessage(
@@ skipping 22 matching lines @@
       base::Bind(
           &PresentationConnectionProxyTest::ExpectSendConnectionMessageCallback,
           base::Unretained(this)));
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationConnectionProxyTest, TestControllerConnectionCallsClose) {
   base::RunLoop run_loop;
   EXPECT_CALL(*controller_connection_, DidClose());
   EXPECT_CALL(*receiver_connection_, DidClose());
+  EXPECT_CALL(mock_connection_closed_callback_, Run());
   controller_connection_proxy_->Close();
   run_loop.RunUntilIdle();
 }
 
 TEST_F(PresentationConnectionProxyTest, TestReceiverConnectionCallsClose) {
   base::RunLoop run_loop;
   EXPECT_CALL(*controller_connection_, DidClose());
   EXPECT_CALL(*receiver_connection_, DidClose());
+  EXPECT_CALL(mock_connection_closed_callback_, Run());
   receiver_connection_proxy_->Close();
   run_loop.RunUntilIdle();
 }
 
 }  // namespace content