[battor agent] Add 'read more bytes' logic in the BattOr connection

tools/battor_agent/battor_connection_impl_unittest.cc

Index: tools/battor_agent/battor_connection_impl_unittest.cc
diff --git a/tools/battor_agent/battor_connection_impl_unittest.cc b/tools/battor_agent/battor_connection_impl_unittest.cc
index 4e19ee123c5609aa21ec1efb33f413d8450b0696..a9027942d8b91a76f77724bd195dc21a4c53a114 100644
--- a/tools/battor_agent/battor_connection_impl_unittest.cc
+++ b/tools/battor_agent/battor_connection_impl_unittest.cc
@@ -230,8 +230,9 @@ TEST_F(BattOrConnectionImplTest, ReadMessageEndsMidMessageByte) {
   SendBytesRaw(data, 5);
   ReadMessage(BATTOR_MESSAGE_TYPE_CONTROL);
 
-  ASSERT_TRUE(IsReadComplete());
-  ASSERT_FALSE(GetReadSuccess());
+  // The first read should recognize that a second read is necessary, but the
+  // second read will hang because no bytes ever come in.
+  ASSERT_FALSE(IsReadComplete());
 }
 
 TEST_F(BattOrConnectionImplTest, ReadMessageMissingEndByte) {
@@ -251,8 +252,9 @@ TEST_F(BattOrConnectionImplTest, ReadMessageMissingEndByte) {
   SendBytesRaw(data, 6);
   ReadMessage(BATTOR_MESSAGE_TYPE_CONTROL);
 
-  ASSERT_TRUE(IsReadComplete());
-  ASSERT_FALSE(GetReadSuccess());
+  // The first read should recognize that a second read is necessary, but the
+  // second read will hang because no bytes ever come in.
+  ASSERT_FALSE(IsReadComplete());
 }
 
 TEST_F(BattOrConnectionImplTest, ReadMessageWithEscapeCharacters) {