OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2012 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 | |
11 #include <memory> | |
12 #include <string> | |
13 | |
14 #include "webrtc/api/test/fakevideotrackrenderer.h" | |
15 #include "webrtc/api/videocapturertracksource.h" | |
16 #include "webrtc/api/videotrack.h" | |
17 #include "webrtc/base/gunit.h" | |
18 #include "webrtc/media/base/fakevideocapturer.h" | |
19 #include "webrtc/media/base/fakemediaengine.h" | |
20 | |
21 using webrtc::FakeVideoTrackRenderer; | |
22 using webrtc::MediaSourceInterface; | |
23 using webrtc::MediaStreamTrackInterface; | |
24 using webrtc::VideoTrackSource; | |
25 using webrtc::VideoTrack; | |
26 using webrtc::VideoTrackInterface; | |
27 | |
28 class VideoTrackTest : public testing::Test { | |
29 public: | |
30 VideoTrackTest() { | |
31 static const char kVideoTrackId[] = "track_id"; | |
32 video_track_source_ = new rtc::RefCountedObject<VideoTrackSource>( | |
33 &capturer_, true /* remote */); | |
34 video_track_ = VideoTrack::Create(kVideoTrackId, video_track_source_); | |
35 capturer_.Start( | |
36 cricket::VideoFormat(640, 480, cricket::VideoFormat::FpsToInterval(30), | |
37 cricket::FOURCC_I420)); | |
38 } | |
39 | |
40 protected: | |
41 cricket::FakeVideoCapturer capturer_; | |
42 rtc::scoped_refptr<VideoTrackSource> video_track_source_; | |
43 rtc::scoped_refptr<VideoTrackInterface> video_track_; | |
44 }; | |
45 | |
46 // Test changing the source state also changes the track state. | |
47 TEST_F(VideoTrackTest, SourceStateChangeTrackState) { | |
48 EXPECT_EQ(MediaStreamTrackInterface::kLive, video_track_->state()); | |
49 video_track_source_->SetState(MediaSourceInterface::kEnded); | |
50 EXPECT_EQ(MediaStreamTrackInterface::kEnded, video_track_->state()); | |
51 } | |
52 | |
53 // Test adding renderers to a video track and render to them by providing | |
54 // frames to the source. | |
55 TEST_F(VideoTrackTest, RenderVideo) { | |
56 // FakeVideoTrackRenderer register itself to |video_track_| | |
57 std::unique_ptr<FakeVideoTrackRenderer> renderer_1( | |
58 new FakeVideoTrackRenderer(video_track_.get())); | |
59 | |
60 capturer_.CaptureFrame(); | |
61 EXPECT_EQ(1, renderer_1->num_rendered_frames()); | |
62 | |
63 // FakeVideoTrackRenderer register itself to |video_track_| | |
64 std::unique_ptr<FakeVideoTrackRenderer> renderer_2( | |
65 new FakeVideoTrackRenderer(video_track_.get())); | |
66 capturer_.CaptureFrame(); | |
67 EXPECT_EQ(2, renderer_1->num_rendered_frames()); | |
68 EXPECT_EQ(1, renderer_2->num_rendered_frames()); | |
69 | |
70 renderer_1.reset(nullptr); | |
71 capturer_.CaptureFrame(); | |
72 EXPECT_EQ(2, renderer_2->num_rendered_frames()); | |
73 } | |
74 | |
75 // Test that disabling the track results in blacked out frames. | |
76 TEST_F(VideoTrackTest, DisableTrackBlackout) { | |
77 std::unique_ptr<FakeVideoTrackRenderer> renderer( | |
78 new FakeVideoTrackRenderer(video_track_.get())); | |
79 | |
80 capturer_.CaptureFrame(); | |
81 EXPECT_EQ(1, renderer->num_rendered_frames()); | |
82 EXPECT_FALSE(renderer->black_frame()); | |
83 | |
84 video_track_->set_enabled(false); | |
85 capturer_.CaptureFrame(); | |
86 EXPECT_EQ(2, renderer->num_rendered_frames()); | |
87 EXPECT_TRUE(renderer->black_frame()); | |
88 | |
89 video_track_->set_enabled(true); | |
90 capturer_.CaptureFrame(); | |
91 EXPECT_EQ(3, renderer->num_rendered_frames()); | |
92 EXPECT_FALSE(renderer->black_frame()); | |
93 } | |
OLD | NEW |