Index: net/quic/quic_sent_packet_manager_test.cc |
diff --git a/net/quic/quic_sent_packet_manager_test.cc b/net/quic/quic_sent_packet_manager_test.cc |
index 189638c8ea8e7d091133a1d35ddf9f9543ef39b6..2f1bef2d511082d134248dbeea1f218fec2a9313 100644 |
--- a/net/quic/quic_sent_packet_manager_test.cc |
+++ b/net/quic/quic_sent_packet_manager_test.cc |
@@ -868,11 +868,7 @@ TEST_F(QuicSentPacketManagerTest, TailLossProbeThenRTO) { |
clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(1000)); |
// The final RTO abandons all of them. |
- EXPECT_CALL(*send_algorithm_, RetransmissionDelay()) |
- .WillOnce(Return(QuicTime::Delta::FromMilliseconds(100))); |
- EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout()); |
- EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)) |
- .Times(kNumSentPackets + 2); |
+ EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); |
manager_.OnRetransmissionTimeout(); |
EXPECT_TRUE(manager_.HasPendingRetransmissions()); |
} |
@@ -926,11 +922,7 @@ TEST_F(QuicSentPacketManagerTest, RetransmissionTimeout) { |
SendDataPacket(i); |
} |
- EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(kNumSentPackets); |
- |
- EXPECT_CALL(*send_algorithm_, RetransmissionDelay()) |
- .WillOnce(Return(QuicTime::Delta::FromMilliseconds(1))); |
- EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout()); |
+ EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); |
manager_.OnRetransmissionTimeout(); |
} |
@@ -1013,8 +1005,7 @@ TEST_F(QuicSentPacketManagerTest, GetTransmissionTimeRTO) { |
EXPECT_EQ(expected_time, manager_.GetRetransmissionTime()); |
// Retransmit the packet by invoking the retransmission timeout. |
- EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(_, _)).Times(2); |
- EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout()); |
+ EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); |
clock_.AdvanceTime(expected_rto_delay); |
manager_.OnRetransmissionTimeout(); |
RetransmitNextPacket(3); |
@@ -1053,8 +1044,7 @@ TEST_F(QuicSentPacketManagerTest, GetTransmissionDelay) { |
EXPECT_EQ(delay, |
QuicSentPacketManagerPeer::GetRetransmissionDelay(&manager_)); |
delay = delay.Add(delay); |
- EXPECT_CALL(*send_algorithm_, OnPacketAbandoned(i + 1, _)); |
- EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout()); |
+ EXPECT_CALL(*send_algorithm_, OnRetransmissionTimeout(true)); |
manager_.OnRetransmissionTimeout(); |
RetransmitNextPacket(i + 2); |
} |