Handle crypto frames in quic::test::ImmediateGoAwaySession
gfe-relnote: n/a (test-only change)
PiperOrigin-RevId: 272967726
Change-Id: I6582fbe3f706686054099cdc69b18e0ea4cb1a4f
diff --git a/quic/test_tools/quic_test_server.cc b/quic/test_tools/quic_test_server.cc
index 9eee97a..3eb4f64 100644
--- a/quic/test_tools/quic_test_server.cc
+++ b/quic/test_tools/quic_test_server.cc
@@ -227,6 +227,11 @@
QuicSimpleServerSession::OnStreamFrame(frame);
}
+void ImmediateGoAwaySession::OnCryptoFrame(const QuicCryptoFrame& frame) {
+ SendGoAway(QUIC_PEER_GOING_AWAY, "");
+ QuicSimpleServerSession::OnCryptoFrame(frame);
+}
+
} // namespace test
} // namespace quic
diff --git a/quic/test_tools/quic_test_server.h b/quic/test_tools/quic_test_server.h
index 3661b7a..f9b677a 100644
--- a/quic/test_tools/quic_test_server.h
+++ b/quic/test_tools/quic_test_server.h
@@ -104,6 +104,7 @@
// Override to send GoAway.
void OnStreamFrame(const QuicStreamFrame& frame) override;
+ void OnCryptoFrame(const QuicCryptoFrame& frame) override;
};
} // namespace test