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 |
(...skipping 11 matching lines...) Expand all Loading... | |
22 class TestProbeBitrateEstimator : public ::testing::Test { | 22 class TestProbeBitrateEstimator : public ::testing::Test { |
23 public: | 23 public: |
24 TestProbeBitrateEstimator() : probe_bitrate_estimator_() {} | 24 TestProbeBitrateEstimator() : probe_bitrate_estimator_() {} |
25 | 25 |
26 void AddPacketFeedback(int probe_cluster_id, | 26 void AddPacketFeedback(int probe_cluster_id, |
27 size_t size_bytes, | 27 size_t size_bytes, |
28 int64_t send_time_ms, | 28 int64_t send_time_ms, |
29 int64_t arrival_time_ms) { | 29 int64_t arrival_time_ms) { |
30 PacketInfo info(arrival_time_ms, send_time_ms, 0, size_bytes, | 30 PacketInfo info(arrival_time_ms, send_time_ms, 0, size_bytes, |
31 probe_cluster_id); | 31 probe_cluster_id); |
32 ProbingResult res = probe_bitrate_estimator_.PacketFeedback(info); | 32 int bps = probe_bitrate_estimator_.HandleProbeAndEstimateBitrate(info); |
33 if (res.valid()) | 33 measured_bps_ = bps > 0 ? bps : 0; |
danilchap
2016/08/23 13:27:34
bps > 0 ? bps : 0
converts invalid value '-1' into
Irfan
2016/08/23 17:27:27
Done.
| |
34 results_.emplace_back(res.bps, res.timestamp); | |
35 } | |
36 | |
37 void CheckResult(size_t index, int bps, int max_diff, int64_t timestamp) { | |
38 ASSERT_GT(results_.size(), index); | |
39 EXPECT_NEAR(results_[index].first, bps, max_diff); | |
40 EXPECT_EQ(results_[index].second, timestamp); | |
41 } | 34 } |
42 | 35 |
43 protected: | 36 protected: |
44 std::vector<std::pair<int, int64_t>> results_; | 37 int measured_bps_ = 0; |
45 ProbeBitrateEstimator probe_bitrate_estimator_; | 38 ProbeBitrateEstimator probe_bitrate_estimator_; |
46 }; | 39 }; |
47 | 40 |
48 TEST_F(TestProbeBitrateEstimator, OneCluster) { | 41 TEST_F(TestProbeBitrateEstimator, OneCluster) { |
49 AddPacketFeedback(0, 1000, 0, 10); | 42 AddPacketFeedback(0, 1000, 0, 10); |
50 AddPacketFeedback(0, 1000, 10, 20); | 43 AddPacketFeedback(0, 1000, 10, 20); |
51 AddPacketFeedback(0, 1000, 20, 30); | 44 AddPacketFeedback(0, 1000, 20, 30); |
52 AddPacketFeedback(0, 1000, 30, 40); | 45 AddPacketFeedback(0, 1000, 30, 40); |
53 | 46 |
54 CheckResult(0, 800000, 10, 40); | 47 EXPECT_NEAR(measured_bps_, 800000, 10); |
55 } | 48 } |
56 | 49 |
57 TEST_F(TestProbeBitrateEstimator, FastReceive) { | 50 TEST_F(TestProbeBitrateEstimator, FastReceive) { |
58 AddPacketFeedback(0, 1000, 0, 15); | 51 AddPacketFeedback(0, 1000, 0, 15); |
59 AddPacketFeedback(0, 1000, 10, 30); | 52 AddPacketFeedback(0, 1000, 10, 30); |
60 AddPacketFeedback(0, 1000, 20, 35); | 53 AddPacketFeedback(0, 1000, 20, 35); |
61 AddPacketFeedback(0, 1000, 30, 40); | 54 AddPacketFeedback(0, 1000, 30, 40); |
62 | 55 |
63 CheckResult(0, 800000, 10, 40); | 56 EXPECT_NEAR(measured_bps_, 800000, 10); |
64 } | 57 } |
65 | 58 |
66 TEST_F(TestProbeBitrateEstimator, TooFastReceive) { | 59 TEST_F(TestProbeBitrateEstimator, TooFastReceive) { |
67 AddPacketFeedback(0, 1000, 0, 19); | 60 AddPacketFeedback(0, 1000, 0, 19); |
68 AddPacketFeedback(0, 1000, 10, 30); | 61 AddPacketFeedback(0, 1000, 10, 30); |
69 AddPacketFeedback(0, 1000, 20, 40); | 62 AddPacketFeedback(0, 1000, 20, 40); |
70 AddPacketFeedback(0, 1000, 40, 50); | 63 AddPacketFeedback(0, 1000, 40, 50); |
71 | 64 |
72 EXPECT_TRUE(results_.empty()); | 65 EXPECT_EQ(measured_bps_, 0); |
73 } | 66 } |
74 | 67 |
75 TEST_F(TestProbeBitrateEstimator, SlowReceive) { | 68 TEST_F(TestProbeBitrateEstimator, SlowReceive) { |
76 AddPacketFeedback(0, 1000, 0, 10); | 69 AddPacketFeedback(0, 1000, 0, 10); |
77 AddPacketFeedback(0, 1000, 10, 40); | 70 AddPacketFeedback(0, 1000, 10, 40); |
78 AddPacketFeedback(0, 1000, 20, 70); | 71 AddPacketFeedback(0, 1000, 20, 70); |
79 AddPacketFeedback(0, 1000, 30, 85); | 72 AddPacketFeedback(0, 1000, 30, 85); |
80 | 73 |
81 CheckResult(0, 320000, 10, 85); | 74 EXPECT_NEAR(measured_bps_, 320000, 10); |
82 } | 75 } |
83 | 76 |
84 TEST_F(TestProbeBitrateEstimator, BurstReceive) { | 77 TEST_F(TestProbeBitrateEstimator, BurstReceive) { |
85 AddPacketFeedback(0, 1000, 0, 50); | 78 AddPacketFeedback(0, 1000, 0, 50); |
86 AddPacketFeedback(0, 1000, 10, 50); | 79 AddPacketFeedback(0, 1000, 10, 50); |
87 AddPacketFeedback(0, 1000, 20, 50); | 80 AddPacketFeedback(0, 1000, 20, 50); |
88 AddPacketFeedback(0, 1000, 40, 50); | 81 AddPacketFeedback(0, 1000, 40, 50); |
89 | 82 |
90 EXPECT_TRUE(results_.empty()); | 83 EXPECT_EQ(measured_bps_, 0); |
91 } | 84 } |
92 | 85 |
93 TEST_F(TestProbeBitrateEstimator, MultipleClusters) { | 86 TEST_F(TestProbeBitrateEstimator, MultipleClusters) { |
94 AddPacketFeedback(0, 1000, 0, 10); | 87 AddPacketFeedback(0, 1000, 0, 10); |
95 AddPacketFeedback(0, 1000, 10, 20); | 88 AddPacketFeedback(0, 1000, 10, 20); |
96 AddPacketFeedback(0, 1000, 20, 30); | 89 AddPacketFeedback(0, 1000, 20, 30); |
97 AddPacketFeedback(0, 1000, 40, 60); | 90 AddPacketFeedback(0, 1000, 40, 60); |
91 | |
92 EXPECT_NEAR(measured_bps_, 480000, 10); | |
93 | |
98 AddPacketFeedback(0, 1000, 50, 60); | 94 AddPacketFeedback(0, 1000, 50, 60); |
99 | 95 |
100 CheckResult(0, 480000, 10, 60); | 96 EXPECT_NEAR(measured_bps_, 640000, 10); |
101 CheckResult(1, 640000, 10, 60); | |
102 | 97 |
103 AddPacketFeedback(1, 1000, 60, 70); | 98 AddPacketFeedback(1, 1000, 60, 70); |
104 AddPacketFeedback(1, 1000, 65, 77); | 99 AddPacketFeedback(1, 1000, 65, 77); |
105 AddPacketFeedback(1, 1000, 70, 84); | 100 AddPacketFeedback(1, 1000, 70, 84); |
106 AddPacketFeedback(1, 1000, 75, 90); | 101 AddPacketFeedback(1, 1000, 75, 90); |
107 | 102 |
108 CheckResult(2, 1200000, 10, 90); | 103 EXPECT_NEAR(measured_bps_, 1200000, 10); |
109 } | 104 } |
110 | 105 |
111 TEST_F(TestProbeBitrateEstimator, OldProbe) { | 106 TEST_F(TestProbeBitrateEstimator, IgnoreOldClusters) { |
112 AddPacketFeedback(0, 1000, 0, 10); | 107 AddPacketFeedback(0, 1000, 0, 10); |
113 AddPacketFeedback(0, 1000, 10, 20); | 108 AddPacketFeedback(0, 1000, 10, 20); |
114 AddPacketFeedback(0, 1000, 20, 30); | 109 AddPacketFeedback(0, 1000, 20, 30); |
115 | 110 |
116 AddPacketFeedback(1, 1000, 60, 70); | 111 AddPacketFeedback(1, 1000, 60, 70); |
117 AddPacketFeedback(1, 1000, 65, 77); | 112 AddPacketFeedback(1, 1000, 65, 77); |
118 AddPacketFeedback(1, 1000, 70, 84); | 113 AddPacketFeedback(1, 1000, 70, 84); |
119 AddPacketFeedback(1, 1000, 75, 90); | 114 AddPacketFeedback(1, 1000, 75, 90); |
120 | 115 |
121 CheckResult(0, 1200000, 10, 90); | 116 EXPECT_NEAR(measured_bps_, 1200000, 10); |
122 | 117 |
123 AddPacketFeedback(0, 1000, 40, 60); | 118 // Coming in 6s later |
119 AddPacketFeedback(0, 1000, 40 + 6000, 60 + 6000); | |
124 | 120 |
125 EXPECT_EQ(1ul, results_.size()); | 121 EXPECT_EQ(measured_bps_, 0); |
126 } | 122 } |
127 | 123 |
128 } // namespace webrtc | 124 } // namespace webrtc |
OLD | NEW |