diff options
Diffstat (limited to 'catapult/common/battor/battor/battor_wrapper_unittest.py')
-rw-r--r-- | catapult/common/battor/battor/battor_wrapper_unittest.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/catapult/common/battor/battor/battor_wrapper_unittest.py b/catapult/common/battor/battor/battor_wrapper_unittest.py index 3e48a336..7e14c349 100644 --- a/catapult/common/battor/battor/battor_wrapper_unittest.py +++ b/catapult/common/battor/battor/battor_wrapper_unittest.py @@ -359,6 +359,28 @@ class BattOrWrapperTest(unittest.TestCase): dependency_manager.DependencyManager._version_return = 'cbaa843' self.assertTrue(self._battor._FlashBattOr()) + def testCollectTraceDataNoStartTime(self): + self._battor = battor_wrapper.BattOrWrapper('linux') + self._DefaultBattOrReplacements() + self._battor.StartShell() + self._battor.StartTracing() + self._battor.GetShellReturnCode = lambda *unused: 0 + self._battor.StopTracing() + self._battor._start_tracing_time = None + with self.assertRaises(battor_error.BattOrError): + self._battor.CollectTraceData() + + def testCollectTraceDataNoStopTime(self): + self._battor = battor_wrapper.BattOrWrapper('linux') + self._DefaultBattOrReplacements() + self._battor.StartShell() + self._battor.StartTracing() + self._battor.GetShellReturnCode = lambda *unused: 0 + self._battor.StopTracing() + self._battor._stop_tracing_time = None + with self.assertRaises(battor_error.BattOrError): + self._battor.CollectTraceData() + if __name__ == '__main__': logging.getLogger().setLevel(logging.DEBUG) |