Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(790)

Side by Side Diff: webrtc/api/videotrack_unittest.cc

Issue 1765423005: Change VideoRtpReceiver to create remote VideoTrack and VideoTrackSource. (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Fixed up the comments. Created 4 years, 9 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « webrtc/api/videotrack.cc ('k') | webrtc/api/videotracksource.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 /* 1 /*
2 * Copyright 2012 The WebRTC project authors. All Rights Reserved. 2 * Copyright 2012 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 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 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 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include <string> 11 #include <string>
12 12
13 #include "webrtc/api/remotevideocapturer.h"
14 #include "webrtc/api/test/fakevideotrackrenderer.h" 13 #include "webrtc/api/test/fakevideotrackrenderer.h"
15 #include "webrtc/api/videocapturertracksource.h" 14 #include "webrtc/api/videocapturertracksource.h"
16 #include "webrtc/api/videotrack.h" 15 #include "webrtc/api/videotrack.h"
17 #include "webrtc/base/gunit.h" 16 #include "webrtc/base/gunit.h"
18 #include "webrtc/base/scoped_ptr.h" 17 #include "webrtc/base/scoped_ptr.h"
18 #include "webrtc/media/base/fakevideocapturer.h"
19 #include "webrtc/media/base/fakemediaengine.h" 19 #include "webrtc/media/base/fakemediaengine.h"
20 #include "webrtc/media/engine/webrtcvideoframe.h" 20 #include "webrtc/media/engine/webrtcvideoframe.h"
21 #include "webrtc/pc/channelmanager.h"
22 21
23 using webrtc::FakeVideoTrackRenderer; 22 using webrtc::FakeVideoTrackRenderer;
24 using webrtc::FakeVideoTrackRendererOld; 23 using webrtc::FakeVideoTrackRendererOld;
25 using webrtc::VideoCapturerTrackSource; 24 using webrtc::VideoTrackSource;
26 using webrtc::VideoTrack; 25 using webrtc::VideoTrack;
27 using webrtc::VideoTrackInterface; 26 using webrtc::VideoTrackInterface;
28 27
29 namespace {
30
31 class WebRtcVideoTestFrame : public cricket::WebRtcVideoFrame {
32 public:
33 using cricket::WebRtcVideoFrame::SetRotation;
34 };
35
36 } // namespace
37 28
38 class VideoTrackTest : public testing::Test { 29 class VideoTrackTest : public testing::Test {
39 public: 30 public:
40 VideoTrackTest() { 31 VideoTrackTest() {
41 static const char kVideoTrackId[] = "track_id"; 32 static const char kVideoTrackId[] = "track_id";
42 video_track_ = VideoTrack::Create( 33 video_track_ = VideoTrack::Create(
43 kVideoTrackId, VideoCapturerTrackSource::Create( 34 kVideoTrackId,
44 rtc::Thread::Current(), 35 new rtc::RefCountedObject<VideoTrackSource>(
45 new webrtc::RemoteVideoCapturer(), NULL, true)); 36 &capturer_, rtc::Thread::Current(), true /* remote */));
37 capturer_.Start(
38 cricket::VideoFormat(640, 480, cricket::VideoFormat::FpsToInterval(30),
39 cricket::FOURCC_I420));
46 } 40 }
47 41
48 protected: 42 protected:
43 cricket::FakeVideoCapturer capturer_;
49 rtc::scoped_refptr<VideoTrackInterface> video_track_; 44 rtc::scoped_refptr<VideoTrackInterface> video_track_;
50 }; 45 };
51 46
52 // Test adding renderers to a video track and render to them by providing 47 // Test adding renderers to a video track and render to them by providing
53 // frames to the source. 48 // frames to the source.
54 TEST_F(VideoTrackTest, RenderVideo) { 49 TEST_F(VideoTrackTest, RenderVideo) {
55 // FakeVideoTrackRenderer register itself to |video_track_| 50 // FakeVideoTrackRenderer register itself to |video_track_|
56 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_1( 51 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_1(
57 new FakeVideoTrackRenderer(video_track_.get())); 52 new FakeVideoTrackRenderer(video_track_.get()));
58 53
59 rtc::VideoSinkInterface<cricket::VideoFrame>* renderer_input = 54 capturer_.CaptureFrame();
60 video_track_->GetSink();
61 ASSERT_FALSE(renderer_input == NULL);
62
63 cricket::WebRtcVideoFrame frame;
64 frame.InitToBlack(123, 123, 0);
65 renderer_input->OnFrame(frame);
66 EXPECT_EQ(1, renderer_1->num_rendered_frames()); 55 EXPECT_EQ(1, renderer_1->num_rendered_frames());
67 56
68 EXPECT_EQ(123, renderer_1->width());
69 EXPECT_EQ(123, renderer_1->height());
70
71 // FakeVideoTrackRenderer register itself to |video_track_| 57 // FakeVideoTrackRenderer register itself to |video_track_|
72 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_2( 58 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_2(
73 new FakeVideoTrackRenderer(video_track_.get())); 59 new FakeVideoTrackRenderer(video_track_.get()));
74 60 capturer_.CaptureFrame();
75 renderer_input->OnFrame(frame);
76
77 EXPECT_EQ(123, renderer_1->width());
78 EXPECT_EQ(123, renderer_1->height());
79 EXPECT_EQ(123, renderer_2->width());
80 EXPECT_EQ(123, renderer_2->height());
81
82 EXPECT_EQ(2, renderer_1->num_rendered_frames()); 61 EXPECT_EQ(2, renderer_1->num_rendered_frames());
83 EXPECT_EQ(1, renderer_2->num_rendered_frames()); 62 EXPECT_EQ(1, renderer_2->num_rendered_frames());
84 63
85 video_track_->RemoveSink(renderer_1.get()); 64 video_track_->RemoveSink(renderer_1.get());
86 renderer_input->OnFrame(frame); 65 capturer_.CaptureFrame();
87
88 EXPECT_EQ(2, renderer_1->num_rendered_frames()); 66 EXPECT_EQ(2, renderer_1->num_rendered_frames());
89 EXPECT_EQ(2, renderer_2->num_rendered_frames()); 67 EXPECT_EQ(2, renderer_2->num_rendered_frames());
90 } 68 }
91 69
92 // Test adding renderers to a video track and render to them by 70 // Test adding renderers to a video track and render to them by
93 // providing frames to the source. Uses the old VideoTrack interface 71 // providing frames to the source. Uses the old VideoTrack interface
94 // with AddRenderer and RemoveRenderer. 72 // with AddRenderer and RemoveRenderer.
95 TEST_F(VideoTrackTest, RenderVideoOld) { 73 TEST_F(VideoTrackTest, RenderVideoOld) {
96 // FakeVideoTrackRenderer register itself to |video_track_| 74 // FakeVideoTrackRenderer register itself to |video_track_|
97 rtc::scoped_ptr<FakeVideoTrackRendererOld> renderer_1( 75 rtc::scoped_ptr<FakeVideoTrackRendererOld> renderer_1(
98 new FakeVideoTrackRendererOld(video_track_.get())); 76 new FakeVideoTrackRendererOld(video_track_.get()));
99 77
100 rtc::VideoSinkInterface<cricket::VideoFrame>* renderer_input = 78 capturer_.CaptureFrame();
101 video_track_->GetSink();
102 ASSERT_FALSE(renderer_input == NULL);
103
104 cricket::WebRtcVideoFrame frame;
105 frame.InitToBlack(123, 123, 0);
106 renderer_input->OnFrame(frame);
107 EXPECT_EQ(1, renderer_1->num_rendered_frames()); 79 EXPECT_EQ(1, renderer_1->num_rendered_frames());
108 80
109 EXPECT_EQ(123, renderer_1->width());
110 EXPECT_EQ(123, renderer_1->height());
111
112 // FakeVideoTrackRenderer register itself to |video_track_| 81 // FakeVideoTrackRenderer register itself to |video_track_|
113 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_2( 82 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer_2(
114 new FakeVideoTrackRenderer(video_track_.get())); 83 new FakeVideoTrackRenderer(video_track_.get()));
115 84
116 renderer_input->OnFrame(frame); 85 capturer_.CaptureFrame();
117
118 EXPECT_EQ(123, renderer_1->width());
119 EXPECT_EQ(123, renderer_1->height());
120 EXPECT_EQ(123, renderer_2->width());
121 EXPECT_EQ(123, renderer_2->height());
122
123 EXPECT_EQ(2, renderer_1->num_rendered_frames()); 86 EXPECT_EQ(2, renderer_1->num_rendered_frames());
124 EXPECT_EQ(1, renderer_2->num_rendered_frames()); 87 EXPECT_EQ(1, renderer_2->num_rendered_frames());
125 88
126 video_track_->RemoveRenderer(renderer_1.get()); 89 video_track_->RemoveRenderer(renderer_1.get());
127 renderer_input->OnFrame(frame); 90 capturer_.CaptureFrame();
128
129 EXPECT_EQ(2, renderer_1->num_rendered_frames()); 91 EXPECT_EQ(2, renderer_1->num_rendered_frames());
130 EXPECT_EQ(2, renderer_2->num_rendered_frames()); 92 EXPECT_EQ(2, renderer_2->num_rendered_frames());
131 } 93 }
132 94
133 // Test that disabling the track results in blacked out frames. 95 // Test that disabling the track results in blacked out frames.
134 TEST_F(VideoTrackTest, DisableTrackBlackout) { 96 TEST_F(VideoTrackTest, DisableTrackBlackout) {
135 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer( 97 rtc::scoped_ptr<FakeVideoTrackRenderer> renderer(
136 new FakeVideoTrackRenderer(video_track_.get())); 98 new FakeVideoTrackRenderer(video_track_.get()));
137 99
138 rtc::VideoSinkInterface<cricket::VideoFrame>* renderer_input = 100 capturer_.CaptureFrame();
139 video_track_->GetSink();
140 ASSERT_FALSE(renderer_input == NULL);
141
142 cricket::WebRtcVideoFrame frame;
143 frame.InitToBlack(100, 200, 0);
144 // Make it not all-black
145 frame.GetUPlane()[0] = 0;
146
147 renderer_input->OnFrame(frame);
148 EXPECT_EQ(1, renderer->num_rendered_frames()); 101 EXPECT_EQ(1, renderer->num_rendered_frames());
149 EXPECT_FALSE(renderer->black_frame()); 102 EXPECT_FALSE(renderer->black_frame());
150 EXPECT_EQ(100, renderer->width());
151 EXPECT_EQ(200, renderer->height());
152 103
153 video_track_->set_enabled(false); 104 video_track_->set_enabled(false);
154 renderer_input->OnFrame(frame); 105 capturer_.CaptureFrame();
155 EXPECT_EQ(2, renderer->num_rendered_frames()); 106 EXPECT_EQ(2, renderer->num_rendered_frames());
156 EXPECT_TRUE(renderer->black_frame()); 107 EXPECT_TRUE(renderer->black_frame());
157 EXPECT_EQ(100, renderer->width());
158 EXPECT_EQ(200, renderer->height());
159 108
160 video_track_->set_enabled(true); 109 video_track_->set_enabled(true);
161 renderer_input->OnFrame(frame); 110 capturer_.CaptureFrame();
162 EXPECT_EQ(3, renderer->num_rendered_frames()); 111 EXPECT_EQ(3, renderer->num_rendered_frames());
163 EXPECT_FALSE(renderer->black_frame()); 112 EXPECT_FALSE(renderer->black_frame());
164 EXPECT_EQ(100, renderer->width());
165 EXPECT_EQ(200, renderer->height());
166 } 113 }
OLDNEW
« no previous file with comments | « webrtc/api/videotrack.cc ('k') | webrtc/api/videotracksource.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698