blob: f84aaaf1c9309d89e67efed83faf185fab2a5058 [file] [log] [blame]
solenbergc7a8b082015-10-16 14:35:07 -07001/*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Fredrik Solenbergea073732015-12-01 11:26:34 +010011#include <string>
ossu20a4b3f2017-04-27 02:08:52 -070012#include <utility>
Fredrik Solenbergea073732015-12-01 11:26:34 +010013#include <vector>
14
Sebastian Jansson5f83cf02018-05-08 14:52:22 +020015#include "api/units/time_delta.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020016#include "audio/audio_send_stream.h"
17#include "audio/audio_state.h"
18#include "audio/conversion.h"
Fredrik Solenberga8b7c7f2018-01-17 11:18:31 +010019#include "audio/mock_voe_channel_proxy.h"
Sebastian Janssonef9daee2018-02-22 14:49:02 +010020#include "call/test/mock_rtp_transport_controller_send.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020021#include "logging/rtc_event_log/mock/mock_rtc_event_log.h"
Fredrik Solenberg2a877972017-12-15 16:42:15 +010022#include "modules/audio_device/include/mock_audio_device.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020023#include "modules/audio_mixer/audio_mixer_impl.h"
Ivo Creusen56d46092017-11-24 17:29:59 +010024#include "modules/audio_processing/include/audio_processing_statistics.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020025#include "modules/audio_processing/include/mock_audio_processing.h"
Sebastian Janssonef9daee2018-02-22 14:49:02 +010026#include "modules/rtp_rtcp/mocks/mock_rtcp_bandwidth_observer.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020027#include "modules/rtp_rtcp/mocks/mock_rtcp_rtt_stats.h"
28#include "modules/rtp_rtcp/mocks/mock_rtp_rtcp.h"
Sam Zackrisson06953ba2018-02-01 16:53:16 +010029#include "rtc_base/fakeclock.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020030#include "rtc_base/ptr_util.h"
31#include "rtc_base/task_queue.h"
32#include "test/gtest.h"
33#include "test/mock_audio_encoder.h"
34#include "test/mock_audio_encoder_factory.h"
Sam Zackrisson06953ba2018-02-01 16:53:16 +010035#include "test/mock_transport.h"
solenbergc7a8b082015-10-16 14:35:07 -070036
37namespace webrtc {
solenberg85a04962015-10-27 03:35:21 -070038namespace test {
Fredrik Solenberg0ccae132015-11-03 10:15:49 +010039namespace {
40
solenberg3a941542015-11-16 07:34:50 -080041using testing::_;
stefan7de8d642017-02-07 07:14:08 -080042using testing::Eq;
43using testing::Ne;
ossu20a4b3f2017-04-27 02:08:52 -070044using testing::Invoke;
solenberg3a941542015-11-16 07:34:50 -080045using testing::Return;
minyue-webrtc8de18262017-07-26 14:18:40 +020046using testing::StrEq;
solenberg3a941542015-11-16 07:34:50 -080047
Fredrik Solenberg0ccae132015-11-03 10:15:49 +010048const uint32_t kSsrc = 1234;
solenberg3a941542015-11-16 07:34:50 -080049const char* kCName = "foo_name";
50const int kAudioLevelId = 2;
Stefan Holmerb86d4e42015-12-07 10:26:18 +010051const int kTransportSequenceNumberId = 4;
Ivo Creusen56d46092017-11-24 17:29:59 +010052const int32_t kEchoDelayMedian = 254;
53const int32_t kEchoDelayStdDev = -3;
54const double kDivergentFilterFraction = 0.2f;
55const double kEchoReturnLoss = -65;
56const double kEchoReturnLossEnhancement = 101;
57const double kResidualEchoLikelihood = -1.0f;
58const double kResidualEchoLikelihoodMax = 23.0f;
Yves Gerey665174f2018-06-19 15:03:05 +020059const CallStatistics kCallStats = {1345, 1678, 1901, 1234, 112,
60 13456, 17890, 1567, -1890, -1123};
solenberg566ef242015-11-06 15:34:49 -080061const ReportBlock kReportBlock = {456, 780, 123, 567, 890, 132, 143, 13354};
Fredrik Solenbergb5727682015-12-04 15:22:19 +010062const int kTelephoneEventPayloadType = 123;
solenbergffbbcac2016-11-17 05:25:37 -080063const int kTelephoneEventPayloadFrequency = 65432;
solenberg8842c3e2016-03-11 03:06:41 -080064const int kTelephoneEventCode = 45;
65const int kTelephoneEventDuration = 6789;
minyue7a973442016-10-20 03:27:12 -070066const CodecInst kIsacCodec = {103, "isac", 16000, 320, 1, 32000};
ossu20a4b3f2017-04-27 02:08:52 -070067constexpr int kIsacPayloadType = 103;
68const SdpAudioFormat kIsacFormat = {"isac", 16000, 1};
69const SdpAudioFormat kOpusFormat = {"opus", 48000, 2};
70const SdpAudioFormat kG722Format = {"g722", 8000, 1};
71const AudioCodecSpec kCodecSpecs[] = {
72 {kIsacFormat, {16000, 1, 32000, 10000, 32000}},
73 {kOpusFormat, {48000, 1, 32000, 6000, 510000}},
74 {kG722Format, {16000, 1, 64000}}};
solenberg566ef242015-11-06 15:34:49 -080075
mflodman86cc6ff2016-07-26 04:44:06 -070076class MockLimitObserver : public BitrateAllocator::LimitObserver {
77 public:
Sebastian Janssonfe617a32018-03-21 12:45:20 +010078 MOCK_METHOD4(OnAllocationLimitsChanged,
mflodman86cc6ff2016-07-26 04:44:06 -070079 void(uint32_t min_send_bitrate_bps,
philipelf69e7682018-02-28 13:06:28 +010080 uint32_t max_padding_bitrate_bps,
Sebastian Janssonfe617a32018-03-21 12:45:20 +010081 uint32_t total_bitrate_bps,
82 bool has_packet_feedback));
mflodman86cc6ff2016-07-26 04:44:06 -070083};
84
ossu20a4b3f2017-04-27 02:08:52 -070085std::unique_ptr<MockAudioEncoder> SetupAudioEncoderMock(
86 int payload_type,
87 const SdpAudioFormat& format) {
88 for (const auto& spec : kCodecSpecs) {
89 if (format == spec.format) {
Sebastian Jansson41f16be2018-02-22 11:09:56 +010090 std::unique_ptr<MockAudioEncoder> encoder(
91 new testing::NiceMock<MockAudioEncoder>());
ossu20a4b3f2017-04-27 02:08:52 -070092 ON_CALL(*encoder.get(), SampleRateHz())
93 .WillByDefault(Return(spec.info.sample_rate_hz));
94 ON_CALL(*encoder.get(), NumChannels())
95 .WillByDefault(Return(spec.info.num_channels));
96 ON_CALL(*encoder.get(), RtpTimestampRateHz())
97 .WillByDefault(Return(spec.format.clockrate_hz));
98 return encoder;
99 }
100 }
101 return nullptr;
102}
103
104rtc::scoped_refptr<MockAudioEncoderFactory> SetupEncoderFactoryMock() {
105 rtc::scoped_refptr<MockAudioEncoderFactory> factory =
106 new rtc::RefCountedObject<MockAudioEncoderFactory>();
107 ON_CALL(*factory.get(), GetSupportedEncoders())
108 .WillByDefault(Return(std::vector<AudioCodecSpec>(
109 std::begin(kCodecSpecs), std::end(kCodecSpecs))));
110 ON_CALL(*factory.get(), QueryAudioEncoder(_))
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100111 .WillByDefault(Invoke(
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200112 [](const SdpAudioFormat& format) -> absl::optional<AudioCodecInfo> {
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100113 for (const auto& spec : kCodecSpecs) {
114 if (format == spec.format) {
115 return spec.info;
116 }
117 }
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200118 return absl::nullopt;
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100119 }));
Karl Wibergd6fbf2a2018-02-27 13:37:31 +0100120 ON_CALL(*factory.get(), MakeAudioEncoderMock(_, _, _, _))
ossu20a4b3f2017-04-27 02:08:52 -0700121 .WillByDefault(Invoke([](int payload_type, const SdpAudioFormat& format,
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200122 absl::optional<AudioCodecPairId> codec_pair_id,
ossu20a4b3f2017-04-27 02:08:52 -0700123 std::unique_ptr<AudioEncoder>* return_value) {
124 *return_value = SetupAudioEncoderMock(payload_type, format);
125 }));
126 return factory;
127}
128
solenberg566ef242015-11-06 15:34:49 -0800129struct ConfigHelper {
ossu20a4b3f2017-04-27 02:08:52 -0700130 ConfigHelper(bool audio_bwe_enabled, bool expect_set_encoder_call)
nisseb8f9a322017-03-27 05:36:15 -0700131 : stream_config_(nullptr),
peaha9cc40b2017-06-29 08:32:09 -0700132 audio_processing_(new rtc::RefCountedObject<MockAudioProcessing>()),
perkj26091b12016-09-01 01:17:40 -0700133 bitrate_allocator_(&limit_observer_),
minyue-webrtc8de18262017-07-26 14:18:40 +0200134 worker_queue_("ConfigHelper_worker_queue"),
135 audio_encoder_(nullptr) {
solenberg13725082015-11-25 08:16:52 -0800136 using testing::Invoke;
solenberg3a941542015-11-16 07:34:50 -0800137
solenberg566ef242015-11-06 15:34:49 -0800138 AudioState::Config config;
aleloi10111bc2016-11-17 06:48:48 -0800139 config.audio_mixer = AudioMixerImpl::Create();
peaha9cc40b2017-06-29 08:32:09 -0700140 config.audio_processing = audio_processing_;
Fredrik Solenberg2a877972017-12-15 16:42:15 +0100141 config.audio_device_module =
142 new rtc::RefCountedObject<MockAudioDeviceModule>();
solenberg566ef242015-11-06 15:34:49 -0800143 audio_state_ = AudioState::Create(config);
solenberg3a941542015-11-16 07:34:50 -0800144
stefan7de8d642017-02-07 07:14:08 -0800145 SetupDefaultChannelProxy(audio_bwe_enabled);
ossu20a4b3f2017-04-27 02:08:52 -0700146 SetupMockForSetupSendCodec(expect_set_encoder_call);
minyue6b825df2016-10-31 04:08:32 -0700147
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100148 // Use ISAC as default codec so as to prevent unnecessary |channel_proxy_|
ossu20a4b3f2017-04-27 02:08:52 -0700149 // calls from the default ctor behavior.
150 stream_config_.send_codec_spec =
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100151 AudioSendStream::Config::SendCodecSpec(kIsacPayloadType, kIsacFormat);
solenberg566ef242015-11-06 15:34:49 -0800152 stream_config_.rtp.ssrc = kSsrc;
solenberg971cab02016-06-14 10:02:41 -0700153 stream_config_.rtp.nack.rtp_history_ms = 200;
solenberg3a941542015-11-16 07:34:50 -0800154 stream_config_.rtp.c_name = kCName;
155 stream_config_.rtp.extensions.push_back(
isheriff6f8d6862016-05-26 11:24:55 -0700156 RtpExtension(RtpExtension::kAudioLevelUri, kAudioLevelId));
stefan7de8d642017-02-07 07:14:08 -0800157 if (audio_bwe_enabled) {
ossu1129df22017-06-30 01:38:56 -0700158 AddBweToConfig(&stream_config_);
stefan7de8d642017-02-07 07:14:08 -0800159 }
ossu20a4b3f2017-04-27 02:08:52 -0700160 stream_config_.encoder_factory = SetupEncoderFactoryMock();
minyue78b4d562016-11-30 04:47:39 -0800161 stream_config_.min_bitrate_bps = 10000;
162 stream_config_.max_bitrate_bps = 65000;
solenberg566ef242015-11-06 15:34:49 -0800163 }
164
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100165 std::unique_ptr<internal::AudioSendStream> CreateAudioSendStream() {
166 return std::unique_ptr<internal::AudioSendStream>(
167 new internal::AudioSendStream(
Sebastian Jansson1896cec2018-02-20 09:06:11 +0100168 stream_config_, audio_state_, &worker_queue_, &rtp_transport_,
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200169 &bitrate_allocator_, &event_log_, &rtcp_rtt_stats_, absl::nullopt,
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100170 &active_lifetime_,
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100171 std::unique_ptr<voe::ChannelProxy>(channel_proxy_)));
172 }
173
solenberg566ef242015-11-06 15:34:49 -0800174 AudioSendStream::Config& config() { return stream_config_; }
ossu20a4b3f2017-04-27 02:08:52 -0700175 MockAudioEncoderFactory& mock_encoder_factory() {
176 return *static_cast<MockAudioEncoderFactory*>(
177 stream_config_.encoder_factory.get());
178 }
solenberg94218532016-06-16 10:53:22 -0700179 MockVoEChannelProxy* channel_proxy() { return channel_proxy_; }
Sebastian Jansson1896cec2018-02-20 09:06:11 +0100180 RtpTransportControllerSendInterface* transport() { return &rtp_transport_; }
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100181 TimeInterval* active_lifetime() { return &active_lifetime_; }
minyue7a973442016-10-20 03:27:12 -0700182
ossu1129df22017-06-30 01:38:56 -0700183 static void AddBweToConfig(AudioSendStream::Config* config) {
Yves Gerey665174f2018-06-19 15:03:05 +0200184 config->rtp.extensions.push_back(RtpExtension(
185 RtpExtension::kTransportSequenceNumberUri, kTransportSequenceNumberId));
ossu1129df22017-06-30 01:38:56 -0700186 config->send_codec_spec->transport_cc_enabled = true;
187 }
188
stefan7de8d642017-02-07 07:14:08 -0800189 void SetupDefaultChannelProxy(bool audio_bwe_enabled) {
Sebastian Jansson41f16be2018-02-22 11:09:56 +0100190 EXPECT_TRUE(channel_proxy_ == nullptr);
minyue6b825df2016-10-31 04:08:32 -0700191 channel_proxy_ = new testing::StrictMock<MockVoEChannelProxy>();
ossu3b9ff382017-04-27 08:03:42 -0700192 EXPECT_CALL(*channel_proxy_, GetRtpRtcp(_, _))
193 .WillRepeatedly(Invoke(
194 [this](RtpRtcp** rtp_rtcp_module, RtpReceiver** rtp_receiver) {
195 *rtp_rtcp_module = &this->rtp_rtcp_;
196 *rtp_receiver = nullptr; // Not deemed necessary for tests yet.
197 }));
minyue6b825df2016-10-31 04:08:32 -0700198 EXPECT_CALL(*channel_proxy_, SetRTCPStatus(true)).Times(1);
199 EXPECT_CALL(*channel_proxy_, SetLocalSSRC(kSsrc)).Times(1);
200 EXPECT_CALL(*channel_proxy_, SetRTCP_CNAME(StrEq(kCName))).Times(1);
201 EXPECT_CALL(*channel_proxy_, SetNACKStatus(true, 10)).Times(1);
202 EXPECT_CALL(*channel_proxy_,
minyue6b825df2016-10-31 04:08:32 -0700203 SetSendAudioLevelIndicationStatus(true, kAudioLevelId))
204 .Times(1);
Sebastian Janssonef9daee2018-02-22 14:49:02 +0100205 EXPECT_CALL(rtp_transport_, GetBandwidthObserver())
206 .WillRepeatedly(Return(&bandwidth_observer_));
stefan7de8d642017-02-07 07:14:08 -0800207 if (audio_bwe_enabled) {
208 EXPECT_CALL(*channel_proxy_,
209 EnableSendTransportSequenceNumber(kTransportSequenceNumberId))
210 .Times(1);
Sebastian Janssonef9daee2018-02-22 14:49:02 +0100211 EXPECT_CALL(*channel_proxy_,
212 RegisterSenderCongestionControlObjects(
213 &rtp_transport_, Eq(&bandwidth_observer_)))
stefan7de8d642017-02-07 07:14:08 -0800214 .Times(1);
215 } else {
nisse559af382017-03-21 06:41:12 -0700216 EXPECT_CALL(*channel_proxy_, RegisterSenderCongestionControlObjects(
Sebastian Jansson1896cec2018-02-20 09:06:11 +0100217 &rtp_transport_, Eq(nullptr)))
stefan7de8d642017-02-07 07:14:08 -0800218 .Times(1);
219 }
nissefdbfdc92017-03-31 05:44:52 -0700220 EXPECT_CALL(*channel_proxy_, ResetSenderCongestionControlObjects())
221 .Times(1);
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100222 {
223 ::testing::InSequence unregister_on_destruction;
224 EXPECT_CALL(*channel_proxy_, RegisterTransport(_)).Times(1);
225 EXPECT_CALL(*channel_proxy_, RegisterTransport(nullptr)).Times(1);
226 }
minyue6b825df2016-10-31 04:08:32 -0700227 EXPECT_CALL(*channel_proxy_, SetRtcEventLog(testing::NotNull())).Times(1);
228 EXPECT_CALL(*channel_proxy_, SetRtcEventLog(testing::IsNull()))
229 .Times(1); // Destructor resets the event log
230 }
231
ossu20a4b3f2017-04-27 02:08:52 -0700232 void SetupMockForSetupSendCodec(bool expect_set_encoder_call) {
233 if (expect_set_encoder_call) {
234 EXPECT_CALL(*channel_proxy_, SetEncoderForMock(_, _))
minyue-webrtc8de18262017-07-26 14:18:40 +0200235 .WillOnce(Invoke(
236 [this](int payload_type, std::unique_ptr<AudioEncoder>* encoder) {
237 this->audio_encoder_ = std::move(*encoder);
238 return true;
239 }));
ossu20a4b3f2017-04-27 02:08:52 -0700240 }
minyue7a973442016-10-20 03:27:12 -0700241 }
ossu20a4b3f2017-04-27 02:08:52 -0700242
minyue-webrtc8de18262017-07-26 14:18:40 +0200243 void SetupMockForModifyEncoder() {
244 // Let ModifyEncoder to invoke mock audio encoder.
245 EXPECT_CALL(*channel_proxy_, ModifyEncoder(_))
246 .WillRepeatedly(Invoke(
247 [this](rtc::FunctionView<void(std::unique_ptr<AudioEncoder>*)>
248 modifier) {
249 if (this->audio_encoder_)
250 modifier(&this->audio_encoder_);
251 }));
252 }
253
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100254 void SetupMockForSendTelephoneEvent() {
255 EXPECT_TRUE(channel_proxy_);
Yves Gerey665174f2018-06-19 15:03:05 +0200256 EXPECT_CALL(*channel_proxy_, SetSendTelephoneEventPayloadType(
257 kTelephoneEventPayloadType,
258 kTelephoneEventPayloadFrequency))
259 .WillOnce(Return(true));
260 EXPECT_CALL(
261 *channel_proxy_,
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100262 SendTelephoneEventOutband(kTelephoneEventCode, kTelephoneEventDuration))
Yves Gerey665174f2018-06-19 15:03:05 +0200263 .WillOnce(Return(true));
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100264 }
265
solenberg566ef242015-11-06 15:34:49 -0800266 void SetupMockForGetStats() {
solenberg3a941542015-11-16 07:34:50 -0800267 using testing::DoAll;
solenbergbd9a77f2017-02-06 12:53:57 -0800268 using testing::SetArgPointee;
solenberg3a941542015-11-16 07:34:50 -0800269 using testing::SetArgReferee;
270
solenberg566ef242015-11-06 15:34:49 -0800271 std::vector<ReportBlock> report_blocks;
272 webrtc::ReportBlock block = kReportBlock;
273 report_blocks.push_back(block); // Has wrong SSRC.
274 block.source_SSRC = kSsrc;
275 report_blocks.push_back(block); // Correct block.
276 block.fraction_lost = 0;
277 report_blocks.push_back(block); // Duplicate SSRC, bad fraction_lost.
278
solenberg358057b2015-11-27 10:46:42 -0800279 EXPECT_TRUE(channel_proxy_);
280 EXPECT_CALL(*channel_proxy_, GetRTCPStatistics())
281 .WillRepeatedly(Return(kCallStats));
282 EXPECT_CALL(*channel_proxy_, GetRemoteRTCPReportBlocks())
283 .WillRepeatedly(Return(report_blocks));
ivoce1198e02017-09-08 08:13:19 -0700284 EXPECT_CALL(*channel_proxy_, GetANAStatistics())
285 .WillRepeatedly(Return(ANAStats()));
solenberg796b8f92017-03-01 17:02:23 -0800286
Ivo Creusen56d46092017-11-24 17:29:59 +0100287 audio_processing_stats_.echo_return_loss = kEchoReturnLoss;
288 audio_processing_stats_.echo_return_loss_enhancement =
289 kEchoReturnLossEnhancement;
290 audio_processing_stats_.delay_median_ms = kEchoDelayMedian;
291 audio_processing_stats_.delay_standard_deviation_ms = kEchoDelayStdDev;
292 audio_processing_stats_.divergent_filter_fraction =
293 kDivergentFilterFraction;
294 audio_processing_stats_.residual_echo_likelihood = kResidualEchoLikelihood;
295 audio_processing_stats_.residual_echo_likelihood_recent_max =
296 kResidualEchoLikelihoodMax;
ivoc7aba0292016-11-14 04:52:06 -0800297
Ivo Creusen56d46092017-11-24 17:29:59 +0100298 EXPECT_CALL(*audio_processing_, GetStatistics(true))
ivoc7aba0292016-11-14 04:52:06 -0800299 .WillRepeatedly(Return(audio_processing_stats_));
solenberg566ef242015-11-06 15:34:49 -0800300 }
301
302 private:
solenberg566ef242015-11-06 15:34:49 -0800303 rtc::scoped_refptr<AudioState> audio_state_;
304 AudioSendStream::Config stream_config_;
solenberg13725082015-11-25 08:16:52 -0800305 testing::StrictMock<MockVoEChannelProxy>* channel_proxy_ = nullptr;
peaha9cc40b2017-06-29 08:32:09 -0700306 rtc::scoped_refptr<MockAudioProcessing> audio_processing_;
Ivo Creusen56d46092017-11-24 17:29:59 +0100307 AudioProcessingStats audio_processing_stats_;
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100308 TimeInterval active_lifetime_;
Sebastian Janssonef9daee2018-02-22 14:49:02 +0100309 testing::StrictMock<MockRtcpBandwidthObserver> bandwidth_observer_;
Sebastian Jansson41f16be2018-02-22 11:09:56 +0100310 testing::NiceMock<MockRtcEventLog> event_log_;
Sebastian Janssonef9daee2018-02-22 14:49:02 +0100311 testing::NiceMock<MockRtpTransportControllerSend> rtp_transport_;
Sebastian Jansson41f16be2018-02-22 11:09:56 +0100312 testing::NiceMock<MockRtpRtcp> rtp_rtcp_;
michaelt9332b7d2016-11-30 07:51:13 -0800313 MockRtcpRttStats rtcp_rtt_stats_;
mflodman86cc6ff2016-07-26 04:44:06 -0700314 testing::NiceMock<MockLimitObserver> limit_observer_;
315 BitrateAllocator bitrate_allocator_;
perkj26091b12016-09-01 01:17:40 -0700316 // |worker_queue| is defined last to ensure all pending tasks are cancelled
317 // and deleted before any other members.
318 rtc::TaskQueue worker_queue_;
minyue-webrtc8de18262017-07-26 14:18:40 +0200319 std::unique_ptr<AudioEncoder> audio_encoder_;
solenberg566ef242015-11-06 15:34:49 -0800320};
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100321} // namespace
solenbergc7a8b082015-10-16 14:35:07 -0700322
323TEST(AudioSendStreamTest, ConfigToString) {
solenbergc7a8b082015-10-16 14:35:07 -0700324 AudioSendStream::Config config(nullptr);
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100325 config.rtp.ssrc = kSsrc;
solenberg3a941542015-11-16 07:34:50 -0800326 config.rtp.c_name = kCName;
minyue10cbb462016-11-07 09:29:22 -0800327 config.min_bitrate_bps = 12000;
328 config.max_bitrate_bps = 34000;
ossu20a4b3f2017-04-27 02:08:52 -0700329 config.send_codec_spec =
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100330 AudioSendStream::Config::SendCodecSpec(kIsacPayloadType, kIsacFormat);
ossu20a4b3f2017-04-27 02:08:52 -0700331 config.send_codec_spec->nack_enabled = true;
332 config.send_codec_spec->transport_cc_enabled = false;
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100333 config.send_codec_spec->cng_payload_type = 42;
ossu20a4b3f2017-04-27 02:08:52 -0700334 config.encoder_factory = MockAudioEncoderFactory::CreateUnusedFactory();
stefanb521aa72016-11-01 03:17:16 -0700335 config.rtp.extensions.push_back(
336 RtpExtension(RtpExtension::kAudioLevelUri, kAudioLevelId));
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100337 EXPECT_EQ(
isheriff6f8d6862016-05-26 11:24:55 -0700338 "{rtp: {ssrc: 1234, extensions: [{uri: "
stefanb521aa72016-11-01 03:17:16 -0700339 "urn:ietf:params:rtp-hdrext:ssrc-audio-level, id: 2}], nack: "
deadbeef922246a2017-02-26 04:18:12 -0800340 "{rtp_history_ms: 0}, c_name: foo_name}, send_transport: null, "
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100341 "min_bitrate_bps: 12000, max_bitrate_bps: 34000, "
solenberg940b6d62016-10-25 11:19:07 -0700342 "send_codec_spec: {nack_enabled: true, transport_cc_enabled: false, "
ossu20a4b3f2017-04-27 02:08:52 -0700343 "cng_payload_type: 42, payload_type: 103, "
344 "format: {name: isac, clockrate_hz: 16000, num_channels: 1, "
345 "parameters: {}}}}",
solenberg85a04962015-10-27 03:35:21 -0700346 config.ToString());
solenbergc7a8b082015-10-16 14:35:07 -0700347}
348
349TEST(AudioSendStreamTest, ConstructDestruct) {
ossu20a4b3f2017-04-27 02:08:52 -0700350 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100351 auto send_stream = helper.CreateAudioSendStream();
solenbergc7a8b082015-10-16 14:35:07 -0700352}
solenberg85a04962015-10-27 03:35:21 -0700353
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100354TEST(AudioSendStreamTest, SendTelephoneEvent) {
ossu20a4b3f2017-04-27 02:08:52 -0700355 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100356 auto send_stream = helper.CreateAudioSendStream();
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100357 helper.SetupMockForSendTelephoneEvent();
Yves Gerey665174f2018-06-19 15:03:05 +0200358 EXPECT_TRUE(send_stream->SendTelephoneEvent(
359 kTelephoneEventPayloadType, kTelephoneEventPayloadFrequency,
360 kTelephoneEventCode, kTelephoneEventDuration));
Fredrik Solenbergb5727682015-12-04 15:22:19 +0100361}
362
solenberg94218532016-06-16 10:53:22 -0700363TEST(AudioSendStreamTest, SetMuted) {
ossu20a4b3f2017-04-27 02:08:52 -0700364 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100365 auto send_stream = helper.CreateAudioSendStream();
solenberg94218532016-06-16 10:53:22 -0700366 EXPECT_CALL(*helper.channel_proxy(), SetInputMute(true));
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100367 send_stream->SetMuted(true);
solenberg94218532016-06-16 10:53:22 -0700368}
369
stefan7de8d642017-02-07 07:14:08 -0800370TEST(AudioSendStreamTest, AudioBweCorrectObjectsOnChannelProxy) {
ossu20a4b3f2017-04-27 02:08:52 -0700371 ConfigHelper helper(true, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100372 auto send_stream = helper.CreateAudioSendStream();
stefan7de8d642017-02-07 07:14:08 -0800373}
374
375TEST(AudioSendStreamTest, NoAudioBweCorrectObjectsOnChannelProxy) {
ossu20a4b3f2017-04-27 02:08:52 -0700376 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100377 auto send_stream = helper.CreateAudioSendStream();
stefan7de8d642017-02-07 07:14:08 -0800378}
379
solenberg85a04962015-10-27 03:35:21 -0700380TEST(AudioSendStreamTest, GetStats) {
ossu20a4b3f2017-04-27 02:08:52 -0700381 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100382 auto send_stream = helper.CreateAudioSendStream();
solenberg566ef242015-11-06 15:34:49 -0800383 helper.SetupMockForGetStats();
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100384 AudioSendStream::Stats stats = send_stream->GetStats(true);
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100385 EXPECT_EQ(kSsrc, stats.local_ssrc);
386 EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesSent), stats.bytes_sent);
387 EXPECT_EQ(kCallStats.packetsSent, stats.packets_sent);
388 EXPECT_EQ(static_cast<int32_t>(kReportBlock.cumulative_num_packets_lost),
solenberg85a04962015-10-27 03:35:21 -0700389 stats.packets_lost);
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100390 EXPECT_EQ(Q8ToFloat(kReportBlock.fraction_lost), stats.fraction_lost);
minyue7a973442016-10-20 03:27:12 -0700391 EXPECT_EQ(std::string(kIsacCodec.plname), stats.codec_name);
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100392 EXPECT_EQ(static_cast<int32_t>(kReportBlock.extended_highest_sequence_number),
solenberg85a04962015-10-27 03:35:21 -0700393 stats.ext_seqnum);
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100394 EXPECT_EQ(static_cast<int32_t>(kReportBlock.interarrival_jitter /
minyue7a973442016-10-20 03:27:12 -0700395 (kIsacCodec.plfreq / 1000)),
Fredrik Solenberg0ccae132015-11-03 10:15:49 +0100396 stats.jitter_ms);
397 EXPECT_EQ(kCallStats.rttMs, stats.rtt_ms);
Fredrik Solenberg2a877972017-12-15 16:42:15 +0100398 EXPECT_EQ(0, stats.audio_level);
399 EXPECT_EQ(0, stats.total_input_energy);
400 EXPECT_EQ(0, stats.total_input_duration);
Ivo Creusen56d46092017-11-24 17:29:59 +0100401 EXPECT_EQ(kEchoDelayMedian, stats.apm_statistics.delay_median_ms);
402 EXPECT_EQ(kEchoDelayStdDev, stats.apm_statistics.delay_standard_deviation_ms);
403 EXPECT_EQ(kEchoReturnLoss, stats.apm_statistics.echo_return_loss);
404 EXPECT_EQ(kEchoReturnLossEnhancement,
405 stats.apm_statistics.echo_return_loss_enhancement);
406 EXPECT_EQ(kDivergentFilterFraction,
407 stats.apm_statistics.divergent_filter_fraction);
408 EXPECT_EQ(kResidualEchoLikelihood,
409 stats.apm_statistics.residual_echo_likelihood);
410 EXPECT_EQ(kResidualEchoLikelihoodMax,
411 stats.apm_statistics.residual_echo_likelihood_recent_max);
Fredrik Solenberg2a877972017-12-15 16:42:15 +0100412 EXPECT_FALSE(stats.typing_noise_detected);
solenberg566ef242015-11-06 15:34:49 -0800413}
minyue7a973442016-10-20 03:27:12 -0700414
minyue-webrtc8de18262017-07-26 14:18:40 +0200415TEST(AudioSendStreamTest, SendCodecAppliesAudioNetworkAdaptor) {
ossu20a4b3f2017-04-27 02:08:52 -0700416 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100417 helper.config().send_codec_spec =
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100418 AudioSendStream::Config::SendCodecSpec(0, kOpusFormat);
minyue-webrtc8de18262017-07-26 14:18:40 +0200419 const std::string kAnaConfigString = "abcde";
420 const std::string kAnaReconfigString = "12345";
421
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100422 helper.config().audio_network_adaptor_config = kAnaConfigString;
ossu20a4b3f2017-04-27 02:08:52 -0700423
Karl Wibergd6fbf2a2018-02-27 13:37:31 +0100424 EXPECT_CALL(helper.mock_encoder_factory(), MakeAudioEncoderMock(_, _, _, _))
minyue-webrtc8de18262017-07-26 14:18:40 +0200425 .WillOnce(Invoke([&kAnaConfigString, &kAnaReconfigString](
426 int payload_type, const SdpAudioFormat& format,
Danil Chapovalovb9b146c2018-06-15 12:28:07 +0200427 absl::optional<AudioCodecPairId> codec_pair_id,
minyue-webrtc8de18262017-07-26 14:18:40 +0200428 std::unique_ptr<AudioEncoder>* return_value) {
ossu20a4b3f2017-04-27 02:08:52 -0700429 auto mock_encoder = SetupAudioEncoderMock(payload_type, format);
minyue-webrtc8de18262017-07-26 14:18:40 +0200430 EXPECT_CALL(*mock_encoder,
431 EnableAudioNetworkAdaptor(StrEq(kAnaConfigString), _))
432 .WillOnce(Return(true));
433 EXPECT_CALL(*mock_encoder,
434 EnableAudioNetworkAdaptor(StrEq(kAnaReconfigString), _))
ossu20a4b3f2017-04-27 02:08:52 -0700435 .WillOnce(Return(true));
436 *return_value = std::move(mock_encoder);
437 }));
438
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100439 auto send_stream = helper.CreateAudioSendStream();
minyue-webrtc8de18262017-07-26 14:18:40 +0200440
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100441 auto stream_config = helper.config();
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100442 stream_config.audio_network_adaptor_config = kAnaReconfigString;
minyue-webrtc8de18262017-07-26 14:18:40 +0200443
444 helper.SetupMockForModifyEncoder();
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100445 send_stream->Reconfigure(stream_config);
minyue7a973442016-10-20 03:27:12 -0700446}
447
448// VAD is applied when codec is mono and the CNG frequency matches the codec
ossu20a4b3f2017-04-27 02:08:52 -0700449// clock rate.
minyue7a973442016-10-20 03:27:12 -0700450TEST(AudioSendStreamTest, SendCodecCanApplyVad) {
ossu20a4b3f2017-04-27 02:08:52 -0700451 ConfigHelper helper(false, false);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100452 helper.config().send_codec_spec =
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100453 AudioSendStream::Config::SendCodecSpec(9, kG722Format);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100454 helper.config().send_codec_spec->cng_payload_type = 105;
ossu20a4b3f2017-04-27 02:08:52 -0700455 using ::testing::Invoke;
456 std::unique_ptr<AudioEncoder> stolen_encoder;
457 EXPECT_CALL(*helper.channel_proxy(), SetEncoderForMock(_, _))
458 .WillOnce(
459 Invoke([&stolen_encoder](int payload_type,
460 std::unique_ptr<AudioEncoder>* encoder) {
461 stolen_encoder = std::move(*encoder);
462 return true;
463 }));
464
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100465 auto send_stream = helper.CreateAudioSendStream();
ossu20a4b3f2017-04-27 02:08:52 -0700466
467 // We cannot truly determine if the encoder created is an AudioEncoderCng. It
468 // is the only reasonable implementation that will return something from
469 // ReclaimContainedEncoders, though.
470 ASSERT_TRUE(stolen_encoder);
471 EXPECT_FALSE(stolen_encoder->ReclaimContainedEncoders().empty());
minyue7a973442016-10-20 03:27:12 -0700472}
473
minyue78b4d562016-11-30 04:47:39 -0800474TEST(AudioSendStreamTest, DoesNotPassHigherBitrateThanMaxBitrate) {
ossu20a4b3f2017-04-27 02:08:52 -0700475 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100476 auto send_stream = helper.CreateAudioSendStream();
minyue78b4d562016-11-30 04:47:39 -0800477 EXPECT_CALL(*helper.channel_proxy(),
478 SetBitrate(helper.config().max_bitrate_bps, _));
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100479 send_stream->OnBitrateUpdated(helper.config().max_bitrate_bps + 5000, 0.0, 50,
480 6000);
minyue78b4d562016-11-30 04:47:39 -0800481}
482
483TEST(AudioSendStreamTest, ProbingIntervalOnBitrateUpdated) {
ossu20a4b3f2017-04-27 02:08:52 -0700484 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100485 auto send_stream = helper.CreateAudioSendStream();
minyue78b4d562016-11-30 04:47:39 -0800486 EXPECT_CALL(*helper.channel_proxy(), SetBitrate(_, 5000));
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100487 send_stream->OnBitrateUpdated(50000, 0.0, 50, 5000);
minyue78b4d562016-11-30 04:47:39 -0800488}
489
ossu20a4b3f2017-04-27 02:08:52 -0700490// Test that AudioSendStream doesn't recreate the encoder unnecessarily.
491TEST(AudioSendStreamTest, DontRecreateEncoder) {
492 ConfigHelper helper(false, false);
493 // WillOnce is (currently) the default used by ConfigHelper if asked to set an
494 // expectation for SetEncoder. Since this behavior is essential for this test
495 // to be correct, it's instead set-up manually here. Otherwise a simple change
496 // to ConfigHelper (say to WillRepeatedly) would silently make this test
497 // useless.
498 EXPECT_CALL(*helper.channel_proxy(), SetEncoderForMock(_, _))
499 .WillOnce(Return(true));
500
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100501 helper.config().send_codec_spec =
Oskar Sundbom2707fb22017-11-16 10:57:35 +0100502 AudioSendStream::Config::SendCodecSpec(9, kG722Format);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100503 helper.config().send_codec_spec->cng_payload_type = 105;
504 auto send_stream = helper.CreateAudioSendStream();
505 send_stream->Reconfigure(helper.config());
ossu20a4b3f2017-04-27 02:08:52 -0700506}
507
ossu1129df22017-06-30 01:38:56 -0700508TEST(AudioSendStreamTest, ReconfigureTransportCcResetsFirst) {
509 ConfigHelper helper(false, true);
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100510 auto send_stream = helper.CreateAudioSendStream();
ossu1129df22017-06-30 01:38:56 -0700511 auto new_config = helper.config();
512 ConfigHelper::AddBweToConfig(&new_config);
513 EXPECT_CALL(*helper.channel_proxy(),
514 EnableSendTransportSequenceNumber(kTransportSequenceNumberId))
515 .Times(1);
516 {
517 ::testing::InSequence seq;
518 EXPECT_CALL(*helper.channel_proxy(), ResetSenderCongestionControlObjects())
519 .Times(1);
520 EXPECT_CALL(*helper.channel_proxy(), RegisterSenderCongestionControlObjects(
Yves Gerey665174f2018-06-19 15:03:05 +0200521 helper.transport(), Ne(nullptr)))
ossu1129df22017-06-30 01:38:56 -0700522 .Times(1);
523 }
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +0100524 send_stream->Reconfigure(new_config);
ossu1129df22017-06-30 01:38:56 -0700525}
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100526
527// Checks that AudioSendStream logs the times at which RTP packets are sent
528// through its interface.
529TEST(AudioSendStreamTest, UpdateLifetime) {
530 ConfigHelper helper(false, true);
531
532 MockTransport mock_transport;
533 helper.config().send_transport = &mock_transport;
534
535 Transport* registered_transport;
536 ON_CALL(*helper.channel_proxy(), RegisterTransport(_))
537 .WillByDefault(Invoke([&registered_transport](Transport* transport) {
538 registered_transport = transport;
539 }));
540
541 rtc::ScopedFakeClock fake_clock;
542 constexpr int64_t kTimeBetweenSendRtpCallsMs = 100;
543 {
544 auto send_stream = helper.CreateAudioSendStream();
545 EXPECT_CALL(mock_transport, SendRtp(_, _, _)).Times(2);
546 const PacketOptions options;
547 registered_transport->SendRtp(nullptr, 0, options);
Sebastian Jansson5f83cf02018-05-08 14:52:22 +0200548 fake_clock.AdvanceTime(TimeDelta::ms(kTimeBetweenSendRtpCallsMs));
Sam Zackrisson06953ba2018-02-01 16:53:16 +0100549 registered_transport->SendRtp(nullptr, 0, options);
550 }
551 EXPECT_TRUE(!helper.active_lifetime()->Empty());
552 EXPECT_EQ(helper.active_lifetime()->Length(), kTimeBetweenSendRtpCallsMs);
553}
solenberg85a04962015-10-27 03:35:21 -0700554} // namespace test
solenbergc7a8b082015-10-16 14:35:07 -0700555} // namespace webrtc