Index: remoting/protocol/message_reader_unittest.cc |
diff --git a/remoting/protocol/message_reader_unittest.cc b/remoting/protocol/message_reader_unittest.cc |
index 02bf273960f828b5e5d3ad435fd9e15a9cbe73d0..f67466a12dad5914abeb7f49cfefdeb24d381e8f 100644 |
--- a/remoting/protocol/message_reader_unittest.cc |
+++ b/remoting/protocol/message_reader_unittest.cc |
@@ -149,7 +149,7 @@ TEST_F(MessageReaderTest, TwoMessages_Separately) { |
// Read() returns error. |
TEST_F(MessageReaderTest, ReadError) { |
- socket_.AppendReadError(net::ERR_FAILED); |
+ socket_.SetReadError(net::ERR_FAILED); |
EXPECT_CALL(callback_, OnMessage()).Times(0); |
@@ -159,6 +159,18 @@ TEST_F(MessageReaderTest, ReadError) { |
EXPECT_FALSE(reader_); |
} |
+// Read() returns 0 (end of stream). |
+TEST_F(MessageReaderTest, EndOfStream) { |
+ socket_.SetReadError(0); |
+ |
+ EXPECT_CALL(callback_, OnMessage()).Times(0); |
+ |
+ InitReader(); |
+ |
+ EXPECT_EQ(0, read_error_); |
+ EXPECT_FALSE(reader_); |
+} |
+ |
// Verify that we the OnMessage callback is not reentered. |
TEST_F(MessageReaderTest, ReadFromCallback) { |
AddMessage(kTestMessage1); |