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 c6cae977958fc927189ce2d894df256c881943ef..c9b81de0c8029230dc92c5d07fe5991766e72b6a 100644 |
--- a/net/quic/test_tools/quic_test_utils.cc |
+++ b/net/quic/test_tools/quic_test_utils.cc |
@@ -201,18 +201,18 @@ QuicRandom* MockConnectionHelper::GetRandomGenerator() { |
return &random_generator_; |
} |
-QuicAlarm* MockConnectionHelper::CreateAlarm(QuicAlarm::Delegate* delegate) { |
- return new MockConnectionHelper::TestAlarm( |
+QuicAlarm* MockAlarmFactory::CreateAlarm(QuicAlarm::Delegate* delegate) { |
+ return new MockAlarmFactory::TestAlarm( |
QuicArenaScopedPtr<QuicAlarm::Delegate>(delegate)); |
} |
-QuicArenaScopedPtr<QuicAlarm> MockConnectionHelper::CreateAlarm( |
+QuicArenaScopedPtr<QuicAlarm> MockAlarmFactory::CreateAlarm( |
QuicArenaScopedPtr<QuicAlarm::Delegate> delegate, |
QuicConnectionArena* arena) { |
if (arena != nullptr) { |
- return arena->New<MockConnectionHelper::TestAlarm>(std::move(delegate)); |
+ return arena->New<MockAlarmFactory::TestAlarm>(std::move(delegate)); |
} else { |
- return QuicArenaScopedPtr<MockConnectionHelper::TestAlarm>( |
+ return QuicArenaScopedPtr<MockAlarmFactory::TestAlarm>( |
new TestAlarm(std::move(delegate))); |
} |
} |
@@ -226,48 +226,58 @@ void MockConnectionHelper::AdvanceTime(QuicTime::Delta delta) { |
} |
MockConnection::MockConnection(MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective) |
: MockConnection(kTestConnectionId, |
IPEndPoint(TestPeerIPAddress(), kTestPort), |
helper, |
+ alarm_factory, |
perspective, |
QuicSupportedVersions()) {} |
MockConnection::MockConnection(IPEndPoint address, |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective) |
: MockConnection(kTestConnectionId, |
address, |
helper, |
+ alarm_factory, |
perspective, |
QuicSupportedVersions()) {} |
MockConnection::MockConnection(QuicConnectionId connection_id, |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective) |
: MockConnection(connection_id, |
IPEndPoint(TestPeerIPAddress(), kTestPort), |
helper, |
+ alarm_factory, |
perspective, |
QuicSupportedVersions()) {} |
MockConnection::MockConnection(MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective, |
const QuicVersionVector& supported_versions) |
: MockConnection(kTestConnectionId, |
IPEndPoint(TestPeerIPAddress(), kTestPort), |
helper, |
+ alarm_factory, |
perspective, |
supported_versions) {} |
MockConnection::MockConnection(QuicConnectionId connection_id, |
IPEndPoint address, |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective, |
const QuicVersionVector& supported_versions) |
: QuicConnection(connection_id, |
address, |
helper, |
+ alarm_factory, |
new testing::NiceMock<MockPacketWriter>(), |
/* owns_writer= */ true, |
perspective, |
@@ -284,14 +294,16 @@ void MockConnection::AdvanceTime(QuicTime::Delta delta) { |
} |
PacketSavingConnection::PacketSavingConnection(MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective) |
- : MockConnection(helper, perspective) {} |
+ : MockConnection(helper, alarm_factory, perspective) {} |
PacketSavingConnection::PacketSavingConnection( |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
Perspective perspective, |
const QuicVersionVector& supported_versions) |
- : MockConnection(helper, perspective, supported_versions) {} |
+ : MockConnection(helper, alarm_factory, perspective, supported_versions) {} |
PacketSavingConnection::~PacketSavingConnection() { |
STLDeleteElements(&encrypted_packets_); |
@@ -761,6 +773,7 @@ void CreateClientSessionForTest(QuicServerId server_id, |
QuicTime::Delta connection_start_time, |
QuicVersionVector supported_versions, |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
QuicCryptoClientConfig* crypto_client_config, |
PacketSavingConnection** client_connection, |
TestQuicSpdyClientSession** client_session) { |
@@ -775,7 +788,7 @@ void CreateClientSessionForTest(QuicServerId server_id, |
? DefaultQuicConfigStatelessRejects() |
: DefaultQuicConfig(); |
*client_connection = new PacketSavingConnection( |
- helper, Perspective::IS_CLIENT, supported_versions); |
+ helper, alarm_factory, Perspective::IS_CLIENT, supported_versions); |
*client_session = new TestQuicSpdyClientSession( |
*client_connection, config, server_id, crypto_client_config); |
(*client_connection)->AdvanceTime(connection_start_time); |
@@ -786,6 +799,7 @@ void CreateServerSessionForTest( |
QuicTime::Delta connection_start_time, |
QuicVersionVector supported_versions, |
MockConnectionHelper* helper, |
+ MockAlarmFactory* alarm_factory, |
QuicCryptoServerConfig* server_crypto_config, |
QuicCompressedCertsCache* compressed_certs_cache, |
PacketSavingConnection** server_connection, |
@@ -798,7 +812,7 @@ void CreateServerSessionForTest( |
<< "strike-register will be unhappy."; |
*server_connection = new PacketSavingConnection( |
- helper, Perspective::IS_SERVER, supported_versions); |
+ helper, alarm_factory, Perspective::IS_SERVER, supported_versions); |
*server_session = new TestQuicSpdyServerSession( |
*server_connection, DefaultQuicConfig(), server_crypto_config, |
compressed_certs_cache); |