Enables PacketGenerator's use of PacketCreator's FEC primitives.

net/quic/test_tools/quic_connection_peer.h

 // Copyright (c) 2012 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_QUIC_CONNECTION_PEER_H_
 #define NET_QUIC_TEST_TOOLS_QUIC_CONNECTION_PEER_H_
 
 #include "base/basictypes.h"
 #include "net/base/ip_endpoint.h"
 #include "net/quic/quic_connection_stats.h"
 #include "net/quic/quic_protocol.h"
 
 namespace net {
 
 struct QuicAckFrame;
 struct QuicPacketHeader;
 class QuicAlarm;
 class QuicConnection;
 class QuicConnectionHelperInterface;
 class QuicConnectionVisitorInterface;
 class QuicEncryptedPacket;
 class QuicFecGroup;
 class QuicFramer;
 class QuicPacketCreator;
+class QuicPacketGenerator;
 class QuicPacketWriter;
 class QuicReceivedPacketManager;
 class QuicSentPacketManager;
 class ReceiveAlgorithmInterface;
 class SendAlgorithmInterface;
 
 namespace test {
 
 // Peer to make public a number of otherwise private QuicConnection methods.
 class QuicConnectionPeer {
  public:
   static void SendAck(QuicConnection* connection);
 
   static void SetReceiveAlgorithm(QuicConnection* connection,
                                   ReceiveAlgorithmInterface* receive_algorithm);
 
   static void SetSendAlgorithm(QuicConnection* connection,
                                SendAlgorithmInterface* send_algorithm);
 
   static QuicAckFrame* CreateAckFrame(QuicConnection* connection);
 
   static QuicConnectionVisitorInterface* GetVisitor(
       QuicConnection* connection);
 
   static QuicPacketCreator* GetPacketCreator(QuicConnection* connection);
 
+  static QuicPacketGenerator* GetPacketGenerator(QuicConnection* connection);
+
   static QuicSentPacketManager* GetSentPacketManager(
       QuicConnection* connection);
 
   static QuicReceivedPacketManager* GetReceivedPacketManager(
       QuicConnection* connection);
 
   static QuicTime::Delta GetNetworkTimeout(QuicConnection* connection);
 
   static bool IsSavedForRetransmission(
       QuicConnection* connection,
@@ skipping 52 matching lines @@
 
  private:
   DISALLOW_COPY_AND_ASSIGN(QuicConnectionPeer);
 };
 
 }  // namespace test
 
 }  // namespace net
 
 #endif  // NET_QUIC_TEST_TOOLS_QUIC_CONNECTION_PEER_H_