OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ | 5 #ifndef NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
6 #define NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ | 6 #define NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "net/quic/quic_crypto_client_stream.h" | 10 #include "net/quic/quic_crypto_client_stream.h" |
11 #include "net/quic/quic_crypto_client_stream_factory.h" | 11 #include "net/quic/quic_crypto_client_stream_factory.h" |
12 #include "net/quic/test_tools/mock_crypto_client_stream.h" | 12 #include "net/quic/test_tools/mock_crypto_client_stream.h" |
13 | 13 |
14 namespace net { | 14 namespace net { |
15 | 15 |
| 16 class QuicSessionKey; |
| 17 |
16 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { | 18 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { |
17 public: | 19 public: |
18 MockCryptoClientStreamFactory(); | 20 MockCryptoClientStreamFactory(); |
19 virtual ~MockCryptoClientStreamFactory() {} | 21 virtual ~MockCryptoClientStreamFactory() {} |
20 | 22 |
21 virtual QuicCryptoClientStream* CreateQuicCryptoClientStream( | 23 virtual QuicCryptoClientStream* CreateQuicCryptoClientStream( |
22 const string& server_hostname, | 24 const QuicSessionKey& server_key, |
23 QuicSession* session, | 25 QuicSession* session, |
24 QuicCryptoClientConfig* crypto_config) OVERRIDE; | 26 QuicCryptoClientConfig* crypto_config) OVERRIDE; |
25 | 27 |
26 void set_handshake_mode( | 28 void set_handshake_mode( |
27 MockCryptoClientStream::HandshakeMode handshake_mode) { | 29 MockCryptoClientStream::HandshakeMode handshake_mode) { |
28 handshake_mode_ = handshake_mode; | 30 handshake_mode_ = handshake_mode; |
29 } | 31 } |
30 | 32 |
31 void set_ssl_info(const SSLInfo* ssl_info) { | 33 void set_ssl_info(const SSLInfo* ssl_info) { |
32 ssl_info_ = ssl_info; | 34 ssl_info_ = ssl_info; |
33 } | 35 } |
34 | 36 |
35 MockCryptoClientStream* last_stream() const { | 37 MockCryptoClientStream* last_stream() const { |
36 return last_stream_; | 38 return last_stream_; |
37 } | 39 } |
38 | 40 |
39 private: | 41 private: |
40 MockCryptoClientStream::HandshakeMode handshake_mode_; | 42 MockCryptoClientStream::HandshakeMode handshake_mode_; |
41 MockCryptoClientStream* last_stream_; | 43 MockCryptoClientStream* last_stream_; |
42 const SSLInfo* ssl_info_; | 44 const SSLInfo* ssl_info_; |
43 }; | 45 }; |
44 | 46 |
45 } // namespace net | 47 } // namespace net |
46 | 48 |
47 #endif // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ | 49 #endif // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
OLD | NEW |