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