QUIC - enable persisting of QUICServerInfo (server config) to disk
[chromium-blink-merge.git] / net / socket / transport_client_socket_pool_test_util.h
blob40ed1d1c453cffc9686071d0637e9054bab07e3c
1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 // Test methods and classes common to transport_client_socket_pool_unittest.cc
6 // and websocket_transport_client_socket_pool_unittest.cc. If you find you need
7 // to use these for another purpose, consider moving them to socket_test_util.h.
9 #ifndef NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_
10 #define NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_
12 #include <queue>
14 #include "base/callback.h"
15 #include "base/compiler_specific.h"
16 #include "base/macros.h"
17 #include "base/memory/scoped_ptr.h"
18 #include "base/time/time.h"
19 #include "net/base/address_list.h"
20 #include "net/base/net_log.h"
21 #include "net/socket/client_socket_factory.h"
22 #include "net/socket/client_socket_handle.h"
23 #include "net/socket/stream_socket.h"
25 namespace net {
27 class ClientSocketHandle;
28 class IPEndPoint;
30 // Make sure |handle| sets load times correctly when it has been assigned a
31 // reused socket. Uses gtest expectations.
32 void TestLoadTimingInfoConnectedReused(const ClientSocketHandle& handle);
34 // Make sure |handle| sets load times correctly when it has been assigned a
35 // fresh socket. Also runs TestLoadTimingInfoConnectedReused, since the owner
36 // of a connection where |is_reused| is false may consider the connection
37 // reused. Uses gtest expectations.
38 void TestLoadTimingInfoConnectedNotReused(const ClientSocketHandle& handle);
40 // Set |address| to 1.1.1.1:80
41 void SetIPv4Address(IPEndPoint* address);
43 // Set |address| to [1:abcd::3:4:ff]:80
44 void SetIPv6Address(IPEndPoint* address);
46 // A ClientSocketFactory that produces sockets with the specified connection
47 // behaviours.
48 class MockTransportClientSocketFactory : public ClientSocketFactory {
49 public:
50 enum ClientSocketType {
51 // Connects successfully, synchronously.
52 MOCK_CLIENT_SOCKET,
53 // Fails to connect, synchronously.
54 MOCK_FAILING_CLIENT_SOCKET,
55 // Connects successfully, asynchronously.
56 MOCK_PENDING_CLIENT_SOCKET,
57 // Fails to connect, asynchronously.
58 MOCK_PENDING_FAILING_CLIENT_SOCKET,
59 // A delayed socket will pause before connecting through the message loop.
60 MOCK_DELAYED_CLIENT_SOCKET,
61 // A delayed socket that fails.
62 MOCK_DELAYED_FAILING_CLIENT_SOCKET,
63 // A stalled socket that never connects at all.
64 MOCK_STALLED_CLIENT_SOCKET,
65 // A socket that can be triggered to connect explicitly, asynchronously.
66 MOCK_TRIGGERABLE_CLIENT_SOCKET,
69 explicit MockTransportClientSocketFactory(NetLog* net_log);
70 virtual ~MockTransportClientSocketFactory();
72 virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket(
73 DatagramSocket::BindType bind_type,
74 const RandIntCallback& rand_int_cb,
75 NetLog* net_log,
76 const NetLog::Source& source) OVERRIDE;
78 virtual scoped_ptr<StreamSocket> CreateTransportClientSocket(
79 const AddressList& addresses,
80 NetLog* /* net_log */,
81 const NetLog::Source& /* source */) OVERRIDE;
83 virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket(
84 scoped_ptr<ClientSocketHandle> transport_socket,
85 const HostPortPair& host_and_port,
86 const SSLConfig& ssl_config,
87 const SSLClientSocketContext& context) OVERRIDE;
89 virtual void ClearSSLSessionCache() OVERRIDE;
91 int allocation_count() const { return allocation_count_; }
93 // Set the default ClientSocketType.
94 void set_default_client_socket_type(ClientSocketType type) {
95 client_socket_type_ = type;
98 // Set a list of ClientSocketTypes to be used.
99 void set_client_socket_types(ClientSocketType* type_list, int num_types);
101 void set_delay(base::TimeDelta delay) { delay_ = delay; }
103 // If one or more MOCK_TRIGGERABLE_CLIENT_SOCKETs has already been created,
104 // then returns a Closure that can be called to cause the first
105 // not-yet-connected one to connect. If no MOCK_TRIGGERABLE_CLIENT_SOCKETs
106 // have been created yet, wait for one to be created before returning the
107 // Closure. This method should be called the same number of times as
108 // MOCK_TRIGGERABLE_CLIENT_SOCKETs are created in the test.
109 base::Closure WaitForTriggerableSocketCreation();
111 private:
112 NetLog* net_log_;
113 int allocation_count_;
114 ClientSocketType client_socket_type_;
115 ClientSocketType* client_socket_types_;
116 int client_socket_index_;
117 int client_socket_index_max_;
118 base::TimeDelta delay_;
119 std::queue<base::Closure> triggerable_sockets_;
120 base::Closure run_loop_quit_closure_;
122 DISALLOW_COPY_AND_ASSIGN(MockTransportClientSocketFactory);
125 } // namespace net
127 #endif // NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_