| Index: net/spdy/spdy_proxy_client_socket_unittest.cc
|
| diff --git a/net/spdy/spdy_proxy_client_socket_unittest.cc b/net/spdy/spdy_proxy_client_socket_unittest.cc
|
| index 7ae59f5e73ff9bc09a93e4a8fb7db7ea503beb9c..1f9747d2f551d928832c573b63a6a29ababbd0b8 100644
|
| --- a/net/spdy/spdy_proxy_client_socket_unittest.cc
|
| +++ b/net/spdy/spdy_proxy_client_socket_unittest.cc
|
| @@ -164,7 +164,6 @@ SpdyProxyClientSocketTest::SpdyProxyClientSocketTest()
|
| }
|
|
|
| void SpdyProxyClientSocketTest::TearDown() {
|
| - sock_.reset(NULL);
|
| if (session_.get() != NULL)
|
| session_->spdy_session_pool()->CloseAllSessions();
|
|
|
| @@ -448,14 +447,15 @@ TEST_P(SpdyProxyClientSocketTest, ConnectWithAuthCredentials) {
|
|
|
| TEST_P(SpdyProxyClientSocketTest, ConnectRedirects) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectRedirectReplyFrame());
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 2), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 3), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -473,6 +473,9 @@ TEST_P(SpdyProxyClientSocketTest, ConnectRedirects) {
|
| std::string location;
|
| ASSERT_TRUE(headers->IsRedirect(&location));
|
| ASSERT_EQ(location, kRedirectUrl);
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| TEST_P(SpdyProxyClientSocketTest, ConnectFails) {
|
| @@ -499,14 +502,15 @@ TEST_P(SpdyProxyClientSocketTest, ConnectFails) {
|
|
|
| TEST_P(SpdyProxyClientSocketTest, WasEverUsedReturnsCorrectValues) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 2), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 3), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -516,6 +520,9 @@ TEST_P(SpdyProxyClientSocketTest, WasEverUsedReturnsCorrectValues) {
|
| EXPECT_TRUE(sock_->WasEverUsed());
|
| sock_->Disconnect();
|
| EXPECT_TRUE(sock_->WasEverUsed());
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // ----------- GetPeerAddress
|
| @@ -987,14 +994,15 @@ TEST_P(SpdyProxyClientSocketTest, PendingReadOnCloseReturnsZero) {
|
| TEST_P(SpdyProxyClientSocketTest,
|
| ReadOnDisconnectSocketReturnsNotConnected) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 2), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 3), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -1005,6 +1013,9 @@ TEST_P(SpdyProxyClientSocketTest,
|
|
|
| ASSERT_EQ(ERR_SOCKET_NOT_CONNECTED,
|
| sock_->Read(NULL, 1, CompletionCallback()));
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // Reading buffered data from an already closed socket should return
|
| @@ -1068,15 +1079,16 @@ TEST_P(SpdyProxyClientSocketTest, WriteOnClosedStream) {
|
| // Calling Write() on a disconnected socket is an error
|
| TEST_P(SpdyProxyClientSocketTest, WriteOnDisconnectedSocket) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| scoped_ptr<SpdyFrame> msg1(ConstructBodyFrame(kMsg1, kLen1));
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 2), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 3), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -1088,6 +1100,9 @@ TEST_P(SpdyProxyClientSocketTest, WriteOnDisconnectedSocket) {
|
| scoped_refptr<IOBufferWithSize> buf(CreateBuffer(kMsg1, kLen1));
|
| EXPECT_EQ(ERR_SOCKET_NOT_CONNECTED,
|
| sock_->Write(buf.get(), buf->size(), CompletionCallback()));
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // If the socket is closed with a pending Write(), the callback
|
| @@ -1124,15 +1139,16 @@ TEST_P(SpdyProxyClientSocketTest, WritePendingOnClose) {
|
| // should not be called.
|
| TEST_P(SpdyProxyClientSocketTest, DisconnectWithWritePending) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| - MockWrite(SYNCHRONOUS, 0, 2), // EOF
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| + MockWrite(SYNCHRONOUS, 0, 3), // EOF
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 3), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 4), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -1149,20 +1165,24 @@ TEST_P(SpdyProxyClientSocketTest, DisconnectWithWritePending) {
|
|
|
| EXPECT_FALSE(sock_->IsConnected());
|
| EXPECT_FALSE(write_callback_.have_result());
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // If the socket is Disconnected with a pending Read(), the callback
|
| // should not be called.
|
| TEST_P(SpdyProxyClientSocketTest, DisconnectWithReadPending) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 2),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - MockRead(ASYNC, 0, 2), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), MockRead(ASYNC, 0, 3), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -1179,6 +1199,9 @@ TEST_P(SpdyProxyClientSocketTest, DisconnectWithReadPending) {
|
|
|
| EXPECT_FALSE(sock_->IsConnected());
|
| EXPECT_FALSE(read_callback_.have_result());
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // If the socket is Reset when both a read and write are pending,
|
| @@ -1220,22 +1243,26 @@ TEST_P(SpdyProxyClientSocketTest, RstWithReadAndWritePending) {
|
| EXPECT_TRUE(sock_.get());
|
| EXPECT_TRUE(read_callback_.have_result());
|
| EXPECT_TRUE(write_callback_.have_result());
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // Makes sure the proxy client socket's source gets the expected NetLog events
|
| // and only the expected NetLog events (No SpdySession events).
|
| TEST_P(SpdyProxyClientSocketTest, NetLog) {
|
| scoped_ptr<SpdyFrame> conn(ConstructConnectRequestFrame());
|
| + scoped_ptr<SpdyFrame> rst(
|
| + spdy_util_.ConstructSpdyRstStream(1, RST_STREAM_CANCEL));
|
| MockWrite writes[] = {
|
| - CreateMockWrite(*conn, 0, SYNCHRONOUS),
|
| + CreateMockWrite(*conn, 0, SYNCHRONOUS), CreateMockWrite(*rst, 3),
|
| };
|
|
|
| scoped_ptr<SpdyFrame> resp(ConstructConnectReplyFrame());
|
| scoped_ptr<SpdyFrame> msg1(ConstructBodyFrame(kMsg1, kLen1));
|
| MockRead reads[] = {
|
| - CreateMockRead(*resp, 1, ASYNC),
|
| - CreateMockRead(*msg1, 2, ASYNC),
|
| - MockRead(ASYNC, 0, 3), // EOF
|
| + CreateMockRead(*resp, 1, ASYNC), CreateMockRead(*msg1, 2, ASYNC),
|
| + MockRead(ASYNC, 0, 4), // EOF
|
| };
|
|
|
| Initialize(reads, arraysize(reads), writes, arraysize(writes));
|
| @@ -1275,6 +1302,9 @@ TEST_P(SpdyProxyClientSocketTest, NetLog) {
|
| NetLog::TYPE_SOCKET_BYTES_RECEIVED,
|
| NetLog::PHASE_NONE));
|
| EXPECT_TRUE(LogContainsEndEvent(entry_list, 9, NetLog::TYPE_SOCKET_ALIVE));
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| // CompletionCallback that causes the SpdyProxyClientSocket to be
|
| @@ -1346,6 +1376,9 @@ TEST_P(SpdyProxyClientSocketTest, RstWithReadAndWritePendingDelete) {
|
| EXPECT_FALSE(sock_.get());
|
| EXPECT_TRUE(read_callback.have_result());
|
| EXPECT_FALSE(write_callback_.have_result());
|
| +
|
| + // Let the RST_STREAM write while |rst| is in-scope.
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| } // namespace net
|
|
|