henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 1 | /* |
kjellander | b24317b | 2016-02-10 15:54:43 | [diff] [blame] | 2 | * Copyright 2012 The WebRTC project authors. All Rights Reserved. |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 3 | * |
kjellander | b24317b | 2016-02-10 15:54:43 | [diff] [blame] | 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 9 | */ |
| 10 | |
Jonas Olsson | a4d8737 | 2019-07-05 17:08:33 | [diff] [blame] | 11 | #include "pc/video_track.h" |
| 12 | |
kwiberg | d1fe281 | 2016-04-27 13:47:29 | [diff] [blame] | 13 | #include <memory> |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 14 | |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 15 | #include "media/base/fake_frame_source.h" |
Steve Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 16 | #include "pc/test/fake_video_track_renderer.h" |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 17 | #include "pc/test/fake_video_track_source.h" |
Steve Anton | 10542f2 | 2019-01-11 17:11:00 | [diff] [blame] | 18 | #include "pc/video_track_source.h" |
Harald Alvestrand | c24a218 | 2022-02-23 13:44:59 | [diff] [blame] | 19 | #include "rtc_base/time_utils.h" |
Yves Gerey | 3e70781 | 2018-11-28 15:47:49 | [diff] [blame] | 20 | #include "test/gtest.h" |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 21 | |
| 22 | using webrtc::FakeVideoTrackRenderer; |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 23 | using webrtc::FakeVideoTrackSource; |
perkj | c8f952d | 2016-03-23 07:33:56 | [diff] [blame] | 24 | using webrtc::MediaSourceInterface; |
| 25 | using webrtc::MediaStreamTrackInterface; |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 26 | using webrtc::VideoTrack; |
| 27 | using webrtc::VideoTrackInterface; |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 28 | using webrtc::VideoTrackSource; |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 29 | |
Mirko Bonadei | 6a489f2 | 2019-04-09 13:11:12 | [diff] [blame] | 30 | class VideoTrackTest : public ::testing::Test { |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 31 | public: |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 32 | VideoTrackTest() : frame_source_(640, 480, rtc::kNumMicrosecsPerSec / 30) { |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 33 | static const char kVideoTrackId[] = "track_id"; |
Tommi | 87f7090 | 2021-04-27 12:43:08 | [diff] [blame] | 34 | video_track_source_ = rtc::make_ref_counted<FakeVideoTrackSource>( |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 35 | /*is_screencast=*/false); |
perkj | 773be36 | 2017-08-01 06:22:01 | [diff] [blame] | 36 | video_track_ = VideoTrack::Create(kVideoTrackId, video_track_source_, |
| 37 | rtc::Thread::Current()); |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 38 | } |
| 39 | |
| 40 | protected: |
Niels Möller | 83830f3 | 2022-05-20 07:12:57 | [diff] [blame] | 41 | rtc::AutoThread main_thread_; |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 42 | rtc::scoped_refptr<FakeVideoTrackSource> video_track_source_; |
Tommi | 34cd1d3 | 2022-02-17 12:28:57 | [diff] [blame] | 43 | rtc::scoped_refptr<VideoTrack> video_track_; |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 44 | cricket::FakeFrameSource frame_source_; |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 45 | }; |
| 46 | |
Tommi | 34cd1d3 | 2022-02-17 12:28:57 | [diff] [blame] | 47 | // 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. |
| 51 | TEST_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 | |
perkj | c8f952d | 2016-03-23 07:33:56 | [diff] [blame] | 56 | // Test changing the source state also changes the track state. |
| 57 | TEST_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.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 63 | // Test adding renderers to a video track and render to them by providing |
wu@webrtc.org | 967bfff | 2013-09-19 05:49:50 | [diff] [blame] | 64 | // frames to the source. |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 65 | TEST_F(VideoTrackTest, RenderVideo) { |
Artem Titov | 880fa81 | 2021-07-30 20:30:23 | [diff] [blame] | 66 | // FakeVideoTrackRenderer register itself to `video_track_` |
kwiberg | d1fe281 | 2016-04-27 13:47:29 | [diff] [blame] | 67 | std::unique_ptr<FakeVideoTrackRenderer> renderer_1( |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 68 | new FakeVideoTrackRenderer(video_track_.get())); |
guoweis@webrtc.org | f9a75d9 | 2015-03-10 06:37:00 | [diff] [blame] | 69 | |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 70 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 71 | EXPECT_EQ(1, renderer_1->num_rendered_frames()); |
| 72 | |
Artem Titov | 880fa81 | 2021-07-30 20:30:23 | [diff] [blame] | 73 | // FakeVideoTrackRenderer register itself to `video_track_` |
kwiberg | d1fe281 | 2016-04-27 13:47:29 | [diff] [blame] | 74 | std::unique_ptr<FakeVideoTrackRenderer> renderer_2( |
guoweis@webrtc.org | 00c509a | 2015-03-12 21:37:26 | [diff] [blame] | 75 | new FakeVideoTrackRenderer(video_track_.get())); |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 76 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
henrike@webrtc.org | 28e2075 | 2013-07-10 00:45:36 | [diff] [blame] | 77 | EXPECT_EQ(2, renderer_1->num_rendered_frames()); |
| 78 | EXPECT_EQ(1, renderer_2->num_rendered_frames()); |
| 79 | |
perkj | d6c3954 | 2016-03-17 09:35:23 | [diff] [blame] | 80 | renderer_1.reset(nullptr); |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 81 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
nisse | db25d2e | 2016-02-26 09:24:58 | [diff] [blame] | 82 | EXPECT_EQ(2, renderer_2->num_rendered_frames()); |
| 83 | } |
| 84 | |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 85 | // Test that disabling the track results in blacked out frames. |
| 86 | TEST_F(VideoTrackTest, DisableTrackBlackout) { |
kwiberg | d1fe281 | 2016-04-27 13:47:29 | [diff] [blame] | 87 | std::unique_ptr<FakeVideoTrackRenderer> renderer( |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 88 | new FakeVideoTrackRenderer(video_track_.get())); |
| 89 | |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 90 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 91 | EXPECT_EQ(1, renderer->num_rendered_frames()); |
| 92 | EXPECT_FALSE(renderer->black_frame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 93 | |
| 94 | video_track_->set_enabled(false); |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 95 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 96 | EXPECT_EQ(2, renderer->num_rendered_frames()); |
| 97 | EXPECT_TRUE(renderer->black_frame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 98 | |
| 99 | video_track_->set_enabled(true); |
Niels Möller | dd6633b | 2019-01-18 13:23:12 | [diff] [blame] | 100 | video_track_source_->InjectFrame(frame_source_.GetFrame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 101 | EXPECT_EQ(3, renderer->num_rendered_frames()); |
| 102 | EXPECT_FALSE(renderer->black_frame()); |
nisse | 81354f5 | 2016-01-19 08:23:24 | [diff] [blame] | 103 | } |