Index: webrtc/p2p/base/tcpport_unittest.cc |
diff --git a/webrtc/p2p/base/tcpport_unittest.cc b/webrtc/p2p/base/tcpport_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..77fb790c537bda82c6ec55171851c966f050c807 |
--- /dev/null |
+++ b/webrtc/p2p/base/tcpport_unittest.cc |
@@ -0,0 +1,85 @@ |
+/* |
+ * Copyright 2016 The WebRTC Project Authors. All rights reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include "webrtc/base/gunit.h" |
+#include "webrtc/base/physicalsocketserver.h" |
+#include "webrtc/base/thread.h" |
+#include "webrtc/base/virtualsocketserver.h" |
+#include "webrtc/p2p/base/basicpacketsocketfactory.h" |
+#include "webrtc/p2p/base/tcpport.h" |
+ |
+using rtc::SocketAddress; |
+using cricket::Connection; |
+using cricket::Port; |
+using cricket::TCPPort; |
+using cricket::ICE_UFRAG_LENGTH; |
+using cricket::ICE_PWD_LENGTH; |
+ |
+static int kTimeout = 1000; |
+static const SocketAddress kLocalAddr("11.11.11.11", 1); |
+static const SocketAddress kRemoteAddr("22.22.22.22", 2); |
+ |
+class TCPPortTest : public testing::Test, public sigslot::has_slots<> { |
+ public: |
+ TCPPortTest() |
+ : main_(rtc::Thread::Current()), |
+ pss_(new rtc::PhysicalSocketServer), |
+ ss_(new rtc::VirtualSocketServer(pss_.get())), |
+ ss_scope_(ss_.get()), |
+ network_("unittest", "unittest", rtc::IPAddress(INADDR_ANY), 32), |
+ socket_factory_(rtc::Thread::Current()), |
+ username_(rtc::CreateRandomString(ICE_UFRAG_LENGTH)), |
+ password_(rtc::CreateRandomString(ICE_PWD_LENGTH)) { |
+ network_.AddIP(rtc::IPAddress(INADDR_ANY)); |
+ } |
+ |
+ void ConnectSignalSocketCreated() { |
+ ss_->SignalSocketCreated.connect(this, &TCPPortTest::OnSocketCreated); |
+ } |
+ |
+ void OnSocketCreated(rtc::VirtualSocket* socket) { |
+ LOG(LS_INFO) << "socket created "; |
+ socket->SignalAddressReady.connect( |
+ this, &TCPPortTest::SetLocalhostAsAlternativeLocalAddress); |
+ } |
+ |
+ void SetLocalhostAsAlternativeLocalAddress(rtc::VirtualSocket* socket, |
+ const SocketAddress& address) { |
+ SocketAddress local_address("127.0.0.1", 2000); |
+ socket->SetAlternativeLocalAddress(local_address); |
+ } |
+ |
+ TCPPort* CreateTCPPort(const SocketAddress& addr) { |
+ return TCPPort::Create(main_, &socket_factory_, &network_, addr.ipaddr(), 0, |
+ 0, username_, password_, true); |
+ } |
+ |
+ protected: |
+ rtc::Thread* main_; |
+ rtc::scoped_ptr<rtc::PhysicalSocketServer> pss_; |
+ rtc::scoped_ptr<rtc::VirtualSocketServer> ss_; |
+ rtc::SocketServerScope ss_scope_; |
+ rtc::Network network_; |
+ rtc::BasicPacketSocketFactory socket_factory_; |
+ std::string username_; |
+ std::string password_; |
+}; |
+ |
+TEST_F(TCPPortTest, TestTCPPortWithLocalhostAddress) { |
+ rtc::scoped_ptr<TCPPort> lport(CreateTCPPort(kLocalAddr)); |
+ rtc::scoped_ptr<TCPPort> rport(CreateTCPPort(kRemoteAddr)); |
+ lport->PrepareAddress(); |
+ rport->PrepareAddress(); |
+ // Start to listen to new socket creation event. |
+ ConnectSignalSocketCreated(); |
+ Connection* conn = |
+ lport->CreateConnection(rport->Candidates()[0], Port::ORIGIN_MESSAGE); |
+ EXPECT_TRUE_WAIT(conn->connected(), kTimeout); |
+} |