Chromium Code Reviews| Index: talk/app/webrtc/peerconnectionfactory_unittest.cc | 
| diff --git a/talk/app/webrtc/peerconnectionfactory_unittest.cc b/talk/app/webrtc/peerconnectionfactory_unittest.cc | 
| index f1d5353abd008a72b650a416cdf303ce3339dbce..c73ce97ad5dce1b683b98a9ff3eba1fbab5b9bf7 100644 | 
| --- a/talk/app/webrtc/peerconnectionfactory_unittest.cc | 
| +++ b/talk/app/webrtc/peerconnectionfactory_unittest.cc | 
| @@ -27,7 +27,6 @@ | 
| #include <string> | 
| -#include "talk/app/webrtc/fakeportallocatorfactory.h" | 
| #include "talk/app/webrtc/mediastreaminterface.h" | 
| #include "talk/app/webrtc/peerconnectionfactory.h" | 
| #include "talk/app/webrtc/test/fakedtlsidentitystore.h" | 
| @@ -39,6 +38,7 @@ | 
| #include "webrtc/base/gunit.h" | 
| #include "webrtc/base/scoped_ptr.h" | 
| #include "webrtc/base/thread.h" | 
| +#include "webrtc/p2p/client/fakeportallocator.h" | 
| using webrtc::DataChannelInterface; | 
| using webrtc::DtlsIdentityStoreInterface; | 
| @@ -47,17 +47,11 @@ using webrtc::MediaStreamInterface; | 
| using webrtc::PeerConnectionFactoryInterface; | 
| using webrtc::PeerConnectionInterface; | 
| using webrtc::PeerConnectionObserver; | 
| -using webrtc::PortAllocatorFactoryInterface; | 
| using webrtc::VideoSourceInterface; | 
| using webrtc::VideoTrackInterface; | 
| namespace { | 
| -typedef std::vector<PortAllocatorFactoryInterface::StunConfiguration> | 
| - StunConfigurations; | 
| -typedef std::vector<PortAllocatorFactoryInterface::TurnConfiguration> | 
| - TurnConfigurations; | 
| - | 
| static const char kStunIceServer[] = "stun:stun.l.google.com:19302"; | 
| static const char kTurnIceServer[] = "turn:test%40hello.com@test.com:1234"; | 
| static const char kTurnIceServerWithTransport[] = | 
| @@ -110,41 +104,39 @@ class PeerConnectionFactoryTest : public testing::Test { | 
| NULL); | 
| ASSERT_TRUE(factory_.get() != NULL); | 
| - allocator_factory_ = webrtc::FakePortAllocatorFactory::Create(); | 
| + port_allocator_.reset( | 
| + new cricket::FakePortAllocator(rtc::Thread::Current(), nullptr)); | 
| + raw_port_allocator_ = port_allocator_.get(); | 
| 
 
pthatcher1
2015/12/15 07:58:44
Why not just have a getter of port_allocator() tha
 
Taylor Brandstetter
2015/12/15 21:31:54
I tried this route, but at the point where I was w
 
 | 
| } | 
| protected: | 
| - void VerifyStunConfigurations(StunConfigurations stun_config) { | 
| - webrtc::FakePortAllocatorFactory* allocator = | 
| - static_cast<webrtc::FakePortAllocatorFactory*>( | 
| - allocator_factory_.get()); | 
| - ASSERT_TRUE(allocator != NULL); | 
| - EXPECT_EQ(stun_config.size(), allocator->stun_configs().size()); | 
| - for (size_t i = 0; i < stun_config.size(); ++i) { | 
| - EXPECT_EQ(stun_config[i].server.ToString(), | 
| - allocator->stun_configs()[i].server.ToString()); | 
| - } | 
| + void VerifyStunServers(cricket::ServerAddresses stun_servers) { | 
| + EXPECT_EQ(stun_servers, raw_port_allocator_->stun_servers()); | 
| } | 
| - void VerifyTurnConfigurations(TurnConfigurations turn_config) { | 
| - webrtc::FakePortAllocatorFactory* allocator = | 
| - static_cast<webrtc::FakePortAllocatorFactory*>( | 
| - allocator_factory_.get()); | 
| - ASSERT_TRUE(allocator != NULL); | 
| - EXPECT_EQ(turn_config.size(), allocator->turn_configs().size()); | 
| - for (size_t i = 0; i < turn_config.size(); ++i) { | 
| - EXPECT_EQ(turn_config[i].server.ToString(), | 
| - allocator->turn_configs()[i].server.ToString()); | 
| - EXPECT_EQ(turn_config[i].username, allocator->turn_configs()[i].username); | 
| - EXPECT_EQ(turn_config[i].password, allocator->turn_configs()[i].password); | 
| - EXPECT_EQ(turn_config[i].transport_type, | 
| - allocator->turn_configs()[i].transport_type); | 
| + void VerifyTurnServers(std::vector<cricket::RelayServerConfig> turn_servers) { | 
| + EXPECT_EQ(turn_servers.size(), raw_port_allocator_->turn_servers().size()); | 
| + for (size_t i = 0; i < turn_servers.size(); ++i) { | 
| + ASSERT_EQ(1u, turn_servers[i].ports.size()); | 
| + EXPECT_EQ(1u, raw_port_allocator_->turn_servers()[i].ports.size()); | 
| + EXPECT_EQ( | 
| + turn_servers[i].ports[0].address.ToString(), | 
| + raw_port_allocator_->turn_servers()[i].ports[0].address.ToString()); | 
| + EXPECT_EQ(turn_servers[i].ports[0].proto, | 
| + raw_port_allocator_->turn_servers()[i].ports[0].proto); | 
| + EXPECT_EQ(turn_servers[i].credentials.username, | 
| + raw_port_allocator_->turn_servers()[i].credentials.username); | 
| + EXPECT_EQ(turn_servers[i].credentials.password, | 
| + raw_port_allocator_->turn_servers()[i].credentials.password); | 
| } | 
| } | 
| rtc::scoped_refptr<PeerConnectionFactoryInterface> factory_; | 
| NullPeerConnectionObserver observer_; | 
| - rtc::scoped_refptr<PortAllocatorFactoryInterface> allocator_factory_; | 
| + rtc::scoped_ptr<cricket::FakePortAllocator> port_allocator_; | 
| + // Since the PC owns the port allocator after it's been initialized, | 
| + // this should only be used when known to be safe. | 
| + cricket::FakePortAllocator* raw_port_allocator_; | 
| }; | 
| // Verify creation of PeerConnection using internal ADM, video factory and | 
| @@ -154,13 +146,12 @@ TEST(PeerConnectionFactoryTestInternal, CreatePCUsingInternalModules) { | 
| webrtc::CreatePeerConnectionFactory()); | 
| NullPeerConnectionObserver observer; | 
| - webrtc::PeerConnectionInterface::IceServers servers; | 
| + webrtc::PeerConnectionInterface::RTCConfiguration config; | 
| rtc::scoped_ptr<FakeDtlsIdentityStore> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory->CreatePeerConnection( | 
| - servers, nullptr, nullptr, dtls_identity_store.Pass(), &observer)); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory->CreatePeerConnection( | 
| + config, nullptr, nullptr, std::move(dtls_identity_store), &observer)); | 
| EXPECT_TRUE(pc.get() != nullptr); | 
| } | 
| @@ -180,25 +171,22 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingIceServers) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - StunConfigurations stun_configs; | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun1( | 
| - "stun.l.google.com", 19302); | 
| - stun_configs.push_back(stun1); | 
| - VerifyStunConfigurations(stun_configs); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn1( | 
| - "test.com", 1234, "test@hello.com", kTurnPassword, "udp", false); | 
| - turn_configs.push_back(turn1); | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn2( | 
| - "hello.com", kDefaultStunPort, "test", kTurnPassword, "tcp", false); | 
| - turn_configs.push_back(turn2); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + cricket::ServerAddresses stun_servers; | 
| + rtc::SocketAddress stun1("stun.l.google.com", 19302); | 
| + stun_servers.insert(stun1); | 
| + VerifyStunServers(stun_servers); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn1("test.com", 1234, "test@hello.com", | 
| + kTurnPassword, cricket::PROTO_UDP, false); | 
| + turn_servers.push_back(turn1); | 
| + cricket::RelayServerConfig turn2("hello.com", kDefaultStunPort, "test", | 
| + kTurnPassword, cricket::PROTO_TCP, false); | 
| + turn_servers.push_back(turn2); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| // This test verifies creation of PeerConnection with valid STUN and TURN | 
| @@ -213,63 +201,22 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingIceServersUrls) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - StunConfigurations stun_configs; | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun1( | 
| - "stun.l.google.com", 19302); | 
| - stun_configs.push_back(stun1); | 
| - VerifyStunConfigurations(stun_configs); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn1( | 
| - "test.com", 1234, "test@hello.com", kTurnPassword, "udp", false); | 
| - turn_configs.push_back(turn1); | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn2( | 
| - "hello.com", kDefaultStunPort, "test", kTurnPassword, "tcp", false); | 
| - turn_configs.push_back(turn2); | 
| - VerifyTurnConfigurations(turn_configs); | 
| -} | 
| - | 
| -// This test verifies creation of PeerConnection with valid STUN and TURN | 
| -// configuration. Also verifies the URL's parsed correctly as expected. | 
| -// This version doesn't use RTCConfiguration. | 
| -// TODO(mallinath) - Remove this method after clients start using RTCConfig. | 
| -TEST_F(PeerConnectionFactoryTest, CreatePCUsingIceServersOldSignature) { | 
| - webrtc::PeerConnectionInterface::IceServers ice_servers; | 
| - webrtc::PeerConnectionInterface::IceServer ice_server; | 
| - ice_server.uri = kStunIceServer; | 
| - ice_servers.push_back(ice_server); | 
| - ice_server.uri = kTurnIceServer; | 
| - ice_server.password = kTurnPassword; | 
| - ice_servers.push_back(ice_server); | 
| - ice_server.uri = kTurnIceServerWithTransport; | 
| - ice_server.password = kTurnPassword; | 
| - ice_servers.push_back(ice_server); | 
| - rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| - new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(ice_servers, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - StunConfigurations stun_configs; | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun1( | 
| - "stun.l.google.com", 19302); | 
| - stun_configs.push_back(stun1); | 
| - VerifyStunConfigurations(stun_configs); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn1( | 
| - "test.com", 1234, "test@hello.com", kTurnPassword, "udp", false); | 
| - turn_configs.push_back(turn1); | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn2( | 
| - "hello.com", kDefaultStunPort, "test", kTurnPassword, "tcp", false); | 
| - turn_configs.push_back(turn2); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + cricket::ServerAddresses stun_servers; | 
| + rtc::SocketAddress stun1("stun.l.google.com", 19302); | 
| + stun_servers.insert(stun1); | 
| + VerifyStunServers(stun_servers); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn1("test.com", 1234, "test@hello.com", | 
| + kTurnPassword, cricket::PROTO_UDP, false); | 
| + turn_servers.push_back(turn1); | 
| + cricket::RelayServerConfig turn2("hello.com", kDefaultStunPort, "test", | 
| + kTurnPassword, cricket::PROTO_TCP, false); | 
| + turn_servers.push_back(turn2); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| TEST_F(PeerConnectionFactoryTest, CreatePCUsingNoUsernameInUri) { | 
| @@ -283,17 +230,15 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingNoUsernameInUri) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn( | 
| - "test.com", 1234, kTurnUsername, kTurnPassword, "udp", false); | 
| - turn_configs.push_back(turn); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn("test.com", 1234, kTurnUsername, | 
| + kTurnPassword, cricket::PROTO_UDP, false); | 
| + turn_servers.push_back(turn); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| // This test verifies the PeerConnection created properly with TURN url which | 
| @@ -306,17 +251,15 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingTurnUrlWithTransportParam) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn( | 
| - "hello.com", kDefaultStunPort, "test", kTurnPassword, "tcp", false); | 
| - turn_configs.push_back(turn); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn("hello.com", kDefaultStunPort, "test", | 
| + kTurnPassword, cricket::PROTO_TCP, false); | 
| + turn_servers.push_back(turn); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| TEST_F(PeerConnectionFactoryTest, CreatePCUsingSecureTurnUrl) { | 
| @@ -333,25 +276,23 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingSecureTurnUrl) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn1( | 
| - "hello.com", kDefaultStunTlsPort, "test", kTurnPassword, "tcp", true); | 
| - turn_configs.push_back(turn1); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn1("hello.com", kDefaultStunTlsPort, "test", | 
| + kTurnPassword, cricket::PROTO_TCP, true); | 
| + turn_servers.push_back(turn1); | 
| // TURNS with transport param should be default to tcp. | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn2( | 
| - "hello.com", 443, "test_no_transport", kTurnPassword, "tcp", true); | 
| - turn_configs.push_back(turn2); | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn3( | 
| - "hello.com", kDefaultStunTlsPort, "test_no_transport", | 
| - kTurnPassword, "tcp", true); | 
| - turn_configs.push_back(turn3); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + cricket::RelayServerConfig turn2("hello.com", 443, "test_no_transport", | 
| + kTurnPassword, cricket::PROTO_TCP, true); | 
| + turn_servers.push_back(turn2); | 
| + cricket::RelayServerConfig turn3("hello.com", kDefaultStunTlsPort, | 
| + "test_no_transport", kTurnPassword, | 
| + cricket::PROTO_TCP, true); | 
| + turn_servers.push_back(turn3); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| TEST_F(PeerConnectionFactoryTest, CreatePCUsingIPLiteralAddress) { | 
| @@ -370,32 +311,26 @@ TEST_F(PeerConnectionFactoryTest, CreatePCUsingIPLiteralAddress) { | 
| config.servers.push_back(ice_server); | 
| rtc::scoped_ptr<DtlsIdentityStoreInterface> dtls_identity_store( | 
| new FakeDtlsIdentityStore()); | 
| - rtc::scoped_refptr<PeerConnectionInterface> pc( | 
| - factory_->CreatePeerConnection(config, nullptr, | 
| - allocator_factory_.get(), | 
| - dtls_identity_store.Pass(), | 
| - &observer_)); | 
| - EXPECT_TRUE(pc.get() != NULL); | 
| - StunConfigurations stun_configs; | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun1( | 
| - "1.2.3.4", 1234); | 
| - stun_configs.push_back(stun1); | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun2( | 
| - "1.2.3.4", 3478); | 
| - stun_configs.push_back(stun2); // Default port | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun3( | 
| - "2401:fa00:4::", 1234); | 
| - stun_configs.push_back(stun3); | 
| - webrtc::PortAllocatorFactoryInterface::StunConfiguration stun4( | 
| - "2401:fa00:4::", 3478); | 
| - stun_configs.push_back(stun4); // Default port | 
| - VerifyStunConfigurations(stun_configs); | 
| + rtc::scoped_refptr<PeerConnectionInterface> pc(factory_->CreatePeerConnection( | 
| + config, nullptr, std::move(port_allocator_), | 
| + std::move(dtls_identity_store), &observer_)); | 
| + ASSERT_TRUE(pc.get() != NULL); | 
| + cricket::ServerAddresses stun_servers; | 
| + rtc::SocketAddress stun1("1.2.3.4", 1234); | 
| + stun_servers.insert(stun1); | 
| + rtc::SocketAddress stun2("1.2.3.4", 3478); | 
| + stun_servers.insert(stun2); // Default port | 
| + rtc::SocketAddress stun3("2401:fa00:4::", 1234); | 
| + stun_servers.insert(stun3); | 
| + rtc::SocketAddress stun4("2401:fa00:4::", 3478); | 
| + stun_servers.insert(stun4); // Default port | 
| + VerifyStunServers(stun_servers); | 
| - TurnConfigurations turn_configs; | 
| - webrtc::PortAllocatorFactoryInterface::TurnConfiguration turn1( | 
| - "2401:fa00:4::", 1234, "test", kTurnPassword, "udp", false); | 
| - turn_configs.push_back(turn1); | 
| - VerifyTurnConfigurations(turn_configs); | 
| + std::vector<cricket::RelayServerConfig> turn_servers; | 
| + cricket::RelayServerConfig turn1("2401:fa00:4::", 1234, "test", kTurnPassword, | 
| + cricket::PROTO_UDP, false); | 
| + turn_servers.push_back(turn1); | 
| + VerifyTurnServers(turn_servers); | 
| } | 
| // This test verifies the captured stream is rendered locally using a |