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 3 // found in the LICENSE file. 4 5 #ifndef NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ 6 #define NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ 7 8 #include <string> 9 10 #include "net/quic/quic_crypto_client_stream.h" 11 #include "net/quic/quic_crypto_client_stream_factory.h" 12 #include "net/quic/test_tools/mock_crypto_client_stream.h" 13 14 namespace net { 15 16 class QuicServerId; 17 18 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory { 19 public: 20 MockCryptoClientStreamFactory(); ~MockCryptoClientStreamFactory()21 virtual ~MockCryptoClientStreamFactory() {} 22 23 virtual QuicCryptoClientStream* CreateQuicCryptoClientStream( 24 const QuicServerId& server_id, 25 QuicClientSession* session, 26 QuicCryptoClientConfig* crypto_config) OVERRIDE; 27 set_handshake_mode(MockCryptoClientStream::HandshakeMode handshake_mode)28 void set_handshake_mode( 29 MockCryptoClientStream::HandshakeMode handshake_mode) { 30 handshake_mode_ = handshake_mode; 31 } 32 set_proof_verify_details(const ProofVerifyDetails * proof_verify_details)33 void set_proof_verify_details( 34 const ProofVerifyDetails* proof_verify_details) { 35 proof_verify_details_ = proof_verify_details; 36 } 37 last_stream()38 MockCryptoClientStream* last_stream() const { 39 return last_stream_; 40 } 41 42 private: 43 MockCryptoClientStream::HandshakeMode handshake_mode_; 44 MockCryptoClientStream* last_stream_; 45 const ProofVerifyDetails* proof_verify_details_; 46 47 DISALLOW_COPY_AND_ASSIGN(MockCryptoClientStreamFactory); 48 }; 49 50 } // namespace net 51 52 #endif // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_ 53