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 <queue>
   9 #include <string>
  10
  11 #include "net/quic/quic_crypto_client_stream.h"
  12 #include "net/quic/quic_crypto_client_stream_factory.h"
  13 #include "net/quic/test_tools/mock_crypto_client_stream.h"
  14
  15 namespace net {
  16
  17 class QuicServerId;
  18
  19 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory  {
  20  public:
  21   MockCryptoClientStreamFactory();
  22   ~MockCryptoClientStreamFactory() override;
  23
  24   QuicCryptoClientStream* CreateQuicCryptoClientStream(
  25       const QuicServerId& server_id,
  26       QuicChromiumClientSession* session,
  27       QuicCryptoClientConfig* crypto_config) override;
  28
  29   void set_handshake_mode(
  30       MockCryptoClientStream::HandshakeMode handshake_mode) {
  31     handshake_mode_ = handshake_mode;
  32   }
  33
  34   // The caller keeps ownership of |proof_verify_details|.
  35   void AddProofVerifyDetails(const ProofVerifyDetails* proof_verify_details) {
  36     proof_verify_details_queue_.push(proof_verify_details);
  37   }
  38
  39   MockCryptoClientStream* last_stream() const {
  40     return last_stream_;
  41   }
  42
  43  private:
  44   MockCryptoClientStream::HandshakeMode handshake_mode_;
  45   MockCryptoClientStream* last_stream_;
  46   std::queue<const ProofVerifyDetails*> proof_verify_details_queue_;
  47
  48   DISALLOW_COPY_AND_ASSIGN(MockCryptoClientStreamFactory);
  49 };
  50
  51 }  // namespace net
  52
  53 #endif  // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_