OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 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 <limits> | 11 #include <limits> |
12 #include <vector> | 12 #include <vector> |
13 | 13 |
14 #include "testing/gmock/include/gmock/gmock.h" | 14 #include "testing/gmock/include/gmock/gmock.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
16 | 16 |
17 #include "webrtc/base/checks.h" | 17 #include "webrtc/base/checks.h" |
18 #include "webrtc/base/scoped_ptr.h" | 18 #include "webrtc/base/scoped_ptr.h" |
19 #include "webrtc/modules/bitrate_controller/include/mock/mock_bitrate_controller
.h" | 19 #include "webrtc/modules/bitrate_controller/include/mock/mock_bitrate_controller
.h" |
20 #include "webrtc/modules/remote_bitrate_estimator/include/mock/mock_remote_bitra
te_estimator.h" | 20 #include "webrtc/modules/remote_bitrate_estimator/include/mock/mock_remote_bitra
te_estimator.h" |
21 #include "webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.h" | 21 #include "webrtc/modules/remote_bitrate_estimator/transport_feedback_adapter.h" |
22 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 22 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
23 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" | 23 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" |
24 #include "webrtc/modules/utility/include/mock/mock_process_thread.h" | |
25 #include "webrtc/system_wrappers/include/clock.h" | 24 #include "webrtc/system_wrappers/include/clock.h" |
26 | 25 |
27 using ::testing::_; | 26 using ::testing::_; |
28 using ::testing::Invoke; | 27 using ::testing::Invoke; |
29 | 28 |
30 namespace webrtc { | 29 namespace webrtc { |
31 namespace test { | 30 namespace test { |
32 | 31 |
33 class TransportFeedbackAdapterTest : public ::testing::Test { | 32 class TransportFeedbackAdapterTest : public ::testing::Test { |
34 public: | 33 public: |
35 TransportFeedbackAdapterTest() | 34 TransportFeedbackAdapterTest() |
36 : clock_(0), | 35 : clock_(0), |
37 bitrate_estimator_(nullptr), | 36 bitrate_estimator_(nullptr), |
38 bitrate_controller_(this), | 37 bitrate_controller_(this), |
39 receiver_estimated_bitrate_(0) {} | 38 receiver_estimated_bitrate_(0) {} |
40 | 39 |
41 virtual ~TransportFeedbackAdapterTest() {} | 40 virtual ~TransportFeedbackAdapterTest() {} |
42 | 41 |
43 virtual void SetUp() { | 42 virtual void SetUp() { |
44 adapter_.reset(new TransportFeedbackAdapter(&bitrate_controller_, &clock_, | 43 adapter_.reset(new TransportFeedbackAdapter(&bitrate_controller_, &clock_)); |
45 &process_thread_)); | |
46 | 44 |
47 bitrate_estimator_ = new MockRemoteBitrateEstimator(); | 45 bitrate_estimator_ = new MockRemoteBitrateEstimator(); |
48 EXPECT_CALL(process_thread_, RegisterModule(bitrate_estimator_)).Times(1); | |
49 adapter_->SetBitrateEstimator(bitrate_estimator_); | 46 adapter_->SetBitrateEstimator(bitrate_estimator_); |
50 } | 47 } |
51 | 48 |
52 virtual void TearDown() { | 49 virtual void TearDown() { |
53 EXPECT_CALL(process_thread_, DeRegisterModule(bitrate_estimator_)).Times(1); | |
54 adapter_.reset(); | 50 adapter_.reset(); |
55 } | 51 } |
56 | 52 |
57 protected: | 53 protected: |
58 // Proxy class used since TransportFeedbackAdapter will own the instance | 54 // Proxy class used since TransportFeedbackAdapter will own the instance |
59 // passed at construction. | 55 // passed at construction. |
60 class MockBitrateControllerAdapter : public MockBitrateController { | 56 class MockBitrateControllerAdapter : public MockBitrateController { |
61 public: | 57 public: |
62 explicit MockBitrateControllerAdapter(TransportFeedbackAdapterTest* owner) | 58 explicit MockBitrateControllerAdapter(TransportFeedbackAdapterTest* owner) |
63 : MockBitrateController(), owner_(owner) {} | 59 : MockBitrateController(), owner_(owner) {} |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
102 | 98 |
103 // Utility method, to reset arrival_time_ms before adding send time. | 99 // Utility method, to reset arrival_time_ms before adding send time. |
104 void OnSentPacket(PacketInfo info) { | 100 void OnSentPacket(PacketInfo info) { |
105 info.arrival_time_ms = 0; | 101 info.arrival_time_ms = 0; |
106 adapter_->AddPacket(info.sequence_number, info.payload_size, | 102 adapter_->AddPacket(info.sequence_number, info.payload_size, |
107 info.was_paced); | 103 info.was_paced); |
108 adapter_->OnSentPacket(info.sequence_number, info.send_time_ms); | 104 adapter_->OnSentPacket(info.sequence_number, info.send_time_ms); |
109 } | 105 } |
110 | 106 |
111 SimulatedClock clock_; | 107 SimulatedClock clock_; |
112 MockProcessThread process_thread_; | |
113 MockRemoteBitrateEstimator* bitrate_estimator_; | 108 MockRemoteBitrateEstimator* bitrate_estimator_; |
114 MockBitrateControllerAdapter bitrate_controller_; | 109 MockBitrateControllerAdapter bitrate_controller_; |
115 rtc::scoped_ptr<TransportFeedbackAdapter> adapter_; | 110 rtc::scoped_ptr<TransportFeedbackAdapter> adapter_; |
116 | 111 |
117 uint32_t receiver_estimated_bitrate_; | 112 uint32_t receiver_estimated_bitrate_; |
118 }; | 113 }; |
119 | 114 |
120 TEST_F(TransportFeedbackAdapterTest, AdaptsFeedbackAndPopulatesSendTimes) { | 115 TEST_F(TransportFeedbackAdapterTest, AdaptsFeedbackAndPopulatesSendTimes) { |
121 std::vector<PacketInfo> packets; | 116 std::vector<PacketInfo> packets; |
122 packets.push_back(PacketInfo(100, 200, 0, 1500, true)); | 117 packets.push_back(PacketInfo(100, 200, 0, 1500, true)); |
(...skipping 192 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
315 })); | 310 })); |
316 adapter_->OnTransportFeedback(*feedback.get()); | 311 adapter_->OnTransportFeedback(*feedback.get()); |
317 | 312 |
318 sent_packets.push_back(info); | 313 sent_packets.push_back(info); |
319 | 314 |
320 ComparePacketVectors(sent_packets, received_feedback); | 315 ComparePacketVectors(sent_packets, received_feedback); |
321 } | 316 } |
322 | 317 |
323 } // namespace test | 318 } // namespace test |
324 } // namespace webrtc | 319 } // namespace webrtc |
OLD | NEW |