OLD | NEW |
---|---|
1 /* | 1 /* |
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/modules/congestion_controller/probe_bitrate_estimator.h" | 11 #include "webrtc/modules/congestion_controller/probe_bitrate_estimator.h" |
12 | 12 |
13 #include <vector> | 13 #include <vector> |
14 #include <utility> | 14 #include <utility> |
15 | 15 |
16 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
18 #include "webrtc/base/random.h" | |
18 #include "webrtc/modules/remote_bitrate_estimator/aimd_rate_control.h" | 19 #include "webrtc/modules/remote_bitrate_estimator/aimd_rate_control.h" |
19 | 20 |
20 namespace webrtc { | 21 namespace webrtc { |
21 | 22 |
22 class TestProbeBitrateEstimator : public ::testing::Test { | 23 class TestProbeBitrateEstimator : public ::testing::Test { |
23 public: | 24 public: |
24 TestProbeBitrateEstimator() : probe_bitrate_estimator_() {} | 25 TestProbeBitrateEstimator() : probe_bitrate_estimator_(), rand_(0x7716123) {} |
25 | 26 |
26 void AddPacketFeedback(int probe_cluster_id, | 27 void AddPacketFeedback(int probe_cluster_id, |
28 int16_t sequence_number, | |
27 size_t size_bytes, | 29 size_t size_bytes, |
28 int64_t send_time_ms, | 30 int64_t send_time_ms, |
29 int64_t arrival_time_ms) { | 31 int64_t arrival_time_ms) { |
30 PacketInfo info(arrival_time_ms, send_time_ms, 0, size_bytes, | 32 PacketInfo info(arrival_time_ms, send_time_ms, sequence_number, size_bytes, |
31 probe_cluster_id); | 33 probe_cluster_id); |
32 ProbingResult res = probe_bitrate_estimator_.PacketFeedback(info); | 34 ProbingResult res = probe_bitrate_estimator_.PacketFeedback(info); |
33 if (res.valid()) | 35 if (res.valid()) |
34 results_.emplace_back(res.bps, res.timestamp); | 36 results_.emplace_back(res.bps, res.timestamp); |
35 } | 37 } |
36 | 38 |
37 void CheckResult(size_t index, int bps, int max_diff, int64_t timestamp) { | 39 void CheckResult(size_t index, int bps, int max_diff, int64_t timestamp) { |
38 ASSERT_GT(results_.size(), index); | 40 ASSERT_GT(results_.size(), index); |
39 EXPECT_NEAR(results_[index].first, bps, max_diff); | 41 EXPECT_NEAR(results_[index].first, bps, max_diff); |
40 EXPECT_EQ(results_[index].second, timestamp); | 42 EXPECT_EQ(results_[index].second, timestamp); |
41 } | 43 } |
42 | 44 |
45 uint16_t Rand() { return rand_.Rand(std::numeric_limits<uint16_t>::max()); } | |
danilchap
2016/08/18 19:29:09
return rand_.Rand<uint16_t>();
if you want to use
| |
46 | |
43 protected: | 47 protected: |
44 std::vector<std::pair<int, int64_t>> results_; | 48 std::vector<std::pair<int, int64_t>> results_; |
45 ProbeBitrateEstimator probe_bitrate_estimator_; | 49 ProbeBitrateEstimator probe_bitrate_estimator_; |
50 Random rand_; | |
danilchap
2016/08/18 19:29:09
May be cleaner use some constant kSeqNo = 7716; fo
| |
46 }; | 51 }; |
47 | 52 |
48 TEST_F(TestProbeBitrateEstimator, OneCluster) { | 53 TEST_F(TestProbeBitrateEstimator, OneCluster) { |
49 AddPacketFeedback(0, 1000, 0, 10); | 54 uint16_t seq_num = Rand(); |
50 AddPacketFeedback(0, 1000, 10, 20); | 55 |
51 AddPacketFeedback(0, 1000, 20, 30); | 56 AddPacketFeedback(0, seq_num++, 1000, 0, 10); |
52 AddPacketFeedback(0, 1000, 30, 40); | 57 AddPacketFeedback(0, seq_num++, 1000, 10, 20); |
58 AddPacketFeedback(0, seq_num++, 1000, 20, 30); | |
59 AddPacketFeedback(0, seq_num++, 1000, 30, 40); | |
53 | 60 |
54 CheckResult(0, 800000, 10, 40); | 61 CheckResult(0, 800000, 10, 40); |
55 } | 62 } |
56 | 63 |
57 TEST_F(TestProbeBitrateEstimator, FastReceive) { | 64 TEST_F(TestProbeBitrateEstimator, FastReceive) { |
58 AddPacketFeedback(0, 1000, 0, 15); | 65 uint16_t seq_num = Rand(); |
59 AddPacketFeedback(0, 1000, 10, 30); | 66 |
60 AddPacketFeedback(0, 1000, 20, 35); | 67 AddPacketFeedback(0, seq_num++, 1000, 0, 15); |
61 AddPacketFeedback(0, 1000, 30, 40); | 68 AddPacketFeedback(0, seq_num++, 1000, 10, 30); |
69 AddPacketFeedback(0, seq_num++, 1000, 20, 35); | |
70 AddPacketFeedback(0, seq_num++, 1000, 30, 40); | |
62 | 71 |
63 CheckResult(0, 800000, 10, 40); | 72 CheckResult(0, 800000, 10, 40); |
64 } | 73 } |
65 | 74 |
66 TEST_F(TestProbeBitrateEstimator, TooFastReceive) { | 75 TEST_F(TestProbeBitrateEstimator, TooFastReceive) { |
67 AddPacketFeedback(0, 1000, 0, 19); | 76 uint16_t seq_num = Rand(); |
68 AddPacketFeedback(0, 1000, 10, 30); | 77 |
69 AddPacketFeedback(0, 1000, 20, 40); | 78 AddPacketFeedback(0, seq_num++, 1000, 0, 19); |
70 AddPacketFeedback(0, 1000, 40, 50); | 79 AddPacketFeedback(0, seq_num++, 1000, 10, 30); |
80 AddPacketFeedback(0, seq_num++, 1000, 20, 40); | |
81 AddPacketFeedback(0, seq_num++, 1000, 40, 50); | |
71 | 82 |
72 EXPECT_TRUE(results_.empty()); | 83 EXPECT_TRUE(results_.empty()); |
73 } | 84 } |
74 | 85 |
75 TEST_F(TestProbeBitrateEstimator, SlowReceive) { | 86 TEST_F(TestProbeBitrateEstimator, SlowReceive) { |
76 AddPacketFeedback(0, 1000, 0, 10); | 87 uint16_t seq_num = Rand(); |
77 AddPacketFeedback(0, 1000, 10, 40); | 88 |
78 AddPacketFeedback(0, 1000, 20, 70); | 89 AddPacketFeedback(0, seq_num++, 1000, 0, 10); |
79 AddPacketFeedback(0, 1000, 30, 85); | 90 AddPacketFeedback(0, seq_num++, 1000, 10, 40); |
91 AddPacketFeedback(0, seq_num++, 1000, 20, 70); | |
92 AddPacketFeedback(0, seq_num++, 1000, 30, 85); | |
80 | 93 |
81 CheckResult(0, 320000, 10, 85); | 94 CheckResult(0, 320000, 10, 85); |
82 } | 95 } |
83 | 96 |
84 TEST_F(TestProbeBitrateEstimator, BurstReceive) { | 97 TEST_F(TestProbeBitrateEstimator, BurstReceive) { |
85 AddPacketFeedback(0, 1000, 0, 50); | 98 uint16_t seq_num = Rand(); |
86 AddPacketFeedback(0, 1000, 10, 50); | 99 |
87 AddPacketFeedback(0, 1000, 20, 50); | 100 AddPacketFeedback(0, seq_num++, 1000, 0, 50); |
88 AddPacketFeedback(0, 1000, 40, 50); | 101 AddPacketFeedback(0, seq_num++, 1000, 10, 50); |
102 AddPacketFeedback(0, seq_num++, 1000, 20, 50); | |
103 AddPacketFeedback(0, seq_num++, 1000, 40, 50); | |
89 | 104 |
90 EXPECT_TRUE(results_.empty()); | 105 EXPECT_TRUE(results_.empty()); |
91 } | 106 } |
92 | 107 |
93 TEST_F(TestProbeBitrateEstimator, MultipleClusters) { | 108 TEST_F(TestProbeBitrateEstimator, MultipleClusters) { |
94 AddPacketFeedback(0, 1000, 0, 10); | 109 uint16_t seq_num = Rand(); |
95 AddPacketFeedback(0, 1000, 10, 20); | 110 |
96 AddPacketFeedback(0, 1000, 20, 30); | 111 AddPacketFeedback(0, seq_num++, 1000, 0, 10); |
97 AddPacketFeedback(0, 1000, 40, 60); | 112 AddPacketFeedback(0, seq_num++, 1000, 10, 20); |
98 AddPacketFeedback(0, 1000, 50, 60); | 113 AddPacketFeedback(0, seq_num++, 1000, 20, 30); |
114 AddPacketFeedback(0, seq_num++, 1000, 40, 60); | |
115 AddPacketFeedback(0, seq_num++, 1000, 50, 60); | |
99 | 116 |
100 CheckResult(0, 480000, 10, 60); | 117 CheckResult(0, 480000, 10, 60); |
101 CheckResult(1, 640000, 10, 60); | 118 CheckResult(1, 640000, 10, 60); |
102 | 119 |
103 AddPacketFeedback(1, 1000, 60, 70); | 120 AddPacketFeedback(1, seq_num++, 1000, 60, 70); |
104 AddPacketFeedback(1, 1000, 65, 77); | 121 AddPacketFeedback(1, seq_num++, 1000, 65, 77); |
105 AddPacketFeedback(1, 1000, 70, 84); | 122 AddPacketFeedback(1, seq_num++, 1000, 70, 84); |
106 AddPacketFeedback(1, 1000, 75, 90); | 123 AddPacketFeedback(1, seq_num++, 1000, 75, 90); |
107 | 124 |
108 CheckResult(2, 1200000, 10, 90); | 125 CheckResult(2, 1200000, 10, 90); |
109 } | 126 } |
110 | 127 |
111 TEST_F(TestProbeBitrateEstimator, OldProbe) { | 128 TEST_F(TestProbeBitrateEstimator, OldProbe) { |
112 AddPacketFeedback(0, 1000, 0, 10); | 129 uint16_t seq_num = Rand(); |
113 AddPacketFeedback(0, 1000, 10, 20); | |
114 AddPacketFeedback(0, 1000, 20, 30); | |
115 | 130 |
116 AddPacketFeedback(1, 1000, 60, 70); | 131 AddPacketFeedback(0, seq_num, 1000, 0, 10); |
117 AddPacketFeedback(1, 1000, 65, 77); | 132 AddPacketFeedback(0, seq_num + 1, 1000, 10, 20); |
118 AddPacketFeedback(1, 1000, 70, 84); | 133 AddPacketFeedback(0, seq_num + 2, 1000, 20, 30); |
119 AddPacketFeedback(1, 1000, 75, 90); | 134 |
135 AddPacketFeedback(1, seq_num + 4, 1000, 60, 70); | |
136 AddPacketFeedback(1, seq_num + 5, 1000, 65, 77); | |
137 AddPacketFeedback(1, seq_num + 6, 1000, 70, 84); | |
138 AddPacketFeedback(1, seq_num + 7, 1000, 75, 90); | |
120 | 139 |
121 CheckResult(0, 1200000, 10, 90); | 140 CheckResult(0, 1200000, 10, 90); |
122 | 141 |
123 AddPacketFeedback(0, 1000, 40, 60); | 142 AddPacketFeedback(0, seq_num + 3, 1000, 40, 60); |
124 | 143 |
125 EXPECT_EQ(1ul, results_.size()); | 144 EXPECT_EQ(1ul, results_.size()); |
126 } | 145 } |
127 | 146 |
147 TEST_F(TestProbeBitrateEstimator, SameProbe) { | |
148 uint16_t seq_num = Rand(); | |
149 | |
150 AddPacketFeedback(0, seq_num + 1, 1000, 0, 10); | |
151 AddPacketFeedback(0, seq_num + 2, 1000, 10, 40); | |
152 AddPacketFeedback(0, seq_num + 2, 1000, 10, 40); | |
153 AddPacketFeedback(0, seq_num + 3, 1000, 20, 70); | |
154 AddPacketFeedback(0, seq_num + 3, 1000, 20, 70); | |
155 AddPacketFeedback(0, seq_num + 4, 1000, 30, 85); | |
156 | |
157 CheckResult(0, 320000, 10, 85); | |
158 } | |
159 | |
128 } // namespace webrtc | 160 } // namespace webrtc |
OLD | NEW |