diff options
Diffstat (limited to 'cros_utils/timeline_test.py')
-rw-r--r-- | cros_utils/timeline_test.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/cros_utils/timeline_test.py b/cros_utils/timeline_test.py new file mode 100644 index 00000000..c93a1274 --- /dev/null +++ b/cros_utils/timeline_test.py @@ -0,0 +1,57 @@ +# Copyright 2012 Google Inc. All Rights Reserved. +"""Tests for time_line.py.""" + +from __future__ import print_function + +__author__ = 'yunlian@google.com (Yunlian Jiang)' + +import time +import unittest + +import timeline + + +class TimeLineTest(unittest.TestCase): + """Tests for the Timeline class.""" + + def testRecord(self): + tl = timeline.Timeline() + tl.Record('A') + t = time.time() + t1 = tl.events[0].timestamp + self.assertEqual(int(t1 - t), 0) + self.assertRaises(AssertionError, tl.Record, 'A') + + def testGetEvents(self): + tl = timeline.Timeline() + tl.Record('A') + e = tl.GetEvents() + self.assertEqual(e, ['A']) + tl.Record('B') + e = tl.GetEvents() + self.assertEqual(e, ['A', 'B']) + + def testGetEventTime(self): + tl = timeline.Timeline() + tl.Record('A') + t = time.time() + t1 = tl.GetEventTime('A') + self.assertEqual(int(t1 - t), 0) + self.assertRaises(IndexError, tl.GetEventTime, 'B') + + def testGetLastEventTime(self): + tl = timeline.Timeline() + self.assertRaises(IndexError, tl.GetLastEventTime) + tl.Record('A') + t = time.time() + t1 = tl.GetLastEventTime() + self.assertEqual(int(t1 - t), 0) + time.sleep(2) + tl.Record('B') + t = time.time() + t1 = tl.GetLastEventTime() + self.assertEqual(int(t1 - t), 0) + + +if __name__ == '__main__': + unittest.main() |