Send HTTP/3 GOAWAY for ImmediateGoAwaySession in tests.

Testing only. not protected.

PiperOrigin-RevId: 310955172
Change-Id: I4031680a40ebdf50a65e5faa6c34e6581343ec4a

quic/test_tools/quic_test_server.cc

diff --git a/quic/test_tools/quic_test_server.cc b/quic/test_tools/quic_test_server.cc
index 11be664..1b6452e 100644
--- a/quic/test_tools/quic_test_server.cc
+++ b/quic/test_tools/quic_test_server.cc
@@ -235,14 +235,27 @@
 }
 
 void ImmediateGoAwaySession::OnCryptoFrame(const QuicCryptoFrame& frame) {
-  // In IETF QUIC, GOAWAY lives up in HTTP/3 layer. Even if it's a immediate
-  // goaway session, goaway shouldn't be sent when crypto frame is received.
+  // In IETF QUIC, GOAWAY lives up in HTTP/3 layer. It's sent in a QUIC stream
+  // and requires encryption. Thus the sending is done in
+  // OnNewEncryptionKeyAvailable().
   if (!VersionUsesHttp3(transport_version())) {
     SendGoAway(QUIC_PEER_GOING_AWAY, "");
   }
   QuicSimpleServerSession::OnCryptoFrame(frame);
 }
 
+void ImmediateGoAwaySession::OnNewEncryptionKeyAvailable(
+    EncryptionLevel level,
+    std::unique_ptr<QuicEncrypter> encrypter) {
+  QuicSimpleServerSession::OnNewEncryptionKeyAvailable(level,
+                                                       std::move(encrypter));
+  if (VersionUsesHttp3(transport_version())) {
+    if (IsEncryptionEstablished() && !http3_goaway_sent()) {
+      SendHttp3GoAway();
+    }
+  }
+}
+
 }  // namespace test
 
 }  // namespace quic

quic/test_tools/quic_test_server.h

diff --git a/quic/test_tools/quic_test_server.h b/quic/test_tools/quic_test_server.h
index d88337a..2fa3e20 100644
--- a/quic/test_tools/quic_test_server.h
+++ b/quic/test_tools/quic_test_server.h
@@ -105,6 +105,9 @@
   // Override to send GoAway.
   void OnStreamFrame(const QuicStreamFrame& frame) override;
   void OnCryptoFrame(const QuicCryptoFrame& frame) override;
+  void OnNewEncryptionKeyAvailable(
+      EncryptionLevel level,
+      std::unique_ptr<QuicEncrypter> encrypter) override;
 };
 
 }  // namespace test