Chromium Code Reviews| 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 <map> | |
| 8 #include <string> | 9 #include <string> |
| 9 | 10 |
| 10 #include "net/quic/quic_crypto_client_stream.h" | 11 #include "net/quic/quic_crypto_client_stream.h" |
| 11 #include "net/quic/quic_crypto_client_stream_factory.h" | 12 #include "net/quic/quic_crypto_client_stream_factory.h" |
| 12 #include "net/quic/test_tools/mock_crypto_client_stream.h" | 13 #include "net/quic/test_tools/mock_crypto_client_stream.h" |
| 13 | 14 |
| 14 namespace net { | 15 namespace net { |
| 15 | 16 |
| 16 class QuicServerId; | 17 class QuicServerId; |
| 17 | 18 |
| 18 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { | 19 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { |
| 19 public: | 20 public: |
| 20 MockCryptoClientStreamFactory(); | 21 MockCryptoClientStreamFactory(); |
| 21 ~MockCryptoClientStreamFactory() override {} | 22 ~MockCryptoClientStreamFactory() override; |
| 22 | 23 |
| 23 QuicCryptoClientStream* CreateQuicCryptoClientStream( | 24 QuicCryptoClientStream* CreateQuicCryptoClientStream( |
| 24 const QuicServerId& server_id, | 25 const QuicServerId& server_id, |
| 25 QuicClientSession* session, | 26 QuicClientSession* session, |
| 26 QuicCryptoClientConfig* crypto_config) override; | 27 QuicCryptoClientConfig* crypto_config) override; |
| 27 | 28 |
| 28 void set_handshake_mode( | 29 void set_handshake_mode( |
| 29 MockCryptoClientStream::HandshakeMode handshake_mode) { | 30 MockCryptoClientStream::HandshakeMode handshake_mode) { |
| 30 handshake_mode_ = handshake_mode; | 31 handshake_mode_ = handshake_mode; |
| 31 } | 32 } |
| 32 | 33 |
| 33 void set_proof_verify_details( | 34 void set_default_proof_verify_details( |
| 34 const ProofVerifyDetails* proof_verify_details) { | 35 const ProofVerifyDetails* default_proof_verify_details) { |
| 35 proof_verify_details_ = proof_verify_details; | 36 default_proof_verify_details_ = default_proof_verify_details; |
| 37 } | |
| 38 | |
| 39 void set_proof_verify_details_for_server( | |
|
Ryan Hamilton
2015/05/15 04:21:21
I'm not in love with having both "default" and "pe
Bence
2015/05/15 12:05:49
Fair enough. I choose the queue approach, because
| |
| 40 const ProofVerifyDetails* proof_verify_details, | |
| 41 const QuicServerId& server_id) { | |
| 42 proof_verify_details_map_[server_id] = proof_verify_details; | |
| 36 } | 43 } |
| 37 | 44 |
| 38 MockCryptoClientStream* last_stream() const { | 45 MockCryptoClientStream* last_stream() const { |
| 39 return last_stream_; | 46 return last_stream_; |
| 40 } | 47 } |
| 41 | 48 |
| 42 private: | 49 private: |
| 43 MockCryptoClientStream::HandshakeMode handshake_mode_; | 50 MockCryptoClientStream::HandshakeMode handshake_mode_; |
| 44 MockCryptoClientStream* last_stream_; | 51 MockCryptoClientStream* last_stream_; |
| 45 const ProofVerifyDetails* proof_verify_details_; | 52 const ProofVerifyDetails* default_proof_verify_details_; |
| 53 typedef std::map<QuicServerId, const ProofVerifyDetails*> | |
| 54 ProofVerifyDetailsMap; | |
| 55 ProofVerifyDetailsMap proof_verify_details_map_; | |
| 46 | 56 |
| 47 DISALLOW_COPY_AND_ASSIGN(MockCryptoClientStreamFactory); | 57 DISALLOW_COPY_AND_ASSIGN(MockCryptoClientStreamFactory); |
| 48 }; | 58 }; |
| 49 | 59 |
| 50 } // namespace net | 60 } // namespace net |
| 51 | 61 |
| 52 #endif // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ | 62 #endif // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
| OLD | NEW |