blob: 388179587c577042b99efeb9e5dbe76bf4ea51dc [file] [log] [blame]
jlmiller@webrtc.org5f93d0a2015-01-20 21:36:13 +00001/*
2 * libjingle
3 * Copyright 2014 Google Inc.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions are met:
7 *
8 * 1. Redistributions of source code must retain the above copyright notice,
9 * this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright notice,
11 * this list of conditions and the following disclaimer in the documentation
12 * and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
17 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
19 * EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
henrike@webrtc.org28e20752013-07-10 00:45:36 +000027
wu@webrtc.orgcadf9042013-08-30 21:24:16 +000028#ifndef TALK_MEDIA_BASE_VIDEOENGINE_UNITTEST_H_ // NOLINT
henrike@webrtc.org28e20752013-07-10 00:45:36 +000029#define TALK_MEDIA_BASE_VIDEOENGINE_UNITTEST_H_
30
31#include <string>
32#include <vector>
33
henrike@webrtc.org28e20752013-07-10 00:45:36 +000034#include "talk/media/base/fakenetworkinterface.h"
35#include "talk/media/base/fakevideocapturer.h"
36#include "talk/media/base/fakevideorenderer.h"
37#include "talk/media/base/mediachannel.h"
38#include "talk/media/base/streamparams.h"
Fredrik Solenberg709ed672015-09-15 12:26:33 +020039#include "talk/media/webrtc/fakewebrtccall.h"
buildbot@webrtc.orga09a9992014-08-13 17:26:08 +000040#include "webrtc/base/bytebuffer.h"
41#include "webrtc/base/gunit.h"
42#include "webrtc/base/timeutils.h"
Fredrik Solenberg709ed672015-09-15 12:26:33 +020043#include "webrtc/call.h"
henrike@webrtc.org28e20752013-07-10 00:45:36 +000044
henrike@webrtc.org28e20752013-07-10 00:45:36 +000045#define EXPECT_FRAME_WAIT(c, w, h, t) \
46 EXPECT_EQ_WAIT((c), renderer_.num_rendered_frames(), (t)); \
47 EXPECT_EQ((w), renderer_.width()); \
48 EXPECT_EQ((h), renderer_.height()); \
49 EXPECT_EQ(0, renderer_.errors()); \
50
51#define EXPECT_FRAME_ON_RENDERER_WAIT(r, c, w, h, t) \
52 EXPECT_EQ_WAIT((c), (r).num_rendered_frames(), (t)); \
53 EXPECT_EQ((w), (r).width()); \
54 EXPECT_EQ((h), (r).height()); \
55 EXPECT_EQ(0, (r).errors()); \
56
wu@webrtc.org9caf2762013-12-11 18:25:07 +000057#define EXPECT_GT_FRAME_ON_RENDERER_WAIT(r, c, w, h, t) \
58 EXPECT_TRUE_WAIT((r).num_rendered_frames() >= (c) && \
59 (w) == (r).width() && \
60 (h) == (r).height(), (t)); \
61 EXPECT_EQ(0, (r).errors()); \
62
henrike@webrtc.org28e20752013-07-10 00:45:36 +000063static const uint32 kTimeout = 5000U;
buildbot@webrtc.org99f63082014-07-18 23:31:30 +000064static const uint32 kDefaultReceiveSsrc = 0;
henrike@webrtc.org28e20752013-07-10 00:45:36 +000065static const uint32 kSsrc = 1234u;
66static const uint32 kRtxSsrc = 4321u;
67static const uint32 kSsrcs4[] = {1, 2, 3, 4};
68
69inline bool IsEqualRes(const cricket::VideoCodec& a, int w, int h, int fps) {
70 return a.width == w && a.height == h && a.framerate == fps;
71}
72
73inline bool IsEqualCodec(const cricket::VideoCodec& a,
74 const cricket::VideoCodec& b) {
75 return a.id == b.id && a.name == b.name &&
76 IsEqualRes(a, b.width, b.height, b.framerate);
77}
78
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +000079namespace std {
henrike@webrtc.org28e20752013-07-10 00:45:36 +000080inline std::ostream& operator<<(std::ostream& s, const cricket::VideoCodec& c) {
81 s << "{" << c.name << "(" << c.id << "), "
82 << c.width << "x" << c.height << "x" << c.framerate << "}";
83 return s;
84}
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +000085} // namespace std
henrike@webrtc.org28e20752013-07-10 00:45:36 +000086
87inline int TimeBetweenSend(const cricket::VideoCodec& codec) {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +000088 return static_cast<int>(
henrike@webrtc.org28e20752013-07-10 00:45:36 +000089 cricket::VideoFormat::FpsToInterval(codec.framerate) /
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +000090 rtc::kNumNanosecsPerMillisec);
henrike@webrtc.org28e20752013-07-10 00:45:36 +000091}
92
93// Fake video engine that makes it possible to test enabling and disabling
94// capturer (checking that the engine state is updated and that the capturer
95// is indeed capturing) without having to create a channel. It also makes it
96// possible to test that the media processors are indeed being called when
97// registered.
98template<class T>
99class VideoEngineOverride : public T {
100 public:
Fredrik Solenberg709ed672015-09-15 12:26:33 +0200101 VideoEngineOverride() : T() {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000102 }
103 virtual ~VideoEngineOverride() {
104 }
105 bool is_camera_on() const { return T::GetVideoCapturer()->IsRunning(); }
106 void set_has_senders(bool has_senders) {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000107 cricket::VideoCapturer* video_capturer = T::GetVideoCapturer();
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000108 if (has_senders) {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000109 video_capturer->SignalVideoFrame.connect(this,
110 &VideoEngineOverride<T>::OnLocalFrame);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000111 } else {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000112 video_capturer->SignalVideoFrame.disconnect(this);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000113 }
114 }
115 void OnLocalFrame(cricket::VideoCapturer*,
116 const cricket::VideoFrame*) {
117 }
118 void OnLocalFrameFormat(cricket::VideoCapturer*,
119 const cricket::VideoFormat*) {
120 }
121
122 void TriggerMediaFrame(
123 uint32 ssrc, cricket::VideoFrame* frame, bool* drop_frame) {
124 T::SignalMediaFrame(ssrc, frame, drop_frame);
125 }
126};
127
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000128template<class E>
129class VideoEngineTest : public testing::Test {
130 protected:
131 // Tests starting and stopping the engine, and creating a channel.
132 void StartupShutdown() {
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000133 EXPECT_TRUE(engine_.Init(rtc::Thread::Current()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000134 cricket::VideoMediaChannel* channel = engine_.CreateChannel(NULL);
135 EXPECT_TRUE(channel != NULL);
136 delete channel;
137 engine_.Terminate();
138 }
139
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000140 void ConstrainNewCodecBody() {
141 cricket::VideoCodec empty, in, out;
142 cricket::VideoCodec max_settings(engine_.codecs()[0].id,
143 engine_.codecs()[0].name,
144 1280, 800, 30, 0);
145
buildbot@webrtc.orgb92f6f92014-07-14 18:22:37 +0000146 // set max settings of 1280x800x30
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000147 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
148 cricket::VideoEncoderConfig(max_settings)));
149
150 // don't constrain the max resolution
151 in = max_settings;
152 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
153 EXPECT_PRED2(IsEqualCodec, out, in);
154
155 // constrain resolution greater than the max and wider aspect,
156 // picking best aspect (16:10)
157 in.width = 1380;
158 in.height = 800;
159 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
160 EXPECT_PRED4(IsEqualRes, out, 1280, 720, 30);
161
162 // constrain resolution greater than the max and narrow aspect,
163 // picking best aspect (16:9)
164 in.width = 1280;
165 in.height = 740;
166 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
167 EXPECT_PRED4(IsEqualRes, out, 1280, 720, 30);
168
169 // constrain resolution greater than the max, picking equal aspect (4:3)
170 in.width = 1280;
171 in.height = 960;
172 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
173 EXPECT_PRED4(IsEqualRes, out, 1280, 800, 30);
174
175 // constrain resolution greater than the max, picking equal aspect (16:10)
176 in.width = 1280;
177 in.height = 800;
178 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
179 EXPECT_PRED4(IsEqualRes, out, 1280, 800, 30);
180
181 // reduce max settings to 640x480x30
182 max_settings.width = 640;
183 max_settings.height = 480;
184 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
185 cricket::VideoEncoderConfig(max_settings)));
186
187 // don't constrain the max resolution
188 in = max_settings;
189 in.width = 640;
190 in.height = 480;
191 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
192 EXPECT_PRED2(IsEqualCodec, out, in);
193
194 // keep 16:10 if they request it
195 in.height = 400;
196 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
197 EXPECT_PRED2(IsEqualCodec, out, in);
198
199 // don't constrain lesser 4:3 resolutions
200 in.width = 320;
201 in.height = 240;
202 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
203 EXPECT_PRED2(IsEqualCodec, out, in);
204
205 // don't constrain lesser 16:10 resolutions
206 in.width = 320;
207 in.height = 200;
208 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
209 EXPECT_PRED2(IsEqualCodec, out, in);
210
211 // requested resolution of 0x0 succeeds
212 in.width = 0;
213 in.height = 0;
214 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
215 EXPECT_PRED2(IsEqualCodec, out, in);
216
217 // constrain resolution lesser than the max and wider aspect,
218 // picking best aspect (16:9)
219 in.width = 350;
220 in.height = 201;
221 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
222 EXPECT_PRED4(IsEqualRes, out, 320, 180, 30);
223
224 // constrain resolution greater than the max and narrow aspect,
225 // picking best aspect (4:3)
226 in.width = 350;
227 in.height = 300;
228 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
229 EXPECT_PRED4(IsEqualRes, out, 320, 240, 30);
230
231 // constrain resolution greater than the max and wider aspect,
232 // picking best aspect (16:9)
233 in.width = 1380;
234 in.height = 800;
235 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
236 EXPECT_PRED4(IsEqualRes, out, 640, 360, 30);
237
238 // constrain resolution greater than the max and narrow aspect,
239 // picking best aspect (4:3)
240 in.width = 1280;
241 in.height = 900;
242 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
243 EXPECT_PRED4(IsEqualRes, out, 640, 480, 30);
244
245 // constrain resolution greater than the max, picking equal aspect (4:3)
246 in.width = 1280;
247 in.height = 960;
248 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
249 EXPECT_PRED4(IsEqualRes, out, 640, 480, 30);
250
251 // constrain resolution greater than the max, picking equal aspect (16:10)
252 in.width = 1280;
253 in.height = 800;
254 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
255 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
256
257 // constrain res & fps greater than the max
258 in.framerate = 50;
259 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
260 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
261
262 // reduce max settings to 160x100x10
263 max_settings.width = 160;
264 max_settings.height = 100;
265 max_settings.framerate = 10;
266 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
267 cricket::VideoEncoderConfig(max_settings)));
268
269 // constrain res & fps to new max
270 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
271 EXPECT_PRED4(IsEqualRes, out, 160, 100, 10);
272
273 // allow 4:3 "comparable" resolutions
274 in.width = 160;
275 in.height = 120;
276 in.framerate = 10;
277 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
278 EXPECT_PRED4(IsEqualRes, out, 160, 120, 10);
279 }
280
buildbot@webrtc.orgb92f6f92014-07-14 18:22:37 +0000281 // This is the new way of constraining codec size, where we no longer maintain
282 // a list of the supported formats. Instead, CanSendCodec will just downscale
283 // the resolution by 2 until the width is below clamp.
284 void ConstrainNewCodec2Body() {
285 cricket::VideoCodec empty, in, out;
286 cricket::VideoCodec max_settings(engine_.codecs()[0].id,
287 engine_.codecs()[0].name,
288 1280, 800, 30, 0);
289
290 // Set max settings of 1280x800x30
291 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
292 cricket::VideoEncoderConfig(max_settings)));
293
294 // Don't constrain the max resolution
295 in = max_settings;
296 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
297 EXPECT_PRED2(IsEqualCodec, out, in);
298
299 // Constrain resolution greater than the max width.
300 in.width = 1380;
301 in.height = 800;
302 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
303 EXPECT_PRED4(IsEqualRes, out, 690, 400, 30);
304
305 // Don't constrain resolution when only the height is greater than max.
306 in.width = 960;
307 in.height = 1280;
308 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
309 EXPECT_PRED4(IsEqualRes, out, 960, 1280, 30);
310
311 // Don't constrain smaller format.
312 in.width = 640;
313 in.height = 480;
314 EXPECT_TRUE(engine_.CanSendCodec(in, empty, &out));
315 EXPECT_PRED4(IsEqualRes, out, 640, 480, 30);
316 }
317
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000318 void ConstrainRunningCodecBody() {
319 cricket::VideoCodec in, out, current;
320 cricket::VideoCodec max_settings(engine_.codecs()[0].id,
321 engine_.codecs()[0].name,
322 1280, 800, 30, 0);
323
324 // set max settings of 1280x960x30
325 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
326 cricket::VideoEncoderConfig(max_settings)));
327
328 // establish current call at 1280x800x30 (16:10)
329 current = max_settings;
330 current.height = 800;
331
332 // Don't constrain current resolution
333 in = current;
334 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
335 EXPECT_PRED2(IsEqualCodec, out, in);
336
337 // requested resolution of 0x0 succeeds
338 in.width = 0;
339 in.height = 0;
340 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
341 EXPECT_PRED2(IsEqualCodec, out, in);
342
343 // Reduce an intermediate resolution down to the next lowest one, preserving
344 // aspect ratio.
345 in.width = 800;
346 in.height = 600;
347 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
348 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
349
350 // Clamping by aspect ratio, but still never return a dimension higher than
351 // requested.
352 in.width = 1280;
353 in.height = 720;
354 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
355 EXPECT_PRED4(IsEqualRes, out, 1280, 720, 30);
356
357 in.width = 1279;
358 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
359 EXPECT_PRED4(IsEqualRes, out, 960, 600, 30);
360
361 in.width = 1281;
362 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
363 EXPECT_PRED4(IsEqualRes, out, 1280, 720, 30);
364
365 // Clamp large resolutions down, always preserving aspect
366 in.width = 1920;
367 in.height = 1080;
368 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
369 EXPECT_PRED4(IsEqualRes, out, 1280, 800, 30);
370
371 in.width = 1921;
372 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
373 EXPECT_PRED4(IsEqualRes, out, 1280, 800, 30);
374
375 in.width = 1919;
376 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
377 EXPECT_PRED4(IsEqualRes, out, 1280, 800, 30);
378
379 // reduce max settings to 640x480x30
380 max_settings.width = 640;
381 max_settings.height = 480;
382 EXPECT_TRUE(engine_.SetDefaultEncoderConfig(
383 cricket::VideoEncoderConfig(max_settings)));
384
385 // establish current call at 640x400x30 (16:10)
386 current = max_settings;
387 current.height = 400;
388
389 // Don't constrain current resolution
390 in = current;
391 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
392 EXPECT_PRED2(IsEqualCodec, out, in);
393
394 // requested resolution of 0x0 succeeds
395 in.width = 0;
396 in.height = 0;
397 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
398 EXPECT_PRED2(IsEqualCodec, out, in);
399
400 // Reduce an intermediate resolution down to the next lowest one, preserving
401 // aspect ratio.
402 in.width = 400;
403 in.height = 300;
404 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
405 EXPECT_PRED4(IsEqualRes, out, 320, 200, 30);
406
407 // Clamping by aspect ratio, but still never return a dimension higher than
408 // requested.
409 in.width = 640;
410 in.height = 360;
411 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
412 EXPECT_PRED4(IsEqualRes, out, 640, 360, 30);
413
414 in.width = 639;
415 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
416 EXPECT_PRED4(IsEqualRes, out, 480, 300, 30);
417
418 in.width = 641;
419 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
420 EXPECT_PRED4(IsEqualRes, out, 640, 360, 30);
421
422 // Clamp large resolutions down, always preserving aspect
423 in.width = 1280;
424 in.height = 800;
425 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
426 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
427
428 in.width = 1281;
429 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
430 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
431
432 in.width = 1279;
433 EXPECT_TRUE(engine_.CanSendCodec(in, current, &out));
434 EXPECT_PRED4(IsEqualRes, out, 640, 400, 30);
435
436 // Should fail for any that are smaller than our supported formats
437 in.width = 80;
438 in.height = 80;
439 EXPECT_FALSE(engine_.CanSendCodec(in, current, &out));
440
441 in.height = 50;
442 EXPECT_FALSE(engine_.CanSendCodec(in, current, &out));
443 }
444
445 VideoEngineOverride<E> engine_;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000446 rtc::scoped_ptr<cricket::FakeVideoCapturer> video_capturer_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000447};
448
449template<class E, class C>
450class VideoMediaChannelTest : public testing::Test,
451 public sigslot::has_slots<> {
452 protected:
Fredrik Solenberg709ed672015-09-15 12:26:33 +0200453 VideoMediaChannelTest<E, C>()
454 : call_(webrtc::Call::Create(webrtc::Call::Config())) {}
455
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000456 virtual cricket::VideoCodec DefaultCodec() = 0;
457
458 virtual cricket::StreamParams DefaultSendStreamParams() {
459 return cricket::StreamParams::CreateLegacy(kSsrc);
460 }
461
462 virtual void SetUp() {
463 cricket::Device device("test", "device");
Fredrik Solenberg9a416bd2015-05-22 09:04:09 +0200464 engine_.Init();
Fredrik Solenberg709ed672015-09-15 12:26:33 +0200465 channel_.reset(
466 engine_.CreateChannel(call_.get(), cricket::VideoOptions()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000467 EXPECT_TRUE(channel_.get() != NULL);
468 ConnectVideoChannelError();
469 network_interface_.SetDestination(channel_.get());
470 channel_->SetInterface(&network_interface_);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000471 media_error_ = cricket::VideoMediaChannel::ERROR_NONE;
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200472 cricket::VideoRecvParameters parameters;
473 parameters.codecs = engine_.codecs();
474 channel_->SetRecvParameters(parameters);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000475 EXPECT_TRUE(channel_->AddSendStream(DefaultSendStreamParams()));
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +0000476 video_capturer_.reset(CreateFakeVideoCapturer());
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000477 cricket::VideoFormat format(640, 480,
478 cricket::VideoFormat::FpsToInterval(30),
479 cricket::FOURCC_I420);
480 EXPECT_EQ(cricket::CS_RUNNING, video_capturer_->Start(format));
481 EXPECT_TRUE(channel_->SetCapturer(kSsrc, video_capturer_.get()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000482 }
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +0000483
484 virtual cricket::FakeVideoCapturer* CreateFakeVideoCapturer() {
485 return new cricket::FakeVideoCapturer();
486 }
487
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000488 // Utility method to setup an additional stream to send and receive video.
489 // Used to test send and recv between two streams.
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000490 void SetUpSecondStream() {
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000491 SetUpSecondStreamWithNoRecv();
492 // Setup recv for second stream.
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000493 EXPECT_TRUE(channel_->AddRecvStream(
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000494 cricket::StreamParams::CreateLegacy(kSsrc + 2)));
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000495 // Make the second renderer available for use by a new stream.
496 EXPECT_TRUE(channel_->SetRenderer(kSsrc + 2, &renderer2_));
497 }
498 // Setup an additional stream just to send video. Defer add recv stream.
499 // This is required if you want to test unsignalled recv of video rtp packets.
500 void SetUpSecondStreamWithNoRecv() {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000501 // SetUp() already added kSsrc make sure duplicate SSRCs cant be added.
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000502 EXPECT_TRUE(channel_->AddRecvStream(
503 cricket::StreamParams::CreateLegacy(kSsrc)));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +0000504 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000505 EXPECT_FALSE(channel_->AddSendStream(
506 cricket::StreamParams::CreateLegacy(kSsrc)));
507 EXPECT_TRUE(channel_->AddSendStream(
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000508 cricket::StreamParams::CreateLegacy(kSsrc + 2)));
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000509 // We dont add recv for the second stream.
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000510
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000511 // Setup the receive and renderer for second stream after send.
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +0000512 video_capturer_2_.reset(CreateFakeVideoCapturer());
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000513 cricket::VideoFormat format(640, 480,
514 cricket::VideoFormat::FpsToInterval(30),
515 cricket::FOURCC_I420);
516 EXPECT_EQ(cricket::CS_RUNNING, video_capturer_2_->Start(format));
517
518 EXPECT_TRUE(channel_->SetCapturer(kSsrc + 2, video_capturer_2_.get()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000519 }
520 virtual void TearDown() {
521 channel_.reset();
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000522 }
523 void ConnectVideoChannelError() {
524 channel_->SignalMediaError.connect(this,
525 &VideoMediaChannelTest<E, C>::OnVideoChannelError);
526 }
527 bool SetDefaultCodec() {
528 return SetOneCodec(DefaultCodec());
529 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000530
531 bool SetOneCodec(int pt, const char* name, int w, int h, int fr) {
532 return SetOneCodec(cricket::VideoCodec(pt, name, w, h, fr, 0));
533 }
534 bool SetOneCodec(const cricket::VideoCodec& codec) {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000535 cricket::VideoFormat capture_format(codec.width, codec.height,
536 cricket::VideoFormat::FpsToInterval(codec.framerate),
537 cricket::FOURCC_I420);
538
539 if (video_capturer_) {
540 EXPECT_EQ(cricket::CS_RUNNING, video_capturer_->Start(capture_format));
541 }
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000542 if (video_capturer_2_) {
543 EXPECT_EQ(cricket::CS_RUNNING, video_capturer_2_->Start(capture_format));
544 }
545
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000546 bool sending = channel_->sending();
547 bool success = SetSend(false);
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200548 if (success) {
549 cricket::VideoSendParameters parameters;
550 parameters.codecs.push_back(codec);
551 success = channel_->SetSendParameters(parameters);
552 }
553 if (success) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000554 success = SetSend(sending);
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200555 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000556 return success;
557 }
558 bool SetSend(bool send) {
559 return channel_->SetSend(send);
560 }
henrike@webrtc.orga7b98182014-02-21 15:51:43 +0000561 bool SetSendStreamFormat(uint32 ssrc, const cricket::VideoCodec& codec) {
562 return channel_->SetSendStreamFormat(ssrc, cricket::VideoFormat(
563 codec.width, codec.height,
564 cricket::VideoFormat::FpsToInterval(codec.framerate),
565 cricket::FOURCC_ANY));
566 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000567 int DrainOutgoingPackets() {
568 int packets = 0;
569 do {
570 packets = NumRtpPackets();
571 // 100 ms should be long enough.
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000572 rtc::Thread::Current()->ProcessMessages(100);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000573 } while (NumRtpPackets() > packets);
574 return NumRtpPackets();
575 }
576 bool SendFrame() {
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000577 if (video_capturer_2_) {
578 video_capturer_2_->CaptureFrame();
579 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000580 return video_capturer_.get() &&
581 video_capturer_->CaptureFrame();
582 }
583 bool WaitAndSendFrame(int wait_ms) {
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000584 bool ret = rtc::Thread::Current()->ProcessMessages(wait_ms);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000585 ret &= SendFrame();
586 return ret;
587 }
588 // Sends frames and waits for the decoder to be fully initialized.
589 // Returns the number of frames that were sent.
590 int WaitForDecoder() {
591#if defined(HAVE_OPENMAX)
592 // Send enough frames for the OpenMAX decoder to continue processing, and
593 // return the number of frames sent.
594 // Send frames for a full kTimeout's worth of 15fps video.
595 int frame_count = 0;
596 while (frame_count < static_cast<int>(kTimeout) / 66) {
597 EXPECT_TRUE(WaitAndSendFrame(66));
598 ++frame_count;
599 }
600 return frame_count;
601#else
602 return 0;
603#endif
604 }
605 bool SendCustomVideoFrame(int w, int h) {
606 if (!video_capturer_.get()) return false;
607 return video_capturer_->CaptureCustomFrame(w, h, cricket::FOURCC_I420);
608 }
609 int NumRtpBytes() {
610 return network_interface_.NumRtpBytes();
611 }
612 int NumRtpBytes(uint32 ssrc) {
613 return network_interface_.NumRtpBytes(ssrc);
614 }
615 int NumRtpPackets() {
616 return network_interface_.NumRtpPackets();
617 }
618 int NumRtpPackets(uint32 ssrc) {
619 return network_interface_.NumRtpPackets(ssrc);
620 }
621 int NumSentSsrcs() {
622 return network_interface_.NumSentSsrcs();
623 }
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000624 const rtc::Buffer* GetRtpPacket(int index) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000625 return network_interface_.GetRtpPacket(index);
626 }
627 int NumRtcpPackets() {
628 return network_interface_.NumRtcpPackets();
629 }
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000630 const rtc::Buffer* GetRtcpPacket(int index) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000631 return network_interface_.GetRtcpPacket(index);
632 }
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000633 static int GetPayloadType(const rtc::Buffer* p) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000634 int pt = -1;
635 ParseRtpPacket(p, NULL, &pt, NULL, NULL, NULL, NULL);
636 return pt;
637 }
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000638 static bool ParseRtpPacket(const rtc::Buffer* p, bool* x, int* pt,
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000639 int* seqnum, uint32* tstamp, uint32* ssrc,
640 std::string* payload) {
Karl Wiberg94784372015-04-20 14:03:07 +0200641 rtc::ByteBuffer buf(*p);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000642 uint8 u08 = 0;
643 uint16 u16 = 0;
644 uint32 u32 = 0;
645
646 // Read X and CC fields.
647 if (!buf.ReadUInt8(&u08)) return false;
648 bool extension = ((u08 & 0x10) != 0);
649 uint8 cc = (u08 & 0x0F);
650 if (x) *x = extension;
651
652 // Read PT field.
653 if (!buf.ReadUInt8(&u08)) return false;
654 if (pt) *pt = (u08 & 0x7F);
655
656 // Read Sequence Number field.
657 if (!buf.ReadUInt16(&u16)) return false;
658 if (seqnum) *seqnum = u16;
659
660 // Read Timestamp field.
661 if (!buf.ReadUInt32(&u32)) return false;
662 if (tstamp) *tstamp = u32;
663
664 // Read SSRC field.
665 if (!buf.ReadUInt32(&u32)) return false;
666 if (ssrc) *ssrc = u32;
667
668 // Skip CSRCs.
669 for (uint8 i = 0; i < cc; ++i) {
670 if (!buf.ReadUInt32(&u32)) return false;
671 }
672
673 // Skip extension header.
674 if (extension) {
675 // Read Profile-specific extension header ID
676 if (!buf.ReadUInt16(&u16)) return false;
677
678 // Read Extension header length
679 if (!buf.ReadUInt16(&u16)) return false;
680 uint16 ext_header_len = u16;
681
682 // Read Extension header
683 for (uint16 i = 0; i < ext_header_len; ++i) {
684 if (!buf.ReadUInt32(&u32)) return false;
685 }
686 }
687
688 if (payload) {
689 return buf.ReadString(payload, buf.Length());
690 }
691 return true;
692 }
693
694 // Parse all RTCP packet, from start_index to stop_index, and count how many
695 // FIR (PT=206 and FMT=4 according to RFC 5104). If successful, set the count
696 // and return true.
697 bool CountRtcpFir(int start_index, int stop_index, int* fir_count) {
698 int count = 0;
699 for (int i = start_index; i < stop_index; ++i) {
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000700 rtc::scoped_ptr<const rtc::Buffer> p(GetRtcpPacket(i));
Karl Wiberg94784372015-04-20 14:03:07 +0200701 rtc::ByteBuffer buf(*p);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000702 size_t total_len = 0;
703 // The packet may be a compound RTCP packet.
kwiberg@webrtc.orgeebcab52015-03-24 09:19:06 +0000704 while (total_len < p->size()) {
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000705 // Read FMT, type and length.
706 uint8 fmt = 0;
707 uint8 type = 0;
708 uint16 length = 0;
709 if (!buf.ReadUInt8(&fmt)) return false;
710 fmt &= 0x1F;
711 if (!buf.ReadUInt8(&type)) return false;
712 if (!buf.ReadUInt16(&length)) return false;
713 buf.Consume(length * 4); // Skip RTCP data.
714 total_len += (length + 1) * 4;
715 if ((192 == type) || ((206 == type) && (4 == fmt))) {
716 ++count;
717 }
718 }
719 }
720
721 if (fir_count) {
722 *fir_count = count;
723 }
724 return true;
725 }
726
727 void OnVideoChannelError(uint32 ssrc,
728 cricket::VideoMediaChannel::Error error) {
729 media_error_ = error;
730 }
731
732 // Test that SetSend works.
733 void SetSend() {
734 EXPECT_FALSE(channel_->sending());
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000735 EXPECT_TRUE(channel_->SetCapturer(kSsrc, video_capturer_.get()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000736 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
737 EXPECT_FALSE(channel_->sending());
738 EXPECT_TRUE(SetSend(true));
739 EXPECT_TRUE(channel_->sending());
740 EXPECT_TRUE(SendFrame());
741 EXPECT_TRUE_WAIT(NumRtpPackets() > 0, kTimeout);
742 EXPECT_TRUE(SetSend(false));
743 EXPECT_FALSE(channel_->sending());
744 }
745 // Test that SetSend fails without codecs being set.
746 void SetSendWithoutCodecs() {
747 EXPECT_FALSE(channel_->sending());
748 EXPECT_FALSE(SetSend(true));
749 EXPECT_FALSE(channel_->sending());
750 }
751 // Test that we properly set the send and recv buffer sizes by the time
752 // SetSend is called.
753 void SetSendSetsTransportBufferSizes() {
754 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
755 EXPECT_TRUE(SetSend(true));
buildbot@webrtc.orgae694ef2014-10-28 17:37:17 +0000756 EXPECT_EQ(64 * 1024, network_interface_.sendbuf_size());
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000757 EXPECT_EQ(64 * 1024, network_interface_.recvbuf_size());
758 }
759 // Tests that we can send frames and the right payload type is used.
760 void Send(const cricket::VideoCodec& codec) {
761 EXPECT_TRUE(SetOneCodec(codec));
762 EXPECT_TRUE(SetSend(true));
763 EXPECT_TRUE(SendFrame());
764 EXPECT_TRUE_WAIT(NumRtpPackets() > 0, kTimeout);
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000765 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000766 EXPECT_EQ(codec.id, GetPayloadType(p.get()));
767 }
768 // Tests that we can send and receive frames.
769 void SendAndReceive(const cricket::VideoCodec& codec) {
770 EXPECT_TRUE(SetOneCodec(codec));
771 EXPECT_TRUE(SetSend(true));
772 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +0000773 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000774 EXPECT_EQ(0, renderer_.num_rendered_frames());
775 EXPECT_TRUE(SendFrame());
776 EXPECT_FRAME_WAIT(1, codec.width, codec.height, kTimeout);
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000777 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000778 EXPECT_EQ(codec.id, GetPayloadType(p.get()));
779 }
780 // Tests that we only get a VideoRenderer::SetSize() callback when needed.
781 void SendManyResizeOnce() {
782 cricket::VideoCodec codec(DefaultCodec());
783 EXPECT_TRUE(SetOneCodec(codec));
784 EXPECT_TRUE(SetSend(true));
785 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +0000786 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000787 EXPECT_EQ(0, renderer_.num_rendered_frames());
788 EXPECT_TRUE(WaitAndSendFrame(30));
789 EXPECT_FRAME_WAIT(1, codec.width, codec.height, kTimeout);
790 EXPECT_TRUE(WaitAndSendFrame(30));
791 EXPECT_FRAME_WAIT(2, codec.width, codec.height, kTimeout);
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000792 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000793 EXPECT_EQ(codec.id, GetPayloadType(p.get()));
794 EXPECT_EQ(1, renderer_.num_set_sizes());
795
796 codec.width /= 2;
797 codec.height /= 2;
798 EXPECT_TRUE(SetOneCodec(codec));
799 EXPECT_TRUE(WaitAndSendFrame(30));
800 EXPECT_FRAME_WAIT(3, codec.width, codec.height, kTimeout);
801 EXPECT_EQ(2, renderer_.num_set_sizes());
802 }
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000803 void SendReceiveManyAndGetStats(const cricket::VideoCodec& codec,
804 int duration_sec, int fps) {
805 EXPECT_TRUE(SetOneCodec(codec));
806 EXPECT_TRUE(SetSend(true));
807 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +0000808 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000809 EXPECT_EQ(0, renderer_.num_rendered_frames());
810 for (int i = 0; i < duration_sec; ++i) {
811 for (int frame = 1; frame <= fps; ++frame) {
812 EXPECT_TRUE(WaitAndSendFrame(1000 / fps));
813 EXPECT_FRAME_WAIT(frame + i * fps, codec.width, codec.height, kTimeout);
814 }
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000815 }
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000816 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000817 EXPECT_EQ(codec.id, GetPayloadType(p.get()));
818 }
819
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000820 // Test that stats work properly for a 1-1 call.
821 void GetStats() {
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000822 const int kDurationSec = 3;
823 const int kFps = 10;
824 SendReceiveManyAndGetStats(DefaultCodec(), kDurationSec, kFps);
825
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000826 cricket::VideoMediaInfo info;
pbos@webrtc.org058b1f12015-03-04 08:54:32 +0000827 EXPECT_TRUE(channel_->GetStats(&info));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000828
829 ASSERT_EQ(1U, info.senders.size());
830 // TODO(whyuan): bytes_sent and bytes_rcvd are different. Are both payload?
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000831 // For webrtc, bytes_sent does not include the RTP header length.
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000832 EXPECT_GT(info.senders[0].bytes_sent, 0);
833 EXPECT_EQ(NumRtpPackets(), info.senders[0].packets_sent);
834 EXPECT_EQ(0.0, info.senders[0].fraction_lost);
835 EXPECT_EQ(0, info.senders[0].firs_rcvd);
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000836 EXPECT_EQ(0, info.senders[0].plis_rcvd);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000837 EXPECT_EQ(0, info.senders[0].nacks_rcvd);
wu@webrtc.org987f2c92014-03-28 16:22:19 +0000838 EXPECT_EQ(DefaultCodec().width, info.senders[0].send_frame_width);
839 EXPECT_EQ(DefaultCodec().height, info.senders[0].send_frame_height);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000840 EXPECT_GT(info.senders[0].framerate_input, 0);
841 EXPECT_GT(info.senders[0].framerate_sent, 0);
842
843 ASSERT_EQ(1U, info.receivers.size());
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000844 EXPECT_EQ(1U, info.senders[0].ssrcs().size());
845 EXPECT_EQ(1U, info.receivers[0].ssrcs().size());
846 EXPECT_EQ(info.senders[0].ssrcs()[0], info.receivers[0].ssrcs()[0]);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000847 EXPECT_EQ(NumRtpBytes(), info.receivers[0].bytes_rcvd);
848 EXPECT_EQ(NumRtpPackets(), info.receivers[0].packets_rcvd);
849 EXPECT_EQ(0.0, info.receivers[0].fraction_lost);
850 EXPECT_EQ(0, info.receivers[0].packets_lost);
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000851 // TODO(asapersson): Not set for webrtc. Handle missing stats.
852 // EXPECT_EQ(0, info.receivers[0].packets_concealed);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000853 EXPECT_EQ(0, info.receivers[0].firs_sent);
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +0000854 EXPECT_EQ(0, info.receivers[0].plis_sent);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000855 EXPECT_EQ(0, info.receivers[0].nacks_sent);
856 EXPECT_EQ(DefaultCodec().width, info.receivers[0].frame_width);
857 EXPECT_EQ(DefaultCodec().height, info.receivers[0].frame_height);
858 EXPECT_GT(info.receivers[0].framerate_rcvd, 0);
859 EXPECT_GT(info.receivers[0].framerate_decoded, 0);
860 EXPECT_GT(info.receivers[0].framerate_output, 0);
861 }
Stefan Holmer586b19b2015-09-18 11:14:31 +0200862
863 cricket::VideoSenderInfo GetSenderStats(size_t i) {
864 cricket::VideoMediaInfo info;
865 EXPECT_TRUE(channel_->GetStats(&info));
866 return info.senders[i];
867 }
868
869 cricket::VideoReceiverInfo GetReceiverStats(size_t i) {
870 cricket::VideoMediaInfo info;
871 EXPECT_TRUE(channel_->GetStats(&info));
872 return info.receivers[i];
873 }
874
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000875 // Test that stats work properly for a conf call with multiple recv streams.
876 void GetStatsMultipleRecvStreams() {
877 cricket::FakeVideoRenderer renderer1, renderer2;
878 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200879 cricket::VideoSendParameters parameters;
880 parameters.codecs.push_back(DefaultCodec());
881 parameters.options.conference_mode.Set(true);
882 EXPECT_TRUE(channel_->SetSendParameters(parameters));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000883 EXPECT_TRUE(SetSend(true));
884 EXPECT_TRUE(channel_->AddRecvStream(
885 cricket::StreamParams::CreateLegacy(1)));
886 EXPECT_TRUE(channel_->AddRecvStream(
887 cricket::StreamParams::CreateLegacy(2)));
888 EXPECT_TRUE(channel_->SetRenderer(1, &renderer1));
889 EXPECT_TRUE(channel_->SetRenderer(2, &renderer2));
890 EXPECT_TRUE(channel_->SetRender(true));
891 EXPECT_EQ(0, renderer1.num_rendered_frames());
892 EXPECT_EQ(0, renderer2.num_rendered_frames());
893 std::vector<uint32> ssrcs;
894 ssrcs.push_back(1);
895 ssrcs.push_back(2);
896 network_interface_.SetConferenceMode(true, ssrcs);
897 EXPECT_TRUE(SendFrame());
898 EXPECT_FRAME_ON_RENDERER_WAIT(
899 renderer1, 1, DefaultCodec().width, DefaultCodec().height, kTimeout);
900 EXPECT_FRAME_ON_RENDERER_WAIT(
901 renderer2, 1, DefaultCodec().width, DefaultCodec().height, kTimeout);
Stefan Holmer586b19b2015-09-18 11:14:31 +0200902
903 EXPECT_TRUE(channel_->SetSend(false));
904
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000905 cricket::VideoMediaInfo info;
pbos@webrtc.org058b1f12015-03-04 08:54:32 +0000906 EXPECT_TRUE(channel_->GetStats(&info));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000907 ASSERT_EQ(1U, info.senders.size());
908 // TODO(whyuan): bytes_sent and bytes_rcvd are different. Are both payload?
buildbot@webrtc.org3eb2c2f2014-06-10 09:39:06 +0000909 // For webrtc, bytes_sent does not include the RTP header length.
Stefan Holmer586b19b2015-09-18 11:14:31 +0200910 EXPECT_GT(GetSenderStats(0).bytes_sent, 0);
911 EXPECT_EQ_WAIT(NumRtpPackets(), GetSenderStats(0).packets_sent, kTimeout);
912 EXPECT_EQ(DefaultCodec().width, GetSenderStats(0).send_frame_width);
913 EXPECT_EQ(DefaultCodec().height, GetSenderStats(0).send_frame_height);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000914
915 ASSERT_EQ(2U, info.receivers.size());
916 for (size_t i = 0; i < info.receivers.size(); ++i) {
Stefan Holmer586b19b2015-09-18 11:14:31 +0200917 EXPECT_EQ(1U, GetReceiverStats(i).ssrcs().size());
918 EXPECT_EQ(i + 1, GetReceiverStats(i).ssrcs()[0]);
919 EXPECT_EQ_WAIT(NumRtpBytes(), GetReceiverStats(i).bytes_rcvd, kTimeout);
920 EXPECT_EQ_WAIT(NumRtpPackets(), GetReceiverStats(i).packets_rcvd,
921 kTimeout);
922 EXPECT_EQ(DefaultCodec().width, GetReceiverStats(i).frame_width);
923 EXPECT_EQ(DefaultCodec().height, GetReceiverStats(i).frame_height);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000924 }
925 }
926 // Test that stats work properly for a conf call with multiple send streams.
927 void GetStatsMultipleSendStreams() {
928 // Normal setup; note that we set the SSRC explicitly to ensure that
929 // it will come first in the senders map.
930 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200931 cricket::VideoSendParameters parameters;
932 parameters.codecs.push_back(DefaultCodec());
933 parameters.options.conference_mode.Set(true);
934 EXPECT_TRUE(channel_->SetSendParameters(parameters));
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000935 EXPECT_TRUE(channel_->AddRecvStream(
buildbot@webrtc.org99f63082014-07-18 23:31:30 +0000936 cricket::StreamParams::CreateLegacy(kSsrc)));
937 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer_));
henrike@webrtc.org28654cb2013-07-22 21:07:49 +0000938 channel_->UpdateAspectRatio(640, 400);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000939 EXPECT_TRUE(SetSend(true));
940 EXPECT_TRUE(channel_->SetRender(true));
941 EXPECT_TRUE(SendFrame());
942 EXPECT_TRUE_WAIT(NumRtpPackets() > 0, kTimeout);
943 EXPECT_FRAME_WAIT(1, DefaultCodec().width, DefaultCodec().height, kTimeout);
944
945 // Add an additional capturer, and hook up a renderer to receive it.
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000946 cricket::FakeVideoRenderer renderer2;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +0000947 rtc::scoped_ptr<cricket::FakeVideoCapturer> capturer(
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +0000948 CreateFakeVideoCapturer());
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000949 capturer->SetScreencast(true);
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +0000950 const int kTestWidth = 160;
951 const int kTestHeight = 120;
952 cricket::VideoFormat format(kTestWidth, kTestHeight,
953 cricket::VideoFormat::FpsToInterval(5),
954 cricket::FOURCC_I420);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000955 EXPECT_EQ(cricket::CS_RUNNING, capturer->Start(format));
956 EXPECT_TRUE(channel_->AddSendStream(
957 cricket::StreamParams::CreateLegacy(5678)));
958 EXPECT_TRUE(channel_->SetCapturer(5678, capturer.get()));
959 EXPECT_TRUE(channel_->AddRecvStream(
960 cricket::StreamParams::CreateLegacy(5678)));
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000961 EXPECT_TRUE(channel_->SetRenderer(5678, &renderer2));
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +0000962 EXPECT_TRUE(capturer->CaptureCustomFrame(
963 kTestWidth, kTestHeight, cricket::FOURCC_I420));
964 EXPECT_FRAME_ON_RENDERER_WAIT(
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000965 renderer2, 1, kTestWidth, kTestHeight, kTimeout);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000966
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000967 // Get stats, and make sure they are correct for two senders. We wait until
968 // the number of expected packets have been sent to avoid races where we
969 // check stats before it has been updated.
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000970 cricket::VideoMediaInfo info;
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000971 for (uint32 i = 0; i < kTimeout; ++i) {
972 rtc::Thread::Current()->ProcessMessages(1);
pbos@webrtc.org058b1f12015-03-04 08:54:32 +0000973 EXPECT_TRUE(channel_->GetStats(&info));
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000974 ASSERT_EQ(2U, info.senders.size());
975 if (info.senders[0].packets_sent + info.senders[1].packets_sent ==
976 NumRtpPackets()) {
977 // Stats have been updated for both sent frames, expectations can be
978 // checked now.
979 break;
980 }
981 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000982 EXPECT_EQ(NumRtpPackets(),
pbos@webrtc.orgec45e3b2015-02-20 10:24:53 +0000983 info.senders[0].packets_sent + info.senders[1].packets_sent)
984 << "Timed out while waiting for packet counts for all sent packets.";
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000985 EXPECT_EQ(1U, info.senders[0].ssrcs().size());
986 EXPECT_EQ(1234U, info.senders[0].ssrcs()[0]);
wu@webrtc.org987f2c92014-03-28 16:22:19 +0000987 EXPECT_EQ(DefaultCodec().width, info.senders[0].send_frame_width);
988 EXPECT_EQ(DefaultCodec().height, info.senders[0].send_frame_height);
sergeyu@chromium.org5bc25c42013-12-05 00:24:06 +0000989 EXPECT_EQ(1U, info.senders[1].ssrcs().size());
990 EXPECT_EQ(5678U, info.senders[1].ssrcs()[0]);
wu@webrtc.org987f2c92014-03-28 16:22:19 +0000991 EXPECT_EQ(kTestWidth, info.senders[1].send_frame_width);
992 EXPECT_EQ(kTestHeight, info.senders[1].send_frame_height);
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000993 // The capturer must be unregistered here as it runs out of it's scope next.
994 EXPECT_TRUE(channel_->SetCapturer(5678, NULL));
995 }
996
sergeyu@chromium.org4b26e2e2014-01-15 23:15:54 +0000997 // Test that we can set the bandwidth.
henrike@webrtc.org28e20752013-07-10 00:45:36 +0000998 void SetSendBandwidth() {
Fredrik Solenbergb071a192015-09-17 16:42:56 +0200999 cricket::VideoSendParameters parameters;
1000 parameters.codecs.push_back(DefaultCodec());
1001 parameters.max_bandwidth_bps = -1; // <= 0 means unlimited.
1002 EXPECT_TRUE(channel_->SetSendParameters(parameters));
1003 parameters.max_bandwidth_bps = 128 * 1024;
1004 EXPECT_TRUE(channel_->SetSendParameters(parameters));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001005 }
1006 // Test that we can set the SSRC for the default send source.
1007 void SetSendSsrc() {
1008 EXPECT_TRUE(SetDefaultCodec());
henrike@webrtc.orga7b98182014-02-21 15:51:43 +00001009 EXPECT_TRUE(SetSendStreamFormat(kSsrc, DefaultCodec()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001010 EXPECT_TRUE(SetSend(true));
1011 EXPECT_TRUE(SendFrame());
1012 EXPECT_TRUE_WAIT(NumRtpPackets() > 0, kTimeout);
1013 uint32 ssrc = 0;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001014 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001015 ParseRtpPacket(p.get(), NULL, NULL, NULL, NULL, &ssrc, NULL);
1016 EXPECT_EQ(kSsrc, ssrc);
1017 EXPECT_EQ(NumRtpPackets(), NumRtpPackets(ssrc));
1018 EXPECT_EQ(NumRtpBytes(), NumRtpBytes(ssrc));
1019 EXPECT_EQ(1, NumSentSsrcs());
1020 EXPECT_EQ(0, NumRtpPackets(kSsrc - 1));
1021 EXPECT_EQ(0, NumRtpBytes(kSsrc - 1));
1022 }
1023 // Test that we can set the SSRC even after codecs are set.
1024 void SetSendSsrcAfterSetCodecs() {
1025 // Remove stream added in Setup.
1026 EXPECT_TRUE(channel_->RemoveSendStream(kSsrc));
1027 EXPECT_TRUE(SetDefaultCodec());
1028 EXPECT_TRUE(channel_->AddSendStream(
1029 cricket::StreamParams::CreateLegacy(999)));
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001030 EXPECT_TRUE(channel_->SetCapturer(999u, video_capturer_.get()));
henrike@webrtc.orga7b98182014-02-21 15:51:43 +00001031 EXPECT_TRUE(SetSendStreamFormat(999u, DefaultCodec()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001032 EXPECT_TRUE(SetSend(true));
1033 EXPECT_TRUE(WaitAndSendFrame(0));
1034 EXPECT_TRUE_WAIT(NumRtpPackets() > 0, kTimeout);
1035 uint32 ssrc = 0;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001036 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001037 ParseRtpPacket(p.get(), NULL, NULL, NULL, NULL, &ssrc, NULL);
1038 EXPECT_EQ(999u, ssrc);
1039 EXPECT_EQ(NumRtpPackets(), NumRtpPackets(ssrc));
1040 EXPECT_EQ(NumRtpBytes(), NumRtpBytes(ssrc));
1041 EXPECT_EQ(1, NumSentSsrcs());
1042 EXPECT_EQ(0, NumRtpPackets(kSsrc));
1043 EXPECT_EQ(0, NumRtpBytes(kSsrc));
1044 }
1045 // Test that we can set the default video renderer before and after
1046 // media is received.
1047 void SetRenderer() {
1048 uint8 data1[] = {
1049 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
1050 };
1051
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001052 rtc::Buffer packet1(data1, sizeof(data1));
1053 rtc::SetBE32(packet1.data() + 8, kSsrc);
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001054 channel_->SetRenderer(kDefaultReceiveSsrc, NULL);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001055 EXPECT_TRUE(SetDefaultCodec());
1056 EXPECT_TRUE(SetSend(true));
1057 EXPECT_TRUE(channel_->SetRender(true));
1058 EXPECT_EQ(0, renderer_.num_rendered_frames());
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001059 channel_->OnPacketReceived(&packet1, rtc::PacketTime());
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001060 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001061 EXPECT_TRUE(SendFrame());
1062 EXPECT_FRAME_WAIT(1, DefaultCodec().width, DefaultCodec().height, kTimeout);
1063 }
1064
1065 // Tests empty StreamParams is rejected.
1066 void RejectEmptyStreamParams() {
1067 // Remove the send stream that was added during Setup.
1068 EXPECT_TRUE(channel_->RemoveSendStream(kSsrc));
1069
1070 cricket::StreamParams empty;
1071 EXPECT_FALSE(channel_->AddSendStream(empty));
1072 EXPECT_TRUE(channel_->AddSendStream(
1073 cricket::StreamParams::CreateLegacy(789u)));
1074 }
1075
1076 // Tests setting up and configuring a send stream.
1077 void AddRemoveSendStreams() {
1078 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
1079 EXPECT_TRUE(SetSend(true));
1080 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001081 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001082 EXPECT_TRUE(SendFrame());
1083 EXPECT_FRAME_WAIT(1, DefaultCodec().width, DefaultCodec().height, kTimeout);
Stefan Holmer586b19b2015-09-18 11:14:31 +02001084 EXPECT_GT(NumRtpPackets(), 0);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001085 uint32 ssrc = 0;
1086 size_t last_packet = NumRtpPackets() - 1;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001087 rtc::scoped_ptr<const rtc::Buffer>
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001088 p(GetRtpPacket(static_cast<int>(last_packet)));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001089 ParseRtpPacket(p.get(), NULL, NULL, NULL, NULL, &ssrc, NULL);
1090 EXPECT_EQ(kSsrc, ssrc);
1091
1092 // Remove the send stream that was added during Setup.
1093 EXPECT_TRUE(channel_->RemoveSendStream(kSsrc));
1094 int rtp_packets = NumRtpPackets();
1095
1096 EXPECT_TRUE(channel_->AddSendStream(
1097 cricket::StreamParams::CreateLegacy(789u)));
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001098 EXPECT_TRUE(channel_->SetCapturer(789u, video_capturer_.get()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001099 EXPECT_EQ(rtp_packets, NumRtpPackets());
1100 // Wait 30ms to guarantee the engine does not drop the frame.
1101 EXPECT_TRUE(WaitAndSendFrame(30));
1102 EXPECT_TRUE_WAIT(NumRtpPackets() > rtp_packets, kTimeout);
1103
1104 last_packet = NumRtpPackets() - 1;
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001105 p.reset(GetRtpPacket(static_cast<int>(last_packet)));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001106 ParseRtpPacket(p.get(), NULL, NULL, NULL, NULL, &ssrc, NULL);
1107 EXPECT_EQ(789u, ssrc);
1108 }
1109
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001110 // Tests setting up and configuring multiple incoming streams.
1111 void AddRemoveRecvStreams() {
1112 cricket::FakeVideoRenderer renderer1, renderer2;
Fredrik Solenbergb071a192015-09-17 16:42:56 +02001113 cricket::VideoSendParameters parameters;
1114 parameters.codecs.push_back(DefaultCodec());
1115 EXPECT_TRUE(channel_->SetSendParameters(parameters));
1116
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001117 // Ensure we can't set the renderer on a non-existent stream.
1118 EXPECT_FALSE(channel_->SetRenderer(1, &renderer1));
1119 EXPECT_FALSE(channel_->SetRenderer(2, &renderer2));
1120 cricket::VideoRenderer* renderer;
1121 EXPECT_FALSE(channel_->GetRenderer(1, &renderer));
1122 EXPECT_FALSE(channel_->GetRenderer(2, &renderer));
1123
1124 // Ensure we can add streams.
1125 EXPECT_TRUE(channel_->AddRecvStream(
1126 cricket::StreamParams::CreateLegacy(1)));
1127 EXPECT_TRUE(channel_->AddRecvStream(
1128 cricket::StreamParams::CreateLegacy(2)));
1129 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001130 EXPECT_TRUE(renderer == NULL);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001131 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1132 EXPECT_TRUE(NULL == renderer);
1133
1134 // Ensure we can now set the renderers.
1135 EXPECT_TRUE(channel_->SetRenderer(1, &renderer1));
1136 EXPECT_TRUE(channel_->SetRenderer(2, &renderer2));
1137 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
1138 EXPECT_TRUE(&renderer1 == renderer);
1139 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1140 EXPECT_TRUE(&renderer2 == renderer);
1141
1142 // Ensure we can change the renderers if needed.
1143 EXPECT_TRUE(channel_->SetRenderer(1, &renderer2));
1144 EXPECT_TRUE(channel_->SetRenderer(2, &renderer1));
1145 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
1146 EXPECT_TRUE(&renderer2 == renderer);
1147 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1148 EXPECT_TRUE(&renderer1 == renderer);
1149
1150 EXPECT_TRUE(channel_->RemoveRecvStream(2));
1151 EXPECT_TRUE(channel_->RemoveRecvStream(1));
1152 EXPECT_FALSE(channel_->GetRenderer(1, &renderer));
1153 EXPECT_FALSE(channel_->GetRenderer(2, &renderer));
1154 }
1155
1156 // Tests setting up and configuring multiple incoming streams in a
1157 // non-conference call.
1158 void AddRemoveRecvStreamsNoConference() {
1159 cricket::FakeVideoRenderer renderer1, renderer2;
1160 // Ensure we can't set the renderer on a non-existent stream.
1161 EXPECT_FALSE(channel_->SetRenderer(1, &renderer1));
1162 EXPECT_FALSE(channel_->SetRenderer(2, &renderer2));
1163 cricket::VideoRenderer* renderer;
1164 EXPECT_FALSE(channel_->GetRenderer(1, &renderer));
1165 EXPECT_FALSE(channel_->GetRenderer(2, &renderer));
1166
1167 // Ensure we can add streams.
1168 EXPECT_TRUE(channel_->AddRecvStream(
1169 cricket::StreamParams::CreateLegacy(1)));
1170 EXPECT_TRUE(channel_->AddRecvStream(
1171 cricket::StreamParams::CreateLegacy(2)));
1172 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
1173 // Verify the first AddRecvStream hook up to the default renderer.
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001174 EXPECT_TRUE(renderer == NULL);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001175 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1176 EXPECT_TRUE(NULL == renderer);
1177
1178 // Ensure we can now set the renderers.
1179 EXPECT_TRUE(channel_->SetRenderer(1, &renderer1));
1180 EXPECT_TRUE(channel_->SetRenderer(2, &renderer2));
1181 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
1182 EXPECT_TRUE(&renderer1 == renderer);
1183 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1184 EXPECT_TRUE(&renderer2 == renderer);
1185
1186 // Ensure we can change the renderers if needed.
1187 EXPECT_TRUE(channel_->SetRenderer(1, &renderer2));
1188 EXPECT_TRUE(channel_->SetRenderer(2, &renderer1));
1189 EXPECT_TRUE(channel_->GetRenderer(1, &renderer));
1190 EXPECT_TRUE(&renderer2 == renderer);
1191 EXPECT_TRUE(channel_->GetRenderer(2, &renderer));
1192 EXPECT_TRUE(&renderer1 == renderer);
1193
1194 EXPECT_TRUE(channel_->RemoveRecvStream(2));
1195 EXPECT_TRUE(channel_->RemoveRecvStream(1));
1196 EXPECT_FALSE(channel_->GetRenderer(1, &renderer));
1197 EXPECT_FALSE(channel_->GetRenderer(2, &renderer));
1198 }
1199
1200 // Test that no frames are rendered after the receive stream have been
1201 // removed.
1202 void AddRemoveRecvStreamAndRender() {
1203 cricket::FakeVideoRenderer renderer1;
1204 EXPECT_TRUE(SetDefaultCodec());
1205 EXPECT_TRUE(SetSend(true));
1206 EXPECT_TRUE(channel_->SetRender(true));
1207 EXPECT_TRUE(channel_->AddRecvStream(
1208 cricket::StreamParams::CreateLegacy(kSsrc)));
1209 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer1));
1210
1211 EXPECT_TRUE(SendFrame());
1212 EXPECT_FRAME_ON_RENDERER_WAIT(
1213 renderer1, 1, DefaultCodec().width, DefaultCodec().height, kTimeout);
1214 EXPECT_TRUE(channel_->RemoveRecvStream(kSsrc));
1215 // Send three more frames. This is to avoid that the test might be flaky
1216 // due to frame dropping.
1217 for (size_t i = 0; i < 3; ++i)
1218 EXPECT_TRUE(WaitAndSendFrame(100));
1219
1220 // Test that no more frames have been rendered.
1221 EXPECT_EQ(1, renderer1.num_rendered_frames());
1222
1223 // Re-add the stream again and make sure it renders.
1224 EXPECT_TRUE(channel_->AddRecvStream(
1225 cricket::StreamParams::CreateLegacy(kSsrc)));
1226 // Force the next frame to be a key frame to make the receiving
1227 // decoder happy.
1228 EXPECT_TRUE(channel_->SendIntraFrame());
1229
1230 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer1));
1231 EXPECT_TRUE(SendFrame());
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001232 // Because the default channel is used, RemoveRecvStream above is not going
1233 // to delete the channel. As a result the engine will continue to receive
1234 // and decode the 3 frames sent above. So it is possible we will receive
1235 // some (e.g. 1) of these 3 frames after the renderer is set again.
1236 EXPECT_GT_FRAME_ON_RENDERER_WAIT(
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001237 renderer1, 2, DefaultCodec().width, DefaultCodec().height, kTimeout);
mallinath@webrtc.org67ee6b92014-02-03 16:57:16 +00001238 // Detach |renderer1| before exit as there might be frames come late.
1239 EXPECT_TRUE(channel_->SetRenderer(kSsrc, NULL));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001240 }
1241
1242 // Tests the behavior of incoming streams in a conference scenario.
1243 void SimulateConference() {
1244 cricket::FakeVideoRenderer renderer1, renderer2;
1245 EXPECT_TRUE(SetDefaultCodec());
Fredrik Solenbergb071a192015-09-17 16:42:56 +02001246 cricket::VideoSendParameters parameters;
1247 parameters.codecs.push_back(DefaultCodec());
1248 parameters.options.conference_mode.Set(true);
1249 EXPECT_TRUE(channel_->SetSendParameters(parameters));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001250 EXPECT_TRUE(SetSend(true));
1251 EXPECT_TRUE(channel_->SetRender(true));
1252 EXPECT_TRUE(channel_->AddRecvStream(
1253 cricket::StreamParams::CreateLegacy(1)));
1254 EXPECT_TRUE(channel_->AddRecvStream(
1255 cricket::StreamParams::CreateLegacy(2)));
1256 EXPECT_TRUE(channel_->SetRenderer(1, &renderer1));
1257 EXPECT_TRUE(channel_->SetRenderer(2, &renderer2));
1258 EXPECT_EQ(0, renderer1.num_rendered_frames());
1259 EXPECT_EQ(0, renderer2.num_rendered_frames());
1260 std::vector<uint32> ssrcs;
1261 ssrcs.push_back(1);
1262 ssrcs.push_back(2);
1263 network_interface_.SetConferenceMode(true, ssrcs);
1264 EXPECT_TRUE(SendFrame());
1265 EXPECT_FRAME_ON_RENDERER_WAIT(
1266 renderer1, 1, DefaultCodec().width, DefaultCodec().height, kTimeout);
1267 EXPECT_FRAME_ON_RENDERER_WAIT(
1268 renderer2, 1, DefaultCodec().width, DefaultCodec().height, kTimeout);
1269
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001270 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001271 EXPECT_EQ(DefaultCodec().id, GetPayloadType(p.get()));
1272 EXPECT_EQ(DefaultCodec().width, renderer1.width());
1273 EXPECT_EQ(DefaultCodec().height, renderer1.height());
1274 EXPECT_EQ(DefaultCodec().width, renderer2.width());
1275 EXPECT_EQ(DefaultCodec().height, renderer2.height());
1276 EXPECT_TRUE(channel_->RemoveRecvStream(2));
1277 EXPECT_TRUE(channel_->RemoveRecvStream(1));
1278 }
1279
1280 // Tests that we can add and remove capturers and frames are sent out properly
1281 void AddRemoveCapturer() {
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +00001282 cricket::VideoCodec codec = DefaultCodec();
1283 codec.width = 320;
1284 codec.height = 240;
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001285 const int time_between_send = TimeBetweenSend(codec);
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +00001286 EXPECT_TRUE(SetOneCodec(codec));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001287 EXPECT_TRUE(SetSend(true));
1288 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001289 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001290 EXPECT_EQ(0, renderer_.num_rendered_frames());
1291 EXPECT_TRUE(SendFrame());
1292 EXPECT_FRAME_WAIT(1, codec.width, codec.height, kTimeout);
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001293 rtc::scoped_ptr<cricket::FakeVideoCapturer> capturer(
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +00001294 CreateFakeVideoCapturer());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001295 capturer->SetScreencast(true);
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +00001296 cricket::VideoFormat format(480, 360,
1297 cricket::VideoFormat::FpsToInterval(30),
1298 cricket::FOURCC_I420);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001299 EXPECT_EQ(cricket::CS_RUNNING, capturer->Start(format));
1300 // All capturers start generating frames with the same timestamp. ViE does
1301 // not allow the same timestamp to be used. Capture one frame before
1302 // associating the capturer with the channel.
1303 EXPECT_TRUE(capturer->CaptureCustomFrame(format.width, format.height,
1304 cricket::FOURCC_I420));
1305
1306 int captured_frames = 1;
1307 for (int iterations = 0; iterations < 2; ++iterations) {
1308 EXPECT_TRUE(channel_->SetCapturer(kSsrc, capturer.get()));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001309 rtc::Thread::Current()->ProcessMessages(time_between_send);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001310 EXPECT_TRUE(capturer->CaptureCustomFrame(format.width, format.height,
1311 cricket::FOURCC_I420));
1312 ++captured_frames;
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001313 // Wait until frame of right size is captured.
1314 EXPECT_TRUE_WAIT(renderer_.num_rendered_frames() >= captured_frames &&
1315 format.width == renderer_.width() &&
henrike@webrtc.org1e09a712013-07-26 19:17:59 +00001316 format.height == renderer_.height() &&
1317 !renderer_.black_frame(), kTimeout);
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001318 EXPECT_GE(renderer_.num_rendered_frames(), captured_frames);
1319 EXPECT_EQ(format.width, renderer_.width());
1320 EXPECT_EQ(format.height, renderer_.height());
1321 captured_frames = renderer_.num_rendered_frames() + 1;
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001322 EXPECT_FALSE(renderer_.black_frame());
1323 EXPECT_TRUE(channel_->SetCapturer(kSsrc, NULL));
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001324 // Make sure a black frame is generated within the specified timeout.
pbos@webrtc.orgb4987bf2015-02-18 10:13:09 +00001325 // The black frame should be the resolution of the previous frame to
1326 // prevent expensive encoder reconfigurations.
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001327 EXPECT_TRUE_WAIT(renderer_.num_rendered_frames() >= captured_frames &&
pbos@webrtc.orgb4987bf2015-02-18 10:13:09 +00001328 format.width == renderer_.width() &&
1329 format.height == renderer_.height() &&
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001330 renderer_.black_frame(), kTimeout);
1331 EXPECT_GE(renderer_.num_rendered_frames(), captured_frames);
pbos@webrtc.orgb4987bf2015-02-18 10:13:09 +00001332 EXPECT_EQ(format.width, renderer_.width());
1333 EXPECT_EQ(format.height, renderer_.height());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001334 EXPECT_TRUE(renderer_.black_frame());
1335
1336 // The black frame has the same timestamp as the next frame since it's
1337 // timestamp is set to the last frame's timestamp + interval. WebRTC will
1338 // not render a frame with the same timestamp so capture another frame
1339 // with the frame capturer to increment the next frame's timestamp.
1340 EXPECT_TRUE(capturer->CaptureCustomFrame(format.width, format.height,
1341 cricket::FOURCC_I420));
1342 }
1343 }
1344
1345 // Tests that if RemoveCapturer is called without a capturer ever being
1346 // added, the plugin shouldn't crash (and no black frame should be sent).
1347 void RemoveCapturerWithoutAdd() {
1348 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
1349 EXPECT_TRUE(SetSend(true));
1350 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001351 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001352 EXPECT_EQ(0, renderer_.num_rendered_frames());
1353 EXPECT_TRUE(SendFrame());
1354 EXPECT_FRAME_WAIT(1, 640, 400, kTimeout);
pbos@webrtc.org776e6f22014-10-29 15:28:39 +00001355 // Wait for one frame so they don't get dropped because we send frames too
1356 // tightly.
1357 rtc::Thread::Current()->ProcessMessages(30);
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001358 // Remove the capturer.
1359 EXPECT_TRUE(channel_->SetCapturer(kSsrc, NULL));
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001360 // Wait for one black frame for removing the capturer.
1361 EXPECT_FRAME_WAIT(2, 640, 400, kTimeout);
1362
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001363 // No capturer was added, so this RemoveCapturer should
1364 // fail.
1365 EXPECT_FALSE(channel_->SetCapturer(kSsrc, NULL));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001366 rtc::Thread::Current()->ProcessMessages(300);
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001367 // Verify no more frames were sent.
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001368 EXPECT_EQ(2, renderer_.num_rendered_frames());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001369 }
1370
1371 // Tests that we can add and remove capturer as unique sources.
1372 void AddRemoveCapturerMultipleSources() {
1373 // WebRTC implementation will drop frames if pushed to quickly. Wait the
1374 // interval time to avoid that.
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001375 // WebRTC implementation will drop frames if pushed to quickly. Wait the
1376 // interval time to avoid that.
1377 // Set up the stream associated with the engine.
1378 EXPECT_TRUE(channel_->AddRecvStream(
1379 cricket::StreamParams::CreateLegacy(kSsrc)));
1380 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer_));
1381 cricket::VideoFormat capture_format; // default format
1382 capture_format.interval = cricket::VideoFormat::FpsToInterval(30);
1383 // Set up additional stream 1.
1384 cricket::FakeVideoRenderer renderer1;
1385 EXPECT_FALSE(channel_->SetRenderer(1, &renderer1));
1386 EXPECT_TRUE(channel_->AddRecvStream(
1387 cricket::StreamParams::CreateLegacy(1)));
1388 EXPECT_TRUE(channel_->SetRenderer(1, &renderer1));
1389 EXPECT_TRUE(channel_->AddSendStream(
1390 cricket::StreamParams::CreateLegacy(1)));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001391 rtc::scoped_ptr<cricket::FakeVideoCapturer> capturer1(
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +00001392 CreateFakeVideoCapturer());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001393 capturer1->SetScreencast(true);
1394 EXPECT_EQ(cricket::CS_RUNNING, capturer1->Start(capture_format));
1395 // Set up additional stream 2.
1396 cricket::FakeVideoRenderer renderer2;
1397 EXPECT_FALSE(channel_->SetRenderer(2, &renderer2));
1398 EXPECT_TRUE(channel_->AddRecvStream(
1399 cricket::StreamParams::CreateLegacy(2)));
1400 EXPECT_TRUE(channel_->SetRenderer(2, &renderer2));
1401 EXPECT_TRUE(channel_->AddSendStream(
1402 cricket::StreamParams::CreateLegacy(2)));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001403 rtc::scoped_ptr<cricket::FakeVideoCapturer> capturer2(
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +00001404 CreateFakeVideoCapturer());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001405 capturer2->SetScreencast(true);
1406 EXPECT_EQ(cricket::CS_RUNNING, capturer2->Start(capture_format));
1407 // State for all the streams.
1408 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
1409 // A limitation in the lmi implementation requires that SetCapturer() is
1410 // called after SetOneCodec().
1411 // TODO(hellner): this seems like an unnecessary constraint, fix it.
1412 EXPECT_TRUE(channel_->SetCapturer(1, capturer1.get()));
1413 EXPECT_TRUE(channel_->SetCapturer(2, capturer2.get()));
1414 EXPECT_TRUE(SetSend(true));
1415 EXPECT_TRUE(channel_->SetRender(true));
1416 // Test capturer associated with engine.
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +00001417 const int kTestWidth = 160;
1418 const int kTestHeight = 120;
1419 EXPECT_TRUE(capturer1->CaptureCustomFrame(
1420 kTestWidth, kTestHeight, cricket::FOURCC_I420));
1421 EXPECT_FRAME_ON_RENDERER_WAIT(
1422 renderer1, 1, kTestWidth, kTestHeight, kTimeout);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001423 // Capture a frame with additional capturer2, frames should be received
henrika@webrtc.orgaebb1ad2014-01-14 10:00:58 +00001424 EXPECT_TRUE(capturer2->CaptureCustomFrame(
1425 kTestWidth, kTestHeight, cricket::FOURCC_I420));
1426 EXPECT_FRAME_ON_RENDERER_WAIT(
1427 renderer2, 1, kTestWidth, kTestHeight, kTimeout);
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001428 // Successfully remove the capturer.
1429 EXPECT_TRUE(channel_->SetCapturer(kSsrc, NULL));
1430 // Fail to re-remove the capturer.
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001431 EXPECT_FALSE(channel_->SetCapturer(kSsrc, NULL));
1432 // The capturers must be unregistered here as it runs out of it's scope
1433 // next.
1434 EXPECT_TRUE(channel_->SetCapturer(1, NULL));
1435 EXPECT_TRUE(channel_->SetCapturer(2, NULL));
1436 }
1437
1438 void HighAspectHighHeightCapturer() {
1439 const int kWidth = 80;
1440 const int kHeight = 10000;
1441 const int kScaledWidth = 20;
1442 const int kScaledHeight = 2500;
1443
1444 cricket::VideoCodec codec(DefaultCodec());
1445 EXPECT_TRUE(SetOneCodec(codec));
1446 EXPECT_TRUE(SetSend(true));
1447
1448 cricket::FakeVideoRenderer renderer;
1449 EXPECT_TRUE(channel_->AddRecvStream(
1450 cricket::StreamParams::CreateLegacy(kSsrc)));
1451 EXPECT_TRUE(channel_->SetRenderer(kSsrc, &renderer));
1452 EXPECT_TRUE(channel_->SetRender(true));
1453 EXPECT_EQ(0, renderer.num_rendered_frames());
1454
1455 EXPECT_TRUE(SendFrame());
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001456 EXPECT_GT_FRAME_ON_RENDERER_WAIT(
1457 renderer, 1, codec.width, codec.height, kTimeout);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001458
1459 // Registering an external capturer is currently the same as screen casting
1460 // (update the test when this changes).
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001461 rtc::scoped_ptr<cricket::FakeVideoCapturer> capturer(
buildbot@webrtc.org4f0d4012014-08-07 04:47:36 +00001462 CreateFakeVideoCapturer());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001463 capturer->SetScreencast(true);
1464 const std::vector<cricket::VideoFormat>* formats =
1465 capturer->GetSupportedFormats();
1466 cricket::VideoFormat capture_format = (*formats)[0];
1467 EXPECT_EQ(cricket::CS_RUNNING, capturer->Start(capture_format));
1468 // Capture frame to not get same frame timestamps as previous capturer.
1469 capturer->CaptureFrame();
1470 EXPECT_TRUE(channel_->SetCapturer(kSsrc, capturer.get()));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001471 EXPECT_TRUE(rtc::Thread::Current()->ProcessMessages(30));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001472 EXPECT_TRUE(capturer->CaptureCustomFrame(kWidth, kHeight,
1473 cricket::FOURCC_ARGB));
wu@webrtc.org9caf2762013-12-11 18:25:07 +00001474 EXPECT_GT_FRAME_ON_RENDERER_WAIT(
1475 renderer, 2, kScaledWidth, kScaledHeight, kTimeout);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001476 EXPECT_TRUE(channel_->SetCapturer(kSsrc, NULL));
1477 }
1478
1479 // Tests that we can adapt video resolution with 16:10 aspect ratio properly.
1480 void AdaptResolution16x10() {
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001481 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001482 cricket::VideoCodec codec(DefaultCodec());
1483 codec.width = 640;
1484 codec.height = 400;
1485 SendAndReceive(codec);
1486 codec.width /= 2;
1487 codec.height /= 2;
1488 // Adapt the resolution.
1489 EXPECT_TRUE(SetOneCodec(codec));
1490 EXPECT_TRUE(WaitAndSendFrame(30));
1491 EXPECT_FRAME_WAIT(2, codec.width, codec.height, kTimeout);
1492 }
1493 // Tests that we can adapt video resolution with 4:3 aspect ratio properly.
1494 void AdaptResolution4x3() {
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001495 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001496 cricket::VideoCodec codec(DefaultCodec());
1497 codec.width = 640;
1498 codec.height = 400;
1499 SendAndReceive(codec);
1500 codec.width /= 2;
1501 codec.height /= 2;
1502 // Adapt the resolution.
1503 EXPECT_TRUE(SetOneCodec(codec));
1504 EXPECT_TRUE(WaitAndSendFrame(30));
1505 EXPECT_FRAME_WAIT(2, codec.width, codec.height, kTimeout);
1506 }
1507 // Tests that we can drop all frames properly.
1508 void AdaptDropAllFrames() {
1509 // Set the channel codec's resolution to 0, which will require the adapter
1510 // to drop all frames.
1511 cricket::VideoCodec codec(DefaultCodec());
1512 codec.width = codec.height = codec.framerate = 0;
1513 EXPECT_TRUE(SetOneCodec(codec));
1514 EXPECT_TRUE(SetSend(true));
1515 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001516 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001517 EXPECT_EQ(0, renderer_.num_rendered_frames());
1518 EXPECT_TRUE(SendFrame());
1519 EXPECT_TRUE(SendFrame());
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001520 rtc::Thread::Current()->ProcessMessages(500);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001521 EXPECT_EQ(0, renderer_.num_rendered_frames());
1522 }
1523 // Tests that we can reduce the frame rate on demand properly.
1524 // TODO(fbarchard): This test is flakey on pulse. Fix and re-enable
1525 void AdaptFramerate() {
1526 cricket::VideoCodec codec(DefaultCodec());
1527 int frame_count = 0;
1528 // The capturer runs at 30 fps. The channel requires 30 fps.
1529 EXPECT_TRUE(SetOneCodec(codec));
1530 EXPECT_TRUE(SetSend(true));
1531 EXPECT_TRUE(channel_->SetRender(true));
1532 EXPECT_EQ(frame_count, renderer_.num_rendered_frames());
1533 EXPECT_TRUE(WaitAndSendFrame(0)); // Should be rendered.
1534 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be rendered.
1535 frame_count += 2;
1536 EXPECT_FRAME_WAIT(frame_count, codec.width, codec.height, kTimeout);
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001537 rtc::scoped_ptr<const rtc::Buffer> p(GetRtpPacket(0));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001538 EXPECT_EQ(codec.id, GetPayloadType(p.get()));
1539
1540 // The channel requires 15 fps.
1541 codec.framerate = 15;
1542 EXPECT_TRUE(SetOneCodec(codec));
1543 EXPECT_TRUE(WaitAndSendFrame(0)); // Should be rendered.
1544 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1545 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be rendered.
1546 frame_count += 2;
1547 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1548
1549 // The channel requires 10 fps.
1550 codec.framerate = 10;
1551 EXPECT_TRUE(SetOneCodec(codec));
1552 EXPECT_TRUE(WaitAndSendFrame(0)); // Should be rendered.
1553 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1554 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1555 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be rendered.
1556 frame_count += 2;
1557 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1558
1559 // The channel requires 8 fps. The adapter adapts to 10 fps, which is the
1560 // closest factor of 30.
1561 codec.framerate = 8;
1562 EXPECT_TRUE(SetOneCodec(codec));
1563 EXPECT_TRUE(WaitAndSendFrame(0)); // Should be rendered.
1564 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1565 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1566 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be rendered.
1567 frame_count += 2;
1568 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1569 }
pbos@webrtc.orgc4175b92014-09-03 15:25:49 +00001570 // Tests that adapted frames won't be upscaled to a higher resolution.
1571 void SendsLowerResolutionOnSmallerFrames() {
1572 cricket::VideoCodec codec = DefaultCodec();
1573 codec.width = 320;
1574 codec.height = 240;
1575 EXPECT_TRUE(SetOneCodec(codec));
1576 EXPECT_TRUE(SetSend(true));
1577 EXPECT_TRUE(channel_->SetRender(true));
1578 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
1579 EXPECT_EQ(0, renderer_.num_rendered_frames());
1580 EXPECT_TRUE(SendFrame());
1581 EXPECT_FRAME_WAIT(1, codec.width, codec.height, kTimeout);
1582
1583 // Check that we send smaller frames at the new resolution.
pbos@webrtc.orgebee4012014-09-03 15:52:02 +00001584 EXPECT_TRUE(rtc::Thread::Current()->ProcessMessages(33));
pbos@webrtc.orgc4175b92014-09-03 15:25:49 +00001585 EXPECT_TRUE(video_capturer_->CaptureCustomFrame(
1586 codec.width / 2, codec.height / 2, cricket::FOURCC_I420));
1587 EXPECT_FRAME_WAIT(2, codec.width / 2, codec.height / 2, kTimeout);
1588 }
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001589 // Tests that we can set the send stream format properly.
1590 void SetSendStreamFormat() {
1591 cricket::VideoCodec codec(DefaultCodec());
1592 SendAndReceive(codec);
1593 int frame_count = 1;
1594 EXPECT_FRAME_WAIT(frame_count, codec.width, codec.height, kTimeout);
1595
1596 // Adapt the resolution and frame rate to half.
1597 cricket::VideoFormat format(
1598 codec.width / 2,
1599 codec.height / 2,
1600 cricket::VideoFormat::FpsToInterval(codec.framerate / 2),
1601 cricket::FOURCC_I420);
1602 // The SSRC differs from the send SSRC.
1603 EXPECT_FALSE(channel_->SetSendStreamFormat(kSsrc - 1, format));
1604 EXPECT_TRUE(channel_->SetSendStreamFormat(kSsrc, format));
1605
1606 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1607 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be rendered.
1608 EXPECT_TRUE(WaitAndSendFrame(30)); // Should be dropped.
1609 frame_count += 1;
1610 EXPECT_FRAME_WAIT(frame_count, format.width, format.height, kTimeout);
1611
1612 // Adapt the resolution to 0x0, which should drop all frames.
1613 format.width = 0;
1614 format.height = 0;
1615 EXPECT_TRUE(channel_->SetSendStreamFormat(kSsrc, format));
1616 EXPECT_TRUE(SendFrame());
1617 EXPECT_TRUE(SendFrame());
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001618 rtc::Thread::Current()->ProcessMessages(500);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001619 EXPECT_EQ(frame_count, renderer_.num_rendered_frames());
1620 }
1621 // Test that setting send stream format to 0x0 resolution will result in
1622 // frames being dropped.
1623 void SetSendStreamFormat0x0() {
1624 EXPECT_TRUE(SetOneCodec(DefaultCodec()));
henrike@webrtc.orga7b98182014-02-21 15:51:43 +00001625 EXPECT_TRUE(SetSendStreamFormat(kSsrc, DefaultCodec()));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001626 EXPECT_TRUE(SetSend(true));
1627 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001628 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001629 EXPECT_EQ(0, renderer_.num_rendered_frames());
1630 // This frame should be received.
1631 EXPECT_TRUE(SendFrame());
1632 EXPECT_FRAME_WAIT(1, DefaultCodec().width, DefaultCodec().height, kTimeout);
1633 const int64 interval = cricket::VideoFormat::FpsToInterval(
1634 DefaultCodec().framerate);
1635 cricket::VideoFormat format(
1636 0,
1637 0,
1638 interval,
1639 cricket::FOURCC_I420);
1640 EXPECT_TRUE(channel_->SetSendStreamFormat(kSsrc, format));
1641 // This frame should not be received.
1642 EXPECT_TRUE(WaitAndSendFrame(
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001643 static_cast<int>(interval/rtc::kNumNanosecsPerMillisec)));
1644 rtc::Thread::Current()->ProcessMessages(500);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001645 EXPECT_EQ(1, renderer_.num_rendered_frames());
1646 }
1647
1648 // Tests that we can mute and unmute the channel properly.
1649 void MuteStream() {
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001650 EXPECT_TRUE(SetDefaultCodec());
1651 cricket::FakeVideoCapturer video_capturer;
1652 video_capturer.Start(
1653 cricket::VideoFormat(
1654 640, 480,
1655 cricket::VideoFormat::FpsToInterval(30),
1656 cricket::FOURCC_I420));
1657 EXPECT_TRUE(channel_->SetCapturer(kSsrc, &video_capturer));
1658 EXPECT_TRUE(SetSend(true));
1659 EXPECT_TRUE(channel_->SetRender(true));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001660 EXPECT_TRUE(channel_->SetRenderer(kDefaultReceiveSsrc, &renderer_));
1661 EXPECT_EQ(0, renderer_.num_rendered_frames());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001662 // Mute the channel and expect black output frame.
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001663 int frame_count = 0;
solenberg1dd98f32015-09-10 01:57:14 -07001664 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, true, nullptr));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001665 EXPECT_TRUE(video_capturer.CaptureFrame());
1666 ++frame_count;
1667 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1668 EXPECT_TRUE(renderer_.black_frame());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001669 // Unmute the channel and expect non-black output frame.
solenberg1dd98f32015-09-10 01:57:14 -07001670 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, false, nullptr));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001671 EXPECT_TRUE(rtc::Thread::Current()->ProcessMessages(30));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001672 EXPECT_TRUE(video_capturer.CaptureFrame());
1673 ++frame_count;
1674 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1675 EXPECT_FALSE(renderer_.black_frame());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001676 // Test that we can also Mute using the correct send stream SSRC.
solenberg1dd98f32015-09-10 01:57:14 -07001677 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, true, nullptr));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001678 EXPECT_TRUE(rtc::Thread::Current()->ProcessMessages(30));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001679 EXPECT_TRUE(video_capturer.CaptureFrame());
1680 ++frame_count;
1681 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1682 EXPECT_TRUE(renderer_.black_frame());
solenberg1dd98f32015-09-10 01:57:14 -07001683 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, false, nullptr));
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001684 EXPECT_TRUE(rtc::Thread::Current()->ProcessMessages(30));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001685 EXPECT_TRUE(video_capturer.CaptureFrame());
1686 ++frame_count;
1687 EXPECT_EQ_WAIT(frame_count, renderer_.num_rendered_frames(), kTimeout);
1688 EXPECT_FALSE(renderer_.black_frame());
pbos@webrtc.orgef8bb8d2014-08-13 21:36:18 +00001689 // Test that muting an existing stream succeeds even if it's muted.
solenberg1dd98f32015-09-10 01:57:14 -07001690 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, true, nullptr));
1691 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, true, nullptr));
pbos@webrtc.orgef8bb8d2014-08-13 21:36:18 +00001692 // Test that unmuting an existing stream succeeds even if it's not muted.
solenberg1dd98f32015-09-10 01:57:14 -07001693 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, false, nullptr));
1694 EXPECT_TRUE(channel_->SetVideoSend(kSsrc, false, nullptr));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001695 // Test that muting an invalid stream fails.
solenberg1dd98f32015-09-10 01:57:14 -07001696 EXPECT_FALSE(channel_->SetVideoSend(kSsrc+1, true, nullptr));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001697 EXPECT_TRUE(channel_->SetCapturer(kSsrc, NULL));
1698 }
1699
1700 // Test that multiple send streams can be created and deleted properly.
1701 void MultipleSendStreams() {
1702 // Remove stream added in Setup. I.e. remove stream corresponding to default
1703 // channel.
1704 EXPECT_TRUE(channel_->RemoveSendStream(kSsrc));
1705 const unsigned int kSsrcsSize = sizeof(kSsrcs4)/sizeof(kSsrcs4[0]);
1706 for (unsigned int i = 0; i < kSsrcsSize; ++i) {
1707 EXPECT_TRUE(channel_->AddSendStream(
1708 cricket::StreamParams::CreateLegacy(kSsrcs4[i])));
1709 }
1710 // Delete one of the non default channel streams, let the destructor delete
1711 // the remaining ones.
1712 EXPECT_TRUE(channel_->RemoveSendStream(kSsrcs4[kSsrcsSize - 1]));
1713 // Stream should already be deleted.
1714 EXPECT_FALSE(channel_->RemoveSendStream(kSsrcs4[kSsrcsSize - 1]));
1715 }
1716
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001717 // Two streams one channel tests.
1718
1719 // Tests that we can send and receive frames.
1720 void TwoStreamsSendAndReceive(const cricket::VideoCodec& codec) {
1721 SetUpSecondStream();
1722 // Test sending and receiving on first stream.
1723 SendAndReceive(codec);
1724 // Test sending and receiving on second stream.
1725 EXPECT_EQ_WAIT(1, renderer2_.num_rendered_frames(), kTimeout);
Stefan Holmer586b19b2015-09-18 11:14:31 +02001726 EXPECT_GT(NumRtpPackets(), 0);
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001727 EXPECT_EQ(1, renderer2_.num_rendered_frames());
1728 }
1729
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001730 // Set up 2 streams where the first stream uses the default channel.
1731 // Then disconnect the first stream and verify default channel becomes
1732 // available.
1733 // Then add a new stream with |new_ssrc|. The new stream should re-use the
1734 // default channel.
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001735 void TwoStreamsReUseFirstStream(const cricket::VideoCodec& codec) {
1736 SetUpSecondStream();
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001737 // Default channel used by the first stream.
buildbot@webrtc.org2c0fb052014-08-13 16:47:12 +00001738 EXPECT_EQ(kSsrc, channel_->GetDefaultSendChannelSsrc());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001739 EXPECT_TRUE(channel_->RemoveRecvStream(kSsrc));
1740 EXPECT_FALSE(channel_->RemoveRecvStream(kSsrc));
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001741 EXPECT_TRUE(channel_->RemoveSendStream(kSsrc));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001742 EXPECT_FALSE(channel_->RemoveSendStream(kSsrc));
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001743 // Default channel is no longer used by a stream.
buildbot@webrtc.org2c0fb052014-08-13 16:47:12 +00001744 EXPECT_EQ(0u, channel_->GetDefaultSendChannelSsrc());
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001745 uint32 new_ssrc = kSsrc + 100;
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001746 EXPECT_TRUE(channel_->AddSendStream(
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001747 cricket::StreamParams::CreateLegacy(new_ssrc)));
1748 // Re-use default channel.
buildbot@webrtc.org2c0fb052014-08-13 16:47:12 +00001749 EXPECT_EQ(new_ssrc, channel_->GetDefaultSendChannelSsrc());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001750 EXPECT_FALSE(channel_->AddSendStream(
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001751 cricket::StreamParams::CreateLegacy(new_ssrc)));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001752 EXPECT_TRUE(channel_->AddRecvStream(
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001753 cricket::StreamParams::CreateLegacy(new_ssrc)));
buildbot@webrtc.org99f63082014-07-18 23:31:30 +00001754 EXPECT_TRUE(channel_->SetRenderer(new_ssrc, &renderer_));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001755 EXPECT_FALSE(channel_->AddRecvStream(
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001756 cricket::StreamParams::CreateLegacy(new_ssrc)));
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001757
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001758 EXPECT_TRUE(channel_->SetCapturer(new_ssrc, video_capturer_.get()));
henrike@webrtc.org28654cb2013-07-22 21:07:49 +00001759
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001760 SendAndReceive(codec);
henrike@webrtc.org6e3dbc22014-03-25 17:09:47 +00001761 EXPECT_TRUE(channel_->RemoveSendStream(new_ssrc));
buildbot@webrtc.org2c0fb052014-08-13 16:47:12 +00001762 EXPECT_EQ(0u, channel_->GetDefaultSendChannelSsrc());
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001763 }
1764
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +00001765 // Tests that we can send and receive frames with early receive.
1766 void TwoStreamsSendAndUnsignalledRecv(const cricket::VideoCodec& codec) {
Fredrik Solenbergb071a192015-09-17 16:42:56 +02001767 cricket::VideoSendParameters parameters;
1768 parameters.options.conference_mode.Set(true);
1769 parameters.options.unsignalled_recv_stream_limit.Set(1);
1770 EXPECT_TRUE(channel_->SetSendParameters(parameters));
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +00001771 SetUpSecondStreamWithNoRecv();
1772 // Test sending and receiving on first stream.
1773 EXPECT_TRUE(channel_->SetRender(true));
1774 Send(codec);
1775 EXPECT_EQ_WAIT(2, NumRtpPackets(), kTimeout);
1776 EXPECT_EQ_WAIT(1, renderer_.num_rendered_frames(), kTimeout);
1777 // The first send is not expected to yield frames, because the ssrc
1778 // is not signalled yet. With unsignalled recv enabled, we will drop frames
1779 // instead of packets.
1780 EXPECT_EQ(0, renderer2_.num_rendered_frames());
1781 // Give a chance for the decoder to process before adding the receiver.
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001782 rtc::Thread::Current()->ProcessMessages(100);
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +00001783 // Test sending and receiving on second stream.
1784 EXPECT_TRUE(channel_->AddRecvStream(
1785 cricket::StreamParams::CreateLegacy(kSsrc + 2)));
1786 EXPECT_TRUE(channel_->SetRenderer(kSsrc + 2, &renderer2_));
1787 SendFrame();
1788 EXPECT_EQ_WAIT(2, renderer_.num_rendered_frames(), kTimeout);
1789 EXPECT_EQ(4, NumRtpPackets());
1790 // The second send is expected to yield frame as the ssrc is signalled now.
1791 // Decode should succeed here, though we received the key frame earlier.
1792 // Without early recv, we would have dropped it and decoding would have
1793 // failed.
1794 EXPECT_EQ_WAIT(1, renderer2_.num_rendered_frames(), kTimeout);
1795 }
1796
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +00001797 // Tests that we drop key frames when conference mode is enabled and we
1798 // receive rtp packets on unsignalled streams. Removal of a unsignalled recv
1799 // stream is successful.
1800 void TwoStreamsAddAndRemoveUnsignalledRecv(
1801 const cricket::VideoCodec& codec) {
1802 cricket::VideoOptions vmo;
1803 vmo.conference_mode.Set(true);
1804 vmo.unsignalled_recv_stream_limit.Set(1);
1805 EXPECT_TRUE(channel_->SetOptions(vmo));
1806 SetUpSecondStreamWithNoRecv();
1807 // Sending and receiving on first stream.
1808 EXPECT_TRUE(channel_->SetRender(true));
1809 Send(codec);
1810 EXPECT_EQ_WAIT(2, NumRtpPackets(), kTimeout);
1811 EXPECT_EQ_WAIT(1, renderer_.num_rendered_frames(), kTimeout);
1812 // The first send is not expected to yield frames, because the ssrc
1813 // is no signalled yet. With unsignalled recv enabled, we will drop frames
1814 // instead of packets.
1815 EXPECT_EQ(0, renderer2_.num_rendered_frames());
1816 // Give a chance for the decoder to process before adding the receiver.
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001817 rtc::Thread::Current()->ProcessMessages(100);
henrike@webrtc.org704bf9e2014-02-27 17:52:04 +00001818 // Ensure that we can remove the unsignalled recv stream that was created
1819 // when the first video packet with unsignalled recv ssrc is received.
1820 EXPECT_TRUE(channel_->RemoveRecvStream(kSsrc + 2));
1821 }
1822
Fredrik Solenberg709ed672015-09-15 12:26:33 +02001823 const rtc::scoped_ptr<webrtc::Call> call_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001824 VideoEngineOverride<E> engine_;
buildbot@webrtc.orgd4e598d2014-07-29 17:36:52 +00001825 rtc::scoped_ptr<cricket::FakeVideoCapturer> video_capturer_;
1826 rtc::scoped_ptr<cricket::FakeVideoCapturer> video_capturer_2_;
1827 rtc::scoped_ptr<C> channel_;
henrike@webrtc.org28e20752013-07-10 00:45:36 +00001828 cricket::FakeNetworkInterface network_interface_;
1829 cricket::FakeVideoRenderer renderer_;
1830 cricket::VideoMediaChannel::Error media_error_;
1831
1832 // Used by test cases where 2 streams are run on the same channel.
1833 cricket::FakeVideoRenderer renderer2_;
1834};
1835
wu@webrtc.orgcadf9042013-08-30 21:24:16 +00001836#endif // TALK_MEDIA_BASE_VIDEOENGINE_UNITTEST_H_ NOLINT