OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ | 5 #ifndef NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ |
6 #define NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ | 6 #define NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ |
7 | 7 |
8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
9 #include "net/socket/client_socket_pool_manager.h" | 9 #include "net/socket/client_socket_pool_manager.h" |
10 #include "net/socket/client_socket_pool_manager_impl.h" | 10 #include "net/socket/client_socket_pool_manager_impl.h" |
11 | 11 |
12 namespace net { | 12 namespace net { |
13 | 13 |
14 class MockClientSocketPoolManager : public ClientSocketPoolManager { | 14 class MockClientSocketPoolManager : public ClientSocketPoolManager { |
15 public: | 15 public: |
16 MockClientSocketPoolManager(); | 16 MockClientSocketPoolManager(); |
17 virtual ~MockClientSocketPoolManager(); | 17 ~MockClientSocketPoolManager() override; |
18 | 18 |
19 // Sets "override" socket pools that get used instead. | 19 // Sets "override" socket pools that get used instead. |
20 void SetTransportSocketPool(TransportClientSocketPool* pool); | 20 void SetTransportSocketPool(TransportClientSocketPool* pool); |
21 void SetSSLSocketPool(SSLClientSocketPool* pool); | 21 void SetSSLSocketPool(SSLClientSocketPool* pool); |
22 void SetSocketPoolForSOCKSProxy(const HostPortPair& socks_proxy, | 22 void SetSocketPoolForSOCKSProxy(const HostPortPair& socks_proxy, |
23 SOCKSClientSocketPool* pool); | 23 SOCKSClientSocketPool* pool); |
24 void SetSocketPoolForHTTPProxy(const HostPortPair& http_proxy, | 24 void SetSocketPoolForHTTPProxy(const HostPortPair& http_proxy, |
25 HttpProxyClientSocketPool* pool); | 25 HttpProxyClientSocketPool* pool); |
26 void SetSocketPoolForSSLWithProxy(const HostPortPair& proxy_server, | 26 void SetSocketPoolForSSLWithProxy(const HostPortPair& proxy_server, |
27 SSLClientSocketPool* pool); | 27 SSLClientSocketPool* pool); |
28 | 28 |
29 // ClientSocketPoolManager methods: | 29 // ClientSocketPoolManager methods: |
30 virtual void FlushSocketPoolsWithError(int error) override; | 30 void FlushSocketPoolsWithError(int error) override; |
31 virtual void CloseIdleSockets() override; | 31 void CloseIdleSockets() override; |
32 virtual TransportClientSocketPool* GetTransportSocketPool() override; | 32 TransportClientSocketPool* GetTransportSocketPool() override; |
33 virtual SSLClientSocketPool* GetSSLSocketPool() override; | 33 SSLClientSocketPool* GetSSLSocketPool() override; |
34 virtual SOCKSClientSocketPool* GetSocketPoolForSOCKSProxy( | 34 SOCKSClientSocketPool* GetSocketPoolForSOCKSProxy( |
35 const HostPortPair& socks_proxy) override; | 35 const HostPortPair& socks_proxy) override; |
36 virtual HttpProxyClientSocketPool* GetSocketPoolForHTTPProxy( | 36 HttpProxyClientSocketPool* GetSocketPoolForHTTPProxy( |
37 const HostPortPair& http_proxy) override; | 37 const HostPortPair& http_proxy) override; |
38 virtual SSLClientSocketPool* GetSocketPoolForSSLWithProxy( | 38 SSLClientSocketPool* GetSocketPoolForSSLWithProxy( |
39 const HostPortPair& proxy_server) override; | 39 const HostPortPair& proxy_server) override; |
40 virtual base::Value* SocketPoolInfoToValue() const override; | 40 base::Value* SocketPoolInfoToValue() const override; |
41 | 41 |
42 private: | 42 private: |
43 typedef internal::OwnedPoolMap<HostPortPair, TransportClientSocketPool*> | 43 typedef internal::OwnedPoolMap<HostPortPair, TransportClientSocketPool*> |
44 TransportSocketPoolMap; | 44 TransportSocketPoolMap; |
45 typedef internal::OwnedPoolMap<HostPortPair, SOCKSClientSocketPool*> | 45 typedef internal::OwnedPoolMap<HostPortPair, SOCKSClientSocketPool*> |
46 SOCKSSocketPoolMap; | 46 SOCKSSocketPoolMap; |
47 typedef internal::OwnedPoolMap<HostPortPair, HttpProxyClientSocketPool*> | 47 typedef internal::OwnedPoolMap<HostPortPair, HttpProxyClientSocketPool*> |
48 HTTPProxySocketPoolMap; | 48 HTTPProxySocketPoolMap; |
49 typedef internal::OwnedPoolMap<HostPortPair, SSLClientSocketPool*> | 49 typedef internal::OwnedPoolMap<HostPortPair, SSLClientSocketPool*> |
50 SSLSocketPoolMap; | 50 SSLSocketPoolMap; |
51 | 51 |
52 scoped_ptr<TransportClientSocketPool> transport_socket_pool_; | 52 scoped_ptr<TransportClientSocketPool> transport_socket_pool_; |
53 scoped_ptr<SSLClientSocketPool> ssl_socket_pool_; | 53 scoped_ptr<SSLClientSocketPool> ssl_socket_pool_; |
54 SOCKSSocketPoolMap socks_socket_pools_; | 54 SOCKSSocketPoolMap socks_socket_pools_; |
55 HTTPProxySocketPoolMap http_proxy_socket_pools_; | 55 HTTPProxySocketPoolMap http_proxy_socket_pools_; |
56 SSLSocketPoolMap ssl_socket_pools_for_proxies_; | 56 SSLSocketPoolMap ssl_socket_pools_for_proxies_; |
57 | 57 |
58 DISALLOW_COPY_AND_ASSIGN(MockClientSocketPoolManager); | 58 DISALLOW_COPY_AND_ASSIGN(MockClientSocketPoolManager); |
59 }; | 59 }; |
60 | 60 |
61 } // namespace net | 61 } // namespace net |
62 | 62 |
63 #endif // NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ | 63 #endif // NET_SOCKET_MOCK_CLIENT_SOCKET_POOL_MANAGER_H_ |
OLD | NEW |