1 // Copyright (c) 2012 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 "base/logging.h"
6 #include "base/memory/scoped_ptr.h"
7 #include "net/quic/congestion_control/leaky_bucket.h"
8 #include "net/quic/test_tools/mock_clock.h"
9 #include "testing/gtest/include/gtest/gtest.h"
10
11 namespace net {
12 namespace test {
13
14 class LeakyBucketTest : public ::testing::Test {
15 protected:
SetUp()16 virtual void SetUp() {
17 leaky_bucket_.reset(new LeakyBucket(QuicBandwidth::Zero()));
18 }
19 MockClock clock_;
20 scoped_ptr<LeakyBucket> leaky_bucket_;
21 };
22
TEST_F(LeakyBucketTest,Basic)23 TEST_F(LeakyBucketTest, Basic) {
24 QuicBandwidth draining_rate = QuicBandwidth::FromBytesPerSecond(200000);
25 leaky_bucket_->SetDrainingRate(clock_.Now(), draining_rate);
26 leaky_bucket_->Add(clock_.Now(), 2000);
27 EXPECT_EQ(2000u, leaky_bucket_->BytesPending(clock_.Now()));
28 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(10),
29 leaky_bucket_->TimeRemaining(clock_.Now()));
30 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
31 EXPECT_EQ(1000u, leaky_bucket_->BytesPending(clock_.Now()));
32 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(5),
33 leaky_bucket_->TimeRemaining(clock_.Now()));
34 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
35 EXPECT_EQ(0u, leaky_bucket_->BytesPending(clock_.Now()));
36 EXPECT_TRUE(leaky_bucket_->TimeRemaining(clock_.Now()).IsZero());
37 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
38 EXPECT_EQ(0u, leaky_bucket_->BytesPending(clock_.Now()));
39 EXPECT_TRUE(leaky_bucket_->TimeRemaining(clock_.Now()).IsZero());
40 leaky_bucket_->Add(clock_.Now(), 2000);
41 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(11));
42 EXPECT_EQ(0u, leaky_bucket_->BytesPending(clock_.Now()));
43 EXPECT_TRUE(leaky_bucket_->TimeRemaining(clock_.Now()).IsZero());
44 leaky_bucket_->Add(clock_.Now(), 2000);
45 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
46 leaky_bucket_->Add(clock_.Now(), 2000);
47 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
48 EXPECT_EQ(2000u, leaky_bucket_->BytesPending(clock_.Now()));
49 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(10),
50 leaky_bucket_->TimeRemaining(clock_.Now()));
51 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(10));
52 EXPECT_EQ(0u, leaky_bucket_->BytesPending(clock_.Now()));
53 EXPECT_TRUE(leaky_bucket_->TimeRemaining(clock_.Now()).IsZero());
54 }
55
TEST_F(LeakyBucketTest,ChangeDrainRate)56 TEST_F(LeakyBucketTest, ChangeDrainRate) {
57 QuicBandwidth draining_rate = QuicBandwidth::FromBytesPerSecond(200000);
58 leaky_bucket_->SetDrainingRate(clock_.Now(), draining_rate);
59 leaky_bucket_->Add(clock_.Now(), 2000);
60 EXPECT_EQ(2000u, leaky_bucket_->BytesPending(clock_.Now()));
61 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(10),
62 leaky_bucket_->TimeRemaining(clock_.Now()));
63 clock_.AdvanceTime(QuicTime::Delta::FromMilliseconds(5));
64 EXPECT_EQ(1000u, leaky_bucket_->BytesPending(clock_.Now()));
65 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(5),
66 leaky_bucket_->TimeRemaining(clock_.Now()));
67 draining_rate = draining_rate.Scale(0.5f); // Cut drain rate in half.
68 leaky_bucket_->SetDrainingRate(clock_.Now(), draining_rate);
69 EXPECT_EQ(1000u, leaky_bucket_->BytesPending(clock_.Now()));
70 EXPECT_EQ(QuicTime::Delta::FromMilliseconds(10),
71 leaky_bucket_->TimeRemaining(clock_.Now()));
72 }
73
74 } // namespace test
75 } // namespace net
76