OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (c) 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 "webrtc/media/base/capturemanager.h" | |
12 | |
13 #include "webrtc/base/arraysize.h" | |
14 #include "webrtc/base/gunit.h" | |
15 #include "webrtc/base/sigslot.h" | |
16 #include "webrtc/media/base/fakevideocapturer.h" | |
17 #include "webrtc/media/base/fakevideorenderer.h" | |
18 | |
19 const int kMsCallbackWait = 50; | |
20 | |
21 const int kFps = 30; | |
22 const cricket::VideoFormatPod kCameraFormats[] = { | |
23 {640, 480, cricket::VideoFormat::FpsToInterval(kFps), cricket::FOURCC_I420}, | |
24 {320, 240, cricket::VideoFormat::FpsToInterval(kFps), cricket::FOURCC_I420} | |
25 }; | |
26 | |
27 class CaptureManagerTest : public ::testing::Test, public sigslot::has_slots<> { | |
28 public: | |
29 CaptureManagerTest() | |
30 : capture_manager_(), | |
31 callback_count_(0), | |
32 format_vga_(kCameraFormats[0]), | |
33 format_qvga_(kCameraFormats[1]) { | |
34 } | |
35 virtual void SetUp() { | |
36 PopulateSupportedFormats(); | |
37 capture_state_ = cricket::CS_STOPPED; | |
38 capture_manager_.SignalCapturerStateChange.connect( | |
39 this, | |
40 &CaptureManagerTest::OnCapturerStateChange); | |
41 } | |
42 void PopulateSupportedFormats() { | |
43 std::vector<cricket::VideoFormat> formats; | |
44 for (uint32_t i = 0; i < arraysize(kCameraFormats); ++i) { | |
45 formats.push_back(cricket::VideoFormat(kCameraFormats[i])); | |
46 } | |
47 video_capturer_.ResetSupportedFormats(formats); | |
48 } | |
49 int NumFramesRendered() { return video_renderer_.num_rendered_frames(); } | |
50 bool WasRenderedResolution(cricket::VideoFormat format) { | |
51 return format.width == video_renderer_.width() && | |
52 format.height == video_renderer_.height(); | |
53 } | |
54 cricket::CaptureState capture_state() { return capture_state_; } | |
55 int callback_count() { return callback_count_; } | |
56 void OnCapturerStateChange(cricket::VideoCapturer* capturer, | |
57 cricket::CaptureState capture_state) { | |
58 capture_state_ = capture_state; | |
59 ++callback_count_; | |
60 } | |
61 | |
62 protected: | |
63 cricket::FakeVideoCapturer video_capturer_; | |
64 cricket::FakeVideoRenderer video_renderer_; | |
65 | |
66 cricket::CaptureManager capture_manager_; | |
67 | |
68 cricket::CaptureState capture_state_; | |
69 int callback_count_; | |
70 cricket::VideoFormat format_vga_; | |
71 cricket::VideoFormat format_qvga_; | |
72 }; | |
73 | |
74 // Incorrect use cases. | |
75 TEST_F(CaptureManagerTest, InvalidAddingRemoving) { | |
76 EXPECT_FALSE(capture_manager_.StopVideoCapture(&video_capturer_, | |
77 cricket::VideoFormat())); | |
78 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
79 format_vga_)); | |
80 EXPECT_EQ_WAIT(cricket::CS_RUNNING, capture_state(), kMsCallbackWait); | |
81 EXPECT_EQ(1, callback_count()); | |
82 // NULL argument currently allowed, and does nothing. | |
83 capture_manager_.AddVideoSink(&video_capturer_, NULL); | |
84 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, format_vga_)); | |
85 } | |
86 | |
87 // Valid use cases | |
88 TEST_F(CaptureManagerTest, KeepFirstResolutionHigh) { | |
89 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
90 format_vga_)); | |
91 EXPECT_EQ_WAIT(cricket::CS_RUNNING, capture_state(), kMsCallbackWait); | |
92 EXPECT_EQ(1, callback_count()); | |
93 capture_manager_.AddVideoSink(&video_capturer_, &video_renderer_); | |
94 EXPECT_TRUE(video_capturer_.CaptureFrame()); | |
95 EXPECT_EQ(1, NumFramesRendered()); | |
96 // Renderer should be fed frames with the resolution of format_vga_. | |
97 EXPECT_TRUE(WasRenderedResolution(format_vga_)); | |
98 | |
99 // Start again with one more format. | |
100 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
101 format_qvga_)); | |
102 // Existing renderers should be fed frames with the resolution of format_vga_. | |
103 EXPECT_TRUE(video_capturer_.CaptureFrame()); | |
104 EXPECT_TRUE(WasRenderedResolution(format_vga_)); | |
105 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, format_vga_)); | |
106 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, | |
107 format_qvga_)); | |
108 EXPECT_FALSE(capture_manager_.StopVideoCapture(&video_capturer_, | |
109 format_vga_)); | |
110 EXPECT_FALSE(capture_manager_.StopVideoCapture(&video_capturer_, | |
111 format_qvga_)); | |
112 } | |
113 | |
114 // Should pick the lowest resolution as the highest resolution is not chosen | |
115 // until after capturing has started. This ensures that no particular resolution | |
116 // is favored over others. | |
117 TEST_F(CaptureManagerTest, KeepFirstResolutionLow) { | |
118 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
119 format_qvga_)); | |
120 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
121 format_vga_)); | |
122 capture_manager_.AddVideoSink(&video_capturer_, &video_renderer_); | |
123 EXPECT_EQ_WAIT(1, callback_count(), kMsCallbackWait); | |
124 EXPECT_TRUE(video_capturer_.CaptureFrame()); | |
125 EXPECT_EQ(1, NumFramesRendered()); | |
126 EXPECT_TRUE(WasRenderedResolution(format_qvga_)); | |
127 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, | |
128 format_qvga_)); | |
129 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, | |
130 format_vga_)); | |
131 } | |
132 | |
133 // Ensure that the reference counting is working when multiple start and | |
134 // multiple stop calls are made. | |
135 TEST_F(CaptureManagerTest, MultipleStartStops) { | |
136 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
137 format_vga_)); | |
138 // Add video capturer but with different format. | |
139 EXPECT_TRUE(capture_manager_.StartVideoCapture(&video_capturer_, | |
140 format_qvga_)); | |
141 EXPECT_EQ_WAIT(cricket::CS_RUNNING, capture_state(), kMsCallbackWait); | |
142 EXPECT_EQ(1, callback_count()); | |
143 capture_manager_.AddVideoSink(&video_capturer_, &video_renderer_); | |
144 // Ensure that a frame can be captured when two start calls have been made. | |
145 EXPECT_TRUE(video_capturer_.CaptureFrame()); | |
146 EXPECT_EQ(1, NumFramesRendered()); | |
147 | |
148 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, format_vga_)); | |
149 // Video should still render since there has been two start calls but only | |
150 // one stop call. | |
151 EXPECT_TRUE(video_capturer_.CaptureFrame()); | |
152 EXPECT_EQ(2, NumFramesRendered()); | |
153 | |
154 EXPECT_TRUE(capture_manager_.StopVideoCapture(&video_capturer_, | |
155 format_qvga_)); | |
156 EXPECT_EQ_WAIT(cricket::CS_STOPPED, capture_state(), kMsCallbackWait); | |
157 EXPECT_EQ(2, callback_count()); | |
158 // Last stop call should fail as it is one more than the number of start | |
159 // calls. | |
160 EXPECT_FALSE(capture_manager_.StopVideoCapture(&video_capturer_, | |
161 format_vga_)); | |
162 } | |
OLD | NEW |