Index: net/quic/quic_connection_test.cc |
diff --git a/net/quic/quic_connection_test.cc b/net/quic/quic_connection_test.cc |
index c29ded9c9ff38515e5858422b7171dd5fc3fa125..2b3fa9537f485f8c6296d25e806686d5f04dd19a 100644 |
--- a/net/quic/quic_connection_test.cc |
+++ b/net/quic/quic_connection_test.cc |
@@ -202,7 +202,8 @@ class TestConnectionHelper : public QuicConnectionHelperInterface { |
public: |
class TestAlarm : public QuicAlarm { |
public: |
- explicit TestAlarm(QuicAlarm::Delegate* delegate) : QuicAlarm(delegate) {} |
+ explicit TestAlarm(QuicArenaScopedPtr<QuicAlarm::Delegate> delegate) |
+ : QuicAlarm(std::move(delegate)) {} |
void SetImpl() override {} |
void CancelImpl() override {} |
@@ -220,7 +221,13 @@ class TestConnectionHelper : public QuicConnectionHelperInterface { |
QuicRandom* GetRandomGenerator() override { return random_generator_; } |
QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) override { |
- return new TestAlarm(delegate); |
+ return new TestAlarm(QuicArenaScopedPtr<QuicAlarm::Delegate>(delegate)); |
+ } |
+ |
+ QuicArenaScopedPtr<QuicAlarm> CreateAlarm( |
+ QuicArenaScopedPtr<QuicAlarm::Delegate> delegate, |
+ QuicConnectionArena* arena) override { |
+ return arena->New<TestAlarm>(std::move(delegate)); |
} |
QuicBufferAllocator* GetBufferAllocator() override { |