Index: webrtc/api/rtpsenderreceiver_unittest.cc |
diff --git a/webrtc/api/rtpsenderreceiver_unittest.cc b/webrtc/api/rtpsenderreceiver_unittest.cc |
index 7f32f29bc4bf40e48a0a3114afed296b57576b47..b652aedeb1e2bb65fb5a28423d5dcde27b05be9c 100644 |
--- a/webrtc/api/rtpsenderreceiver_unittest.cc |
+++ b/webrtc/api/rtpsenderreceiver_unittest.cc |
@@ -70,16 +70,15 @@ class MockAudioProvider : public AudioProviderInterface { |
class MockVideoProvider : public VideoProviderInterface { |
public: |
virtual ~MockVideoProvider() {} |
- MOCK_METHOD2(SetCaptureDevice, |
- bool(uint32_t ssrc, cricket::VideoCapturer* camera)); |
MOCK_METHOD3(SetVideoPlayout, |
void(uint32_t ssrc, |
bool enable, |
rtc::VideoSinkInterface<cricket::VideoFrame>* sink)); |
- MOCK_METHOD3(SetVideoSend, |
+ MOCK_METHOD4(SetVideoSend, |
void(uint32_t ssrc, |
bool enable, |
- const cricket::VideoOptions* options)); |
+ const cricket::VideoOptions* options, |
+ cricket::VideoCapturer* camera)); |
MOCK_CONST_METHOD1(GetVideoRtpParameters, RtpParameters(uint32_t ssrc)); |
MOCK_METHOD2(SetVideoRtpParameters, |
@@ -112,9 +111,8 @@ class RtpSenderReceiverTest : public testing::Test { |
void CreateVideoRtpSender() { |
AddVideoTrack(); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice( |
- kVideoSsrc, video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
video_rtp_sender_ = new VideoRtpSender(stream_->GetVideoTracks()[0], |
stream_->label(), &video_provider_); |
video_rtp_sender_->SetSsrc(kVideoSsrc); |
@@ -127,8 +125,8 @@ class RtpSenderReceiverTest : public testing::Test { |
} |
void DestroyVideoRtpSender() { |
- EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, NULL)).Times(1); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .Times(1); |
video_rtp_sender_ = nullptr; |
} |
@@ -227,10 +225,14 @@ TEST_F(RtpSenderReceiverTest, RemoteAudioTrackDisable) { |
TEST_F(RtpSenderReceiverTest, LocalVideoTrackDisable) { |
CreateVideoRtpSender(); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)); |
+ EXPECT_CALL(video_provider_, |
+ SetVideoSend(kVideoSsrc, false, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
video_track_->set_enabled(false); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ EXPECT_CALL(video_provider_, |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
video_track_->set_enabled(true); |
DestroyVideoRtpSender(); |
@@ -346,14 +348,13 @@ TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupSsrcThenTrack) { |
new VideoRtpSender(&video_provider_); |
sender->SetSsrc(kVideoSsrc); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
sender->SetTrack(video_track_); |
// Calls expected from destructor. |
- EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .Times(1); |
} |
// Test that a video sender calls the expected methods on the provider once |
@@ -364,14 +365,13 @@ TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupTrackThenSsrc) { |
new VideoRtpSender(&video_provider_); |
sender->SetTrack(video_track_); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
sender->SetSsrc(kVideoSsrc); |
// Calls expected from destructor. |
- EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc, nullptr)).Times(1); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .Times(1); |
} |
// Test that the sender is disconnected from the provider when its SSRC is |
@@ -397,21 +397,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcSetToZero) { |
TEST_F(RtpSenderReceiverTest, VideoSenderSsrcSetToZero) { |
AddVideoTrack(); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
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_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .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_, SetVideoSend(_, _, _)).Times(0); |
+ EXPECT_CALL(video_provider_, SetVideoSend(_, _, _, _)).Times(0); |
} |
TEST_F(RtpSenderReceiverTest, AudioSenderTrackSetToNull) { |
@@ -433,21 +431,19 @@ TEST_F(RtpSenderReceiverTest, AudioSenderTrackSetToNull) { |
TEST_F(RtpSenderReceiverTest, VideoSenderTrackSetToNull) { |
AddVideoTrack(); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
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_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .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_, SetVideoSend(_, _, _)).Times(0); |
+ EXPECT_CALL(video_provider_, SetVideoSend(_, _, _, _)).Times(0); |
} |
TEST_F(RtpSenderReceiverTest, AudioSenderSsrcChanged) { |
@@ -470,24 +466,23 @@ TEST_F(RtpSenderReceiverTest, AudioSenderSsrcChanged) { |
TEST_F(RtpSenderReceiverTest, VideoSenderSsrcChanged) { |
AddVideoTrack(); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
+ SetVideoSend(kVideoSsrc, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())); |
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_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, false, _, nullptr)) |
+ .Times(1); |
EXPECT_CALL(video_provider_, |
- SetCaptureDevice(kVideoSsrc2, |
- video_track_->GetSource()->GetVideoCapturer())); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, true, _)); |
+ SetVideoSend(kVideoSsrc2, true, _, |
+ video_track_->GetSource()->GetVideoCapturer())) |
+ .Times(1); |
sender->SetSsrc(kVideoSsrc2); |
// Calls expected from destructor. |
- EXPECT_CALL(video_provider_, SetCaptureDevice(kVideoSsrc2, nullptr)).Times(1); |
- EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, false, _)).Times(1); |
+ EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc2, false, _, nullptr)) |
+ .Times(1); |
} |
TEST_F(RtpSenderReceiverTest, AudioSenderCanSetParameters) { |