| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/congestion_control/inter_arrival_sender.h" | 5 #include "net/quic/congestion_control/inter_arrival_sender.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/stl_util.h" | 9 #include "base/stl_util.h" |
| 10 #include "net/quic/test_tools/mock_clock.h" | 10 #include "net/quic/test_tools/mock_clock.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 | 12 |
| 13 using std::pair; | 13 using std::pair; |
| 14 | 14 |
| 15 namespace net { | 15 namespace net { |
| 16 namespace test { | 16 namespace test { |
| 17 | 17 |
| 18 class InterArrivalSenderTest : public ::testing::Test { | 18 class InterArrivalSenderTest : public ::testing::Test { |
| 19 protected: | 19 protected: |
| 20 InterArrivalSenderTest() | 20 InterArrivalSenderTest() |
| 21 : rtt_(QuicTime::Delta::FromMilliseconds(60)), | 21 : one_ms_(QuicTime::Delta::FromMilliseconds(1)), |
| 22 one_ms_(QuicTime::Delta::FromMilliseconds(1)), | |
| 23 one_s_(QuicTime::Delta::FromMilliseconds(1000)), | 22 one_s_(QuicTime::Delta::FromMilliseconds(1000)), |
| 24 nine_ms_(QuicTime::Delta::FromMilliseconds(9)), | 23 nine_ms_(QuicTime::Delta::FromMilliseconds(9)), |
| 25 send_start_time_(send_clock_.Now()), | 24 send_start_time_(send_clock_.Now()), |
| 26 sender_(&send_clock_), | 25 sender_(&send_clock_), |
| 27 sequence_number_(1), | 26 sequence_number_(1), |
| 28 acked_sequence_number_(1), | 27 acked_sequence_number_(1), |
| 29 feedback_sequence_number_(1) { | 28 feedback_sequence_number_(1) { |
| 30 send_clock_.AdvanceTime(one_ms_); | 29 send_clock_.AdvanceTime(one_ms_); |
| 31 receive_clock_.AdvanceTime(one_ms_); | 30 receive_clock_.AdvanceTime(one_ms_); |
| 32 } | 31 } |
| (...skipping 14 matching lines...) Expand all Loading... |
| 47 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA); | 46 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA); |
| 48 sequence_number_++; | 47 sequence_number_++; |
| 49 } | 48 } |
| 50 EXPECT_FALSE(sender_.TimeUntilSend(send_clock_.Now(), | 49 EXPECT_FALSE(sender_.TimeUntilSend(send_clock_.Now(), |
| 51 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); | 50 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); |
| 52 } | 51 } |
| 53 | 52 |
| 54 void AckNPackets(int n) { | 53 void AckNPackets(int n) { |
| 55 for (int i = 0; i < n; ++i) { | 54 for (int i = 0; i < n; ++i) { |
| 56 sender_.OnPacketAcked( | 55 sender_.OnPacketAcked( |
| 57 acked_sequence_number_++, kDefaultMaxPacketSize, rtt_); | 56 acked_sequence_number_++, kDefaultMaxPacketSize); |
| 58 } | 57 } |
| 59 } | 58 } |
| 60 | 59 |
| 61 void SendDelaySpikeFeedbackMessage(QuicTime::Delta spike_time) { | 60 void SendDelaySpikeFeedbackMessage(QuicTime::Delta spike_time) { |
| 62 QuicCongestionFeedbackFrame feedback; | 61 QuicCongestionFeedbackFrame feedback; |
| 63 feedback.type = kInterArrival; | 62 feedback.type = kInterArrival; |
| 64 feedback.inter_arrival.accumulated_number_of_lost_packets = 0; | 63 feedback.inter_arrival.accumulated_number_of_lost_packets = 0; |
| 65 receive_clock_.AdvanceTime(spike_time); | 64 receive_clock_.AdvanceTime(spike_time); |
| 66 QuicTime receive_time = receive_clock_.ApproximateNow(); | 65 QuicTime receive_time = receive_clock_.ApproximateNow(); |
| 67 feedback.inter_arrival.received_packet_times.insert( | 66 feedback.inter_arrival.received_packet_times.insert( |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 99 feedback_sequence_number_++; | 98 feedback_sequence_number_++; |
| 100 } | 99 } |
| 101 sender_.OnIncomingQuicCongestionFeedbackFrame(feedback, send_clock_.Now(), | 100 sender_.OnIncomingQuicCongestionFeedbackFrame(feedback, send_clock_.Now(), |
| 102 sent_packets_); | 101 sent_packets_); |
| 103 } | 102 } |
| 104 | 103 |
| 105 QuicTime::Delta SenderDeltaSinceStart() { | 104 QuicTime::Delta SenderDeltaSinceStart() { |
| 106 return send_clock_.ApproximateNow().Subtract(send_start_time_); | 105 return send_clock_.ApproximateNow().Subtract(send_start_time_); |
| 107 } | 106 } |
| 108 | 107 |
| 109 const QuicTime::Delta rtt_; | |
| 110 const QuicTime::Delta one_ms_; | 108 const QuicTime::Delta one_ms_; |
| 111 const QuicTime::Delta one_s_; | 109 const QuicTime::Delta one_s_; |
| 112 const QuicTime::Delta nine_ms_; | 110 const QuicTime::Delta nine_ms_; |
| 113 MockClock send_clock_; | 111 MockClock send_clock_; |
| 114 MockClock receive_clock_; | 112 MockClock receive_clock_; |
| 115 const QuicTime send_start_time_; | 113 const QuicTime send_start_time_; |
| 116 InterArrivalSender sender_; | 114 InterArrivalSender sender_; |
| 117 QuicPacketSequenceNumber sequence_number_; | 115 QuicPacketSequenceNumber sequence_number_; |
| 118 QuicPacketSequenceNumber acked_sequence_number_; | 116 QuicPacketSequenceNumber acked_sequence_number_; |
| 119 QuicPacketSequenceNumber feedback_sequence_number_; | 117 QuicPacketSequenceNumber feedback_sequence_number_; |
| (...skipping 377 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 497 // our minimum bitrate. | 495 // our minimum bitrate. |
| 498 SendFeedbackMessageNPackets(10, one_s_, one_s_); | 496 SendFeedbackMessageNPackets(10, one_s_, one_s_); |
| 499 send_clock_.AdvanceTime(sender_.TimeUntilSend(send_clock_.Now(), | 497 send_clock_.AdvanceTime(sender_.TimeUntilSend(send_clock_.Now(), |
| 500 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE)); | 498 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE)); |
| 501 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), | 499 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), |
| 502 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); | 500 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); |
| 503 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); | 501 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); |
| 504 } | 502 } |
| 505 | 503 |
| 506 TEST_F(InterArrivalSenderTest, MinBitrateDueToLoss) { | 504 TEST_F(InterArrivalSenderTest, MinBitrateDueToLoss) { |
| 505 sender_.UpdateRtt(QuicTime::Delta::FromMilliseconds(60)); |
| 507 QuicBandwidth expected_min_bitrate = QuicBandwidth::FromKBitsPerSecond(10); | 506 QuicBandwidth expected_min_bitrate = QuicBandwidth::FromKBitsPerSecond(10); |
| 508 QuicCongestionFeedbackFrame feedback; | 507 QuicCongestionFeedbackFrame feedback; |
| 509 // At startup make sure we can send. | 508 // At startup make sure we can send. |
| 510 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), | 509 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), |
| 511 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); | 510 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); |
| 512 | 511 |
| 513 // Send 5 bursts. | 512 // Send 5 bursts. |
| 514 for (int i = 0; i < 4; ++i) { | 513 for (int i = 0; i < 4; ++i) { |
| 515 SendAvailableCongestionWindow(); | 514 SendAvailableCongestionWindow(); |
| 516 send_clock_.AdvanceTime(sender_.TimeUntilSend(send_clock_.Now(), | 515 send_clock_.AdvanceTime(sender_.TimeUntilSend(send_clock_.Now(), |
| (...skipping 17 matching lines...) Expand all Loading... |
| 534 sender_.BandwidthEstimate().ToBytesPerSecond(), 1000); | 533 sender_.BandwidthEstimate().ToBytesPerSecond(), 1000); |
| 535 | 534 |
| 536 for (int i = 0; i < 15; ++i) { | 535 for (int i = 0; i < 15; ++i) { |
| 537 SendAvailableCongestionWindow(); | 536 SendAvailableCongestionWindow(); |
| 538 QuicTime::Delta time_until_send = sender_.TimeUntilSend(send_clock_.Now(), | 537 QuicTime::Delta time_until_send = sender_.TimeUntilSend(send_clock_.Now(), |
| 539 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE); | 538 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE); |
| 540 send_clock_.AdvanceTime(time_until_send); | 539 send_clock_.AdvanceTime(time_until_send); |
| 541 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), | 540 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), |
| 542 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); | 541 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); |
| 543 sender_.OnPacketLost(acked_sequence_number_ - 1, send_clock_.Now()); | 542 sender_.OnPacketLost(acked_sequence_number_ - 1, send_clock_.Now()); |
| 544 sender_.OnPacketAcked(acked_sequence_number_, kDefaultMaxPacketSize, rtt_); | 543 sender_.OnPacketAcked(acked_sequence_number_, kDefaultMaxPacketSize); |
| 545 acked_sequence_number_ += 2; // Create a loss by not acking both packets. | 544 acked_sequence_number_ += 2; // Create a loss by not acking both packets. |
| 546 SendFeedbackMessageNPackets(2, nine_ms_, nine_ms_); | 545 SendFeedbackMessageNPackets(2, nine_ms_, nine_ms_); |
| 547 } | 546 } |
| 548 // Test that our exponentail back off stop at expected_min_bitrate. | 547 // Test that our exponentail back off stop at expected_min_bitrate. |
| 549 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); | 548 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); |
| 550 | 549 |
| 551 for (int i = 0; i < 50; ++i) { | 550 for (int i = 0; i < 50; ++i) { |
| 552 SendAvailableCongestionWindow(); | 551 SendAvailableCongestionWindow(); |
| 553 QuicTime::Delta time_until_send = sender_.TimeUntilSend(send_clock_.Now(), | 552 QuicTime::Delta time_until_send = sender_.TimeUntilSend(send_clock_.Now(), |
| 554 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE); | 553 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE); |
| 555 send_clock_.AdvanceTime(time_until_send); | 554 send_clock_.AdvanceTime(time_until_send); |
| 556 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), | 555 EXPECT_TRUE(sender_.TimeUntilSend(send_clock_.Now(), |
| 557 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); | 556 NOT_RETRANSMISSION, HAS_RETRANSMITTABLE_DATA, NOT_HANDSHAKE).IsZero()); |
| 558 sender_.OnPacketLost(acked_sequence_number_ - 1, send_clock_.Now()); | 557 sender_.OnPacketLost(acked_sequence_number_ - 1, send_clock_.Now()); |
| 559 sender_.OnPacketAcked(acked_sequence_number_, kDefaultMaxPacketSize, rtt_); | 558 sender_.OnPacketAcked(acked_sequence_number_, kDefaultMaxPacketSize); |
| 560 acked_sequence_number_ += 2; // Create a loss by not acking both packets. | 559 acked_sequence_number_ += 2; // Create a loss by not acking both packets. |
| 561 SendFeedbackMessageNPackets(2, nine_ms_, nine_ms_); | 560 SendFeedbackMessageNPackets(2, nine_ms_, nine_ms_); |
| 562 | 561 |
| 563 // Make sure our bitrate is fixed at the expected_min_bitrate. | 562 // Make sure our bitrate is fixed at the expected_min_bitrate. |
| 564 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); | 563 EXPECT_EQ(expected_min_bitrate, sender_.BandwidthEstimate()); |
| 565 } | 564 } |
| 566 } | 565 } |
| 567 | 566 |
| 568 } // namespace test | 567 } // namespace test |
| 569 } // namespace net | 568 } // namespace net |
| OLD | NEW |