Chromium Code Reviews| Index: talk/app/webrtc/rtpsenderreceiver_unittest.cc |
| diff --git a/talk/app/webrtc/rtpsenderreceiver_unittest.cc b/talk/app/webrtc/rtpsenderreceiver_unittest.cc |
| index c9d7e008c300475dd8c06aa09d4e4cd141381987..557d452f1601ac1ed040989efb0f8c8f83da2931 100644 |
| --- a/talk/app/webrtc/rtpsenderreceiver_unittest.cc |
| +++ b/talk/app/webrtc/rtpsenderreceiver_unittest.cc |
| @@ -120,8 +120,10 @@ class RtpSenderReceiverTest : public testing::Test { |
| audio_track_ = AudioTrack::Create(kAudioTrackId, NULL); |
| EXPECT_TRUE(stream_->AddTrack(audio_track_)); |
| EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| - audio_rtp_sender_ = new AudioRtpSender(stream_->GetAudioTracks()[0], |
| - kAudioSsrc, &audio_provider_); |
| + audio_rtp_sender_ = |
| + new AudioRtpSender(stream_->GetAudioTracks()[0], stream_->label(), |
| + &audio_provider_, nullptr); |
| + audio_rtp_sender_->SetSsrc(kAudioSsrc); |
| } |
| void CreateVideoRtpSender() { |
| @@ -130,7 +132,8 @@ class RtpSenderReceiverTest : public testing::Test { |
| kVideoSsrc, video_track_->GetSource()->GetVideoCapturer())); |
| EXPECT_CALL(video_provider_, SetVideoSend(kVideoSsrc, true, _)); |
| video_rtp_sender_ = new VideoRtpSender(stream_->GetVideoTracks()[0], |
| - kVideoSsrc, &video_provider_); |
| + stream_->label(), &video_provider_); |
| + video_rtp_sender_->SetSsrc(kVideoSsrc); |
| } |
| void DestroyAudioRtpSender() { |
| @@ -280,4 +283,121 @@ TEST_F(RtpSenderReceiverTest, RemoteAudioTrackSetVolume) { |
| DestroyAudioRtpReceiver(); |
| } |
| +TEST_F(RtpSenderReceiverTest, AudioSenderWithoutTrack) { |
| + rtc::scoped_refptr<AudioRtpSender> sender = |
| + new AudioRtpSender(&audio_provider_, nullptr); |
| + // Just let it get destroyed and make sure it doesn't call any methods on the |
| + // provider interface. |
| +} |
| + |
| +TEST_F(RtpSenderReceiverTest, VideoSenderWithoutTrack) { |
| + rtc::scoped_refptr<VideoRtpSender> sender = |
| + new VideoRtpSender(&video_provider_); |
| + // Just let it get destroyed and make sure it doesn't call any methods on the |
| + // provider interface. |
| +} |
| + |
| +// Test that an audio sender calls the expected methods on the provider once |
| +// it has a track and SSRC, when the SSRC is set first. |
| +TEST_F(RtpSenderReceiverTest, AudioSenderEarlyWarmupSsrcThenTrack) { |
| + rtc::scoped_refptr<AudioRtpSender> sender = |
| + new AudioRtpSender(&audio_provider_, nullptr); |
| + rtc::scoped_refptr<AudioTrackInterface> track = |
| + AudioTrack::Create(kAudioTrackId, nullptr); |
| + sender->SetSsrc(kAudioSsrc); |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| + sender->SetTrack(track); |
| + |
| + // Calls expected from destructor. |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
|
pthatcher1
2015/10/20 17:42:49
Should we test that this gets called even if SetTr
Taylor Brandstetter
2015/10/21 00:22:09
It shouldn't get called if SetTrack is never calle
|
| +} |
| + |
| +// Test that an audio sender calls the expected methods on the provider once |
| +// it has a track and SSRC, when the SSRC is set last. |
| +TEST_F(RtpSenderReceiverTest, AudioSenderEarlyWarmupTrackThenSsrc) { |
| + rtc::scoped_refptr<AudioRtpSender> sender = |
| + new AudioRtpSender(&audio_provider_, nullptr); |
| + rtc::scoped_refptr<AudioTrackInterface> track = |
| + AudioTrack::Create(kAudioTrackId, nullptr); |
| + sender->SetTrack(track); |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| + sender->SetSsrc(kAudioSsrc); |
| + |
| + // Calls expected from destructor. |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| +} |
| + |
| +// Test that a video sender calls the expected methods on the provider once |
| +// it has a track and SSRC, when the SSRC is set first. |
| +TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupSsrcThenTrack) { |
| + rtc::scoped_refptr<VideoRtpSender> sender = |
| + new VideoRtpSender(&video_provider_); |
| + sender->SetSsrc(kVideoSsrc); |
| + EXPECT_CALL(video_provider_, |
| + SetCaptureDevice(kVideoSsrc, |
| + video_track_->GetSource()->GetVideoCapturer())); |
| + 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_, SetVideoSend(kVideoSsrc, false, _)).Times(1); |
| +} |
| + |
| +// Test that a video sender calls the expected methods on the provider once |
| +// it has a track and SSRC, when the SSRC is set last. |
| +TEST_F(RtpSenderReceiverTest, VideoSenderEarlyWarmupTrackThenSsrc) { |
| + rtc::scoped_refptr<VideoRtpSender> sender = |
| + 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, _)); |
| + 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); |
| +} |
|
pthatcher1
2015/10/20 17:42:49
It seems you don't have any tests for setting the
Taylor Brandstetter
2015/10/21 00:22:09
No, I just wasn't thorough enough. I added some mo
|
| + |
| +// Test that the sender is disconnected from the provider when its SSRC is |
| +// set to 0. |
| +TEST_F(RtpSenderReceiverTest, AudioSenderSsrcSetToZero) { |
| + rtc::scoped_refptr<AudioTrackInterface> track = |
| + AudioTrack::Create(kAudioTrackId, nullptr); |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, true, _, _)); |
| + rtc::scoped_refptr<AudioRtpSender> sender = |
| + new AudioRtpSender(track, kStreamLabel1, &audio_provider_, nullptr); |
| + sender->SetSsrc(kAudioSsrc); |
| + |
| + EXPECT_CALL(audio_provider_, SetAudioSend(kAudioSsrc, false, _, _)).Times(1); |
| + sender->SetSsrc(0); |
| + |
| + // Make sure it's SetSsrc that called methods on the provider, and not the |
| + // destructor. |
| + EXPECT_CALL(audio_provider_, SetAudioSend(_, _, _, _)).Times(0); |
| +} |
| + |
| +// Test that the sender is disconnected from the provider when its SSRC is |
| +// set to 0. |
| +TEST_F(RtpSenderReceiverTest, VideoSenderSsrcSetToZero) { |
| + EXPECT_CALL(video_provider_, |
| + SetCaptureDevice(kVideoSsrc, |
| + video_track_->GetSource()->GetVideoCapturer())); |
| + 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_, 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_, SetVideoSend(_, _, _)).Times(0); |
| +} |
| + |
| } // namespace webrtc |