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