| Index: webrtc/api/rtpsenderreceiver_unittest.cc
|
| diff --git a/webrtc/api/rtpsenderreceiver_unittest.cc b/webrtc/api/rtpsenderreceiver_unittest.cc
|
| index fd1c1f33b1556ec9c2f72c5d2ae65cc22298d29f..4442e524359222489833f2d82e7a1a6ba449427c 100644
|
| --- a/webrtc/api/rtpsenderreceiver_unittest.cc
|
| +++ b/webrtc/api/rtpsenderreceiver_unittest.cc
|
| @@ -70,8 +70,9 @@ class MockAudioProvider : public AudioProviderInterface {
|
| class MockVideoProvider : public VideoProviderInterface {
|
| public:
|
| virtual ~MockVideoProvider() {}
|
| - MOCK_METHOD2(SetCaptureDevice,
|
| - bool(uint32_t ssrc, cricket::VideoCapturer* camera));
|
| + MOCK_METHOD2(SetSource,
|
| + bool(uint32_t ssrc,
|
| + rtc::VideoSourceInterface<cricket::VideoFrame>* source));
|
| MOCK_METHOD3(SetVideoPlayout,
|
| void(uint32_t ssrc,
|
| bool enable,
|
| @@ -112,8 +113,7 @@ class RtpSenderReceiverTest : public testing::Test {
|
| void CreateVideoRtpSender() {
|
| AddVideoTrack();
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(
|
| - kVideoSsrc, video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| video_rtp_sender_ = new VideoRtpSender(stream_->GetVideoTracks()[0],
|
| stream_->label(), &video_provider_);
|
| @@ -127,7 +127,7 @@ class RtpSenderReceiverTest : public testing::Test {
|
| }
|
|
|
| void DestroyVideoRtpSender() {
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, NULL)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, NULL)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| video_rtp_sender_ = nullptr;
|
| }
|
| @@ -345,13 +345,12 @@ TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupSsrcThenTrack) {
|
| new VideoRtpSender(&video_provider_);
|
| sender->SetSsrc(kVideoSsrc);
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| sender->SetTrack(video_track_);
|
|
|
| // Calls expected from destructor.
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| }
|
|
|
| @@ -363,13 +362,12 @@ TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupTrackThenSsrc) {
|
| new VideoRtpSender(&video_provider_);
|
| sender->SetTrack(video_track_);
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| sender->SetSsrc(kVideoSsrc);
|
|
|
| // Calls expected from destructor.
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| }
|
|
|
| @@ -396,20 +394,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcSetToZero) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderSsrcSetToZero) {
|
| AddVideoTrack();
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| rtc::scoped_refptr<VideoRtpSender> sender =
|
| new VideoRtpSender(video_track_, kStreamLabel1, &video_provider_);
|
| sender->SetSsrc(kVideoSsrc);
|
|
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| sender->SetSsrc(0);
|
|
|
| // Make sure it's SetSsrc that called methods on the provider, and not the
|
| // destructor.
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(_, _)).Times(0);
|
| + EXPECT_CALL(video_provider_, SetSource(_, _)).Times(0);
|
| EXPECT_CALL(video_provider_, SetVideoSend(_, _, _)).Times(0);
|
| }
|
|
|
| @@ -432,20 +429,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderTrackSetToNull) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderTrackSetToNull) {
|
| AddVideoTrack();
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| rtc::scoped_refptr<VideoRtpSender> sender =
|
| new VideoRtpSender(video_track_, kStreamLabel1, &video_provider_);
|
| sender->SetSsrc(kVideoSsrc);
|
|
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| EXPECT_TRUE(sender->SetTrack(nullptr));
|
|
|
| // Make sure it's SetTrack that called methods on the provider, and not the
|
| // destructor.
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(_, _)).Times(0);
|
| + EXPECT_CALL(video_provider_, SetSource(_, _)).Times(0);
|
| EXPECT_CALL(video_provider_, SetVideoSend(_, _, _)).Times(0);
|
| }
|
|
|
| @@ -469,23 +465,21 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcChanged) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderSsrcChanged) {
|
| AddVideoTrack();
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _));
|
| rtc::scoped_refptr<VideoRtpSender> sender =
|
| new VideoRtpSender(video_track_, kStreamLabel1, &video_provider_);
|
| sender->SetSsrc(kVideoSsrc);
|
|
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1);
|
| EXPECT_CALL(video_provider_,
|
| - SetCaptureDevice(kVideoSsrc2,
|
| - video_track_->GetSource()->GetVideoCapturer()));
|
| + SetSource(kVideoSsrc2, video_track_.get()));
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, true, _));
|
| sender->SetSsrc(kVideoSsrc2);
|
|
|
| // Calls expected from destructor.
|
| - EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc2, nullptr)).Times(1);
|
| + EXPECT_CALL(video_provider_, SetSource(kVideoSsrc2, nullptr)).Times(1);
|
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, false, _)).Times(1);
|
| }
|
|
|
|
|