src/net/quic/test_tools/mock_crypto_client_stream_factory.h

   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   ~MockCryptoClientStreamFactory() override {}
  22
  23   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_