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
11using testing::_;
12
13namespace net {
14namespace test {
15namespace {
16
17class QuicAckNotifierTest : public ::testing::Test {
18 protected:
19  QuicAckNotifierTest() : zero_(QuicTime::Delta::Zero()) {}
20
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.
36TEST_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.
44TEST_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.
53TEST_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.
65TEST_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