Refactoring: Create per-connection packet writers in QuicDispatcher.

net/tools/quic/test_tools/quic_dispatcher_peer.h

 // Copyright 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_TOOLS_QUIC_TEST_TOOLS_QUIC_DISPATCHER_PEER_H_
 #define NET_TOOLS_QUIC_TEST_TOOLS_QUIC_DISPATCHER_PEER_H_
 
 #include "net/tools/quic/quic_dispatcher.h"
 
 #include "net/base/ip_endpoint.h"
 
 namespace net {
 namespace tools {
 
 class QuicPacketWriterWrapper;
 
 namespace test {
 
 class QuicDispatcherPeer {
  public:
   static void SetTimeWaitListManager(
       QuicDispatcher* dispatcher,
       QuicTimeWaitListManager* time_wait_list_manager);
 
-  // Injects |writer| into |dispatcher| as the top level writer.
+  // Injects |writer| into |dispatcher| as the shared writer.
   static void UseWriter(QuicDispatcher* dispatcher,
                         QuicPacketWriterWrapper* writer);
 
   static QuicPacketWriter* GetWriter(QuicDispatcher* dispatcher);
 
+  static void SetPacketWriterFactory(
+      QuicDispatcher* dispatcher,
+      QuicDispatcher::PacketWriterFactory* packet_writer_factory);
+
   static QuicEpollConnectionHelper* GetHelper(QuicDispatcher* dispatcher);
 
   static QuicConnection* CreateQuicConnection(
       QuicDispatcher* dispatcher,
       QuicConnectionId connection_id,
       const IPEndPoint& server,
       const IPEndPoint& client);
 
   static QuicDispatcher::WriteBlockedList* GetWriteBlockedList(
       QuicDispatcher* dispatcher);
 
  private:
   DISALLOW_COPY_AND_ASSIGN(QuicDispatcherPeer);
 };
 
 }  // namespace test
 }  // namespace tools
 }  // namespace net
 
 #endif  // NET_TOOLS_QUIC_TEST_TOOLS_QUIC_DISPATCHER_PEER_H_