Handle multiple simultanous wss: connections.

net/socket/websocket_transport_client_socket_pool_unittest.cc

 // Copyright 2014 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 "net/socket/websocket_transport_client_socket_pool.h"
 
 #include <queue>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 1092 matching lines @@
 
   connect_trigger.Run();  // Calls InvokeUserCallbackLater()
 
   request(0)->handle()->Reset();  // calls CancelRequest()
 
   // We should now be able to create a new connection without blocking on the
   // endpoint lock.
   EXPECT_EQ(OK, StartRequest("a", kDefaultPriority));
 }
 
+// A handshake completing and then the WebSocket closing should only release one
+// Endpoint, not two.
+TEST_F(WebSocketTransportClientSocketPoolTest, EndpointLockIsOnlyReleasedOnce) {
+  host_resolver_->set_synchronous_mode(true);
+  EXPECT_EQ(OK, StartRequest("a", kDefaultPriority));
+  EXPECT_EQ(ERR_IO_PENDING, StartRequest("a", kDefaultPriority));
+  EXPECT_EQ(ERR_IO_PENDING, StartRequest("a", kDefaultPriority));
+  // First socket completes handshake.
+  WebSocketTransportClientSocketPool::UnlockEndpoint(request(0)->handle());
+  // First socket is closed.
+  request(0)->handle()->Reset();
+  // Second socket should have been released.
+  EXPECT_EQ(OK, request(1)->WaitForResult());
+  // Third socket should still be waiting for endpoint.
+  ASSERT_FALSE(request(2)->handle()->is_initialized());
+  EXPECT_EQ(LOAD_STATE_WAITING_FOR_AVAILABLE_SOCKET,
+            request(2)->handle()->GetLoadState());
+}
+
 }  // namespace
 
 }  // namespace net