diff --git a/p2p/base/async_stun_tcp_socket_unittest.cc b/p2p/base/async_stun_tcp_socket_unittest.cc index 64c9a3488b..571db83943 100644 --- a/p2p/base/async_stun_tcp_socket_unittest.cc +++ b/p2p/base/async_stun_tcp_socket_unittest.cc @@ -17,6 +17,7 @@ #include #include +#include "absl/memory/memory.h" #include "rtc_base/network/sent_packet.h" #include "rtc_base/socket.h" #include "rtc_base/third_party/sigslot/sigslot.h" @@ -78,13 +79,13 @@ class AsyncStunTCPSocketTest : public ::testing::Test, void CreateSockets() { rtc::Socket* server = vss_->CreateSocket(kServerAddr.family(), SOCK_STREAM); server->Bind(kServerAddr); - recv_socket_.reset(new AsyncStunServerTCPSocket(server)); - recv_socket_->SignalNewConnection.connect( + listen_socket_ = std::make_unique(server); + listen_socket_->SignalNewConnection.connect( this, &AsyncStunTCPSocketTest::OnNewConnection); rtc::Socket* client = vss_->CreateSocket(kClientAddr.family(), SOCK_STREAM); send_socket_.reset(AsyncStunTCPSocket::Create( - client, kClientAddr, recv_socket_->GetLocalAddress())); + client, kClientAddr, listen_socket_->GetLocalAddress())); send_socket_->SignalSentPacket.connect( this, &AsyncStunTCPSocketTest::OnSentPacket); ASSERT_TRUE(send_socket_.get() != NULL); @@ -106,7 +107,7 @@ class AsyncStunTCPSocketTest : public ::testing::Test, void OnNewConnection(rtc::AsyncPacketSocket* /*server*/, rtc::AsyncPacketSocket* new_socket) { - listen_socket_.reset(new_socket); + recv_socket_ = absl::WrapUnique(new_socket); new_socket->SignalReadPacket.connect(this, &AsyncStunTCPSocketTest::OnReadPacket); } @@ -132,8 +133,8 @@ class AsyncStunTCPSocketTest : public ::testing::Test, std::unique_ptr vss_; rtc::AutoSocketServerThread thread_; std::unique_ptr send_socket_; - std::unique_ptr recv_socket_; std::unique_ptr listen_socket_; + std::unique_ptr recv_socket_; std::list recv_packets_; int sent_packets_ = 0; };