Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(15)

Side by Side Diff: webrtc/audio/audio_receive_stream_unittest.cc

Issue 2669463006: Move RemoteBitrateEstimator::RemoveStream calls from receive streams to Call. (Closed)
Patch Set: Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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
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
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
OLDNEW
« no previous file with comments | « webrtc/audio/audio_receive_stream.cc ('k') | webrtc/call/call.cc » ('j') | webrtc/call/call.cc » ('J')

Powered by Google App Engine
This is Rietveld 408576698