Home | History | Annotate | Download | only in test_tools 
      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();
     21   virtual ~MockCryptoClientStreamFactory() {}
     22 
     23   virtual QuicCryptoClientStream* CreateQuicCryptoClientStream(
     24       const QuicServerId& server_id,
     25       QuicClientSession* session,
     26       QuicCryptoClientConfig* crypto_config) OVERRIDE;
     27 
     28   void set_handshake_mode(
     29       MockCryptoClientStream::HandshakeMode handshake_mode) {
     30     handshake_mode_ = handshake_mode;
     31   }
     32 
     33   void set_proof_verify_details(
     34       const ProofVerifyDetails* proof_verify_details) {
     35     proof_verify_details_ = proof_verify_details;
     36   }
     37 
     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