Index: webrtc/p2p/base/relayserver_unittest.cc |
diff --git a/webrtc/p2p/base/relayserver_unittest.cc b/webrtc/p2p/base/relayserver_unittest.cc |
index 0c98286af9507d6d097a824691becf38f9d9407a..a53a66668a5b932f77b06df0ef5c528c44106e91 100644 |
--- a/webrtc/p2p/base/relayserver_unittest.cc |
+++ b/webrtc/p2p/base/relayserver_unittest.cc |
@@ -11,16 +11,17 @@ |
#include <memory> |
#include <string> |
-#include "webrtc/p2p/base/relayserver.h" |
#include "webrtc/base/gunit.h" |
#include "webrtc/base/helpers.h" |
#include "webrtc/base/logging.h" |
#include "webrtc/base/physicalsocketserver.h" |
+#include "webrtc/base/ptr_util.h" |
#include "webrtc/base/socketaddress.h" |
#include "webrtc/base/ssladapter.h" |
#include "webrtc/base/testclient.h" |
#include "webrtc/base/thread.h" |
#include "webrtc/base/virtualsocketserver.h" |
+#include "webrtc/p2p/base/relayserver.h" |
using rtc::SocketAddress; |
using namespace cricket; |
@@ -55,9 +56,9 @@ class RelayServerTest : public testing::Test { |
rtc::AsyncUDPSocket::Create(ss_.get(), server_ext_addr)); |
client1_.reset(new rtc::TestClient( |
- rtc::AsyncUDPSocket::Create(ss_.get(), client1_addr))); |
+ WrapUnique(rtc::AsyncUDPSocket::Create(ss_.get(), client1_addr)))); |
client2_.reset(new rtc::TestClient( |
- rtc::AsyncUDPSocket::Create(ss_.get(), client2_addr))); |
+ WrapUnique(rtc::AsyncUDPSocket::Create(ss_.get(), client2_addr)))); |
} |
void Allocate() { |
@@ -116,24 +117,22 @@ class RelayServerTest : public testing::Test { |
} |
StunMessage* Receive(rtc::TestClient* client) { |
StunMessage* msg = NULL; |
- rtc::TestClient::Packet* packet = |
+ std::unique_ptr<rtc::TestClient::Packet> packet = |
client->NextPacket(rtc::TestClient::kTimeoutMs); |
if (packet) { |
rtc::ByteBufferWriter buf(packet->buf, packet->size); |
rtc::ByteBufferReader read_buf(buf); |
msg = new RelayMessage(); |
msg->Read(&read_buf); |
- delete packet; |
} |
return msg; |
} |
std::string ReceiveRaw(rtc::TestClient* client) { |
std::string raw; |
- rtc::TestClient::Packet* packet = |
+ std::unique_ptr<rtc::TestClient::Packet> packet = |
client->NextPacket(rtc::TestClient::kTimeoutMs); |
if (packet) { |
raw = std::string(packet->buf, packet->size); |
- delete packet; |
} |
return raw; |
} |