1 files changed, 5 insertions, 2 deletions
diff --git a/net/socket_stream/socket_stream_unittest.cc b/net/socket_stream/socket_stream_unittest.cc
index d8e2093..497020d 100644
--- a/net/socket_stream/socket_stream_unittest.cc
+++ b/net/socket_stream/socket_stream_unittest.cc
@@ -238,11 +238,13 @@ TEST_F(SocketStreamTest, CloseFlushPendingWrite) {
   delegate->SetOnReceivedData(NewCallback(
       test, &SocketStreamTest::DoCloseFlushPendingWriteTest));
 
+  MockHostResolver host_resolver;
+
   scoped_refptr<SocketStream> socket_stream =
       new SocketStream(GURL("ws://example.com/demo"), delegate.get());
 
   socket_stream->set_context(new TestURLRequestContext());
-  socket_stream->SetHostResolver(new MockHostResolver());
+  socket_stream->SetHostResolver(&host_resolver);
 
   MockWrite data_writes[] = {
     MockWrite(SocketStreamTest::kWebSocketHandshakeRequest),
@@ -333,7 +335,8 @@ TEST_F(SocketStreamTest, BasicAuthProxy) {
       new SocketStream(GURL("ws://example.com/demo"), delegate.get());
 
   socket_stream->set_context(new TestURLRequestContext("myproxy:70"));
-  socket_stream->SetHostResolver(new MockHostResolver());
+  MockHostResolver host_resolver;
+  socket_stream->SetHostResolver(&host_resolver);
   socket_stream->SetClientSocketFactory(&mock_socket_factory);
 
   socket_stream->Connect();