Index: webrtc/p2p/base/turnport_unittest.cc |
diff --git a/webrtc/p2p/base/turnport_unittest.cc b/webrtc/p2p/base/turnport_unittest.cc |
index 3172ba252f2930156792fa57049d31c460f5c045..00a62fa70b444b4148ca583151a5ee7ca6621e08 100644 |
--- a/webrtc/p2p/base/turnport_unittest.cc |
+++ b/webrtc/p2p/base/turnport_unittest.cc |
@@ -627,6 +627,28 @@ TEST_F(TurnPortTest, TestTurnTcpAllocateMismatch) { |
EXPECT_NE(first_addr, turn_port_->socket()->GetLocalAddress()); |
} |
+// Test that CreateConnection will return null if port becomes disconnected. |
+TEST_F(TurnPortTest, TestCreateConnectionWhenSocketClosed) { |
+ turn_server_.AddInternalSocket(kTurnTcpIntAddr, cricket::PROTO_TCP); |
+ CreateTurnPort(kTurnUsername, kTurnPassword, kTurnTcpProtoAddr); |
+ turn_port_->PrepareAddress(); |
+ ASSERT_TRUE_WAIT(turn_ready_, kTimeout); |
+ |
+ CreateUdpPort(); |
+ udp_port_->PrepareAddress(); |
+ ASSERT_TRUE_WAIT(udp_ready_, kTimeout); |
+ // Create a connection. |
+ Connection* conn1 = turn_port_->CreateConnection(udp_port_->Candidates()[0], |
+ Port::ORIGIN_MESSAGE); |
+ ASSERT_TRUE(conn1 != NULL); |
+ |
+ // Close the socket and create a connection again. |
+ turn_port_->OnSocketClose(turn_port_->socket(), 1); |
+ conn1 = turn_port_->CreateConnection(udp_port_->Candidates()[0], |
+ Port::ORIGIN_MESSAGE); |
+ ASSERT_TRUE(conn1 == NULL); |
+} |
+ |
// Test try-alternate-server feature. |
TEST_F(TurnPortTest, TestTurnAlternateServerUDP) { |
TestTurnAlternateServer(cricket::PROTO_UDP); |