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 <string> | 11 #include <string> |
12 #include <vector> | 12 #include <vector> |
13 | 13 |
14 #include "webrtc/api/test/mock_audio_mixer.h" | 14 #include "webrtc/api/test/mock_audio_mixer.h" |
15 #include "webrtc/audio/audio_receive_stream.h" | 15 #include "webrtc/audio/audio_receive_stream.h" |
16 #include "webrtc/audio/conversion.h" | 16 #include "webrtc/audio/conversion.h" |
17 #include "webrtc/logging/rtc_event_log/mock/mock_rtc_event_log.h" | 17 #include "webrtc/logging/rtc_event_log/mock/mock_rtc_event_log.h" |
18 #include "webrtc/modules/audio_coding/codecs/mock/mock_audio_decoder_factory.h" | 18 #include "webrtc/modules/audio_coding/codecs/mock/mock_audio_decoder_factory.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/pacing/packet_router.h" | 20 #include "webrtc/modules/pacing/packet_router.h" |
21 #include "webrtc/modules/remote_bitrate_estimator/include/mock/mock_remote_bitra
te_estimator.h" | |
22 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" | 21 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
23 #include "webrtc/test/gtest.h" | 22 #include "webrtc/test/gtest.h" |
24 #include "webrtc/test/mock_voe_channel_proxy.h" | 23 #include "webrtc/test/mock_voe_channel_proxy.h" |
25 #include "webrtc/test/mock_voice_engine.h" | 24 #include "webrtc/test/mock_voice_engine.h" |
26 | 25 |
27 namespace webrtc { | 26 namespace webrtc { |
28 namespace test { | 27 namespace test { |
29 namespace { | 28 namespace { |
30 | 29 |
31 using testing::_; | 30 using testing::_; |
(...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
119 stream_config_.rtp.remote_ssrc = kRemoteSsrc; | 118 stream_config_.rtp.remote_ssrc = kRemoteSsrc; |
120 stream_config_.rtp.nack.rtp_history_ms = 300; | 119 stream_config_.rtp.nack.rtp_history_ms = 300; |
121 stream_config_.rtp.extensions.push_back( | 120 stream_config_.rtp.extensions.push_back( |
122 RtpExtension(RtpExtension::kAudioLevelUri, kAudioLevelId)); | 121 RtpExtension(RtpExtension::kAudioLevelUri, kAudioLevelId)); |
123 stream_config_.rtp.extensions.push_back(RtpExtension( | 122 stream_config_.rtp.extensions.push_back(RtpExtension( |
124 RtpExtension::kTransportSequenceNumberUri, kTransportSequenceNumberId)); | 123 RtpExtension::kTransportSequenceNumberUri, kTransportSequenceNumberId)); |
125 stream_config_.decoder_factory = decoder_factory_; | 124 stream_config_.decoder_factory = decoder_factory_; |
126 } | 125 } |
127 | 126 |
128 PacketRouter* packet_router() { return &packet_router_; } | 127 PacketRouter* packet_router() { return &packet_router_; } |
129 MockRemoteBitrateEstimator* remote_bitrate_estimator() { | |
130 return &remote_bitrate_estimator_; | |
131 } | |
132 MockRtcEventLog* event_log() { return &event_log_; } | 128 MockRtcEventLog* event_log() { return &event_log_; } |
133 AudioReceiveStream::Config& config() { return stream_config_; } | 129 AudioReceiveStream::Config& config() { return stream_config_; } |
134 rtc::scoped_refptr<AudioState> audio_state() { return audio_state_; } | 130 rtc::scoped_refptr<AudioState> audio_state() { return audio_state_; } |
135 rtc::scoped_refptr<MockAudioMixer> audio_mixer() { return audio_mixer_; } | 131 rtc::scoped_refptr<MockAudioMixer> audio_mixer() { return audio_mixer_; } |
136 MockVoiceEngine& voice_engine() { return voice_engine_; } | 132 MockVoiceEngine& voice_engine() { return voice_engine_; } |
137 MockVoEChannelProxy* channel_proxy() { return channel_proxy_; } | 133 MockVoEChannelProxy* channel_proxy() { return channel_proxy_; } |
138 | 134 |
139 void SetupMockForBweFeedback(bool send_side_bwe) { | |
140 EXPECT_CALL(remote_bitrate_estimator_, | |
141 RemoveStream(stream_config_.rtp.remote_ssrc)); | |
142 } | |
143 | |
144 void SetupMockForGetStats() { | 135 void SetupMockForGetStats() { |
145 using testing::DoAll; | 136 using testing::DoAll; |
146 using testing::SetArgReferee; | 137 using testing::SetArgReferee; |
147 | 138 |
148 ASSERT_TRUE(channel_proxy_); | 139 ASSERT_TRUE(channel_proxy_); |
149 EXPECT_CALL(*channel_proxy_, GetRTCPStatistics()) | 140 EXPECT_CALL(*channel_proxy_, GetRTCPStatistics()) |
150 .WillOnce(Return(kCallStats)); | 141 .WillOnce(Return(kCallStats)); |
151 EXPECT_CALL(*channel_proxy_, GetDelayEstimate()) | 142 EXPECT_CALL(*channel_proxy_, GetDelayEstimate()) |
152 .WillOnce(Return(kJitterBufferDelay + kPlayoutBufferDelay)); | 143 .WillOnce(Return(kJitterBufferDelay + kPlayoutBufferDelay)); |
153 EXPECT_CALL(*channel_proxy_, GetSpeechOutputLevelFullRange()) | 144 EXPECT_CALL(*channel_proxy_, GetSpeechOutputLevelFullRange()) |
154 .WillOnce(Return(kSpeechOutputLevel)); | 145 .WillOnce(Return(kSpeechOutputLevel)); |
155 EXPECT_CALL(*channel_proxy_, GetNetworkStatistics()) | 146 EXPECT_CALL(*channel_proxy_, GetNetworkStatistics()) |
156 .WillOnce(Return(kNetworkStats)); | 147 .WillOnce(Return(kNetworkStats)); |
157 EXPECT_CALL(*channel_proxy_, GetDecodingCallStatistics()) | 148 EXPECT_CALL(*channel_proxy_, GetDecodingCallStatistics()) |
158 .WillOnce(Return(kAudioDecodeStats)); | 149 .WillOnce(Return(kAudioDecodeStats)); |
159 | 150 |
160 EXPECT_CALL(voice_engine_, GetRecCodec(kChannelId, _)) | 151 EXPECT_CALL(voice_engine_, GetRecCodec(kChannelId, _)) |
161 .WillOnce(DoAll(SetArgReferee<1>(kCodecInst), Return(0))); | 152 .WillOnce(DoAll(SetArgReferee<1>(kCodecInst), Return(0))); |
162 } | 153 } |
163 | 154 |
164 private: | 155 private: |
165 PacketRouter packet_router_; | 156 PacketRouter packet_router_; |
166 rtc::scoped_refptr<AudioDecoderFactory> decoder_factory_; | 157 rtc::scoped_refptr<AudioDecoderFactory> decoder_factory_; |
167 MockRemoteBitrateEstimator remote_bitrate_estimator_; | |
168 MockRtcEventLog event_log_; | 158 MockRtcEventLog event_log_; |
169 testing::StrictMock<MockVoiceEngine> voice_engine_; | 159 testing::StrictMock<MockVoiceEngine> voice_engine_; |
170 rtc::scoped_refptr<AudioState> audio_state_; | 160 rtc::scoped_refptr<AudioState> audio_state_; |
171 rtc::scoped_refptr<MockAudioMixer> audio_mixer_; | 161 rtc::scoped_refptr<MockAudioMixer> audio_mixer_; |
172 AudioReceiveStream::Config stream_config_; | 162 AudioReceiveStream::Config stream_config_; |
173 testing::StrictMock<MockVoEChannelProxy>* channel_proxy_ = nullptr; | 163 testing::StrictMock<MockVoEChannelProxy>* channel_proxy_ = nullptr; |
174 }; | 164 }; |
175 | 165 |
176 void BuildOneByteExtension(std::vector<uint8_t>::iterator it, | 166 void BuildOneByteExtension(std::vector<uint8_t>::iterator it, |
177 int id, | 167 int id, |
(...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
237 "{rtp_history_ms: 0}, extensions: [{uri: " | 227 "{rtp_history_ms: 0}, extensions: [{uri: " |
238 "urn:ietf:params:rtp-hdrext:ssrc-audio-level, id: 3}]}, " | 228 "urn:ietf:params:rtp-hdrext:ssrc-audio-level, id: 3}]}, " |
239 "rtcp_send_transport: nullptr, voe_channel_id: 2}", | 229 "rtcp_send_transport: nullptr, voe_channel_id: 2}", |
240 config.ToString()); | 230 config.ToString()); |
241 } | 231 } |
242 | 232 |
243 TEST(AudioReceiveStreamTest, ConstructDestruct) { | 233 TEST(AudioReceiveStreamTest, ConstructDestruct) { |
244 ConfigHelper helper; | 234 ConfigHelper helper; |
245 internal::AudioReceiveStream recv_stream( | 235 internal::AudioReceiveStream recv_stream( |
246 helper.packet_router(), | 236 helper.packet_router(), |
247 helper.remote_bitrate_estimator(), | |
248 helper.config(), helper.audio_state(), helper.event_log()); | 237 helper.config(), helper.audio_state(), helper.event_log()); |
249 } | 238 } |
250 | 239 |
251 TEST(AudioReceiveStreamTest, ReceiveRtpPacket) { | 240 TEST(AudioReceiveStreamTest, ReceiveRtpPacket) { |
252 ConfigHelper helper; | 241 ConfigHelper helper; |
253 helper.config().rtp.transport_cc = true; | 242 helper.config().rtp.transport_cc = true; |
254 helper.SetupMockForBweFeedback(true); | |
255 internal::AudioReceiveStream recv_stream( | 243 internal::AudioReceiveStream recv_stream( |
256 helper.packet_router(), | 244 helper.packet_router(), |
257 helper.remote_bitrate_estimator(), | |
258 helper.config(), helper.audio_state(), helper.event_log()); | 245 helper.config(), helper.audio_state(), helper.event_log()); |
259 const int kTransportSequenceNumberValue = 1234; | 246 const int kTransportSequenceNumberValue = 1234; |
260 std::vector<uint8_t> rtp_packet = CreateRtpHeaderWithOneByteExtension( | 247 std::vector<uint8_t> rtp_packet = CreateRtpHeaderWithOneByteExtension( |
261 kTransportSequenceNumberId, kTransportSequenceNumberValue, 2); | 248 kTransportSequenceNumberId, kTransportSequenceNumberValue, 2); |
262 PacketTime packet_time(5678000, 0); | 249 PacketTime packet_time(5678000, 0); |
263 EXPECT_CALL(*helper.channel_proxy(), | 250 EXPECT_CALL(*helper.channel_proxy(), |
264 ReceivedRTPPacket(&rtp_packet[0], | 251 ReceivedRTPPacket(&rtp_packet[0], |
265 rtp_packet.size(), | 252 rtp_packet.size(), |
266 _)) | 253 _)) |
267 .WillOnce(Return(true)); | 254 .WillOnce(Return(true)); |
268 EXPECT_TRUE( | 255 EXPECT_TRUE( |
269 recv_stream.DeliverRtp(&rtp_packet[0], rtp_packet.size(), packet_time)); | 256 recv_stream.DeliverRtp(&rtp_packet[0], rtp_packet.size(), packet_time)); |
270 } | 257 } |
271 | 258 |
272 TEST(AudioReceiveStreamTest, ReceiveRtcpPacket) { | 259 TEST(AudioReceiveStreamTest, ReceiveRtcpPacket) { |
273 ConfigHelper helper; | 260 ConfigHelper helper; |
274 helper.config().rtp.transport_cc = true; | 261 helper.config().rtp.transport_cc = true; |
275 helper.SetupMockForBweFeedback(true); | |
276 internal::AudioReceiveStream recv_stream( | 262 internal::AudioReceiveStream recv_stream( |
277 helper.packet_router(), | 263 helper.packet_router(), |
278 helper.remote_bitrate_estimator(), | |
279 helper.config(), helper.audio_state(), helper.event_log()); | 264 helper.config(), helper.audio_state(), helper.event_log()); |
280 | 265 |
281 std::vector<uint8_t> rtcp_packet = CreateRtcpSenderReport(); | 266 std::vector<uint8_t> rtcp_packet = CreateRtcpSenderReport(); |
282 EXPECT_CALL(*helper.channel_proxy(), | 267 EXPECT_CALL(*helper.channel_proxy(), |
283 ReceivedRTCPPacket(&rtcp_packet[0], rtcp_packet.size())) | 268 ReceivedRTCPPacket(&rtcp_packet[0], rtcp_packet.size())) |
284 .WillOnce(Return(true)); | 269 .WillOnce(Return(true)); |
285 EXPECT_TRUE(recv_stream.DeliverRtcp(&rtcp_packet[0], rtcp_packet.size())); | 270 EXPECT_TRUE(recv_stream.DeliverRtcp(&rtcp_packet[0], rtcp_packet.size())); |
286 } | 271 } |
287 | 272 |
288 TEST(AudioReceiveStreamTest, GetStats) { | 273 TEST(AudioReceiveStreamTest, GetStats) { |
289 ConfigHelper helper; | 274 ConfigHelper helper; |
290 internal::AudioReceiveStream recv_stream( | 275 internal::AudioReceiveStream recv_stream( |
291 helper.packet_router(), | 276 helper.packet_router(), |
292 helper.remote_bitrate_estimator(), | |
293 helper.config(), helper.audio_state(), helper.event_log()); | 277 helper.config(), helper.audio_state(), helper.event_log()); |
294 helper.SetupMockForGetStats(); | 278 helper.SetupMockForGetStats(); |
295 AudioReceiveStream::Stats stats = recv_stream.GetStats(); | 279 AudioReceiveStream::Stats stats = recv_stream.GetStats(); |
296 EXPECT_EQ(kRemoteSsrc, stats.remote_ssrc); | 280 EXPECT_EQ(kRemoteSsrc, stats.remote_ssrc); |
297 EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesReceived), stats.bytes_rcvd); | 281 EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesReceived), stats.bytes_rcvd); |
298 EXPECT_EQ(static_cast<uint32_t>(kCallStats.packetsReceived), | 282 EXPECT_EQ(static_cast<uint32_t>(kCallStats.packetsReceived), |
299 stats.packets_rcvd); | 283 stats.packets_rcvd); |
300 EXPECT_EQ(kCallStats.cumulativeLost, stats.packets_lost); | 284 EXPECT_EQ(kCallStats.cumulativeLost, stats.packets_lost); |
301 EXPECT_EQ(Q8ToFloat(kCallStats.fractionLost), stats.fraction_lost); | 285 EXPECT_EQ(Q8ToFloat(kCallStats.fractionLost), stats.fraction_lost); |
302 EXPECT_EQ(std::string(kCodecInst.plname), stats.codec_name); | 286 EXPECT_EQ(std::string(kCodecInst.plname), stats.codec_name); |
(...skipping 25 matching lines...) Expand all Loading... |
328 EXPECT_EQ(kAudioDecodeStats.decoded_muted_output, | 312 EXPECT_EQ(kAudioDecodeStats.decoded_muted_output, |
329 stats.decoding_muted_output); | 313 stats.decoding_muted_output); |
330 EXPECT_EQ(kCallStats.capture_start_ntp_time_ms_, | 314 EXPECT_EQ(kCallStats.capture_start_ntp_time_ms_, |
331 stats.capture_start_ntp_time_ms); | 315 stats.capture_start_ntp_time_ms); |
332 } | 316 } |
333 | 317 |
334 TEST(AudioReceiveStreamTest, SetGain) { | 318 TEST(AudioReceiveStreamTest, SetGain) { |
335 ConfigHelper helper; | 319 ConfigHelper helper; |
336 internal::AudioReceiveStream recv_stream( | 320 internal::AudioReceiveStream recv_stream( |
337 helper.packet_router(), | 321 helper.packet_router(), |
338 helper.remote_bitrate_estimator(), | |
339 helper.config(), helper.audio_state(), helper.event_log()); | 322 helper.config(), helper.audio_state(), helper.event_log()); |
340 EXPECT_CALL(*helper.channel_proxy(), | 323 EXPECT_CALL(*helper.channel_proxy(), |
341 SetChannelOutputVolumeScaling(FloatEq(0.765f))); | 324 SetChannelOutputVolumeScaling(FloatEq(0.765f))); |
342 recv_stream.SetGain(0.765f); | 325 recv_stream.SetGain(0.765f); |
343 } | 326 } |
344 | 327 |
345 TEST(AudioReceiveStreamTest, StreamShouldNotBeAddedToMixerWhenVoEReturnsError) { | 328 TEST(AudioReceiveStreamTest, StreamShouldNotBeAddedToMixerWhenVoEReturnsError) { |
346 ConfigHelper helper; | 329 ConfigHelper helper; |
347 internal::AudioReceiveStream recv_stream( | 330 internal::AudioReceiveStream recv_stream( |
348 helper.packet_router(), | 331 helper.packet_router(), |
349 helper.remote_bitrate_estimator(), | |
350 helper.config(), helper.audio_state(), helper.event_log()); | 332 helper.config(), helper.audio_state(), helper.event_log()); |
351 | 333 |
352 EXPECT_CALL(helper.voice_engine(), StartPlayout(_)).WillOnce(Return(-1)); | 334 EXPECT_CALL(helper.voice_engine(), StartPlayout(_)).WillOnce(Return(-1)); |
353 EXPECT_CALL(*helper.audio_mixer(), AddSource(_)).Times(0); | 335 EXPECT_CALL(*helper.audio_mixer(), AddSource(_)).Times(0); |
354 | 336 |
355 recv_stream.Start(); | 337 recv_stream.Start(); |
356 } | 338 } |
357 | 339 |
358 TEST(AudioReceiveStreamTest, StreamShouldBeAddedToMixerOnStart) { | 340 TEST(AudioReceiveStreamTest, StreamShouldBeAddedToMixerOnStart) { |
359 ConfigHelper helper; | 341 ConfigHelper helper; |
360 internal::AudioReceiveStream recv_stream( | 342 internal::AudioReceiveStream recv_stream( |
361 helper.packet_router(), | 343 helper.packet_router(), |
362 helper.remote_bitrate_estimator(), | |
363 helper.config(), helper.audio_state(), helper.event_log()); | 344 helper.config(), helper.audio_state(), helper.event_log()); |
364 | 345 |
365 EXPECT_CALL(helper.voice_engine(), StartPlayout(_)).WillOnce(Return(0)); | 346 EXPECT_CALL(helper.voice_engine(), StartPlayout(_)).WillOnce(Return(0)); |
366 EXPECT_CALL(helper.voice_engine(), StopPlayout(_)); | 347 EXPECT_CALL(helper.voice_engine(), StopPlayout(_)); |
367 EXPECT_CALL(*helper.audio_mixer(), AddSource(&recv_stream)) | 348 EXPECT_CALL(*helper.audio_mixer(), AddSource(&recv_stream)) |
368 .WillOnce(Return(true)); | 349 .WillOnce(Return(true)); |
369 | 350 |
370 recv_stream.Start(); | 351 recv_stream.Start(); |
371 } | 352 } |
372 } // namespace test | 353 } // namespace test |
373 } // namespace webrtc | 354 } // namespace webrtc |
OLD | NEW |