[Presentation API] Remove closed connections from PresentationConnectionList

third_party/WebKit/Source/modules/presentation/PresentationReceiverTest.cpp

 // 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 "modules/presentation/PresentationReceiver.h"
 
 #include <memory>
 
 #include "bindings/core/v8/ScriptPromiseResolver.h"
 #include "bindings/core/v8/V8BindingForTesting.h"
@@ skipping 210 matching lines @@
 }
 
 TEST_F(PresentationReceiverTest, CreateReceiver) {
   MockWebPresentationClient client;
   EXPECT_CALL(client, SetReceiver(testing::_));
 
   V8TestingScope scope;
   new PresentationReceiver(&scope.GetFrame(), &client);
 }
 
+TEST_F(PresentationReceiverTest, TestRemoveConnection) {
+  V8TestingScope scope;
+  auto receiver = new PresentationReceiver(&scope.GetFrame(), nullptr);
+
+  // Receive first connection.
+  WebPresentationInfo presentation_info1(KURL(KURL(), "http://example1.com"),
+                                         "id1");
+  auto* connection1 =
+      receiver->OnReceiverConnectionAvailable(presentation_info1);
+  EXPECT_TRUE(connection1);
+
+  // Receive second connection.
+  WebPresentationInfo presentation_info2(KURL(KURL(), "http://example2.com"),
+                                         "id2");
+  auto* connection2 =
+      receiver->OnReceiverConnectionAvailable(presentation_info2);
+  EXPECT_TRUE(connection2);
+
+  receiver->connectionList(scope.GetScriptState());
+  VerifyConnectionListSize(2, receiver);
+
+  receiver->RemoveConnection(connection1);
+  VerifyConnectionListSize(1, receiver);
+}
+
 }  // namespace blink