| OLD | NEW | 
|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 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/quic_ack_notifier.h" | 5 #include "net/quic/quic_ack_notifier.h" | 
| 6 | 6 | 
| 7 #include "net/quic/test_tools/quic_test_utils.h" | 7 #include "net/quic/test_tools/quic_test_utils.h" | 
| 8 #include "testing/gmock/include/gmock/gmock.h" | 8 #include "testing/gmock/include/gmock/gmock.h" | 
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" | 
| 10 | 10 | 
| (...skipping 16 matching lines...) Expand all  Loading... | 
| 27     notifier_->AddSequenceNumber(1234, 3); | 27     notifier_->AddSequenceNumber(1234, 3); | 
| 28   } | 28   } | 
| 29 | 29 | 
| 30   MockAckNotifierDelegate* delegate_; | 30   MockAckNotifierDelegate* delegate_; | 
| 31   scoped_ptr<QuicAckNotifier> notifier_; | 31   scoped_ptr<QuicAckNotifier> notifier_; | 
| 32   QuicTime::Delta zero_; | 32   QuicTime::Delta zero_; | 
| 33 }; | 33 }; | 
| 34 | 34 | 
| 35 // Should trigger callback when we receive acks for all the registered seqnums. | 35 // Should trigger callback when we receive acks for all the registered seqnums. | 
| 36 TEST_F(QuicAckNotifierTest, TriggerCallback) { | 36 TEST_F(QuicAckNotifierTest, TriggerCallback) { | 
| 37   EXPECT_CALL(*delegate_, OnAckNotification(3, 123, 0, 0, zero_)).Times(1); | 37   EXPECT_CALL(*delegate_, OnAckNotification(0, 0, zero_)).Times(1); | 
| 38   EXPECT_FALSE(notifier_->OnAck(26, zero_)); | 38   EXPECT_FALSE(notifier_->OnAck(26, zero_)); | 
| 39   EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 39   EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 
| 40   EXPECT_TRUE(notifier_->OnAck(1234, zero_)); | 40   EXPECT_TRUE(notifier_->OnAck(1234, zero_)); | 
| 41 } | 41 } | 
| 42 | 42 | 
| 43 // Should not trigger callback if we never provide all the seqnums. | 43 // Should not trigger callback if we never provide all the seqnums. | 
| 44 TEST_F(QuicAckNotifierTest, DoesNotTrigger) { | 44 TEST_F(QuicAckNotifierTest, DoesNotTrigger) { | 
| 45   // Should not trigger callback as not all packets have been seen. | 45   // Should not trigger callback as not all packets have been seen. | 
| 46   EXPECT_CALL(*delegate_, OnAckNotification(_, _, _, _, _)).Times(0); | 46   EXPECT_CALL(*delegate_, OnAckNotification(_, _, _)).Times(0); | 
| 47   EXPECT_FALSE(notifier_->OnAck(26, zero_)); | 47   EXPECT_FALSE(notifier_->OnAck(26, zero_)); | 
| 48   EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 48   EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 
| 49 } | 49 } | 
| 50 | 50 | 
| 51 // Should trigger even after updating sequence numbers and receiving ACKs for | 51 // Should trigger even after updating sequence numbers and receiving ACKs for | 
| 52 // new sequeunce numbers. | 52 // new sequeunce numbers. | 
| 53 TEST_F(QuicAckNotifierTest, UpdateSeqNums) { | 53 TEST_F(QuicAckNotifierTest, UpdateSeqNums) { | 
| 54   // Update a couple of the sequence numbers (i.e. retransmitted packets) | 54   // Update a couple of the sequence numbers (i.e. retransmitted packets) | 
| 55   notifier_->UpdateSequenceNumber(99, 3000); | 55   notifier_->UpdateSequenceNumber(99, 3000); | 
| 56   notifier_->UpdateSequenceNumber(1234, 3001); | 56   notifier_->UpdateSequenceNumber(1234, 3001); | 
| 57 | 57 | 
| 58   EXPECT_CALL(*delegate_, OnAckNotification(3, 123, 2, 20 + 3, _)).Times(1); | 58   EXPECT_CALL(*delegate_, OnAckNotification(2, 20 + 3, _)).Times(1); | 
| 59   EXPECT_FALSE(notifier_->OnAck(26, zero_));    // original | 59   EXPECT_FALSE(notifier_->OnAck(26, zero_));    // original | 
| 60   EXPECT_FALSE(notifier_->OnAck(3000, zero_));  // updated | 60   EXPECT_FALSE(notifier_->OnAck(3000, zero_));  // updated | 
| 61   EXPECT_TRUE(notifier_->OnAck(3001, zero_));   // updated | 61   EXPECT_TRUE(notifier_->OnAck(3001, zero_));   // updated | 
| 62 } | 62 } | 
| 63 | 63 | 
| 64 // Make sure the delegate is called with the delta time from the last ACK. | 64 // Make sure the delegate is called with the delta time from the last ACK. | 
| 65 TEST_F(QuicAckNotifierTest, DeltaTime) { | 65 TEST_F(QuicAckNotifierTest, DeltaTime) { | 
| 66   const QuicTime::Delta first_delta = QuicTime::Delta::FromSeconds(5); | 66   const QuicTime::Delta first_delta = QuicTime::Delta::FromSeconds(5); | 
| 67   const QuicTime::Delta second_delta = QuicTime::Delta::FromSeconds(33); | 67   const QuicTime::Delta second_delta = QuicTime::Delta::FromSeconds(33); | 
| 68   const QuicTime::Delta third_delta = QuicTime::Delta::FromSeconds(10); | 68   const QuicTime::Delta third_delta = QuicTime::Delta::FromSeconds(10); | 
| 69 | 69 | 
| 70   EXPECT_CALL(*delegate_, | 70   EXPECT_CALL(*delegate_, OnAckNotification(0, 0, third_delta)).Times(1); | 
| 71               OnAckNotification(3, 123, 0, 0, third_delta)) |  | 
| 72       .Times(1); |  | 
| 73   EXPECT_FALSE(notifier_->OnAck(26, first_delta)); | 71   EXPECT_FALSE(notifier_->OnAck(26, first_delta)); | 
| 74   EXPECT_FALSE(notifier_->OnAck(99, second_delta)); | 72   EXPECT_FALSE(notifier_->OnAck(99, second_delta)); | 
| 75   EXPECT_TRUE(notifier_->OnAck(1234, third_delta)); | 73   EXPECT_TRUE(notifier_->OnAck(1234, third_delta)); | 
| 76 } | 74 } | 
| 77 | 75 | 
| 78 }  // namespace | 76 }  // namespace | 
| 79 }  // namespace test | 77 }  // namespace test | 
| 80 }  // namespace net | 78 }  // namespace net | 
| OLD | NEW | 
|---|