Index: blimp/net/stream_packet_writer_unittest.cc |
diff --git a/blimp/net/stream_packet_writer_unittest.cc b/blimp/net/stream_packet_writer_unittest.cc |
index c8014f9481f8bcc3ddc3c3ec7220e5f89a7bb04d..564b9b91ea4f2e1bf8d89bd5c2cef28f04755a57 100644 |
--- a/blimp/net/stream_packet_writer_unittest.cc |
+++ b/blimp/net/stream_packet_writer_unittest.cc |
@@ -60,8 +60,7 @@ TEST_F(StreamPacketWriterTest, TestWriteAsync) { |
EXPECT_CALL(socket_, Write(BufferEquals(EncodeHeader(test_data_str_.size())), |
kPacketHeaderSizeBytes, _)) |
.WillOnce(DoAll(SaveArg<2>(&header_cb), Return(net::ERR_IO_PENDING))); |
- EXPECT_EQ(net::ERR_IO_PENDING, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
Mock::VerifyAndClearExpectations(&socket_); |
// Write payload. |
@@ -100,8 +99,7 @@ TEST_F(StreamPacketWriterTest, TestPartialWriteAsync) { |
.WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING))) |
.RetiresOnSaturation(); |
- EXPECT_EQ(net::ERR_IO_PENDING, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
// Header is written - first one byte, then the remainder. |
header_cb.Run(1); |
@@ -127,8 +125,7 @@ TEST_F(StreamPacketWriterTest, TestWriteErrorAsync) { |
Write(BufferEquals(test_data_str_), test_data_str_.size(), _)) |
.WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING))); |
- EXPECT_EQ(net::ERR_IO_PENDING, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
header_cb.Run(kPacketHeaderSizeBytes); |
payload_cb.Run(net::ERR_CONNECTION_RESET); |
@@ -144,9 +141,8 @@ TEST_F(StreamPacketWriterTest, TestWriteSync) { |
EXPECT_CALL(socket_, |
Write(BufferEquals(test_data_str_), test_data_str_.size(), _)) |
.WillOnce(Return(test_data_str_.size())); |
- EXPECT_EQ(net::OK, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
- EXPECT_FALSE(writer_cb.have_result()); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
+ EXPECT_EQ(net::OK, writer_cb.WaitForResult()); |
} |
// Successful write with 2 sync header writes and 2 sync payload writes. |
@@ -167,19 +163,18 @@ TEST_F(StreamPacketWriterTest, TestPartialWriteSync) { |
payload.size() - 1, _)) |
.WillOnce(Return(payload.size() - 1)); |
- EXPECT_EQ(net::OK, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
- EXPECT_FALSE(writer_cb.have_result()); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
+ EXPECT_EQ(net::OK, writer_cb.WaitForResult()); |
} |
// Verify that zero-length packets are rejected. |
TEST_F(StreamPacketWriterTest, TestZeroLengthPacketsRejected) { |
net::TestCompletionCallback writer_cb; |
- EXPECT_EQ(net::ERR_INVALID_ARGUMENT, |
- message_writer_.WritePacket( |
- new net::DrainableIOBuffer(new net::IOBuffer(0), 0), |
- writer_cb.callback())); |
+ message_writer_.WritePacket( |
+ new net::DrainableIOBuffer(new net::IOBuffer(0), 0), |
+ writer_cb.callback()); |
+ EXPECT_EQ(net::ERR_INVALID_ARGUMENT, writer_cb.WaitForResult()); |
EXPECT_FALSE(writer_cb.have_result()); |
} |
@@ -192,8 +187,8 @@ TEST_F(StreamPacketWriterTest, TestWriteHeaderErrorSync) { |
kPacketHeaderSizeBytes, _)) |
.WillOnce(Return(net::ERR_FAILED)); |
- EXPECT_EQ(net::ERR_FAILED, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
+ EXPECT_EQ(net::ERR_FAILED, writer_cb.WaitForResult()); |
EXPECT_EQ(net::ERR_EMPTY_RESPONSE, |
writer_cb.GetResult(net::ERR_EMPTY_RESPONSE)); |
@@ -211,9 +206,8 @@ TEST_F(StreamPacketWriterTest, TestWritePayloadErrorSync) { |
Write(BufferEquals(test_data_str_), test_data_str_.size(), _)) |
.WillOnce(Return(net::ERR_FAILED)); |
- EXPECT_EQ(net::ERR_FAILED, |
- message_writer_.WritePacket(test_data_, writer_cb.callback())); |
- EXPECT_FALSE(writer_cb.have_result()); |
+ message_writer_.WritePacket(test_data_, writer_cb.callback()); |
+ EXPECT_EQ(net::ERR_FAILED, writer_cb.WaitForResult()); |
} |
// Verify that asynchronous header write completions don't cause a |
@@ -228,8 +222,7 @@ TEST_F(StreamPacketWriterTest, DeletedDuringHeaderWrite) { |
EXPECT_CALL(socket_, Write(BufferEquals(EncodeHeader(test_data_str_.size())), |
kPacketHeaderSizeBytes, _)) |
.WillOnce(DoAll(SaveArg<2>(&header_cb), Return(net::ERR_IO_PENDING))); |
- EXPECT_EQ(net::ERR_IO_PENDING, |
- writer->WritePacket(test_data_, writer_cb.callback())); |
+ writer->WritePacket(test_data_, writer_cb.callback()); |
Mock::VerifyAndClearExpectations(&socket_); |
// Header write completion callback is invoked after the writer died. |
@@ -252,8 +245,7 @@ TEST_F(StreamPacketWriterTest, DeletedDuringPayloadWrite) { |
Write(BufferEquals(test_data_str_), test_data_str_.size(), _)) |
.WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING))); |
- EXPECT_EQ(net::ERR_IO_PENDING, |
- writer->WritePacket(test_data_, writer_cb.callback())); |
+ writer->WritePacket(test_data_, writer_cb.callback()); |
// Header write completes successfully. |
header_cb.Run(kPacketHeaderSizeBytes); |