tools/battor_agent: Adds missing RunUntilIdles in unit test

tools/battor_agent/battor_agent_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "tools/battor_agent/battor_agent.h"
 
 #include "base/test/test_simple_task_runner.h"
 #include "base/thread_task_runner_handle.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 438 matching lines @@
 
   RunStopTracingTo(BattOrAgentState::SAMPLES_REQUEST_SENT);
 
   // Send the calibration frame.
   BattOrFrameHeader cal_frame_header{0, 2 * sizeof(RawBattOrSample)};
   RawBattOrSample cal_frame[] = {
       RawBattOrSample{1, 1}, RawBattOrSample{2, 2},
   };
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(cal_frame_header, cal_frame, 2));
+  GetTaskRunner()->RunUntilIdle();
 
   // Send the two real data frames.
   BattOrFrameHeader frame_header1{0, 3 * sizeof(RawBattOrSample)};
   RawBattOrSample frame1[] = {
       RawBattOrSample{1, 1}, RawBattOrSample{2, 2}, RawBattOrSample{3, 3},
   };
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(frame_header1, frame1, 3));
+  GetTaskRunner()->RunUntilIdle();
 
   BattOrFrameHeader frame_header2{0, 1 * sizeof(RawBattOrSample)};
   RawBattOrSample frame2[] = {RawBattOrSample{1, 1}};
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(frame_header2, frame2, 1));
+  GetTaskRunner()->RunUntilIdle();
 
   // Send an empty last frame to indicate that we're done.
   BattOrFrameHeader frame_header3{0, 0 * sizeof(RawBattOrSample)};
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(frame_header3, nullptr, 0));
+  GetTaskRunner()->RunUntilIdle();
 
   EXPECT_TRUE(IsCommandComplete());
   EXPECT_EQ(BATTOR_ERROR_NONE, GetCommandError());
   EXPECT_EQ("0.00 -0.3 -0.6\n1.00 0.3 0.6\n2.00 0.9 1.8\n3.00 -0.3 -0.6\n",
             GetTrace());
 }
 
 TEST_F(BattOrAgentTest, StopTracingFailsWithoutConnection) {
   GetAgent()->StopTracing();
   GetTaskRunner()->RunUntilIdle();
@@ skipping 182 matching lines @@
   RunStopTracingTo(BattOrAgentState::SAMPLES_REQUEST_SENT);
 
   // Send a calibration frame with a mismatch between the frame length in the
   // header and the actual frame length.
   BattOrFrameHeader cal_frame_header{0, 1 * sizeof(RawBattOrSample)};
   RawBattOrSample cal_frame[] = {
       RawBattOrSample{1, 1}, RawBattOrSample{2, 2},
   };
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(cal_frame_header, cal_frame, 2));
+  GetTaskRunner()->RunUntilIdle();
 
   EXPECT_TRUE(IsCommandComplete());
   EXPECT_EQ(BATTOR_ERROR_UNEXPECTED_MESSAGE, GetCommandError());
 }
 
 TEST_F(BattOrAgentTest, StopTracingFailsIfDataFrameHasWrongLength) {
   RunStopTracingTo(BattOrAgentState::SAMPLES_REQUEST_SENT);
 
   BattOrFrameHeader cal_frame_header{0, 1 * sizeof(RawBattOrSample)};
   RawBattOrSample cal_frame[] = {
       RawBattOrSample{1, 1},
   };
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(cal_frame_header, cal_frame, 1));
+  GetTaskRunner()->RunUntilIdle();
 
   // Send a data frame with a mismatch between the frame length in the
   // header and the actual frame length.
   BattOrFrameHeader frame_header{0, 2 * sizeof(RawBattOrSample)};
   RawBattOrSample frame[] = {RawBattOrSample{1, 1}};
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(frame_header, frame, 1));
   GetTaskRunner()->RunUntilIdle();
 
   EXPECT_TRUE(IsCommandComplete());
   EXPECT_EQ(BATTOR_ERROR_UNEXPECTED_MESSAGE, GetCommandError());
 }
 
 TEST_F(BattOrAgentTest, StopTracingFailsIfCalibrationFrameMissingByte) {
   RunStopTracingTo(BattOrAgentState::SAMPLES_REQUEST_SENT);
 
   BattOrFrameHeader cal_frame_header{0, 2 * sizeof(RawBattOrSample)};
   RawBattOrSample cal_frame[] = {
       RawBattOrSample{1, 1}, RawBattOrSample{2, 2},
   };
 
   // Remove the last byte from the frame to make it invalid.
   scoped_ptr<vector<char>> cal_frame_bytes =
       CreateFrame(cal_frame_header, cal_frame, 2);
   cal_frame_bytes->pop_back();
 
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             std::move(cal_frame_bytes));
+  GetTaskRunner()->RunUntilIdle();
 
   EXPECT_TRUE(IsCommandComplete());
   EXPECT_EQ(BATTOR_ERROR_UNEXPECTED_MESSAGE, GetCommandError());
 }
 
 TEST_F(BattOrAgentTest, StopTracingFailsIfDataFrameMissingByte) {
   RunStopTracingTo(BattOrAgentState::SAMPLES_REQUEST_SENT);
 
   BattOrFrameHeader cal_frame_header{0, 1 * sizeof(RawBattOrSample)};
   RawBattOrSample cal_frame[] = {
       RawBattOrSample{1, 1},
   };
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             CreateFrame(cal_frame_header, cal_frame, 1));
+  GetTaskRunner()->RunUntilIdle();
 
   BattOrFrameHeader frame_header{0, 1};
   RawBattOrSample frame[] = {RawBattOrSample{1, 1}};
 
   // Remove the last byte from the frame to make it invalid.
   scoped_ptr<vector<char>> frame_bytes = CreateFrame(frame_header, frame, 2);
   frame_bytes->pop_back();
 
   GetAgent()->OnMessageRead(true, BATTOR_MESSAGE_TYPE_SAMPLES,
                             std::move(frame_bytes));
   GetTaskRunner()->RunUntilIdle();
 
   EXPECT_TRUE(IsCommandComplete());
   EXPECT_EQ(BATTOR_ERROR_UNEXPECTED_MESSAGE, GetCommandError());
 }
 
 }  // namespace battor