Index: webrtc/media/engine/webrtcvoiceengine_unittest.cc |
diff --git a/webrtc/media/engine/webrtcvoiceengine_unittest.cc b/webrtc/media/engine/webrtcvoiceengine_unittest.cc |
index e17aea34b874853d425740f914845600666811bc..60aa869a4b04e0314095e2c417c0e3d40a545174 100644 |
--- a/webrtc/media/engine/webrtcvoiceengine_unittest.cc |
+++ b/webrtc/media/engine/webrtcvoiceengine_unittest.cc |
@@ -164,6 +164,16 @@ class WebRtcVoiceEngineTestFake : public testing::Test { |
return GetRecvStream(ssrc).GetConfig(); |
} |
+ void SetSend(cricket::VoiceMediaChannel* channel, bool enable) { |
+ ASSERT_TRUE(channel); |
+ if (enable) { |
+ EXPECT_CALL(adm_, RecordingIsInitialized()).WillOnce(Return(false)); |
+ EXPECT_CALL(adm_, Recording()).WillOnce(Return(false)); |
+ EXPECT_CALL(adm_, InitRecording()).WillOnce(Return(0)); |
+ } |
+ channel->SetSend(enable); |
+ } |
+ |
void TestInsertDtmf(uint32_t ssrc, bool caller) { |
EXPECT_TRUE(SetupChannel()); |
if (caller) { |
@@ -175,7 +185,7 @@ class WebRtcVoiceEngineTestFake : public testing::Test { |
// Test we can only InsertDtmf when the other side supports telephone-event. |
EXPECT_TRUE(channel_->SetSendParameters(send_parameters_)); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_FALSE(channel_->CanInsertDtmf()); |
EXPECT_FALSE(channel_->InsertDtmf(ssrc, 1, 111)); |
send_parameters_.codecs.push_back(kTelephoneEventCodec); |
@@ -2154,9 +2164,9 @@ TEST_F(WebRtcVoiceEngineTestFake, RecvAbsoluteSendTimeHeaderExtensions) { |
TEST_F(WebRtcVoiceEngineTestFake, Send) { |
EXPECT_TRUE(SetupSendStream()); |
EXPECT_TRUE(channel_->SetSendParameters(send_parameters_)); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
- channel_->SetSend(false); |
+ SetSend(channel_, false); |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
} |
@@ -2166,7 +2176,7 @@ TEST_F(WebRtcVoiceEngineTestFake, SendStateWithAndWithoutSource) { |
EXPECT_TRUE(SetupSendStream()); |
EXPECT_TRUE(channel_->SetSendParameters(send_parameters_)); |
EXPECT_TRUE(channel_->SetAudioSend(kSsrc1, true, nullptr, nullptr)); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
EXPECT_TRUE(channel_->SetAudioSend(kSsrc1, true, nullptr, &fake_source_)); |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
@@ -2180,7 +2190,7 @@ TEST_F(WebRtcVoiceEngineTestFake, SendStateWhenStreamsAreRecreated) { |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
// Turn on sending. |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
// Changing RTP header extensions will recreate the AudioSendStream. |
@@ -2190,7 +2200,7 @@ TEST_F(WebRtcVoiceEngineTestFake, SendStateWhenStreamsAreRecreated) { |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
// Turn off sending. |
- channel_->SetSend(false); |
+ SetSend(channel_, false); |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
// Changing RTP header extensions will recreate the AudioSendStream. |
@@ -2215,7 +2225,7 @@ TEST_F(WebRtcVoiceEngineTestFake, CreateAndDeleteMultipleSendStreams) { |
SetupForMultiSendStream(); |
// Set the global state for sending. |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
for (uint32_t ssrc : kSsrcs4) { |
EXPECT_TRUE(channel_->AddSendStream( |
@@ -2286,14 +2296,14 @@ TEST_F(WebRtcVoiceEngineTestFake, SetSendWithMultipleSendStreams) { |
} |
// Set the global state for starting sending. |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
for (uint32_t ssrc : kSsrcs4) { |
// Verify that we are in a sending state for all the send streams. |
EXPECT_TRUE(GetSendStream(ssrc).IsSending()); |
} |
// Set the global state for stopping sending. |
- channel_->SetSend(false); |
+ SetSend(channel_, false); |
for (uint32_t ssrc : kSsrcs4) { |
// Verify that we are in a stop state for all the send streams. |
EXPECT_FALSE(GetSendStream(ssrc).IsSending()); |
@@ -2372,7 +2382,7 @@ TEST_F(WebRtcVoiceEngineTestFake, PlayoutWithMultipleStreams) { |
EXPECT_TRUE( |
channel_->AddRecvStream(cricket::StreamParams::CreateLegacy(kSsrc2))); |
int channel_num2 = voe_.GetLastChannel(); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
// Make sure only the new stream is played out. |
@@ -2388,7 +2398,7 @@ TEST_F(WebRtcVoiceEngineTestFake, PlayoutWithMultipleStreams) { |
EXPECT_TRUE(voe_.GetPlayout(channel_num3)); |
// Stop sending. |
- channel_->SetSend(false); |
+ SetSend(channel_, false); |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
// Stop playout. |
@@ -2420,11 +2430,11 @@ TEST_F(WebRtcVoiceEngineTestFake, CodianSend) { |
EXPECT_EQ(0, agc_config.targetLeveldBOv); |
send_parameters_.options = options_adjust_agc; |
EXPECT_TRUE(channel_->SetSendParameters(send_parameters_)); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_TRUE(GetSendStream(kSsrc1).IsSending()); |
EXPECT_EQ(0, voe_.GetAgcConfig(agc_config)); |
EXPECT_EQ(agc_config.targetLeveldBOv, 10); // level was attenuated |
- channel_->SetSend(false); |
+ SetSend(channel_, false); |
EXPECT_FALSE(GetSendStream(kSsrc1).IsSending()); |
EXPECT_EQ(0, voe_.GetAgcConfig(agc_config)); |
} |
@@ -2499,7 +2509,7 @@ TEST_F(WebRtcVoiceEngineTestFake, GetStats) { |
// Start sending - this affects some reported stats. |
{ |
cricket::VoiceMediaInfo info; |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_EQ(true, channel_->GetStats(&info)); |
VerifyVoiceSenderInfo(info.senders[0], true); |
} |
@@ -2755,7 +2765,7 @@ TEST_F(WebRtcVoiceEngineTestFake, InsertDtmfOnSendStreamAsCallee) { |
TEST_F(WebRtcVoiceEngineTestFake, TestSetPlayoutError) { |
EXPECT_TRUE(SetupSendStream()); |
EXPECT_TRUE(channel_->SetSendParameters(send_parameters_)); |
- channel_->SetSend(true); |
+ SetSend(channel_, true); |
EXPECT_TRUE(channel_->AddRecvStream(cricket::StreamParams::CreateLegacy(2))); |
EXPECT_TRUE(channel_->AddRecvStream(cricket::StreamParams::CreateLegacy(3))); |
EXPECT_TRUE(channel_->SetPlayout(true)); |
@@ -3036,7 +3046,7 @@ TEST_F(WebRtcVoiceEngineTestFake, SetOptionOverridesViaChannels) { |
EXPECT_TRUE(agc_enabled); |
EXPECT_TRUE(ns_enabled); |
- channel1->SetSend(true); |
+ SetSend(channel1.get(), true); |
voe_.GetEcStatus(ec_enabled, ec_mode); |
voe_.GetAgcStatus(agc_enabled, agc_mode); |
voe_.GetNsStatus(ns_enabled, ns_mode); |
@@ -3044,7 +3054,7 @@ TEST_F(WebRtcVoiceEngineTestFake, SetOptionOverridesViaChannels) { |
EXPECT_TRUE(agc_enabled); |
EXPECT_FALSE(ns_enabled); |
- channel2->SetSend(true); |
+ SetSend(channel2.get(), true); |
voe_.GetEcStatus(ec_enabled, ec_mode); |
voe_.GetAgcStatus(agc_enabled, agc_mode); |
voe_.GetNsStatus(ns_enabled, ns_mode); |