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 |
11 using testing::_; | 11 using testing::_; |
12 | 12 |
13 namespace net { | 13 namespace net { |
14 namespace test { | 14 namespace test { |
15 namespace { | 15 namespace { |
16 | 16 |
17 class QuicAckNotifierTest : public ::testing::Test { | 17 class QuicAckNotifierTest : public ::testing::Test { |
18 protected: | 18 protected: |
19 QuicAckNotifierTest() : zero_(QuicTime::Delta::Zero()) {} | 19 QuicAckNotifierTest() : zero_(QuicTime::Delta::Zero()) {} |
20 | 20 |
21 void SetUp() override { | 21 void SetUp() override { |
22 delegate_ = new MockAckNotifierDelegate; | 22 delegate_ = new MockAckNotifierDelegate; |
23 notifier_.reset(new QuicAckNotifier(delegate_)); | 23 notifier_.reset(new QuicAckNotifier(delegate_)); |
24 | 24 |
25 notifier_->AddSequenceNumber(26, 100); | 25 notifier_->OnSerializedPacket(); |
26 notifier_->AddSequenceNumber(99, 20); | 26 notifier_->OnSerializedPacket(); |
27 notifier_->AddSequenceNumber(1234, 3); | 27 notifier_->OnSerializedPacket(); |
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(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(zero_)); |
39 EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 39 EXPECT_FALSE(notifier_->OnAck(zero_)); |
40 EXPECT_TRUE(notifier_->OnAck(1234, zero_)); | 40 EXPECT_TRUE(notifier_->OnAck(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(zero_)); |
48 EXPECT_FALSE(notifier_->OnAck(99, zero_)); | 48 EXPECT_FALSE(notifier_->OnAck(zero_)); |
| 49 } |
| 50 |
| 51 // Should not trigger callback if we abandon all three packets. |
| 52 TEST_F(QuicAckNotifierTest, AbandonDoesNotTrigger) { |
| 53 // Should not trigger callback as not all packets have been seen. |
| 54 EXPECT_CALL(*delegate_, OnAckNotification(_, _, _)).Times(0); |
| 55 EXPECT_FALSE(notifier_->OnPacketAbandoned()); |
| 56 EXPECT_FALSE(notifier_->OnPacketAbandoned()); |
| 57 EXPECT_TRUE(notifier_->OnPacketAbandoned()); |
49 } | 58 } |
50 | 59 |
51 // Should trigger even after updating sequence numbers and receiving ACKs for | 60 // Should trigger even after updating sequence numbers and receiving ACKs for |
52 // new sequeunce numbers. | 61 // new sequeunce numbers. |
53 TEST_F(QuicAckNotifierTest, UpdateSeqNums) { | 62 TEST_F(QuicAckNotifierTest, UpdateSeqNums) { |
54 // Update a couple of the sequence numbers (i.e. retransmitted packets) | 63 // Update a couple of the sequence numbers (i.e. retransmitted packets) |
55 notifier_->OnPacketRetransmitted(20); | 64 notifier_->OnPacketRetransmitted(20); |
56 notifier_->OnPacketRetransmitted(3); | 65 notifier_->OnPacketRetransmitted(3); |
57 | 66 |
58 EXPECT_CALL(*delegate_, OnAckNotification(2, 20 + 3, _)).Times(1); | 67 EXPECT_CALL(*delegate_, OnAckNotification(2, 20 + 3, _)).Times(1); |
59 EXPECT_FALSE(notifier_->OnAck(26, zero_)); // original | 68 EXPECT_FALSE(notifier_->OnAck(zero_)); // original |
60 EXPECT_FALSE(notifier_->OnAck(3000, zero_)); // updated | 69 EXPECT_FALSE(notifier_->OnAck(zero_)); // updated |
61 EXPECT_TRUE(notifier_->OnAck(3001, zero_)); // updated | 70 EXPECT_TRUE(notifier_->OnAck(zero_)); // updated |
62 } | 71 } |
63 | 72 |
64 // Make sure the delegate is called with the delta time from the last ACK. | 73 // Make sure the delegate is called with the delta time from the last ACK. |
65 TEST_F(QuicAckNotifierTest, DeltaTime) { | 74 TEST_F(QuicAckNotifierTest, DeltaTime) { |
66 const QuicTime::Delta first_delta = QuicTime::Delta::FromSeconds(5); | 75 const QuicTime::Delta first_delta = QuicTime::Delta::FromSeconds(5); |
67 const QuicTime::Delta second_delta = QuicTime::Delta::FromSeconds(33); | 76 const QuicTime::Delta second_delta = QuicTime::Delta::FromSeconds(33); |
68 const QuicTime::Delta third_delta = QuicTime::Delta::FromSeconds(10); | 77 const QuicTime::Delta third_delta = QuicTime::Delta::FromSeconds(10); |
69 | 78 |
70 EXPECT_CALL(*delegate_, OnAckNotification(0, 0, third_delta)).Times(1); | 79 EXPECT_CALL(*delegate_, OnAckNotification(0, 0, third_delta)).Times(1); |
71 EXPECT_FALSE(notifier_->OnAck(26, first_delta)); | 80 EXPECT_FALSE(notifier_->OnAck(first_delta)); |
72 EXPECT_FALSE(notifier_->OnAck(99, second_delta)); | 81 EXPECT_FALSE(notifier_->OnAck(second_delta)); |
73 EXPECT_TRUE(notifier_->OnAck(1234, third_delta)); | 82 EXPECT_TRUE(notifier_->OnAck(third_delta)); |
74 } | 83 } |
75 | 84 |
76 } // namespace | 85 } // namespace |
77 } // namespace test | 86 } // namespace test |
78 } // namespace net | 87 } // namespace net |
OLD | NEW |