| Index: net/socket/socks_client_socket_unittest.cc
|
| diff --git a/net/socket/socks_client_socket_unittest.cc b/net/socket/socks_client_socket_unittest.cc
|
| index ef11107a6f89c3045c0b74774d37654194f09ada..eb2ab6ad9c71ad3eb794ceec4fa38ebde492b6d1 100644
|
| --- a/net/socket/socks_client_socket_unittest.cc
|
| +++ b/net/socket/socks_client_socket_unittest.cc
|
| @@ -1,4 +1,4 @@
|
| -// Copyright (c) 2009 The Chromium Authors. All rights reserved.
|
| +// Copyright (c) 2010 The Chromium Authors. All rights reserved.
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| @@ -32,7 +32,8 @@ class SOCKSClientSocketTest : public PlatformTest {
|
| SOCKSClientSocket* BuildMockSocket(MockRead reads[], size_t reads_count,
|
| MockWrite writes[], size_t writes_count,
|
| HostResolver* host_resolver,
|
| - const std::string& hostname, int port);
|
| + const std::string& hostname, int port,
|
| + NetLog* net_log);
|
| virtual void SetUp();
|
|
|
| protected:
|
| @@ -63,14 +64,15 @@ SOCKSClientSocket* SOCKSClientSocketTest::BuildMockSocket(
|
| size_t writes_count,
|
| HostResolver* host_resolver,
|
| const std::string& hostname,
|
| - int port) {
|
| + int port,
|
| + NetLog* net_log) {
|
|
|
| TestCompletionCallback callback;
|
| data_.reset(new StaticSocketDataProvider(reads, reads_count,
|
| writes, writes_count));
|
| - tcp_sock_ = new MockTCPClientSocket(address_list_, data_.get());
|
| + tcp_sock_ = new MockTCPClientSocket(address_list_, net_log, data_.get());
|
|
|
| - int rv = tcp_sock_->Connect(&callback, NULL);
|
| + int rv = tcp_sock_->Connect(&callback);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| rv = callback.WaitForResult();
|
| EXPECT_EQ(OK, rv);
|
| @@ -129,17 +131,17 @@ TEST_F(SOCKSClientSocketTest, CompleteHandshake) {
|
| MockRead data_reads[] = {
|
| MockRead(true, kSOCKSOkReply, arraysize(kSOCKSOkReply)),
|
| MockRead(true, payload_read.data(), payload_read.size()) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, "localhost", 80));
|
| + host_resolver_, "localhost", 80, &log));
|
|
|
| // At this state the TCP connection is completed but not the SOCKS handshake.
|
| EXPECT_TRUE(tcp_sock_->IsConnected());
|
| EXPECT_FALSE(user_sock_->IsConnected());
|
|
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(
|
| LogContainsBeginEvent(log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -197,13 +199,13 @@ TEST_F(SOCKSClientSocketTest, HandshakeFailures) {
|
| MockWrite(false, kSOCKSOkRequest, arraysize(kSOCKSOkRequest)) };
|
| MockRead data_reads[] = {
|
| MockRead(false, tests[i].fail_reply, arraysize(tests[i].fail_reply)) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, "localhost", 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| + host_resolver_, "localhost", 80, &log));
|
|
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -227,14 +229,13 @@ TEST_F(SOCKSClientSocketTest, PartialServerReads) {
|
| MockRead data_reads[] = {
|
| MockRead(true, kSOCKSPartialReply1, arraysize(kSOCKSPartialReply1)),
|
| MockRead(true, kSOCKSPartialReply2, arraysize(kSOCKSPartialReply2)) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, "localhost", 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| + host_resolver_, "localhost", 80, &log));
|
|
|
| -
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -260,13 +261,13 @@ TEST_F(SOCKSClientSocketTest, PartialClientWrites) {
|
| arraysize(kSOCKSPartialRequest2)) };
|
| MockRead data_reads[] = {
|
| MockRead(true, kSOCKSOkReply, arraysize(kSOCKSOkReply)) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, "localhost", 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| + host_resolver_, "localhost", 80, &log));
|
|
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -286,14 +287,13 @@ TEST_F(SOCKSClientSocketTest, FailedSocketRead) {
|
| MockRead(true, kSOCKSOkReply, arraysize(kSOCKSOkReply) - 2),
|
| // close connection unexpectedly
|
| MockRead(false, 0) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, "localhost", 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| -
|
| + host_resolver_, "localhost", 80, &log));
|
|
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -318,13 +318,13 @@ TEST_F(SOCKSClientSocketTest, SOCKS4AFailedDNS) {
|
| MockWrite(false, request.data(), request.size()) };
|
| MockRead data_reads[] = {
|
| MockRead(false, kSOCKSOkReply, arraysize(kSOCKSOkReply)) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, hostname, 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| + host_resolver_, hostname, 80, &log));
|
|
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -351,13 +351,13 @@ TEST_F(SOCKSClientSocketTest, SOCKS4AIfDomainInIPv6) {
|
| MockWrite(false, request.data(), request.size()) };
|
| MockRead data_reads[] = {
|
| MockRead(false, kSOCKSOkReply, arraysize(kSOCKSOkReply)) };
|
| + CapturingNetLog log(CapturingNetLog::kUnbounded);
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - host_resolver_, hostname, 80));
|
| - CapturingBoundNetLog log(CapturingNetLog::kUnbounded);
|
| + host_resolver_, hostname, 80, &log));
|
|
|
| - int rv = user_sock_->Connect(&callback_, log.bound());
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
| EXPECT_TRUE(LogContainsBeginEvent(
|
| log.entries(), 0, NetLog::TYPE_SOCKS_CONNECT));
|
| @@ -381,10 +381,10 @@ TEST_F(SOCKSClientSocketTest, DisconnectWhileHostResolveInProgress) {
|
|
|
| user_sock_.reset(BuildMockSocket(data_reads, arraysize(data_reads),
|
| data_writes, arraysize(data_writes),
|
| - hanging_resolver, "foo", 80));
|
| + hanging_resolver, "foo", 80, NULL));
|
|
|
| // Start connecting (will get stuck waiting for the host to resolve).
|
| - int rv = user_sock_->Connect(&callback_, NULL);
|
| + int rv = user_sock_->Connect(&callback_);
|
| EXPECT_EQ(ERR_IO_PENDING, rv);
|
|
|
| EXPECT_FALSE(user_sock_->IsConnected());
|
|
|