1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <cmath>
#include "base/logging.h"
#include "base/memory/scoped_ptr.h"
#include "net/quic/congestion_control/fix_rate_receiver.h"
#include "net/quic/congestion_control/fix_rate_sender.h"
#include "net/quic/test_tools/mock_clock.h"
#include "net/quic/quic_protocol.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
class FixRateTest : public ::testing::Test {
protected:
void SetUp() {
sender_.reset(new FixRateSender(&clock_));
receiver_.reset(new FixRateReceiver());
}
MockClock clock_;
scoped_ptr<FixRateSender> sender_;
scoped_ptr<FixRateReceiver> receiver_;
};
TEST_F(FixRateTest, ReceiverAPI) {
CongestionInfo info;
receiver_->SetBitrate(300000); // Bytes per second.
receiver_->RecordIncomingPacket(1, 1, 1, false);
ASSERT_TRUE(receiver_->GenerateCongestionInfo(&info));
EXPECT_EQ(kFixRate, info.type);
EXPECT_EQ(300000u, info.fix_rate.bitrate_in_bytes_per_second);
}
TEST_F(FixRateTest, SenderAPI) {
CongestionInfo info;
info.type = kFixRate;
info.fix_rate.bitrate_in_bytes_per_second = 300000;
sender_->OnIncomingCongestionInfo(info);
EXPECT_EQ(0, sender_->TimeUntilSend(false));
EXPECT_EQ(kMaxPacketSize, sender_->AvailableCongestionWindow());
sender_->SentPacket(1, 1200, false);
EXPECT_EQ(0u, sender_->AvailableCongestionWindow());
EXPECT_EQ(300000, sender_->BandwidthEstimate());
EXPECT_EQ(4000, sender_->TimeUntilSend(false));
clock_.AdvanceTime(0.002);
EXPECT_EQ(2000, sender_->TimeUntilSend(false));
clock_.AdvanceTime(0.002);
EXPECT_EQ(0, sender_->TimeUntilSend(false));
}
TEST_F(FixRateTest, Pacing) {
const int packet_size = 1200;
const int rtt_us = 30000;
CongestionInfo info;
receiver_->SetBitrate(240000); // Bytes per second.
ASSERT_TRUE(receiver_->GenerateCongestionInfo(&info));
sender_->OnIncomingCongestionInfo(info);
double acc_advance_time = 0.0;
for (int i = 0; i < 100; ++i) {
EXPECT_EQ(0, sender_->TimeUntilSend(false));
EXPECT_EQ(kMaxPacketSize, sender_->AvailableCongestionWindow());
sender_->SentPacket(i, packet_size, false);
double advance_time = sender_->TimeUntilSend(false) / 1000000.0;
clock_.AdvanceTime(advance_time);
sender_->OnIncomingAck(i, packet_size, rtt_us);
acc_advance_time += advance_time;
}
EXPECT_EQ(500, floor((acc_advance_time * 1000) + 0.5));
}
} // namespace net
|
