Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(3266)

Issues Search
    My Issues | Starred     Open | Closed | All

Unified Diff: blimp/net/stream_packet_reader_unittest.cc

Issue 1452823011: Make PacketReader/PacketWriter interfaces async-only. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« blimp/net/packet_writer.h ('K') | « blimp/net/stream_packet_reader.cc ('k') | blimp/net/stream_packet_writer.h » ('j') | blimp/net/stream_packet_writer.cc » ('J')
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: blimp/net/stream_packet_reader_unittest.cc
diff --git a/blimp/net/stream_packet_reader_unittest.cc b/blimp/net/stream_packet_reader_unittest.cc
index 61419cfa412862906821716c67d75180a4f55f76..6061e41f5f8ca177108088007974d43b3c22c282 100644
--- a/blimp/net/stream_packet_reader_unittest.cc
+++ b/blimp/net/stream_packet_reader_unittest.cc
@@ -41,9 +41,7 @@ class StreamPacketReaderTest : public testing::Test {
~StreamPacketReaderTest() override {}
- int ReadPacket() {
- return data_reader_.ReadPacket(buffer_, callback_.callback());
- }
+ void ReadPacket() { data_reader_.ReadPacket(buffer_, callback_.callback()); }
protected:
scoped_refptr<net::GrowableIOBuffer> buffer_;
@@ -65,12 +63,10 @@ TEST_F(StreamPacketReaderTest, ReadAsyncHeaderAsyncPayload) {
.WillOnce(DoAll(FillBufferFromString<0>(test_msg_),
SaveArg<2>(&socket_cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
socket_cb.Run(kPacketHeaderSizeBytes);
socket_cb.Run(test_msg_.size());
- int rv = callback_.WaitForResult();
-
- EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
+ EXPECT_EQ(static_cast<int>(test_msg_.size()), callback_.WaitForResult());
EXPECT_TRUE(BufferStartsWith(buffer_.get(), test_msg_));
}
@@ -89,13 +85,11 @@ TEST_F(StreamPacketReaderTest, ReadAsyncHeaderSyncPayload) {
.WillOnce(
DoAll(FillBufferFromString<0>(test_msg_), Return(test_msg_.size())));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
EXPECT_FALSE(callback_.have_result());
socket_cb.Run(kPacketHeaderSizeBytes);
- int rv = callback_.WaitForResult();
- EXPECT_GT(rv, 0);
- EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
+ EXPECT_EQ(static_cast<int>(test_msg_.size()), callback_.WaitForResult());
EXPECT_TRUE(BufferStartsWith(buffer_.get(), test_msg_));
}
@@ -110,12 +104,9 @@ TEST_F(StreamPacketReaderTest, ReadSyncHeaderAsyncPayload) {
.WillOnce(DoAll(FillBufferFromString<0>(test_msg_),
SaveArg<2>(&socket_cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
socket_cb.Run(test_msg_.size());
- int rv = callback_.WaitForResult();
-
- EXPECT_GT(rv, 0);
- EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
+ EXPECT_EQ(static_cast<int>(test_msg_.size()), callback_.WaitForResult());
EXPECT_TRUE(BufferStartsWith(buffer_.get(), test_msg_));
}
@@ -130,12 +121,9 @@ TEST_F(StreamPacketReaderTest, ReadSyncHeaderSyncPayload) {
.WillOnce(
DoAll(FillBufferFromString<0>(test_msg_), Return(test_msg_.size())));
- int rv = ReadPacket();
-
- EXPECT_GT(rv, 0);
- EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
+ ReadPacket();
+ EXPECT_EQ(static_cast<int>(test_msg_.size()), callback_.WaitForResult());
EXPECT_TRUE(BufferStartsWith(buffer_.get(), test_msg_));
- EXPECT_FALSE(callback_.have_result());
}
// Successful read of 2 messages, header and payload reads all completing
@@ -168,13 +156,12 @@ TEST_F(StreamPacketReaderTest, ReadMultipleMessagesSync) {
DoAll(FillBufferFromString<0>(test_msg2), Return(test_msg2.size())))
.RetiresOnSaturation();
- int rv = ReadPacket();
+ ReadPacket();
+ EXPECT_EQ(callback_.WaitForResult(), static_cast<int>(test_msg_.size()));
+
+ ReadPacket();
+ EXPECT_EQ(callback_.WaitForResult(), static_cast<int>(test_msg2.size()));
- EXPECT_GT(rv, 0);
- EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
- rv = ReadPacket();
- EXPECT_GT(rv, 0);
- EXPECT_EQ(rv, static_cast<int>(test_msg2.size()));
EXPECT_TRUE(BufferStartsWith(buffer_.get(), test_msg_));
EXPECT_FALSE(callback_.have_result());
}
@@ -210,8 +197,7 @@ TEST_F(StreamPacketReaderTest, ReadMultipleMessagesAsync) {
SaveArg<2>(&socket_cb), Return(net::ERR_IO_PENDING)))
.RetiresOnSaturation();
- EXPECT_EQ(net::ERR_IO_PENDING,
- data_reader_.ReadPacket(buffer_, read_cb1.callback()));
+ data_reader_.ReadPacket(buffer_, read_cb1.callback());
socket_cb.Run(kPacketHeaderSizeBytes);
socket_cb.Run(test_msg_.size());
int rv = read_cb1.WaitForResult();
@@ -219,8 +205,7 @@ TEST_F(StreamPacketReaderTest, ReadMultipleMessagesAsync) {
EXPECT_GT(rv, 0);
EXPECT_EQ(rv, static_cast<int>(test_msg_.size()));
- EXPECT_EQ(net::ERR_IO_PENDING,
- data_reader_.ReadPacket(buffer_, read_cb2.callback()));
+ data_reader_.ReadPacket(buffer_, read_cb2.callback());
socket_cb.Run(kPacketHeaderSizeBytes);
socket_cb.Run(test_msg_.size());
rv = read_cb2.WaitForResult();
@@ -250,7 +235,7 @@ TEST_F(StreamPacketReaderTest, PartialHeaderReadAsync) {
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size(), _))
.WillOnce(Return(net::ERR_IO_PENDING));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
cb.Run(1);
cb.Run(kPacketHeaderSizeBytes - 1);
}
@@ -268,7 +253,7 @@ TEST_F(StreamPacketReaderTest, PartialPayloadReadAsync) {
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size(), _))
.WillOnce(DoAll(FillBufferFromString<0>(test_msg_.substr(0, 1)),
SaveArg<2>(&cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size() - 1, _))
.WillOnce(DoAll(
FillBufferFromString<0>(test_msg_.substr(1, test_msg_.size() - 1)),
@@ -276,10 +261,7 @@ TEST_F(StreamPacketReaderTest, PartialPayloadReadAsync) {
cb.Run(1);
cb.Run(test_msg_.size() - 1);
- int rv = callback_.WaitForResult();
-
- EXPECT_GT(rv, 0);
- EXPECT_EQ(static_cast<int>(test_msg_.size()), rv);
+ EXPECT_EQ(static_cast<int>(test_msg_.size()), callback_.WaitForResult());
}
// Verify that synchronous header read errors are reported correctly.
@@ -287,7 +269,8 @@ TEST_F(StreamPacketReaderTest, ReadHeaderErrorSync) {
net::CompletionCallback cb;
EXPECT_CALL(socket_, Read(NotNull(), kPacketHeaderSizeBytes, _))
.WillOnce(Return(net::ERR_FAILED));
- EXPECT_EQ(net::ERR_FAILED, ReadPacket());
+ ReadPacket();
+ EXPECT_EQ(net::ERR_FAILED, callback_.WaitForResult());
}
// Verify that synchronous payload read errors are reported correctly.
@@ -300,7 +283,8 @@ TEST_F(StreamPacketReaderTest, ReadPayloadErrorSync) {
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size(), _))
.WillOnce(Return(net::ERR_FAILED));
- EXPECT_EQ(net::ERR_FAILED, ReadPacket());
+ ReadPacket();
+ EXPECT_EQ(net::ERR_FAILED, callback_.WaitForResult());
}
// Verify that async header read errors are reported correctly.
@@ -312,7 +296,7 @@ TEST_F(StreamPacketReaderTest, ReadHeaderErrorAsync) {
.WillOnce(DoAll(FillBufferFromString<0>(EncodeHeader(test_msg_.size())),
SaveArg<2>(&cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
cb.Run(net::ERR_FAILED);
EXPECT_EQ(net::ERR_FAILED, callback_.WaitForResult());
}
@@ -327,7 +311,7 @@ TEST_F(StreamPacketReaderTest, ReadPayloadErrorAsync) {
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size(), _))
.WillOnce(DoAll(SaveArg<2>(&cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING, ReadPacket());
+ ReadPacket();
cb.Run(net::ERR_FAILED);
EXPECT_EQ(net::ERR_FAILED, callback_.WaitForResult());
}
@@ -343,8 +327,7 @@ TEST_F(StreamPacketReaderTest, ReaderDeletedDuringAsyncHeaderRead) {
.WillOnce(DoAll(FillBufferFromString<0>(EncodeHeader(test_msg_.size())),
SaveArg<2>(&cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING,
- reader->ReadPacket(buffer_, callback_.callback()));
+ reader->ReadPacket(buffer_, callback_.callback());
reader.reset(); // Delete the reader object.
cb.Run(kPacketHeaderSizeBytes); // Complete the socket operation.
}
@@ -360,8 +343,7 @@ TEST_F(StreamPacketReaderTest, ReaderDeletedDuringAsyncPayloadRead) {
Return(kPacketHeaderSizeBytes)));
EXPECT_CALL(socket_, Read(NotNull(), test_msg_.size(), _))
.WillOnce(DoAll(SaveArg<2>(&cb), Return(net::ERR_IO_PENDING)));
- EXPECT_EQ(net::ERR_IO_PENDING,
- reader->ReadPacket(buffer_, callback_.callback()));
+ reader->ReadPacket(buffer_, callback_.callback());
reader.reset(); // Delete the reader object.
cb.Run(net::ERR_FAILED); // Complete the socket operation.
@@ -374,7 +356,8 @@ TEST_F(StreamPacketReaderTest, ReadWhatIsThisAPacketForAnts) {
Return(kPacketHeaderSizeBytes)))
.RetiresOnSaturation();
- EXPECT_EQ(net::ERR_INVALID_RESPONSE, ReadPacket());
+ ReadPacket();
+ EXPECT_EQ(net::ERR_INVALID_RESPONSE, callback_.WaitForResult());
}
// Verify that an illegally large payloads is reported as an erroneous inputs.
@@ -384,7 +367,8 @@ TEST_F(StreamPacketReaderTest, ReadErrorIllegallyLargePayload) {
DoAll(FillBufferFromString<0>(EncodeHeader(kTestMaxBufferSize + 1)),
Return(kPacketHeaderSizeBytes)));
- EXPECT_EQ(net::ERR_INVALID_RESPONSE, ReadPacket());
+ ReadPacket();
+ EXPECT_EQ(net::ERR_INVALID_RESPONSE, callback_.WaitForResult());
}
} // namespace
« blimp/net/packet_writer.h ('K') | « blimp/net/stream_packet_reader.cc ('k') | blimp/net/stream_packet_writer.h » ('j') | blimp/net/stream_packet_writer.cc » ('J')

Powered by Google App Engine
This is Rietveld 408576698