Chromium Code Reviews| Index: webrtc/pc/channelmanager_unittest.cc |
| diff --git a/webrtc/pc/channelmanager_unittest.cc b/webrtc/pc/channelmanager_unittest.cc |
| index 74c359012c188a2405b1154c2700e44180ce84a3..31991eaaa50b128e5953e6d3f2106cc141550e85 100644 |
| --- a/webrtc/pc/channelmanager_unittest.cc |
| +++ b/webrtc/pc/channelmanager_unittest.cc |
| @@ -8,6 +8,8 @@ |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| +#include <memory> |
| + |
| #include "webrtc/base/gunit.h" |
| #include "webrtc/base/logging.h" |
| #include "webrtc/base/thread.h" |
| @@ -20,8 +22,7 @@ |
| #include "webrtc/pc/channelmanager.h" |
| #include "webrtc/pc/fakemediacontroller.h" |
| -namespace cricket { |
| -const bool kDefaultRtcpMuxRequired = true; |
| +namespace { |
| const bool kDefaultSrtpRequired = true; |
| } |
| @@ -40,34 +41,28 @@ class ChannelManagerTest : public testing::Test { |
| ChannelManagerTest() |
| : fme_(new cricket::FakeMediaEngine()), |
| fdme_(new cricket::FakeDataEngine()), |
| - cm_(new cricket::ChannelManager(fme_, fdme_, rtc::Thread::Current())), |
| + cm_(new cricket::ChannelManager( |
| + std::unique_ptr<MediaEngineInterface>(fme_), |
| + std::unique_ptr<DataEngineInterface>(fdme_), |
| + rtc::Thread::Current())), |
| fake_call_(webrtc::Call::Config(&event_log_)), |
| - fake_mc_(cm_, &fake_call_), |
| + fake_mc_(cm_.get(), &fake_call_), |
| transport_controller_( |
| - new cricket::FakeTransportController(ICEROLE_CONTROLLING)) {} |
| - |
| - virtual void SetUp() { |
| + new cricket::FakeTransportController(ICEROLE_CONTROLLING)) { |
| fme_->SetAudioCodecs(MAKE_VECTOR(kAudioCodecs)); |
| fme_->SetVideoCodecs(MAKE_VECTOR(kVideoCodecs)); |
| } |
| - virtual void TearDown() { |
| - delete transport_controller_; |
| - delete cm_; |
| - cm_ = NULL; |
| - fdme_ = NULL; |
| - fme_ = NULL; |
| - } |
| - |
| webrtc::RtcEventLogNullImpl event_log_; |
| rtc::Thread network_; |
| rtc::Thread worker_; |
| + // |fme_| and |fdme_| are actually owned by |cm_|. |
| cricket::FakeMediaEngine* fme_; |
| cricket::FakeDataEngine* fdme_; |
| - cricket::ChannelManager* cm_; |
| + std::unique_ptr<cricket::ChannelManager> cm_; |
| cricket::FakeCall fake_call_; |
| cricket::FakeMediaController fake_mc_; |
| - cricket::FakeTransportController* transport_controller_; |
| + std::unique_ptr<cricket::FakeTransportController> transport_controller_; |
| }; |
| // Test that we startup/shutdown properly. |
| @@ -107,18 +102,17 @@ TEST_F(ChannelManagerTest, CreateDestroyChannels) { |
| cricket::CN_AUDIO, cricket::ICE_CANDIDATE_COMPONENT_RTP); |
| cricket::VoiceChannel* voice_channel = cm_->CreateVoiceChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_AUDIO, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired, AudioOptions()); |
| + rtc::Thread::Current(), cricket::CN_AUDIO, kDefaultSrtpRequired, |
| + AudioOptions()); |
| EXPECT_TRUE(voice_channel != nullptr); |
| cricket::VideoChannel* video_channel = cm_->CreateVideoChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_VIDEO, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired, VideoOptions()); |
| + rtc::Thread::Current(), cricket::CN_VIDEO, kDefaultSrtpRequired, |
| + VideoOptions()); |
| EXPECT_TRUE(video_channel != nullptr); |
| cricket::RtpDataChannel* rtp_data_channel = cm_->CreateRtpDataChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_DATA, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired); |
| + rtc::Thread::Current(), cricket::CN_DATA, kDefaultSrtpRequired); |
| EXPECT_TRUE(rtp_data_channel != nullptr); |
| cm_->DestroyVideoChannel(video_channel); |
| cm_->DestroyVoiceChannel(voice_channel); |
| @@ -133,26 +127,24 @@ TEST_F(ChannelManagerTest, CreateDestroyChannelsOnThread) { |
| EXPECT_TRUE(cm_->set_worker_thread(&worker_)); |
| EXPECT_TRUE(cm_->set_network_thread(&network_)); |
| EXPECT_TRUE(cm_->Init()); |
| - delete transport_controller_; |
| - transport_controller_ = |
| - new cricket::FakeTransportController(&network_, ICEROLE_CONTROLLING); |
| + transport_controller_.reset( |
| + new cricket::FakeTransportController(&network_, ICEROLE_CONTROLLING)); |
| cricket::DtlsTransportInternal* rtp_transport = |
| transport_controller_->CreateDtlsTransport( |
| cricket::CN_AUDIO, cricket::ICE_CANDIDATE_COMPONENT_RTP); |
| cricket::VoiceChannel* voice_channel = cm_->CreateVoiceChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_AUDIO, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired, AudioOptions()); |
| + rtc::Thread::Current(), cricket::CN_AUDIO, kDefaultSrtpRequired, |
| + AudioOptions()); |
| EXPECT_TRUE(voice_channel != nullptr); |
| cricket::VideoChannel* video_channel = cm_->CreateVideoChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_VIDEO, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired, VideoOptions()); |
| + rtc::Thread::Current(), cricket::CN_VIDEO, kDefaultSrtpRequired, |
| + VideoOptions()); |
| EXPECT_TRUE(video_channel != nullptr); |
| cricket::RtpDataChannel* rtp_data_channel = cm_->CreateRtpDataChannel( |
| &fake_mc_, rtp_transport, nullptr /*rtcp_transport*/, |
| - rtc::Thread::Current(), cricket::CN_DATA, nullptr, |
| - kDefaultRtcpMuxRequired, kDefaultSrtpRequired); |
| + rtc::Thread::Current(), cricket::CN_DATA, kDefaultSrtpRequired); |
|
pthatcher1
2017/02/08 01:33:49
It feels like it would make sense to do this in a
Taylor Brandstetter
2017/02/10 00:19:46
Done.
|
| EXPECT_TRUE(rtp_data_channel != nullptr); |
| cm_->DestroyVideoChannel(video_channel); |
| cm_->DestroyVoiceChannel(voice_channel); |