1 /*
2 * Copyright 2010 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "webrtc/base/gunit.h"
12 #include "webrtc/base/ratetracker.h"
13
14 namespace rtc {
15
16 class RateTrackerForTest : public RateTracker {
17 public:
RateTrackerForTest()18 RateTrackerForTest() : time_(0) {}
Time() const19 virtual uint32 Time() const { return time_; }
AdvanceTime(uint32 delta)20 void AdvanceTime(uint32 delta) { time_ += delta; }
21
22 private:
23 uint32 time_;
24 };
25
TEST(RateTrackerTest,TestBasics)26 TEST(RateTrackerTest, TestBasics) {
27 RateTrackerForTest tracker;
28 EXPECT_EQ(0U, tracker.total_units());
29 EXPECT_EQ(0U, tracker.units_second());
30
31 // Add a sample.
32 tracker.Update(1234);
33 // Advance the clock by 100 ms.
34 tracker.AdvanceTime(100);
35 // total_units should advance, but units_second should stay 0.
36 EXPECT_EQ(1234U, tracker.total_units());
37 EXPECT_EQ(0U, tracker.units_second());
38
39 // Repeat.
40 tracker.Update(1234);
41 tracker.AdvanceTime(100);
42 EXPECT_EQ(1234U * 2, tracker.total_units());
43 EXPECT_EQ(0U, tracker.units_second());
44
45 // Advance the clock by 800 ms, so we've elapsed a full second.
46 // units_second should now be filled in properly.
47 tracker.AdvanceTime(800);
48 EXPECT_EQ(1234U * 2, tracker.total_units());
49 EXPECT_EQ(1234U * 2, tracker.units_second());
50
51 // Poll the tracker again immediately. The reported rate should stay the same.
52 EXPECT_EQ(1234U * 2, tracker.total_units());
53 EXPECT_EQ(1234U * 2, tracker.units_second());
54
55 // Do nothing and advance by a second. We should drop down to zero.
56 tracker.AdvanceTime(1000);
57 EXPECT_EQ(1234U * 2, tracker.total_units());
58 EXPECT_EQ(0U, tracker.units_second());
59
60 // Send a bunch of data at a constant rate for 5.5 "seconds".
61 // We should report the rate properly.
62 for (int i = 0; i < 5500; i += 100) {
63 tracker.Update(9876U);
64 tracker.AdvanceTime(100);
65 }
66 EXPECT_EQ(9876U * 10, tracker.units_second());
67
68 // Advance the clock by 500 ms. Since we sent nothing over this half-second,
69 // the reported rate should be reduced by half.
70 tracker.AdvanceTime(500);
71 EXPECT_EQ(9876U * 5, tracker.units_second());
72 }
73
74 } // namespace rtc
75