Index: net/quic/test_tools/quic_test_utils.cc |
diff --git a/net/quic/test_tools/quic_test_utils.cc b/net/quic/test_tools/quic_test_utils.cc |
index a2ba09f9e0470b766171644ada095868e2de95f4..7fd347fdb75d1aad960fa4246df3ae6b814bc889 100644 |
--- a/net/quic/test_tools/quic_test_utils.cc |
+++ b/net/quic/test_tools/quic_test_utils.cc |
@@ -194,7 +194,19 @@ QuicRandom* MockConnectionHelper::GetRandomGenerator() { |
} |
QuicAlarm* MockConnectionHelper::CreateAlarm(QuicAlarm::Delegate* delegate) { |
- return new MockConnectionHelper::TestAlarm(delegate); |
+ return new MockConnectionHelper::TestAlarm( |
+ QuicArenaScopedPtr<QuicAlarm::Delegate>(delegate)); |
+} |
+ |
+QuicArenaScopedPtr<QuicAlarm> MockConnectionHelper::CreateAlarm( |
+ QuicArenaScopedPtr<QuicAlarm::Delegate> delegate, |
+ QuicConnectionArena* arena) { |
+ if (arena != nullptr) { |
+ return arena->New<MockConnectionHelper::TestAlarm>(std::move(delegate)); |
+ } else { |
+ return QuicArenaScopedPtr<MockConnectionHelper::TestAlarm>( |
+ new TestAlarm(std::move(delegate))); |
+ } |
} |
QuicBufferAllocator* MockConnectionHelper::GetBufferAllocator() { |