Chromium Code Reviews| Index: net/quic/test_tools/mock_crypto_client_stream_factory.h |
| diff --git a/net/quic/test_tools/mock_crypto_client_stream_factory.h b/net/quic/test_tools/mock_crypto_client_stream_factory.h |
| index b138c70ee87ee992d48b849035cc3217005b8fa7..dbe2422e3ef04218cf356670435d48a4f25c6871 100644 |
| --- a/net/quic/test_tools/mock_crypto_client_stream_factory.h |
| +++ b/net/quic/test_tools/mock_crypto_client_stream_factory.h |
| @@ -5,6 +5,7 @@ |
| #ifndef NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
| #define NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ |
| +#include <map> |
| #include <string> |
| #include "net/quic/quic_crypto_client_stream.h" |
| @@ -18,7 +19,7 @@ class QuicServerId; |
| class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { |
| public: |
| MockCryptoClientStreamFactory(); |
| - ~MockCryptoClientStreamFactory() override {} |
| + ~MockCryptoClientStreamFactory() override; |
| QuicCryptoClientStream* CreateQuicCryptoClientStream( |
| const QuicServerId& server_id, |
| @@ -30,9 +31,15 @@ class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { |
| handshake_mode_ = handshake_mode; |
| } |
| - void set_proof_verify_details( |
| - const ProofVerifyDetails* proof_verify_details) { |
| - proof_verify_details_ = proof_verify_details; |
| + void set_default_proof_verify_details( |
| + const ProofVerifyDetails* default_proof_verify_details) { |
| + default_proof_verify_details_ = default_proof_verify_details; |
| + } |
| + |
| + 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
|
| + const ProofVerifyDetails* proof_verify_details, |
| + const QuicServerId& server_id) { |
| + proof_verify_details_map_[server_id] = proof_verify_details; |
| } |
| MockCryptoClientStream* last_stream() const { |
| @@ -42,7 +49,10 @@ class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { |
| private: |
| MockCryptoClientStream::HandshakeMode handshake_mode_; |
| MockCryptoClientStream* last_stream_; |
| - const ProofVerifyDetails* proof_verify_details_; |
| + const ProofVerifyDetails* default_proof_verify_details_; |
| + typedef std::map<QuicServerId, const ProofVerifyDetails*> |
| + ProofVerifyDetailsMap; |
| + ProofVerifyDetailsMap proof_verify_details_map_; |
| DISALLOW_COPY_AND_ASSIGN(MockCryptoClientStreamFactory); |
| }; |