QUIC: retransmit packets with the correct encryption.

net/quic/test_tools/mock_crypto_client_stream_factory.h

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_
 #define NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_
 
 #include <string>
 
 #include "net/quic/quic_crypto_client_stream.h"
 #include "net/quic/quic_crypto_client_stream_factory.h"
 #include "net/quic/quic_session.h"
 
 namespace net {
 
 class MockCryptoClientStreamFactory : public QuicCryptoClientStreamFactory  {
  public:
+  // HandshakeMode enumerates the handshake mode MockCryptoClientStream should

[Ryan Hamilton, 2013/05/01 23:47:32]

nit: please move this enum to the mock stream (out of the factory)  (This will
let you remove the new include in the mock_stream.h

[ramant (doing other things), 2013/05/02 00:08:05]

Done.

+  // mock in CryptoConnect.
+  enum HandshakeMode {
+    // CONFIRM_HANDSHAKE indicates that during CryptoConnect, we will set
+    // encryption_established_ to true but handshake_confirmed_ to true.

[Ryan Hamilton, 2013/05/01 23:47:32]

// CONFIRM_HANDSHAKE indicates that CryptoConnect will immediately confirm the
handshake and establish encryption.

[ramant (doing other things), 2013/05/02 00:08:05]

Done.

+    CONFIRM_HANDSHAKE,
+
+    // ZERO_RTT indicates that during CryptoConnect, we will set
+    // encryption_established_ to true but handshake_confirmed_ to false.

[Ryan Hamilton, 2013/05/01 23:47:32]

// ZERO_RTT indicates that CryptoConnect will establish encryption but will not
confirm the handshake.

[ramant (doing other things), 2013/05/02 00:08:05]

Done.

+    ZERO_RTT,
+  };
+
+  MockCryptoClientStreamFactory();
+
   virtual ~MockCryptoClientStreamFactory() {}
 
   virtual QuicCryptoClientStream* CreateQuicCryptoClientStream(
       const string& server_hostname,
       const QuicConfig& config,
       QuicSession* session,
       QuicCryptoClientConfig* crypto_config) OVERRIDE;
+
+  void set_handshake_mode(HandshakeMode handshake_mode) {
+    handshake_mode_ = handshake_mode;
+  }
+
+ private:
+  HandshakeMode handshake_mode_;
 };
 
 }  // namespace net
 
 #endif  // NET_QUIC_TEST_TOOLS_MOCK_CRYPTO_CLIENT_STREAM_FACTORY_H_