Add a merely pass-through QuicMultipathSentPacketManager. Protected behind blocked flag FLAGS_quic_…

net/quic/test_tools/quic_connection_peer.cc

 // 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.
 
 #include "net/quic/test_tools/quic_connection_peer.h"
 
 #include "base/stl_util.h"
 #include "net/quic/congestion_control/send_algorithm_interface.h"
+#include "net/quic/quic_multipath_sent_packet_manager.h"
 #include "net/quic/quic_packet_writer.h"
 #include "net/quic/quic_received_packet_manager.h"
 #include "net/quic/test_tools/quic_framer_peer.h"
 #include "net/quic/test_tools/quic_packet_generator_peer.h"
 #include "net/quic/test_tools/quic_sent_packet_manager_peer.h"
 
 namespace net {
 namespace test {
 
 // static
 void QuicConnectionPeer::SendAck(QuicConnection* connection) {
   connection->SendAck();
 }
 
 // static
 void QuicConnectionPeer::SetSendAlgorithm(
     QuicConnection* connection,
+    QuicPathId path_id,
     SendAlgorithmInterface* send_algorithm) {
-  // TODO(fayang): Remove this method when there is a MockSentPacketManager.
-  static_cast<QuicSentPacketManager*>(connection->sent_packet_manager_.get())
+  GetSentPacketManager(connection, path_id)
       ->send_algorithm_.reset(send_algorithm);
 }
 
 // static
+void QuicConnectionPeer::SetLossAlgorithm(
+    QuicConnection* connection,
+    QuicPathId path_id,
+    LossDetectionInterface* loss_algorithm) {
+  GetSentPacketManager(connection, path_id)
+      ->loss_algorithm_.reset(loss_algorithm);
+}
+
+// static
 const QuicFrame QuicConnectionPeer::GetUpdatedAckFrame(
     QuicConnection* connection) {
   return connection->GetUpdatedAckFrame();
 }
 
 // static
 void QuicConnectionPeer::PopulateStopWaitingFrame(
     QuicConnection* connection,
     QuicStopWaitingFrame* stop_waiting) {
   connection->PopulateStopWaitingFrame(stop_waiting);
@@ skipping 13 matching lines @@
 }
 
 // static
 QuicPacketGenerator* QuicConnectionPeer::GetPacketGenerator(
     QuicConnection* connection) {
   return &connection->packet_generator_;
 }
 
 // static
 QuicSentPacketManager* QuicConnectionPeer::GetSentPacketManager(
-    QuicConnection* connection) {
-  // TODO(fayang): Remove this method when there is a MockSentPacketManager.
+    QuicConnection* connection,
+    QuicPathId path_id) {
+  if (FLAGS_quic_enable_multipath) {
+    return static_cast<QuicSentPacketManager*>(
+        static_cast<QuicMultipathSentPacketManager*>(
+            connection->sent_packet_manager_.get())
+            ->MaybeGetSentPacketManagerForPath(path_id));
+  }
   return static_cast<QuicSentPacketManager*>(
       connection->sent_packet_manager_.get());
 }
 
 // static
 QuicTime::Delta QuicConnectionPeer::GetNetworkTimeout(
     QuicConnection* connection) {
   return connection->idle_network_timeout_;
 }
 
@@ skipping 195 matching lines @@
 }
 
 // static
 void QuicConnectionPeer::SetAckDecimationDelay(QuicConnection* connection,
                                                float ack_decimation_delay) {
   connection->ack_decimation_delay_ = ack_decimation_delay;
 }
 
 }  // namespace test
 }  // namespace net