commit f9e025b15f34b4cbe190e5617e99deefc454fedf
author wub <wub@google.com> Wed Sep 01 07:53:35 2021 -0700
committer Copybara-Service <copybara-worker@google.com> Wed Sep 01 07:55:09 2021 -0700
tree d5d78f22a004339914aa4755150c8bbec2d1877f
parent 8d5eb27ee2e3f009f7180e8ace0ff97830d9c3e9

If a QuicAlarm belongs to a single QuicConnection, make sure the alarm fires under the corresponding QuicConnectionContext.

Protected by FLAGS_quic_reloadable_flag_quic_restore_connection_context_in_alarms.

PiperOrigin-RevId: 394230753

quic/test_tools/packet_dropping_test_writer.cc

diff --git a/quic/test_tools/packet_dropping_test_writer.cc b/quic/test_tools/packet_dropping_test_writer.cc
index 76dd778..079b228 100644
--- a/quic/test_tools/packet_dropping_test_writer.cc
+++ b/quic/test_tools/packet_dropping_test_writer.cc
@@ -18,7 +18,7 @@
 
 // An alarm that is scheduled if a blocked socket is simulated to indicate
 // it's writable again.
-class WriteUnblockedAlarm : public QuicAlarm::Delegate {
+class WriteUnblockedAlarm : public QuicAlarm::DelegateWithoutContext {
  public:
   explicit WriteUnblockedAlarm(PacketDroppingTestWriter* writer)
       : writer_(writer) {}
@@ -34,7 +34,7 @@
 
 // An alarm that is scheduled every time a new packet is to be written at a
 // later point.
-class DelayAlarm : public QuicAlarm::Delegate {
+class DelayAlarm : public QuicAlarm::DelegateWithoutContext {
  public:
   explicit DelayAlarm(PacketDroppingTestWriter* writer) : writer_(writer) {}