blob: e75fd034b3715fbc75dd5bb03eb2f825211107b2 [file] [log] [blame]
henrike@webrtc.org28e20752013-07-10 00:45:361/*
kjellanderb24317b2016-02-10 15:54:432 * Copyright 2012 The WebRTC project authors. All Rights Reserved.
henrike@webrtc.org28e20752013-07-10 00:45:363 *
kjellanderb24317b2016-02-10 15:54:434 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
henrike@webrtc.org28e20752013-07-10 00:45:369 */
10
Jonas Olssona4d87372019-07-05 17:08:3311#include "pc/video_track.h"
12
kwibergd1fe2812016-04-27 13:47:2913#include <memory>
henrike@webrtc.org28e20752013-07-10 00:45:3614
Niels Möllerdd6633b2019-01-18 13:23:1215#include "media/base/fake_frame_source.h"
Steve Anton10542f22019-01-11 17:11:0016#include "pc/test/fake_video_track_renderer.h"
Niels Möllerdd6633b2019-01-18 13:23:1217#include "pc/test/fake_video_track_source.h"
Steve Anton10542f22019-01-11 17:11:0018#include "pc/video_track_source.h"
Harald Alvestrandc24a2182022-02-23 13:44:5919#include "rtc_base/time_utils.h"
Yves Gerey3e707812018-11-28 15:47:4920#include "test/gtest.h"
henrike@webrtc.org28e20752013-07-10 00:45:3621
22using webrtc::FakeVideoTrackRenderer;
Niels Möllerdd6633b2019-01-18 13:23:1223using webrtc::FakeVideoTrackSource;
perkjc8f952d2016-03-23 07:33:5624using webrtc::MediaSourceInterface;
25using webrtc::MediaStreamTrackInterface;
henrike@webrtc.org28e20752013-07-10 00:45:3626using webrtc::VideoTrack;
27using webrtc::VideoTrackInterface;
Niels Möllerdd6633b2019-01-18 13:23:1228using webrtc::VideoTrackSource;
henrike@webrtc.org28e20752013-07-10 00:45:3629
Mirko Bonadei6a489f22019-04-09 13:11:1230class VideoTrackTest : public ::testing::Test {
guoweis@webrtc.org00c509a2015-03-12 21:37:2631 public:
Niels Möllerdd6633b2019-01-18 13:23:1232 VideoTrackTest() : frame_source_(640, 480, rtc::kNumMicrosecsPerSec / 30) {
guoweis@webrtc.org00c509a2015-03-12 21:37:2633 static const char kVideoTrackId[] = "track_id";
Tommi87f70902021-04-27 12:43:0834 video_track_source_ = rtc::make_ref_counted<FakeVideoTrackSource>(
Niels Möllerdd6633b2019-01-18 13:23:1235 /*is_screencast=*/false);
perkj773be362017-08-01 06:22:0136 video_track_ = VideoTrack::Create(kVideoTrackId, video_track_source_,
37 rtc::Thread::Current());
guoweis@webrtc.org00c509a2015-03-12 21:37:2638 }
39
40 protected:
Niels Möller83830f32022-05-20 07:12:5741 rtc::AutoThread main_thread_;
Niels Möllerdd6633b2019-01-18 13:23:1242 rtc::scoped_refptr<FakeVideoTrackSource> video_track_source_;
Tommi34cd1d32022-02-17 12:28:5743 rtc::scoped_refptr<VideoTrack> video_track_;
Niels Möllerdd6633b2019-01-18 13:23:1244 cricket::FakeFrameSource frame_source_;
guoweis@webrtc.org00c509a2015-03-12 21:37:2645};
46
Tommi34cd1d32022-02-17 12:28:5747// VideoTrack::Create will create an API proxy around the source object.
48// The `GetSource` method provides access to the proxy object intented for API
49// use while the GetSourceInternal() provides direct access to the source object
50// as provided to the `VideoTrack::Create` factory function.
51TEST_F(VideoTrackTest, CheckApiProxyAndInternalSource) {
52 EXPECT_NE(video_track_->GetSource(), video_track_source_.get());
53 EXPECT_EQ(video_track_->GetSourceInternal(), video_track_source_.get());
54}
55
perkjc8f952d2016-03-23 07:33:5656// Test changing the source state also changes the track state.
57TEST_F(VideoTrackTest, SourceStateChangeTrackState) {
58 EXPECT_EQ(MediaStreamTrackInterface::kLive, video_track_->state());
59 video_track_source_->SetState(MediaSourceInterface::kEnded);
60 EXPECT_EQ(MediaStreamTrackInterface::kEnded, video_track_->state());
61}
62
henrike@webrtc.org28e20752013-07-10 00:45:3663// Test adding renderers to a video track and render to them by providing
wu@webrtc.org967bfff2013-09-19 05:49:5064// frames to the source.
guoweis@webrtc.org00c509a2015-03-12 21:37:2665TEST_F(VideoTrackTest, RenderVideo) {
Artem Titov880fa812021-07-30 20:30:2366 // FakeVideoTrackRenderer register itself to `video_track_`
kwibergd1fe2812016-04-27 13:47:2967 std::unique_ptr<FakeVideoTrackRenderer> renderer_1(
guoweis@webrtc.org00c509a2015-03-12 21:37:2668 new FakeVideoTrackRenderer(video_track_.get()));
guoweis@webrtc.orgf9a75d92015-03-10 06:37:0069
Niels Möllerdd6633b2019-01-18 13:23:1270 video_track_source_->InjectFrame(frame_source_.GetFrame());
henrike@webrtc.org28e20752013-07-10 00:45:3671 EXPECT_EQ(1, renderer_1->num_rendered_frames());
72
Artem Titov880fa812021-07-30 20:30:2373 // FakeVideoTrackRenderer register itself to `video_track_`
kwibergd1fe2812016-04-27 13:47:2974 std::unique_ptr<FakeVideoTrackRenderer> renderer_2(
guoweis@webrtc.org00c509a2015-03-12 21:37:2675 new FakeVideoTrackRenderer(video_track_.get()));
Niels Möllerdd6633b2019-01-18 13:23:1276 video_track_source_->InjectFrame(frame_source_.GetFrame());
henrike@webrtc.org28e20752013-07-10 00:45:3677 EXPECT_EQ(2, renderer_1->num_rendered_frames());
78 EXPECT_EQ(1, renderer_2->num_rendered_frames());
79
perkjd6c39542016-03-17 09:35:2380 renderer_1.reset(nullptr);
Niels Möllerdd6633b2019-01-18 13:23:1281 video_track_source_->InjectFrame(frame_source_.GetFrame());
nissedb25d2e2016-02-26 09:24:5882 EXPECT_EQ(2, renderer_2->num_rendered_frames());
83}
84
nisse81354f52016-01-19 08:23:2485// Test that disabling the track results in blacked out frames.
86TEST_F(VideoTrackTest, DisableTrackBlackout) {
kwibergd1fe2812016-04-27 13:47:2987 std::unique_ptr<FakeVideoTrackRenderer> renderer(
nisse81354f52016-01-19 08:23:2488 new FakeVideoTrackRenderer(video_track_.get()));
89
Niels Möllerdd6633b2019-01-18 13:23:1290 video_track_source_->InjectFrame(frame_source_.GetFrame());
nisse81354f52016-01-19 08:23:2491 EXPECT_EQ(1, renderer->num_rendered_frames());
92 EXPECT_FALSE(renderer->black_frame());
nisse81354f52016-01-19 08:23:2493
94 video_track_->set_enabled(false);
Niels Möllerdd6633b2019-01-18 13:23:1295 video_track_source_->InjectFrame(frame_source_.GetFrame());
nisse81354f52016-01-19 08:23:2496 EXPECT_EQ(2, renderer->num_rendered_frames());
97 EXPECT_TRUE(renderer->black_frame());
nisse81354f52016-01-19 08:23:2498
99 video_track_->set_enabled(true);
Niels Möllerdd6633b2019-01-18 13:23:12100 video_track_source_->InjectFrame(frame_source_.GetFrame());
nisse81354f52016-01-19 08:23:24101 EXPECT_EQ(3, renderer->num_rendered_frames());
102 EXPECT_FALSE(renderer->black_frame());
nisse81354f52016-01-19 08:23:24103}