| Index: webrtc/base/nat_unittest.cc
|
| diff --git a/webrtc/base/nat_unittest.cc b/webrtc/base/nat_unittest.cc
|
| index f585effdd95337e7313d2e96229f4d84e8d3b1a7..1295a3463751c72ce94bd66f538785e0c906cfa2 100644
|
| --- a/webrtc/base/nat_unittest.cc
|
| +++ b/webrtc/base/nat_unittest.cc
|
| @@ -12,6 +12,7 @@
|
| #include <memory>
|
| #include <string>
|
|
|
| +#include "webrtc/base/asynctcpsocket.h"
|
| #include "webrtc/base/gunit.h"
|
| #include "webrtc/base/logging.h"
|
| #include "webrtc/base/natserver.h"
|
| @@ -19,8 +20,8 @@
|
| #include "webrtc/base/nethelpers.h"
|
| #include "webrtc/base/network.h"
|
| #include "webrtc/base/physicalsocketserver.h"
|
| +#include "webrtc/base/ptr_util.h"
|
| #include "webrtc/base/testclient.h"
|
| -#include "webrtc/base/asynctcpsocket.h"
|
| #include "webrtc/base/virtualsocketserver.h"
|
|
|
| using namespace rtc;
|
| @@ -34,13 +35,12 @@ bool CheckReceive(
|
|
|
| TestClient* CreateTestClient(
|
| SocketFactory* factory, const SocketAddress& local_addr) {
|
| - AsyncUDPSocket* socket = AsyncUDPSocket::Create(factory, local_addr);
|
| - return new TestClient(socket);
|
| + return new TestClient(
|
| + WrapUnique(AsyncUDPSocket::Create(factory, local_addr)));
|
| }
|
|
|
| TestClient* CreateTCPTestClient(AsyncSocket* socket) {
|
| - AsyncTCPSocket* packet_socket = new AsyncTCPSocket(socket, false);
|
| - return new TestClient(packet_socket);
|
| + return new TestClient(MakeUnique<AsyncTCPSocket>(socket, false));
|
| }
|
|
|
| // Tests that when sending from internal_addr to external_addrs through the
|
|
|