Create WebSocketTransportClientSocketPool

net/socket/transport_client_socket_pool_test_util.h

Index: net/socket/transport_client_socket_pool_test_util.h
diff --git a/net/socket/transport_client_socket_pool_test_util.h b/net/socket/transport_client_socket_pool_test_util.h
new file mode 100644
index 0000000000000000000000000000000000000000..d48193c688e8d5031626339c69fb72d9b836c1a4
--- /dev/null
+++ b/net/socket/transport_client_socket_pool_test_util.h
@@ -0,0 +1,127 @@
+// 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.
+
+// Test methods and classes common to transport_client_socket_pool_unittest.cc
+// and websocket_transport_client_socket_pool_unittest.cc. If you find you need
+// to use these for another purpose, consider moving them to socket_test_util.h.
+
+#ifndef NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_
+#define NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_
+
+#include <queue>
+
+#include "base/callback.h"
+#include "base/compiler_specific.h"
+#include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/time/time.h"
+#include "net/base/address_list.h"
+#include "net/base/net_log.h"
+#include "net/socket/client_socket_factory.h"
+#include "net/socket/client_socket_handle.h"
+#include "net/socket/stream_socket.h"
+
+namespace net {
+
+class ClientSocketHandle;
+class IPEndPoint;
+
+// Make sure |handle| sets load times correctly when it has been assigned a
+// reused socket. Uses gtest expectations.
+void TestLoadTimingInfoConnectedReused(const ClientSocketHandle& handle);
+
+// Make sure |handle| sets load times correctly when it has been assigned a
+// fresh socket.  Also runs TestLoadTimingInfoConnectedReused, since the owner
+// of a connection where |is_reused| is false may consider the connection
+// reused. Uses gtest expectations.
+void TestLoadTimingInfoConnectedNotReused(const ClientSocketHandle& handle);
+
+// Set |address| to 1.1.1.1:80
+void SetIPv4Address(IPEndPoint* address);
+
+// Set |address| to [1:abcd::3:4:ff]:80
+void SetIPv6Address(IPEndPoint* address);
+
+// A ClientSocketFactory that produces sockets with the specified connection
+// behaviours.
+class MockTransportClientSocketFactory : public ClientSocketFactory {
+ public:
+  enum ClientSocketType {
+    // Connects successfully, synchronously.
+    MOCK_CLIENT_SOCKET,
+    // Fails to connect, synchronously.
+    MOCK_FAILING_CLIENT_SOCKET,
+    // Connects successfully, asynchronously.
+    MOCK_PENDING_CLIENT_SOCKET,
+    // Fails to connect, asynchronously.
+    MOCK_PENDING_FAILING_CLIENT_SOCKET,
+    // A delayed socket will pause before connecting through the message loop.
+    MOCK_DELAYED_CLIENT_SOCKET,
+    // A delayed socket that fails.
+    MOCK_DELAYED_FAILING_CLIENT_SOCKET,
+    // A stalled socket that never connects at all.
+    MOCK_STALLED_CLIENT_SOCKET,
+    // A socket that can be triggered to connect explicitly, asynchronously.
+    MOCK_TRIGGERABLE_CLIENT_SOCKET,
+  };
+
+  explicit MockTransportClientSocketFactory(NetLog* net_log);
+  virtual ~MockTransportClientSocketFactory();
+
+  virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket(
+      DatagramSocket::BindType bind_type,
+      const RandIntCallback& rand_int_cb,
+      NetLog* net_log,
+      const NetLog::Source& source) OVERRIDE;
+
+  virtual scoped_ptr<StreamSocket> CreateTransportClientSocket(
+      const AddressList& addresses,
+      NetLog* /* net_log */,
+      const NetLog::Source& /* source */) OVERRIDE;
+
+  virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket(
+      scoped_ptr<ClientSocketHandle> transport_socket,
+      const HostPortPair& host_and_port,
+      const SSLConfig& ssl_config,
+      const SSLClientSocketContext& context) OVERRIDE;
+
+  virtual void ClearSSLSessionCache() OVERRIDE;
+
+  int allocation_count() const { return allocation_count_; }
+
+  // Set the default ClientSocketType.
+  void set_client_socket_type(ClientSocketType type) {
+    client_socket_type_ = type;
+  }
+
+  // Set a list of ClientSocketTypes to be used.
+  void set_client_socket_types(ClientSocketType* type_list, int num_types);
+
+  void set_delay(base::TimeDelta delay) { delay_ = delay; }
+
+  // If one or more MOCK_TRIGGERABLE_CLIENT_SOCKETs has already been created,
+  // then returns a Closure that can be called to cause the first
+  // not-yet-connected one to connect. If no MOCK_TRIGGERABLE_CLIENT_SOCKETs
+  // have been created yet, wait for one to be created before returning the
+  // Closure. This method should be called the same number of times as
+  // MOCK_TRIGGERABLE_CLIENT_SOCKETs are created in the test.
+  base::Closure WaitForTriggerableSocketCreation();
+
+ private:
+  NetLog* net_log_;
+  int allocation_count_;
+  ClientSocketType client_socket_type_;
+  ClientSocketType* client_socket_types_;
+  int client_socket_index_;
+  int client_socket_index_max_;
+  base::TimeDelta delay_;
+  std::queue<base::Closure> triggerable_sockets_;
+  base::Closure run_loop_quit_closure_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockTransportClientSocketFactory);
+};
+
+}  // namespace net
+
+#endif  // NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_