| Index: telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py
|
| diff --git a/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py b/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py
|
| index 8fe1c4571e814343efd78e077b05b25ad491a723..3e06bfe31ae7141c10d9f5fcd6c73467371d751c 100644
|
| --- a/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py
|
| +++ b/telemetry/telemetry/internal/platform/tracing_agent/battor_tracing_agent_unittest.py
|
| @@ -169,9 +169,9 @@ class BattOrTracingAgentTest(unittest.TestCase):
|
| self.android_agent.CollectAgentTraceData(builder)
|
| self.assertTrue(self.android_agent._battor._collect_trace_data_called)
|
| builder = builder.AsData()
|
| - self.assertTrue(builder.HasTraceFor(trace_data.BATTOR_TRACE_PART))
|
| - data_from_builder = builder.GetTraceFor(trace_data.BATTOR_TRACE_PART)
|
| - self.assertEqual(_BATTOR_RETURN, data_from_builder)
|
| + self.assertTrue(builder.HasTracesFor(trace_data.BATTOR_TRACE_PART))
|
| + data_from_builder = builder.GetTracesFor(trace_data.BATTOR_TRACE_PART)
|
| + self.assertEqual([_BATTOR_RETURN], data_from_builder)
|
|
|
| def testAndroidCharging(self):
|
| self.assertTrue(self.android_agent._battery.GetCharging())
|
|
|