OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 2016 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/modules/congestion_controller/probe_bitrate_estimator.h" | |
12 | |
13 #include <vector> | |
14 #include <utility> | |
15 | |
16 #include "testing/gmock/include/gmock/gmock.h" | |
17 #include "testing/gtest/include/gtest/gtest.h" | |
18 #include "webrtc/modules/remote_bitrate_estimator/aimd_rate_control.h" | |
19 | |
20 namespace webrtc { | |
21 | |
22 class TestProbeBitrateEstimator : public ::testing::Test { | |
23 public: | |
24 TestProbeBitrateEstimator() : probe_bitrate_estimator_() {} | |
25 | |
26 void AddPacketFeedback(int probe_cluster_id, | |
27 size_t size, | |
28 int64_t send_time_ms, | |
29 int64_t arrival_time_ms) { | |
30 PacketInfo info(arrival_time_ms, send_time_ms, 0, size, probe_cluster_id); | |
31 ProbingResult res = probe_bitrate_estimator_.PacketFeedback(info); | |
32 if (res.bps != ProbingResult::kNoEstimate) | |
33 results_.emplace_back(res.bps, res.timestamp); | |
34 } | |
35 | |
36 void CheckResult(size_t index, int bps, int max_diff, int64_t timestamp) { | |
37 ASSERT_GT(results_.size(), index); | |
38 EXPECT_NEAR(results_[index].first, bps, max_diff); | |
39 EXPECT_EQ(results_[index].second, timestamp); | |
40 } | |
41 | |
42 protected: | |
43 std::vector<std::pair<int, int64_t>> results_; | |
44 ProbeBitrateEstimator probe_bitrate_estimator_; | |
45 }; | |
46 | |
47 TEST_F(TestProbeBitrateEstimator, OneCluster) { | |
48 AddPacketFeedback(0, 1000, 0, 10); | |
49 AddPacketFeedback(0, 1000, 10, 20); | |
50 AddPacketFeedback(0, 1000, 20, 30); | |
51 AddPacketFeedback(0, 1000, 40, 50); | |
52 | |
53 CheckResult(0, 100000, 10, 50); | |
54 } | |
55 | |
56 TEST_F(TestProbeBitrateEstimator, FastReceive) { | |
57 AddPacketFeedback(0, 1000, 0, 15); | |
58 AddPacketFeedback(0, 1000, 10, 30); | |
59 AddPacketFeedback(0, 1000, 20, 40); | |
60 AddPacketFeedback(0, 1000, 40, 50); | |
61 | |
62 CheckResult(0, 100000, 10, 50); | |
63 } | |
64 | |
65 TEST_F(TestProbeBitrateEstimator, TooFastReceive) { | |
66 AddPacketFeedback(0, 1000, 0, 19); | |
67 AddPacketFeedback(0, 1000, 10, 30); | |
68 AddPacketFeedback(0, 1000, 20, 40); | |
69 AddPacketFeedback(0, 1000, 40, 50); | |
70 | |
71 EXPECT_TRUE(results_.empty()); | |
72 } | |
73 | |
74 TEST_F(TestProbeBitrateEstimator, SlowReceive) { | |
75 AddPacketFeedback(0, 1000, 0, 10); | |
76 AddPacketFeedback(0, 1000, 10, 40); | |
77 AddPacketFeedback(0, 1000, 20, 70); | |
78 AddPacketFeedback(0, 1000, 40, 110); | |
79 | |
80 CheckResult(0, 40000, 10, 110); | |
81 } | |
82 | |
83 TEST_F(TestProbeBitrateEstimator, BurstReceive) { | |
84 AddPacketFeedback(0, 1000, 0, 50); | |
85 AddPacketFeedback(0, 1000, 10, 50); | |
86 AddPacketFeedback(0, 1000, 20, 50); | |
87 AddPacketFeedback(0, 1000, 40, 50); | |
88 | |
89 EXPECT_TRUE(results_.empty()); | |
90 } | |
91 | |
92 TEST_F(TestProbeBitrateEstimator, MultipleClusters) { | |
93 AddPacketFeedback(0, 1000, 0, 10); | |
94 AddPacketFeedback(0, 1000, 10, 20); | |
95 AddPacketFeedback(0, 1000, 20, 30); | |
96 AddPacketFeedback(0, 1000, 40, 60); | |
97 AddPacketFeedback(0, 1000, 50, 60); | |
98 | |
99 CheckResult(0, 80000, 10, 60); | |
100 CheckResult(1, 100000, 10, 60); | |
101 | |
102 AddPacketFeedback(1, 1000, 60, 70); | |
103 AddPacketFeedback(1, 1000, 65, 77); | |
104 AddPacketFeedback(1, 1000, 70, 84); | |
105 AddPacketFeedback(1, 1000, 75, 90); | |
106 | |
107 CheckResult(2, 200000, 10, 90); | |
108 } | |
109 | |
110 TEST_F(TestProbeBitrateEstimator, OldProbe) { | |
111 AddPacketFeedback(0, 1000, 0, 10); | |
112 AddPacketFeedback(0, 1000, 10, 20); | |
113 AddPacketFeedback(0, 1000, 20, 30); | |
114 | |
115 AddPacketFeedback(1, 1000, 60, 70); | |
116 AddPacketFeedback(1, 1000, 65, 77); | |
117 AddPacketFeedback(1, 1000, 70, 84); | |
118 AddPacketFeedback(1, 1000, 75, 90); | |
119 | |
120 CheckResult(0, 200000, 10, 90); | |
121 | |
122 AddPacketFeedback(0, 1000, 40, 60); | |
123 | |
124 EXPECT_EQ(1ul, results_.size()); | |
125 } | |
126 | |
127 } // namespace webrtc | |
OLD | NEW |