| OLD | NEW |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 import unittest | 5 import unittest |
| 6 | 6 |
| 7 from battor import battor_error | 7 from battor import battor_error |
| 8 from battor import battor_wrapper | 8 from battor import battor_wrapper |
| 9 from devil.android import battery_utils | 9 from devil.android import battery_utils |
| 10 from telemetry.internal.platform.tracing_agent import battor_tracing_agent | 10 from telemetry.internal.platform.tracing_agent import battor_tracing_agent |
| 11 from telemetry.timeline import trace_data | 11 from telemetry.timeline import trace_data |
| 12 from telemetry.timeline import tracing_config | 12 from telemetry.timeline import tracing_config |
| 13 | 13 |
| 14 _BATTOR_RETURN = ['fake', 'battor', 'data'] | 14 _BATTOR_RETURN = 'fake\nbattor\ndata' |
| 15 | 15 |
| 16 | 16 |
| 17 class FakeBatteryUtils(object): | 17 class FakeBatteryUtils(object): |
| 18 def __init__(self, device): | 18 def __init__(self, device): |
| 19 self._device = device | 19 self._device = device |
| 20 self._charging_state = True | 20 self._charging_state = True |
| 21 | 21 |
| 22 def SetCharging(self, state): | 22 def SetCharging(self, state): |
| 23 self._charging_state = state | 23 self._charging_state = state |
| 24 | 24 |
| (...skipping 138 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 163 self.android_agent.StopAgentTracing() | 163 self.android_agent.StopAgentTracing() |
| 164 self.assertTrue(self.android_agent._battor._stop_tracing_called) | 164 self.assertTrue(self.android_agent._battor._stop_tracing_called) |
| 165 | 165 |
| 166 def testCollectAgentTraceData(self): | 166 def testCollectAgentTraceData(self): |
| 167 builder = trace_data.TraceDataBuilder() | 167 builder = trace_data.TraceDataBuilder() |
| 168 self.android_agent.CollectAgentTraceData(builder) | 168 self.android_agent.CollectAgentTraceData(builder) |
| 169 self.assertTrue(self.android_agent._battor._collect_trace_data_called) | 169 self.assertTrue(self.android_agent._battor._collect_trace_data_called) |
| 170 builder = builder.AsData() | 170 builder = builder.AsData() |
| 171 self.assertTrue(builder.HasTraceFor(trace_data.BATTOR_TRACE_PART)) | 171 self.assertTrue(builder.HasTraceFor(trace_data.BATTOR_TRACE_PART)) |
| 172 data_from_builder = builder.GetTraceFor(trace_data.BATTOR_TRACE_PART) | 172 data_from_builder = builder.GetTraceFor(trace_data.BATTOR_TRACE_PART) |
| 173 self.assertEqual('\n'.join(_BATTOR_RETURN), data_from_builder) | 173 self.assertEqual(_BATTOR_RETURN, data_from_builder) |
| 174 | 174 |
| 175 def testAndroidCharging(self): | 175 def testAndroidCharging(self): |
| 176 self.assertTrue(self.android_agent._battery.GetCharging()) | 176 self.assertTrue(self.android_agent._battery.GetCharging()) |
| 177 self.assertTrue(self.android_agent.StartAgentTracing(self._config, 0)) | 177 self.assertTrue(self.android_agent.StartAgentTracing(self._config, 0)) |
| 178 self.assertFalse(self.android_agent._battery.GetCharging()) | 178 self.assertFalse(self.android_agent._battery.GetCharging()) |
| 179 self.android_agent.StopAgentTracing() | 179 self.android_agent.StopAgentTracing() |
| 180 self.assertTrue(self.android_agent._battery.GetCharging()) | 180 self.assertTrue(self.android_agent._battery.GetCharging()) |
| 181 | 181 |
| 182 def testRecordClockSyncMarker(self): | 182 def testRecordClockSyncMarker(self): |
| 183 def callback_with_exception(a, b): | 183 def callback_with_exception(a, b): |
| 184 del a # unused | 184 del a # unused |
| 185 del b # unused | 185 del b # unused |
| 186 raise Exception | 186 raise Exception |
| 187 def callback_without_exception(a, b): | 187 def callback_without_exception(a, b): |
| 188 del a # unused | 188 del a # unused |
| 189 del b # unused | 189 del b # unused |
| 190 | 190 |
| 191 self.android_agent.RecordClockSyncMarker('123', callback_without_exception) | 191 self.android_agent.RecordClockSyncMarker('123', callback_without_exception) |
| 192 with self.assertRaises(Exception): | 192 with self.assertRaises(Exception): |
| 193 self.android_agent.RecordClockSyncMarker('abc', callback_with_exception) | 193 self.android_agent.RecordClockSyncMarker('abc', callback_with_exception) |
| OLD | NEW |