commit 56c8644cb240d042aab8c6d3fb0447c4b57289dc
author fayang <fayang@google.com> Wed Apr 07 12:09:43 2021 -0700
committer Copybara-Service <copybara-worker@google.com> Wed Apr 07 12:10:53 2021 -0700
tree c8305eb346cabf04a96e3c1732d899ea0650025c
parent 8fbe1bd8824689e7f92d4a8f0df0d581a2eb4b4a

Fix QuicConnectionTest.LostDataThenGetAcknowledged.

PiperOrigin-RevId: 367269823
Change-Id: I4a8f8e60d22122325fd5022ad6670f2708c4c96b

quic/core/quic_connection_test.cc

diff --git a/quic/core/quic_connection_test.cc b/quic/core/quic_connection_test.cc
index c27f96c..e2fba5b 100644
--- a/quic/core/quic_connection_test.cc
+++ b/quic/core/quic_connection_test.cc
@@ -14259,15 +14259,19 @@
           InvokeWithoutArgs(&notifier_, &SimpleSessionNotifier::OnCanWrite));
   QuicIpAddress ip_address;
   ASSERT_TRUE(ip_address.FromString("127.0.52.223"));
-  EXPECT_QUIC_BUG(ProcessFramesPacketWithAddresses(
-                      frames, kSelfAddress, QuicSocketAddress(ip_address, 1000),
-                      ENCRYPTION_FORWARD_SECURE),
-                  "Try to write mid packet processing");
   if (GetQuicReloadableFlag(quic_donot_write_mid_packet_processing)) {
+    EXPECT_QUIC_BUG(
+        ProcessFramesPacketWithAddresses(frames, kSelfAddress,
+                                         QuicSocketAddress(ip_address, 1000),
+                                         ENCRYPTION_FORWARD_SECURE),
+        "Try to write mid packet processing");
     EXPECT_EQ(1u, writer_->path_challenge_frames().size());
     // Verify stream frame will not be retransmitted.
     EXPECT_TRUE(writer_->stream_frames().empty());
   } else {
+    ProcessFramesPacketWithAddresses(frames, kSelfAddress,
+                                     QuicSocketAddress(ip_address, 1000),
+                                     ENCRYPTION_FORWARD_SECURE);
     // In prod, this would cause FAILED_TO_SERIALIZE_PACKET since the stream
     // data has been freed, but simple_data_producer does not free data.
     EXPECT_EQ(1u, writer_->stream_frames().size());