OLD | NEW |
---|---|
(Empty) | |
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. | |
4 | |
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. | |
8 | |
9 #ifndef NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_ | |
10 #define NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_ | |
11 | |
12 #include <queue> | |
13 | |
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" | |
24 | |
25 namespace net { | |
26 | |
27 class ClientSocketHandle; | |
28 class IPEndPoint; | |
29 | |
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); | |
33 | |
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); | |
39 | |
40 // Set |address| to 1.1.1.1:80 | |
41 void SetIPv4Address(IPEndPoint* address); | |
42 | |
43 // Set |address| to [1:abcd::3:4:ff]:80 | |
44 void SetIPv6Address(IPEndPoint* address); | |
45 | |
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, | |
67 }; | |
Ryan Sleevi
2014/07/08 18:50:03
We already have utilities in Chromium to simulate
tyoshino (SeeGerritForStatus)
2014/07/09 00:17:40
FYI, this is refactored version of MockClientSocke
| |
68 | |
69 explicit MockTransportClientSocketFactory(NetLog* net_log); | |
70 virtual ~MockTransportClientSocketFactory(); | |
71 | |
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; | |
77 | |
78 virtual scoped_ptr<StreamSocket> CreateTransportClientSocket( | |
79 const AddressList& addresses, | |
80 NetLog* /* net_log */, | |
81 const NetLog::Source& /* source */) OVERRIDE; | |
82 | |
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; | |
88 | |
89 virtual void ClearSSLSessionCache() OVERRIDE; | |
90 | |
91 int allocation_count() const { return allocation_count_; } | |
92 | |
93 // Set the default ClientSocketType. | |
94 void set_client_socket_type(ClientSocketType type) { | |
95 client_socket_type_ = type; | |
96 } | |
97 | |
98 // Set a list of ClientSocketTypes to be used. | |
99 void set_client_socket_types(ClientSocketType* type_list, int num_types); | |
100 | |
101 void set_delay(base::TimeDelta delay) { delay_ = delay; } | |
102 | |
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(); | |
110 | |
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_; | |
121 | |
122 DISALLOW_COPY_AND_ASSIGN(MockTransportClientSocketFactory); | |
123 }; | |
124 | |
125 } // namespace net | |
126 | |
127 #endif // NET_SOCKET_TRANSPORT_CLIENT_SOCKET_POOL_TEST_UTIL_H_ | |
OLD | NEW |