| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright 2016 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 "webrtc/base/gunit.h" | 11 #include "webrtc/base/gunit.h" |
| 12 #include "webrtc/media/base/fakevideorenderer.h" | 12 #include "webrtc/media/base/fakevideorenderer.h" |
| 13 #include "webrtc/media/base/videobroadcaster.h" | 13 #include "webrtc/media/base/videobroadcaster.h" |
| 14 #include "webrtc/video_frame.h" | 14 #include "webrtc/media/engine/webrtcvideoframe.h" |
| 15 | 15 |
| 16 using rtc::VideoBroadcaster; | 16 using rtc::VideoBroadcaster; |
| 17 using rtc::VideoSinkWants; | 17 using rtc::VideoSinkWants; |
| 18 using cricket::FakeVideoRenderer; | 18 using cricket::FakeVideoRenderer; |
| 19 using cricket::WebRtcVideoFrame; |
| 19 | 20 |
| 20 | 21 |
| 21 TEST(VideoBroadcasterTest, frame_wanted) { | 22 TEST(VideoBroadcasterTest, frame_wanted) { |
| 22 VideoBroadcaster broadcaster; | 23 VideoBroadcaster broadcaster; |
| 23 EXPECT_FALSE(broadcaster.frame_wanted()); | 24 EXPECT_FALSE(broadcaster.frame_wanted()); |
| 24 | 25 |
| 25 FakeVideoRenderer sink; | 26 FakeVideoRenderer sink; |
| 26 broadcaster.AddOrUpdateSink(&sink, rtc::VideoSinkWants()); | 27 broadcaster.AddOrUpdateSink(&sink, rtc::VideoSinkWants()); |
| 27 EXPECT_TRUE(broadcaster.frame_wanted()); | 28 EXPECT_TRUE(broadcaster.frame_wanted()); |
| 28 | 29 |
| 29 broadcaster.RemoveSink(&sink); | 30 broadcaster.RemoveSink(&sink); |
| 30 EXPECT_FALSE(broadcaster.frame_wanted()); | 31 EXPECT_FALSE(broadcaster.frame_wanted()); |
| 31 } | 32 } |
| 32 | 33 |
| 33 TEST(VideoBroadcasterTest, OnFrame) { | 34 TEST(VideoBroadcasterTest, OnFrame) { |
| 34 VideoBroadcaster broadcaster; | 35 VideoBroadcaster broadcaster; |
| 35 | 36 |
| 36 FakeVideoRenderer sink1; | 37 FakeVideoRenderer sink1; |
| 37 FakeVideoRenderer sink2; | 38 FakeVideoRenderer sink2; |
| 38 broadcaster.AddOrUpdateSink(&sink1, rtc::VideoSinkWants()); | 39 broadcaster.AddOrUpdateSink(&sink1, rtc::VideoSinkWants()); |
| 39 broadcaster.AddOrUpdateSink(&sink2, rtc::VideoSinkWants()); | 40 broadcaster.AddOrUpdateSink(&sink2, rtc::VideoSinkWants()); |
| 40 | 41 |
| 41 webrtc::VideoFrame frame; | 42 WebRtcVideoFrame frame; |
| 42 | 43 |
| 43 broadcaster.OnFrame(frame); | 44 broadcaster.OnFrame(frame); |
| 44 EXPECT_EQ(1, sink1.num_rendered_frames()); | 45 EXPECT_EQ(1, sink1.num_rendered_frames()); |
| 45 EXPECT_EQ(1, sink2.num_rendered_frames()); | 46 EXPECT_EQ(1, sink2.num_rendered_frames()); |
| 46 | 47 |
| 47 broadcaster.RemoveSink(&sink1); | 48 broadcaster.RemoveSink(&sink1); |
| 48 broadcaster.OnFrame(frame); | 49 broadcaster.OnFrame(frame); |
| 49 EXPECT_EQ(1, sink1.num_rendered_frames()); | 50 EXPECT_EQ(1, sink1.num_rendered_frames()); |
| 50 EXPECT_EQ(2, sink2.num_rendered_frames()); | 51 EXPECT_EQ(2, sink2.num_rendered_frames()); |
| 51 | 52 |
| (...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 131 FakeVideoRenderer sink2; | 132 FakeVideoRenderer sink2; |
| 132 VideoSinkWants wants2; | 133 VideoSinkWants wants2; |
| 133 wants2.black_frames = false; | 134 wants2.black_frames = false; |
| 134 broadcaster.AddOrUpdateSink(&sink2, wants2); | 135 broadcaster.AddOrUpdateSink(&sink2, wants2); |
| 135 | 136 |
| 136 rtc::scoped_refptr<webrtc::I420Buffer> buffer( | 137 rtc::scoped_refptr<webrtc::I420Buffer> buffer( |
| 137 new rtc::RefCountedObject<webrtc::I420Buffer>(100, 200)); | 138 new rtc::RefCountedObject<webrtc::I420Buffer>(100, 200)); |
| 138 // Makes it not all black. | 139 // Makes it not all black. |
| 139 buffer->InitializeData(); | 140 buffer->InitializeData(); |
| 140 | 141 |
| 141 webrtc::VideoFrame frame1(buffer, webrtc::kVideoRotation_0, | 142 cricket::WebRtcVideoFrame frame1(buffer, webrtc::kVideoRotation_0, |
| 142 10 /* timestamp_us */); | 143 10 /* timestamp_us */); |
| 143 broadcaster.OnFrame(frame1); | 144 broadcaster.OnFrame(frame1); |
| 144 EXPECT_TRUE(sink1.black_frame()); | 145 EXPECT_TRUE(sink1.black_frame()); |
| 145 EXPECT_EQ(10, sink1.timestamp_us()); | 146 EXPECT_EQ(10, sink1.timestamp_us()); |
| 146 EXPECT_FALSE(sink2.black_frame()); | 147 EXPECT_FALSE(sink2.black_frame()); |
| 147 EXPECT_EQ(10, sink2.timestamp_us()); | 148 EXPECT_EQ(10, sink2.timestamp_us()); |
| 148 | 149 |
| 149 // Switch the sink wants. | 150 // Switch the sink wants. |
| 150 wants1.black_frames = false; | 151 wants1.black_frames = false; |
| 151 broadcaster.AddOrUpdateSink(&sink1, wants1); | 152 broadcaster.AddOrUpdateSink(&sink1, wants1); |
| 152 wants2.black_frames = true; | 153 wants2.black_frames = true; |
| 153 broadcaster.AddOrUpdateSink(&sink2, wants2); | 154 broadcaster.AddOrUpdateSink(&sink2, wants2); |
| 154 | 155 |
| 155 webrtc::VideoFrame frame2(buffer, webrtc::kVideoRotation_0, | 156 cricket::WebRtcVideoFrame frame2(buffer, webrtc::kVideoRotation_0, |
| 156 30 /* timestamp_us */); | 157 30 /* timestamp_us */); |
| 157 broadcaster.OnFrame(frame2); | 158 broadcaster.OnFrame(frame2); |
| 158 EXPECT_FALSE(sink1.black_frame()); | 159 EXPECT_FALSE(sink1.black_frame()); |
| 159 EXPECT_EQ(30, sink1.timestamp_us()); | 160 EXPECT_EQ(30, sink1.timestamp_us()); |
| 160 EXPECT_TRUE(sink2.black_frame()); | 161 EXPECT_TRUE(sink2.black_frame()); |
| 161 EXPECT_EQ(30, sink2.timestamp_us()); | 162 EXPECT_EQ(30, sink2.timestamp_us()); |
| 162 } | 163 } |
| OLD | NEW |