Index: tools/battor_agent/battor_agent_unittest.cc |
diff --git a/tools/battor_agent/battor_agent_unittest.cc b/tools/battor_agent/battor_agent_unittest.cc |
index 291c109c9ae016f7a3620e3bedbf1678cca3e631..611f809dc893612485c05f17f43161ca226a1dd1 100644 |
--- a/tools/battor_agent/battor_agent_unittest.cc |
+++ b/tools/battor_agent/battor_agent_unittest.cc |
@@ -301,6 +301,15 @@ class BattOrAgentTest : public testing::Test, public BattOrAgent::Listener { |
return; |
OnBytesSent(true); |
+ if (end_state == BattOrAgentState::INIT_SENT) |
+ return; |
+ |
+ OnMessageRead(true, BATTOR_MESSAGE_TYPE_CONTROL_ACK, |
+ ToCharVector(kInitAck)); |
+ if (end_state == BattOrAgentState::INIT_ACKED) |
+ return; |
+ |
+ OnBytesSent(true); |
if (end_state == BattOrAgentState::GIT_FIRMWARE_HASH_REQUEST_SENT) |
return; |