OLD | NEW |
(Empty) | |
| 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is govered by a BSD-style |
| 3 # license that can be found in the LICENSE file or at |
| 4 # https://developers.google.com/open-source/licenses/bsd |
| 5 |
| 6 """Unittest for timestr module.""" |
| 7 |
| 8 import datetime |
| 9 import time |
| 10 import unittest |
| 11 |
| 12 from framework import timestr |
| 13 |
| 14 |
| 15 class TimeStrTest(unittest.TestCase): |
| 16 """Unit tests for timestr routines.""" |
| 17 |
| 18 def testFormatAbsoluteDate(self): |
| 19 now = datetime.datetime(2008, 1, 1) |
| 20 |
| 21 def GetDate(*args): |
| 22 date = datetime.datetime(*args) |
| 23 return timestr.FormatAbsoluteDate( |
| 24 time.mktime(date.timetuple()), clock=lambda: now) |
| 25 |
| 26 self.assertEquals(GetDate(2008, 1, 1), 'Today') |
| 27 self.assertEquals(GetDate(2007, 12, 31), 'Yesterday') |
| 28 self.assertEquals(GetDate(2007, 12, 30), 'Dec 30') |
| 29 self.assertEquals(GetDate(2007, 1, 1), 'Jan 2007') |
| 30 self.assertEquals(GetDate(2007, 1, 2), 'Jan 2007') |
| 31 self.assertEquals(GetDate(2007, 12, 31), 'Yesterday') |
| 32 self.assertEquals(GetDate(2006, 12, 31), 'Dec 2006') |
| 33 self.assertEquals(GetDate(2007, 7, 1), 'Jul 1') |
| 34 self.assertEquals(GetDate(2007, 6, 30), 'Jun 2007') |
| 35 self.assertEquals(GetDate(2008, 1, 3), 'Jan 2008') |
| 36 |
| 37 # Leap year fun |
| 38 now = datetime.datetime(2008, 3, 1) |
| 39 self.assertEquals(GetDate(2008, 2, 29), 'Yesterday') |
| 40 |
| 41 # Clock skew |
| 42 now = datetime.datetime(2008, 1, 1, 23, 59, 59) |
| 43 self.assertEquals(GetDate(2008, 1, 2), 'Today') |
| 44 now = datetime.datetime(2007, 12, 31, 23, 59, 59) |
| 45 self.assertEquals(GetDate(2008, 1, 1), 'Today') |
| 46 self.assertEquals(GetDate(2008, 1, 2), 'Jan 2008') |
| 47 |
| 48 def testFormatRelativeDate(self): |
| 49 now = time.mktime(datetime.datetime(2008, 1, 1).timetuple()) |
| 50 |
| 51 def TestSecsAgo(secs_ago, expected, expected_recent_only): |
| 52 test_time = now - secs_ago |
| 53 actual = timestr.FormatRelativeDate( |
| 54 test_time, clock=lambda: now) |
| 55 self.assertEquals(actual, expected) |
| 56 actual_recent_only = timestr.FormatRelativeDate( |
| 57 test_time, clock=lambda: now, recent_only=True) |
| 58 self.assertEquals(actual_recent_only, expected_recent_only) |
| 59 |
| 60 TestSecsAgo(10 * 24 * 60 * 60, ', 2007', '') |
| 61 TestSecsAgo(5 * 24 * 60 * 60 - 1, ' (4 days ago)', '4 days ago') |
| 62 TestSecsAgo(5 * 60 * 60 - 1, ' (4 hours ago)', '4 hours ago') |
| 63 TestSecsAgo(5 * 60 - 1, ' (4 minutes ago)', '4 minutes ago') |
| 64 TestSecsAgo(2 * 60 - 1, ' (1 minute ago)', '1 minute ago') |
| 65 TestSecsAgo(60 - 1, ' (moments ago)', 'moments ago') |
| 66 TestSecsAgo(0, ' (moments ago)', 'moments ago') |
| 67 TestSecsAgo(-10, ' (moments ago)', 'moments ago') |
| 68 TestSecsAgo(-100, ' (in the future)', '') |
| 69 |
| 70 def testGetHumanScaleDate(self): |
| 71 """Tests GetHumanScaleDate().""" |
| 72 now = time.mktime(datetime.datetime(2008, 4, 10, 20, 50, 30).timetuple()) |
| 73 |
| 74 def GetDate(*args): |
| 75 date = datetime.datetime(*args) |
| 76 timestamp = time.mktime(date.timetuple()) |
| 77 return timestr.GetHumanScaleDate(timestamp, now=now) |
| 78 |
| 79 self.assertEquals( |
| 80 GetDate(2008, 4, 10, 15), ('Today', '5 hours ago')) |
| 81 self.assertEquals( |
| 82 GetDate(2008, 4, 10, 19, 55), ('Today', '55 min ago')) |
| 83 self.assertEquals( |
| 84 GetDate(2008, 4, 10, 20, 48, 35), ('Today', '1 min ago')) |
| 85 self.assertEquals( |
| 86 GetDate(2008, 4, 10, 20, 49, 35), ('Today', 'moments ago')) |
| 87 self.assertEquals( |
| 88 GetDate(2008, 4, 10, 20, 50, 55), ('Today', 'moments ago')) |
| 89 self.assertEquals( |
| 90 GetDate(2008, 4, 9, 15), ('Yesterday', '29 hours ago')) |
| 91 self.assertEquals( |
| 92 GetDate(2008, 4, 5, 15), ('Last 7 days', 'Apr 05, 2008')) |
| 93 self.assertEquals( |
| 94 GetDate(2008, 3, 22, 15), ('Last 30 days', 'Mar 22, 2008')) |
| 95 self.assertEquals( |
| 96 GetDate(2008, 1, 2, 15), ('Earlier this year', 'Jan 02, 2008')) |
| 97 self.assertEquals( |
| 98 GetDate(2007, 12, 31, 15), ('Older', 'Dec 31, 2007')) |
| 99 self.assertEquals( |
| 100 GetDate(2008, 4, 11, 20, 49, 35), ('Future', 'Later')) |
| 101 |
| 102 |
| 103 if __name__ == '__main__': |
| 104 unittest.main() |
OLD | NEW |