Fix QuicConnectionTest.LostDataThenGetAcknowledged. PiperOrigin-RevId: 367269823 Change-Id: I4a8f8e60d22122325fd5022ad6670f2708c4c96b

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());