blob: 641f03c3d6ca2faa96756f62cec2cafb753bab20 [file] [log] [blame]
pbos@webrtc.org994d0b72014-06-27 08:47:52 +00001/*
2 * Copyright (c) 2014 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 */
brandtr841de6a2016-11-15 07:10:52 -080010
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#include "test/call_test.h"
brandtr841de6a2016-11-15 07:10:52 -080012
13#include <algorithm>
14
Karl Wiberg918f50c2018-07-05 11:40:33 +020015#include "absl/memory/memory.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020016#include "api/audio_codecs/builtin_audio_decoder_factory.h"
17#include "api/audio_codecs/builtin_audio_encoder_factory.h"
Niels Möller0a8f4352018-05-18 11:37:23 +020018#include "api/video_codecs/video_encoder_config.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020019#include "call/rtp_transport_controller_send.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020020#include "modules/audio_mixer/audio_mixer_impl.h"
21#include "rtc_base/checks.h"
22#include "rtc_base/event.h"
Niels Möller4db138e2018-04-19 09:04:13 +020023#include "test/fake_encoder.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020024#include "test/testsupport/fileutils.h"
eladalon413ee9a2017-08-22 04:02:52 -070025
pbos@webrtc.org994d0b72014-06-27 08:47:52 +000026namespace webrtc {
27namespace test {
28
Guo-wei Shieh2c370782015-04-08 13:00:10 -070029namespace {
30const int kVideoRotationRtpExtensionId = 4;
31}
32
pbos@webrtc.org994d0b72014-06-27 08:47:52 +000033CallTest::CallTest()
pbos@webrtc.org2bb1bda2014-07-07 13:06:48 +000034 : clock_(Clock::GetRealTimeClock()),
Sebastian Jansson8e6602f2018-07-13 10:43:20 +020035 send_event_log_(RtcEventLog::CreateNull()),
36 recv_event_log_(RtcEventLog::CreateNull()),
sprangdb2a9fc2017-08-09 06:42:32 -070037 sender_call_transport_controller_(nullptr),
Stefan Holmer9fea80f2016-01-07 17:43:18 +010038 audio_send_config_(nullptr),
39 audio_send_stream_(nullptr),
Niels Möller4db138e2018-04-19 09:04:13 +020040 fake_encoder_factory_([this]() {
Karl Wiberg918f50c2018-07-05 11:40:33 +020041 auto encoder = absl::make_unique<test::FakeEncoder>(clock_);
Niels Möller4db138e2018-04-19 09:04:13 +020042 encoder->SetMaxBitrate(fake_encoder_max_bitrate_);
43 return encoder;
44 }),
Stefan Holmer04cb7632016-01-14 20:34:30 +010045 num_video_streams_(1),
Stefan Holmer9fea80f2016-01-07 17:43:18 +010046 num_audio_streams_(0),
brandtr841de6a2016-11-15 07:10:52 -080047 num_flexfec_streams_(0),
Niels Möller2784a032018-03-28 14:16:04 +020048 audio_decoder_factory_(CreateBuiltinAudioDecoderFactory()),
49 audio_encoder_factory_(CreateBuiltinAudioEncoderFactory()),
Sebastian Janssonf33905d2018-07-13 09:49:00 +020050 task_queue_("CallTestTaskQueue"),
51 video_send_config_(nullptr),
52 video_send_stream_(nullptr) {}
pbos@webrtc.org32452b22014-10-22 12:15:24 +000053
pbos@webrtc.org994d0b72014-06-27 08:47:52 +000054CallTest::~CallTest() {
eladalon413ee9a2017-08-22 04:02:52 -070055 task_queue_.SendTask([this]() {
Fredrik Solenberg2a877972017-12-15 16:42:15 +010056 fake_send_audio_device_ = nullptr;
57 fake_recv_audio_device_ = nullptr;
eladalon413ee9a2017-08-22 04:02:52 -070058 frame_generator_capturer_.reset();
59 });
pbos@webrtc.org994d0b72014-06-27 08:47:52 +000060}
61
stefane74eef12016-01-08 06:47:13 -080062void CallTest::RunBaseTest(BaseTest* test) {
eladalon413ee9a2017-08-22 04:02:52 -070063 task_queue_.SendTask([this, test]() {
64 num_video_streams_ = test->GetNumVideoStreams();
65 num_audio_streams_ = test->GetNumAudioStreams();
66 num_flexfec_streams_ = test->GetNumFlexfecStreams();
67 RTC_DCHECK(num_video_streams_ > 0 || num_audio_streams_ > 0);
68 Call::Config send_config(test->GetSenderCallConfig());
Stefan Holmer9fea80f2016-01-07 17:43:18 +010069 if (num_audio_streams_ > 0) {
eladalon413ee9a2017-08-22 04:02:52 -070070 CreateFakeAudioDevices(test->CreateCapturer(), test->CreateRenderer());
71 test->OnFakeAudioDevicesCreated(fake_send_audio_device_.get(),
72 fake_recv_audio_device_.get());
Ivo Creusen62337e52018-01-09 14:17:33 +010073 apm_send_ = AudioProcessingBuilder().Create();
74 apm_recv_ = AudioProcessingBuilder().Create();
Fredrik Solenberg8f5787a2018-01-11 13:52:30 +010075 EXPECT_EQ(0, fake_send_audio_device_->Init());
76 EXPECT_EQ(0, fake_recv_audio_device_->Init());
Stefan Holmer9fea80f2016-01-07 17:43:18 +010077 AudioState::Config audio_state_config;
aleloi10111bc2016-11-17 06:48:48 -080078 audio_state_config.audio_mixer = AudioMixerImpl::Create();
eladalon413ee9a2017-08-22 04:02:52 -070079 audio_state_config.audio_processing = apm_send_;
Fredrik Solenberg2a877972017-12-15 16:42:15 +010080 audio_state_config.audio_device_module = fake_send_audio_device_;
eladalon413ee9a2017-08-22 04:02:52 -070081 send_config.audio_state = AudioState::Create(audio_state_config);
Fredrik Solenbergd3195342017-11-21 20:33:05 +010082 fake_send_audio_device_->RegisterAudioCallback(
83 send_config.audio_state->audio_transport());
Stefan Holmer9fea80f2016-01-07 17:43:18 +010084 }
eladalon413ee9a2017-08-22 04:02:52 -070085 CreateSenderCall(send_config);
86 if (sender_call_transport_controller_ != nullptr) {
87 test->OnRtpTransportControllerSendCreated(
88 sender_call_transport_controller_);
89 }
90 if (test->ShouldCreateReceivers()) {
91 Call::Config recv_config(test->GetReceiverCallConfig());
92 if (num_audio_streams_ > 0) {
93 AudioState::Config audio_state_config;
eladalon413ee9a2017-08-22 04:02:52 -070094 audio_state_config.audio_mixer = AudioMixerImpl::Create();
95 audio_state_config.audio_processing = apm_recv_;
Fredrik Solenberg2a877972017-12-15 16:42:15 +010096 audio_state_config.audio_device_module = fake_recv_audio_device_;
eladalon413ee9a2017-08-22 04:02:52 -070097 recv_config.audio_state = AudioState::Create(audio_state_config);
Fredrik Solenbergd3195342017-11-21 20:33:05 +010098 fake_recv_audio_device_->RegisterAudioCallback(
Yves Gerey665174f2018-06-19 15:03:05 +020099 recv_config.audio_state->audio_transport());
100 }
eladalon413ee9a2017-08-22 04:02:52 -0700101 CreateReceiverCall(recv_config);
102 }
103 test->OnCallsCreated(sender_call_.get(), receiver_call_.get());
104 receive_transport_.reset(test->CreateReceiveTransport(&task_queue_));
105 send_transport_.reset(
106 test->CreateSendTransport(&task_queue_, sender_call_.get()));
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000107
eladalon413ee9a2017-08-22 04:02:52 -0700108 if (test->ShouldCreateReceivers()) {
109 send_transport_->SetReceiver(receiver_call_->Receiver());
110 receive_transport_->SetReceiver(sender_call_->Receiver());
111 if (num_video_streams_ > 0)
112 receiver_call_->SignalChannelNetworkState(MediaType::VIDEO, kNetworkUp);
113 if (num_audio_streams_ > 0)
114 receiver_call_->SignalChannelNetworkState(MediaType::AUDIO, kNetworkUp);
115 } else {
116 // Sender-only call delivers to itself.
117 send_transport_->SetReceiver(sender_call_->Receiver());
118 receive_transport_->SetReceiver(nullptr);
119 }
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000120
eladalon413ee9a2017-08-22 04:02:52 -0700121 CreateSendConfig(num_video_streams_, num_audio_streams_,
122 num_flexfec_streams_, send_transport_.get());
123 if (test->ShouldCreateReceivers()) {
124 CreateMatchingReceiveConfigs(receive_transport_.get());
125 }
126 if (num_video_streams_ > 0) {
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200127 test->ModifyVideoConfigs(GetVideoSendConfig(), &video_receive_configs_,
128 GetVideoEncoderConfig());
eladalon413ee9a2017-08-22 04:02:52 -0700129 }
130 if (num_audio_streams_ > 0) {
131 test->ModifyAudioConfigs(&audio_send_config_, &audio_receive_configs_);
132 }
133 if (num_flexfec_streams_ > 0) {
134 test->ModifyFlexfecConfigs(&flexfec_receive_configs_);
135 }
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100136
eladalon413ee9a2017-08-22 04:02:52 -0700137 if (num_flexfec_streams_ > 0) {
138 CreateFlexfecStreams();
139 test->OnFlexfecStreamsCreated(flexfec_receive_streams_);
140 }
141 if (num_video_streams_ > 0) {
142 CreateVideoStreams();
143 test->OnVideoStreamsCreated(video_send_stream_, video_receive_streams_);
144 }
145 if (num_audio_streams_ > 0) {
146 CreateAudioStreams();
147 test->OnAudioStreamsCreated(audio_send_stream_, audio_receive_streams_);
148 }
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000149
eladalon413ee9a2017-08-22 04:02:52 -0700150 if (num_video_streams_ > 0) {
151 int width = kDefaultWidth;
152 int height = kDefaultHeight;
153 int frame_rate = kDefaultFramerate;
154 test->ModifyVideoCaptureStartResolution(&width, &height, &frame_rate);
155 CreateFrameGeneratorCapturer(frame_rate, width, height);
156 test->OnFrameGeneratorCapturerCreated(frame_generator_capturer_.get());
157 }
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000158
eladalon413ee9a2017-08-22 04:02:52 -0700159 Start();
160 });
161
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000162 test->PerformTest();
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000163
Fredrik Solenberg73276ad2017-09-14 14:46:47 +0200164 task_queue_.SendTask([this, test]() {
eladalon413ee9a2017-08-22 04:02:52 -0700165 Stop();
Fredrik Solenberg73276ad2017-09-14 14:46:47 +0200166 test->OnStreamsStopped();
eladalon413ee9a2017-08-22 04:02:52 -0700167 DestroyStreams();
168 send_transport_.reset();
169 receive_transport_.reset();
170 DestroyCalls();
eladalon413ee9a2017-08-22 04:02:52 -0700171 });
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000172}
173
Sebastian Jansson8e6602f2018-07-13 10:43:20 +0200174void CallTest::CreateCalls() {
175 CreateCalls(Call::Config(send_event_log_.get()),
176 Call::Config(recv_event_log_.get()));
177}
178
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000179void CallTest::CreateCalls(const Call::Config& sender_config,
180 const Call::Config& receiver_config) {
181 CreateSenderCall(sender_config);
182 CreateReceiverCall(receiver_config);
183}
184
Sebastian Jansson8e6602f2018-07-13 10:43:20 +0200185void CallTest::CreateSenderCall() {
186 CreateSenderCall(Call::Config(send_event_log_.get()));
187}
188
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000189void CallTest::CreateSenderCall(const Call::Config& config) {
Sebastian Jansson9a03dd82018-02-22 10:31:14 +0100190 std::unique_ptr<RtpTransportControllerSend> controller_send =
Karl Wiberg918f50c2018-07-05 11:40:33 +0200191 absl::make_unique<RtpTransportControllerSend>(
Sebastian Janssondfce03a2018-05-18 18:05:10 +0200192 Clock::GetRealTimeClock(), config.event_log,
193 config.network_controller_factory, config.bitrate_config);
Sebastian Jansson9a03dd82018-02-22 10:31:14 +0100194 sender_call_transport_controller_ = controller_send.get();
195 sender_call_.reset(Call::Create(config, std::move(controller_send)));
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000196}
197
198void CallTest::CreateReceiverCall(const Call::Config& config) {
199 receiver_call_.reset(Call::Create(config));
200}
201
Fredrik Solenberg4f4ec0a2015-10-22 10:49:27 +0200202void CallTest::DestroyCalls() {
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100203 sender_call_.reset();
204 receiver_call_.reset();
Fredrik Solenberg4f4ec0a2015-10-22 10:49:27 +0200205}
206
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100207void CallTest::CreateVideoSendConfig(VideoSendStream::Config* video_config,
208 size_t num_video_streams,
209 size_t num_used_ssrcs,
210 Transport* send_transport) {
211 RTC_DCHECK_LE(num_video_streams + num_used_ssrcs, kNumSsrcs);
212 *video_config = VideoSendStream::Config(send_transport);
Niels Möller4db138e2018-04-19 09:04:13 +0200213 video_config->encoder_settings.encoder_factory = &fake_encoder_factory_;
Niels Möller259a4972018-04-05 15:36:51 +0200214 video_config->rtp.payload_name = "FAKE";
215 video_config->rtp.payload_type = kFakeVideoSendPayloadType;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100216 video_config->rtp.extensions.push_back(
217 RtpExtension(RtpExtension::kTransportSequenceNumberUri,
218 kTransportSequenceNumberExtensionId));
219 video_config->rtp.extensions.push_back(RtpExtension(
220 RtpExtension::kVideoContentTypeUri, kVideoContentTypeExtensionId));
Niels Möller259a4972018-04-05 15:36:51 +0200221 FillEncoderConfiguration(kVideoCodecGeneric, num_video_streams,
222 &video_encoder_config_);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100223
224 for (size_t i = 0; i < num_video_streams; ++i)
225 video_config->rtp.ssrcs.push_back(kVideoSendSsrcs[num_used_ssrcs + i]);
226 video_config->rtp.extensions.push_back(RtpExtension(
227 RtpExtension::kVideoRotationUri, kVideoRotationRtpExtensionId));
228}
229
230void CallTest::CreateAudioAndFecSendConfigs(size_t num_audio_streams,
231 size_t num_flexfec_streams,
232 Transport* send_transport) {
kwibergaf476c72016-11-28 15:21:39 -0800233 RTC_DCHECK_LE(num_audio_streams, 1);
234 RTC_DCHECK_LE(num_flexfec_streams, 1);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100235 if (num_audio_streams > 0) {
236 audio_send_config_ = AudioSendStream::Config(send_transport);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100237 audio_send_config_.rtp.ssrc = kAudioSendSsrc;
Oskar Sundbom3f6804d2017-11-16 10:54:58 +0100238 audio_send_config_.send_codec_spec = AudioSendStream::Config::SendCodecSpec(
239 kAudioSendPayloadType, {"opus", 48000, 2, {{"stereo", "1"}}});
Niels Möller2784a032018-03-28 14:16:04 +0200240 audio_send_config_.encoder_factory = audio_encoder_factory_;
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100241 }
brandtr841de6a2016-11-15 07:10:52 -0800242
243 // TODO(brandtr): Update this when we support multistream protection.
244 if (num_flexfec_streams > 0) {
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200245 GetVideoSendConfig()->rtp.flexfec.payload_type = kFlexfecPayloadType;
246 GetVideoSendConfig()->rtp.flexfec.ssrc = kFlexfecSendSsrc;
247 GetVideoSendConfig()->rtp.flexfec.protected_media_ssrcs = {
248 kVideoSendSsrcs[0]};
brandtr841de6a2016-11-15 07:10:52 -0800249 }
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000250}
251
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100252void CallTest::CreateSendConfig(size_t num_video_streams,
253 size_t num_audio_streams,
254 size_t num_flexfec_streams,
255 Transport* send_transport) {
256 if (num_video_streams > 0) {
257 CreateVideoSendConfig(&video_send_config_, num_video_streams, 0,
258 send_transport);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100259 }
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100260 CreateAudioAndFecSendConfigs(num_audio_streams, num_flexfec_streams,
261 send_transport);
262}
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100263
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100264std::vector<VideoReceiveStream::Config>
265CallTest::CreateMatchingVideoReceiveConfigs(
266 const VideoSendStream::Config& video_send_config,
267 Transport* rtcp_send_transport) {
268 std::vector<VideoReceiveStream::Config> result;
269 RTC_DCHECK(!video_send_config.rtp.ssrcs.empty());
270 VideoReceiveStream::Config video_config(rtcp_send_transport);
271 video_config.rtp.remb = false;
272 video_config.rtp.transport_cc = true;
273 video_config.rtp.local_ssrc = kReceiverLocalVideoSsrc;
274 for (const RtpExtension& extension : video_send_config.rtp.extensions)
275 video_config.rtp.extensions.push_back(extension);
276 video_config.renderer = &fake_renderer_;
277 for (size_t i = 0; i < video_send_config.rtp.ssrcs.size(); ++i) {
278 VideoReceiveStream::Decoder decoder =
Niels Möller259a4972018-04-05 15:36:51 +0200279 test::CreateMatchingDecoder(video_send_config);
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100280 allocated_decoders_.push_back(
281 std::unique_ptr<VideoDecoder>(decoder.decoder));
282 video_config.decoders.clear();
283 video_config.decoders.push_back(decoder);
284 video_config.rtp.remote_ssrc = video_send_config.rtp.ssrcs[i];
285 result.push_back(video_config.Copy());
286 }
287 result[0].rtp.protected_by_flexfec = (num_flexfec_streams_ == 1);
288 return result;
289}
290
291void CallTest::CreateMatchingAudioAndFecConfigs(
292 Transport* rtcp_send_transport) {
kwibergaf476c72016-11-28 15:21:39 -0800293 RTC_DCHECK_GE(1, num_audio_streams_);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100294 if (num_audio_streams_ == 1) {
295 AudioReceiveStream::Config audio_config;
296 audio_config.rtp.local_ssrc = kReceiverLocalAudioSsrc;
297 audio_config.rtcp_send_transport = rtcp_send_transport;
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100298 audio_config.rtp.remote_ssrc = audio_send_config_.rtp.ssrc;
Niels Möller2784a032018-03-28 14:16:04 +0200299 audio_config.decoder_factory = audio_decoder_factory_;
minyue20c84cc2017-04-10 16:57:57 -0700300 audio_config.decoder_map = {{kAudioSendPayloadType, {"opus", 48000, 2}}};
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100301 audio_receive_configs_.push_back(audio_config);
pbos@webrtc.orgbe9d2a42014-06-30 13:19:09 +0000302 }
brandtr841de6a2016-11-15 07:10:52 -0800303
304 // TODO(brandtr): Update this when we support multistream protection.
305 RTC_DCHECK(num_flexfec_streams_ <= 1);
306 if (num_flexfec_streams_ == 1) {
brandtr8313a6f2017-01-13 07:41:19 -0800307 FlexfecReceiveStream::Config config(rtcp_send_transport);
brandtr1cfbd602016-12-08 04:17:53 -0800308 config.payload_type = kFlexfecPayloadType;
309 config.remote_ssrc = kFlexfecSendSsrc;
310 config.protected_media_ssrcs = {kVideoSendSsrcs[0]};
brandtrfa5a3682017-01-17 01:33:54 -0800311 config.local_ssrc = kReceiverLocalVideoSsrc;
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200312 for (const RtpExtension& extension : GetVideoSendConfig()->rtp.extensions)
brandtrb29e6522016-12-21 06:37:18 -0800313 config.rtp_header_extensions.push_back(extension);
brandtr1cfbd602016-12-08 04:17:53 -0800314 flexfec_receive_configs_.push_back(config);
brandtr841de6a2016-11-15 07:10:52 -0800315 }
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000316}
317
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100318void CallTest::CreateMatchingReceiveConfigs(Transport* rtcp_send_transport) {
319 video_receive_configs_.clear();
320 allocated_decoders_.clear();
321 if (num_video_streams_ > 0) {
322 std::vector<VideoReceiveStream::Config> new_configs =
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200323 CreateMatchingVideoReceiveConfigs(*GetVideoSendConfig(),
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100324 rtcp_send_transport);
325 for (VideoReceiveStream::Config& config : new_configs) {
326 video_receive_configs_.push_back(config.Copy());
327 }
328 }
329 CreateMatchingAudioAndFecConfigs(rtcp_send_transport);
330}
331
danilchap9c6a0c72016-02-10 10:54:47 -0800332void CallTest::CreateFrameGeneratorCapturerWithDrift(Clock* clock,
perkjfa10b552016-10-02 23:45:26 -0700333 float speed,
334 int framerate,
335 int width,
336 int height) {
danilchap9c6a0c72016-02-10 10:54:47 -0800337 frame_generator_capturer_.reset(test::FrameGeneratorCapturer::Create(
Danil Chapovalov431abd92018-06-18 12:54:17 +0200338 width, height, absl::nullopt, absl::nullopt, framerate * speed, clock));
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200339 ConnectVideoSourcesToStreams();
danilchap9c6a0c72016-02-10 10:54:47 -0800340}
341
perkjfa10b552016-10-02 23:45:26 -0700342void CallTest::CreateFrameGeneratorCapturer(int framerate,
343 int width,
344 int height) {
Emircan Uysaler03e6ec92018-03-09 15:03:26 -0800345 frame_generator_capturer_.reset(test::FrameGeneratorCapturer::Create(
Danil Chapovalov431abd92018-06-18 12:54:17 +0200346 width, height, absl::nullopt, absl::nullopt, framerate, clock_));
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200347 ConnectVideoSourcesToStreams();
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000348}
pbosf1828e82015-07-28 08:20:59 -0700349
oprypin92220ff2017-03-23 03:40:03 -0700350void CallTest::CreateFakeAudioDevices(
Artem Titov3faa8322018-03-07 14:44:00 +0100351 std::unique_ptr<TestAudioDeviceModule::Capturer> capturer,
352 std::unique_ptr<TestAudioDeviceModule::Renderer> renderer) {
353 fake_send_audio_device_ = TestAudioDeviceModule::CreateTestAudioDeviceModule(
Fredrik Solenberg2a877972017-12-15 16:42:15 +0100354 std::move(capturer), nullptr, 1.f);
Artem Titov3faa8322018-03-07 14:44:00 +0100355 fake_recv_audio_device_ = TestAudioDeviceModule::CreateTestAudioDeviceModule(
Fredrik Solenberg2a877972017-12-15 16:42:15 +0100356 nullptr, std::move(renderer), 1.f);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100357}
358
359void CallTest::CreateVideoStreams() {
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100360 RTC_DCHECK(video_receive_streams_.empty());
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200361 CreateVideoSendStreams();
stefanff483612015-12-21 03:14:00 -0800362 for (size_t i = 0; i < video_receive_configs_.size(); ++i) {
Tommi733b5472016-06-10 17:58:01 +0200363 video_receive_streams_.push_back(receiver_call_->CreateVideoReceiveStream(
364 video_receive_configs_[i].Copy()));
pbos@webrtc.orgbe9d2a42014-06-30 13:19:09 +0000365 }
eladalonc0d481a2017-08-02 07:39:07 -0700366
367 AssociateFlexfecStreamsWithVideoStreams();
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000368}
369
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200370void CallTest::CreateVideoSendStreams() {
371 RTC_DCHECK(video_send_stream_ == nullptr);
372 video_send_stream_ = sender_call_->CreateVideoSendStream(
373 GetVideoSendConfig()->Copy(), GetVideoEncoderConfig()->Copy());
374}
375
376void CallTest::CreateVideoSendStream(const VideoEncoderConfig& encoder_config) {
377 RTC_DCHECK(video_send_stream_ == nullptr);
378 video_send_stream_ = sender_call_->CreateVideoSendStream(
379 GetVideoSendConfig()->Copy(), encoder_config.Copy());
380}
381
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100382void CallTest::CreateAudioStreams() {
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100383 RTC_DCHECK(audio_send_stream_ == nullptr);
384 RTC_DCHECK(audio_receive_streams_.empty());
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100385 audio_send_stream_ = sender_call_->CreateAudioSendStream(audio_send_config_);
386 for (size_t i = 0; i < audio_receive_configs_.size(); ++i) {
387 audio_receive_streams_.push_back(
388 receiver_call_->CreateAudioReceiveStream(audio_receive_configs_[i]));
389 }
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100390}
391
brandtr841de6a2016-11-15 07:10:52 -0800392void CallTest::CreateFlexfecStreams() {
393 for (size_t i = 0; i < flexfec_receive_configs_.size(); ++i) {
394 flexfec_receive_streams_.push_back(
395 receiver_call_->CreateFlexfecReceiveStream(
396 flexfec_receive_configs_[i]));
397 }
eladalonc0d481a2017-08-02 07:39:07 -0700398
399 AssociateFlexfecStreamsWithVideoStreams();
400}
401
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200402void CallTest::ConnectVideoSourcesToStreams() {
403 GetVideoSendStream()->SetSource(frame_generator_capturer_.get(),
404 degradation_preference_);
405}
406
eladalonc0d481a2017-08-02 07:39:07 -0700407void CallTest::AssociateFlexfecStreamsWithVideoStreams() {
408 // All FlexFEC streams protect all of the video streams.
409 for (FlexfecReceiveStream* flexfec_recv_stream : flexfec_receive_streams_) {
410 for (VideoReceiveStream* video_recv_stream : video_receive_streams_) {
411 video_recv_stream->AddSecondarySink(flexfec_recv_stream);
412 }
413 }
414}
415
416void CallTest::DissociateFlexfecStreamsFromVideoStreams() {
417 for (FlexfecReceiveStream* flexfec_recv_stream : flexfec_receive_streams_) {
418 for (VideoReceiveStream* video_recv_stream : video_receive_streams_) {
419 video_recv_stream->RemoveSecondarySink(flexfec_recv_stream);
420 }
421 }
brandtr841de6a2016-11-15 07:10:52 -0800422}
423
eladalon445f1a12017-08-09 01:52:36 -0700424void CallTest::Start() {
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200425 StartVideoStreams();
eladalon445f1a12017-08-09 01:52:36 -0700426 if (audio_send_stream_) {
427 audio_send_stream_->Start();
428 }
429 for (AudioReceiveStream* audio_recv_stream : audio_receive_streams_)
430 audio_recv_stream->Start();
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200431 StartVideoCapture();
432}
433
434void CallTest::StartVideoStreams() {
435 if (video_send_stream_)
436 video_send_stream_->Start();
437 for (VideoReceiveStream* video_recv_stream : video_receive_streams_)
438 video_recv_stream->Start();
439}
440
441void CallTest::StartVideoCapture() {
eladalon445f1a12017-08-09 01:52:36 -0700442 if (frame_generator_capturer_.get() != NULL)
443 frame_generator_capturer_->Start();
444}
445
446void CallTest::Stop() {
447 if (frame_generator_capturer_.get() != NULL)
448 frame_generator_capturer_->Stop();
eladalon445f1a12017-08-09 01:52:36 -0700449 for (AudioReceiveStream* audio_recv_stream : audio_receive_streams_)
450 audio_recv_stream->Stop();
451 if (audio_send_stream_) {
452 audio_send_stream_->Stop();
453 }
454 for (VideoReceiveStream* video_recv_stream : video_receive_streams_)
455 video_recv_stream->Stop();
456 if (video_send_stream_)
457 video_send_stream_->Stop();
458}
459
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000460void CallTest::DestroyStreams() {
eladalonc0d481a2017-08-02 07:39:07 -0700461 DissociateFlexfecStreamsFromVideoStreams();
462
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100463 if (audio_send_stream_)
464 sender_call_->DestroyAudioSendStream(audio_send_stream_);
465 audio_send_stream_ = nullptr;
466 for (AudioReceiveStream* audio_recv_stream : audio_receive_streams_)
467 receiver_call_->DestroyAudioReceiveStream(audio_recv_stream);
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100468
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200469 DestroyVideoSendStreams();
ilnika014cc52017-03-07 04:21:04 -0800470
brandtrfb45c6c2017-01-27 06:47:55 -0800471 for (VideoReceiveStream* video_recv_stream : video_receive_streams_)
472 receiver_call_->DestroyVideoReceiveStream(video_recv_stream);
473
brandtr841de6a2016-11-15 07:10:52 -0800474 for (FlexfecReceiveStream* flexfec_recv_stream : flexfec_receive_streams_)
475 receiver_call_->DestroyFlexfecReceiveStream(flexfec_recv_stream);
476
477 video_receive_streams_.clear();
pbos@webrtc.org776e6f22014-10-29 15:28:39 +0000478 allocated_decoders_.clear();
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000479}
480
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200481void CallTest::DestroyVideoSendStreams() {
482 if (video_send_stream_)
483 sender_call_->DestroyVideoSendStream(video_send_stream_);
484 video_send_stream_ = nullptr;
485}
486
eladalon445f1a12017-08-09 01:52:36 -0700487void CallTest::SetFakeVideoCaptureRotation(VideoRotation rotation) {
488 frame_generator_capturer_->SetFakeRotation(rotation);
489}
490
Sebastian Janssonf33905d2018-07-13 09:49:00 +0200491void CallTest::SetVideoDegradation(DegradationPreference preference) {
492 GetVideoSendStream()->SetSource(frame_generator_capturer_.get(), preference);
493}
494
495VideoSendStream::Config* CallTest::GetVideoSendConfig() {
496 return &video_send_config_;
497}
498
499void CallTest::SetVideoSendConfig(const VideoSendStream::Config& config) {
500 video_send_config_ = config.Copy();
501}
502
503VideoEncoderConfig* CallTest::GetVideoEncoderConfig() {
504 return &video_encoder_config_;
505}
506
507void CallTest::SetVideoEncoderConfig(const VideoEncoderConfig& config) {
508 video_encoder_config_ = config.Copy();
509}
510
511VideoSendStream* CallTest::GetVideoSendStream() {
512 return video_send_stream_;
513}
514
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100515constexpr size_t CallTest::kNumSsrcs;
perkjfa10b552016-10-02 23:45:26 -0700516const int CallTest::kDefaultWidth;
517const int CallTest::kDefaultHeight;
518const int CallTest::kDefaultFramerate;
Peter Boström5811a392015-12-10 13:02:50 +0100519const int CallTest::kDefaultTimeoutMs = 30 * 1000;
520const int CallTest::kLongTimeoutMs = 120 * 1000;
Ilya Nikolaevskiy255d1cd2017-12-21 18:02:59 +0100521const uint32_t CallTest::kSendRtxSsrcs[kNumSsrcs] = {
522 0xBADCAFD, 0xBADCAFE, 0xBADCAFF, 0xBADCB00, 0xBADCB01, 0xBADCB02};
523const uint32_t CallTest::kVideoSendSsrcs[kNumSsrcs] = {
524 0xC0FFED, 0xC0FFEE, 0xC0FFEF, 0xC0FFF0, 0xC0FFF1, 0xC0FFF2};
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100525const uint32_t CallTest::kAudioSendSsrc = 0xDEADBEEF;
brandtr841de6a2016-11-15 07:10:52 -0800526const uint32_t CallTest::kFlexfecSendSsrc = 0xBADBEEF;
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100527const uint32_t CallTest::kReceiverLocalVideoSsrc = 0x123456;
528const uint32_t CallTest::kReceiverLocalAudioSsrc = 0x1234567;
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000529const int CallTest::kNackRtpHistoryMs = 1000;
530
sprangd2702ef2017-07-10 08:41:10 -0700531const uint8_t CallTest::kDefaultKeepalivePayloadType =
532 RtpKeepAliveConfig().payload_type;
533
minyue20c84cc2017-04-10 16:57:57 -0700534const std::map<uint8_t, MediaType> CallTest::payload_type_map_ = {
535 {CallTest::kVideoSendPayloadType, MediaType::VIDEO},
536 {CallTest::kFakeVideoSendPayloadType, MediaType::VIDEO},
537 {CallTest::kSendRtxPayloadType, MediaType::VIDEO},
538 {CallTest::kRedPayloadType, MediaType::VIDEO},
539 {CallTest::kRtxRedPayloadType, MediaType::VIDEO},
540 {CallTest::kUlpfecPayloadType, MediaType::VIDEO},
541 {CallTest::kFlexfecPayloadType, MediaType::VIDEO},
sprangd2702ef2017-07-10 08:41:10 -0700542 {CallTest::kAudioSendPayloadType, MediaType::AUDIO},
543 {CallTest::kDefaultKeepalivePayloadType, MediaType::ANY}};
minyue20c84cc2017-04-10 16:57:57 -0700544
philipel4fb651d2017-04-10 03:54:05 -0700545BaseTest::BaseTest() : event_log_(RtcEventLog::CreateNull()) {}
philipele828c962017-03-21 03:24:27 -0700546
philipel4fb651d2017-04-10 03:54:05 -0700547BaseTest::BaseTest(unsigned int timeout_ms)
548 : RtpRtcpObserver(timeout_ms), event_log_(RtcEventLog::CreateNull()) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000549
Yves Gerey665174f2018-06-19 15:03:05 +0200550BaseTest::~BaseTest() {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000551
Artem Titov3faa8322018-03-07 14:44:00 +0100552std::unique_ptr<TestAudioDeviceModule::Capturer> BaseTest::CreateCapturer() {
553 return TestAudioDeviceModule::CreatePulsedNoiseCapturer(256, 48000);
oprypin92220ff2017-03-23 03:40:03 -0700554}
555
Artem Titov3faa8322018-03-07 14:44:00 +0100556std::unique_ptr<TestAudioDeviceModule::Renderer> BaseTest::CreateRenderer() {
557 return TestAudioDeviceModule::CreateDiscardRenderer(48000);
oprypin92220ff2017-03-23 03:40:03 -0700558}
559
Artem Titov3faa8322018-03-07 14:44:00 +0100560void BaseTest::OnFakeAudioDevicesCreated(
561 TestAudioDeviceModule* send_audio_device,
562 TestAudioDeviceModule* recv_audio_device) {}
oprypin92220ff2017-03-23 03:40:03 -0700563
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000564Call::Config BaseTest::GetSenderCallConfig() {
philipel4fb651d2017-04-10 03:54:05 -0700565 return Call::Config(event_log_.get());
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000566}
567
568Call::Config BaseTest::GetReceiverCallConfig() {
philipel4fb651d2017-04-10 03:54:05 -0700569 return Call::Config(event_log_.get());
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000570}
571
sprangdb2a9fc2017-08-09 06:42:32 -0700572void BaseTest::OnRtpTransportControllerSendCreated(
573 RtpTransportControllerSend* controller) {}
574
Yves Gerey665174f2018-06-19 15:03:05 +0200575void BaseTest::OnCallsCreated(Call* sender_call, Call* receiver_call) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000576
eladalon413ee9a2017-08-22 04:02:52 -0700577test::PacketTransport* BaseTest::CreateSendTransport(
578 SingleThreadedTaskQueueForTesting* task_queue,
579 Call* sender_call) {
580 return new PacketTransport(
581 task_queue, sender_call, this, test::PacketTransport::kSender,
582 CallTest::payload_type_map_, FakeNetworkPipe::Config());
stefane74eef12016-01-08 06:47:13 -0800583}
584
eladalon413ee9a2017-08-22 04:02:52 -0700585test::PacketTransport* BaseTest::CreateReceiveTransport(
586 SingleThreadedTaskQueueForTesting* task_queue) {
587 return new PacketTransport(
588 task_queue, nullptr, this, test::PacketTransport::kReceiver,
589 CallTest::payload_type_map_, FakeNetworkPipe::Config());
stefane74eef12016-01-08 06:47:13 -0800590}
stefanf116bd02015-10-27 08:29:42 -0700591
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100592size_t BaseTest::GetNumVideoStreams() const {
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000593 return 1;
594}
595
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100596size_t BaseTest::GetNumAudioStreams() const {
597 return 0;
598}
599
brandtr841de6a2016-11-15 07:10:52 -0800600size_t BaseTest::GetNumFlexfecStreams() const {
601 return 0;
602}
603
stefanff483612015-12-21 03:14:00 -0800604void BaseTest::ModifyVideoConfigs(
pbos@webrtc.orgbe9d2a42014-06-30 13:19:09 +0000605 VideoSendStream::Config* send_config,
606 std::vector<VideoReceiveStream::Config>* receive_configs,
stefanff483612015-12-21 03:14:00 -0800607 VideoEncoderConfig* encoder_config) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000608
perkjfa10b552016-10-02 23:45:26 -0700609void BaseTest::ModifyVideoCaptureStartResolution(int* width,
610 int* heigt,
611 int* frame_rate) {}
612
stefanff483612015-12-21 03:14:00 -0800613void BaseTest::OnVideoStreamsCreated(
pbos@webrtc.orgbe9d2a42014-06-30 13:19:09 +0000614 VideoSendStream* send_stream,
stefanff483612015-12-21 03:14:00 -0800615 const std::vector<VideoReceiveStream*>& receive_streams) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000616
Stefan Holmer9fea80f2016-01-07 17:43:18 +0100617void BaseTest::ModifyAudioConfigs(
618 AudioSendStream::Config* send_config,
619 std::vector<AudioReceiveStream::Config>* receive_configs) {}
620
621void BaseTest::OnAudioStreamsCreated(
622 AudioSendStream* send_stream,
623 const std::vector<AudioReceiveStream*>& receive_streams) {}
624
brandtr841de6a2016-11-15 07:10:52 -0800625void BaseTest::ModifyFlexfecConfigs(
626 std::vector<FlexfecReceiveStream::Config>* receive_configs) {}
627
628void BaseTest::OnFlexfecStreamsCreated(
629 const std::vector<FlexfecReceiveStream*>& receive_streams) {}
630
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000631void BaseTest::OnFrameGeneratorCapturerCreated(
Yves Gerey665174f2018-06-19 15:03:05 +0200632 FrameGeneratorCapturer* frame_generator_capturer) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000633
Yves Gerey665174f2018-06-19 15:03:05 +0200634void BaseTest::OnStreamsStopped() {}
oprypin92220ff2017-03-23 03:40:03 -0700635
Yves Gerey665174f2018-06-19 15:03:05 +0200636SendTest::SendTest(unsigned int timeout_ms) : BaseTest(timeout_ms) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000637
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000638bool SendTest::ShouldCreateReceivers() const {
639 return false;
640}
641
philipele828c962017-03-21 03:24:27 -0700642EndToEndTest::EndToEndTest() {}
643
Yves Gerey665174f2018-06-19 15:03:05 +0200644EndToEndTest::EndToEndTest(unsigned int timeout_ms) : BaseTest(timeout_ms) {}
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000645
pbos@webrtc.org994d0b72014-06-27 08:47:52 +0000646bool EndToEndTest::ShouldCreateReceivers() const {
647 return true;
648}
649
650} // namespace test
651} // namespace webrtc