| Index: webrtc/api/rtpsenderreceiver_unittest.cc
|
| diff --git a/webrtc/api/rtpsenderreceiver_unittest.cc b/webrtc/api/rtpsenderreceiver_unittest.cc
|
| index b26526fffd939e7507e5da2a31ec101857462ba3..966a15177078db30f25ff10dd273fa9fd227b926 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,
|
| @@ -128,8 +129,7 @@ class RtpSenderReceiverTest : public testing::Test {
|
| void CreateVideoRtpSender() {
|
| AddVideoTrack(false);
|
| 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_);
|
| @@ -143,7 +143,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;
|
| }
|
| @@ -361,13 +361,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);
|
| }
|
|
|
| @@ -379,13 +378,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);
|
| }
|
|
|
| @@ -412,20 +410,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcSetToZero) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderSsrcSetToZero) {
|
| AddVideoTrack(false);
|
| 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);
|
| }
|
|
|
| @@ -448,20 +445,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderTrackSetToNull) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderTrackSetToNull) {
|
| AddVideoTrack(false);
|
| 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);
|
| }
|
|
|
| @@ -485,23 +481,21 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcChanged) {
|
| TEST_F(RtpSenderReceiverTest, VideoSenderSsrcChanged) {
|
| AddVideoTrack(false);
|
| 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);
|
| }
|
|
|
|
|