Index: net/quic/test_tools/mock_crypto_client_stream_factory.cc |
diff --git a/net/quic/test_tools/mock_crypto_client_stream_factory.cc b/net/quic/test_tools/mock_crypto_client_stream_factory.cc |
index 5088c4347c8ef27cd4d761230c20cca6aa6681b2..3119ed4f0772c12c3e864b3702da65ea7298c2a8 100644 |
--- a/net/quic/test_tools/mock_crypto_client_stream_factory.cc |
+++ b/net/quic/test_tools/mock_crypto_client_stream_factory.cc |
@@ -13,10 +13,13 @@ using std::string; |
namespace net { |
+MockCryptoClientStreamFactory::~MockCryptoClientStreamFactory() { |
+} |
+ |
MockCryptoClientStreamFactory::MockCryptoClientStreamFactory() |
: handshake_mode_(MockCryptoClientStream::CONFIRM_HANDSHAKE), |
last_stream_(nullptr), |
- proof_verify_details_(nullptr) { |
+ default_proof_verify_details_(nullptr) { |
} |
QuicCryptoClientStream* |
@@ -24,9 +27,14 @@ MockCryptoClientStreamFactory::CreateQuicCryptoClientStream( |
const QuicServerId& server_id, |
QuicClientSession* session, |
QuicCryptoClientConfig* crypto_config) { |
- last_stream_ = new MockCryptoClientStream( |
- server_id, session, nullptr, crypto_config, handshake_mode_, |
- proof_verify_details_); |
+ ProofVerifyDetailsMap::iterator it = |
+ proof_verify_details_map_.find(server_id); |
+ const ProofVerifyDetails* proof_verify_details = |
+ (it == proof_verify_details_map_.end() ? default_proof_verify_details_ |
+ : it->second); |
+ last_stream_ = |
+ new MockCryptoClientStream(server_id, session, nullptr, crypto_config, |
+ handshake_mode_, proof_verify_details); |
return last_stream_; |
} |