# Copyright (C) 2007, 2009, 2010, 2011, 2016 Canonical Ltd # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA import random import time from breezy import ( tests, timestamp, ) from breezy.osutils import local_time_offset class TestPatchHeader(tests.TestCase): def test_format_patch_date(self): # epoch is always in utc self.assertEqual('1970-01-01 00:00:00 +0000', timestamp.format_patch_date(0)) self.assertEqual('1970-01-01 00:00:00 +0000', timestamp.format_patch_date(0, 5 * 3600)) self.assertEqual('1970-01-01 00:00:00 +0000', timestamp.format_patch_date(0, -5 * 3600)) # regular timestamp with typical timezone self.assertEqual('2007-03-06 10:04:19 -0500', timestamp.format_patch_date(1173193459, -5 * 3600)) # the timezone part is HHMM self.assertEqual('2007-03-06 09:34:19 -0530', timestamp.format_patch_date(1173193459, -5.5 * 3600)) # timezones can be offset by single minutes (but no less) self.assertEqual('2007-03-06 15:05:19 +0001', timestamp.format_patch_date(1173193459, +1 * 60)) def test_parse_patch_date(self): self.assertEqual( (0, 0), timestamp.parse_patch_date('1970-01-01 00:00:00 +0000')) # even though we don't emit pre-epoch dates, we can parse them self.assertEqual( (0, -5 * 3600), timestamp.parse_patch_date('1969-12-31 19:00:00 -0500')) self.assertEqual( (0, +5 * 3600), timestamp.parse_patch_date('1970-01-01 05:00:00 +0500')) self.assertEqual( (1173193459, -5 * 3600), timestamp.parse_patch_date('2007-03-06 10:04:19 -0500')) # offset of three minutes self.assertEqual( (1173193459, +3 * 60), timestamp.parse_patch_date('2007-03-06 15:07:19 +0003')) # No space between time and offset self.assertEqual( (1173193459, -5 * 3600), timestamp.parse_patch_date('2007-03-06 10:04:19-0500')) # Extra spacing self.assertEqual( (1173193459, -5 * 3600), timestamp.parse_patch_date('2007-03-06 10:04:19 -0500')) def test_parse_patch_date_bad(self): self.assertRaises(ValueError, timestamp.parse_patch_date, 'NOT A TIME') # Extra data at end self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 -0500x') # Missing day self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03 10:04:19 -0500') # Missing seconds self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04 -0500') # Missing offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19') # Missing plus or minus in offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 0500') # Invalid hour in offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 +2400') self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 -2400') # Invalid minute in offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 -0560') # Too many digits in offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 79500') # Minus sign in middle of offset self.assertRaises(ValueError, timestamp.parse_patch_date, '2007-03-06 10:04:19 +05-5') class UnpackHighresDateTests(tests.TestCase): def test_unpack_highres_date(self): self.assertEqual( (1120153132.3508501, -18000), timestamp.unpack_highres_date( 'Thu 2005-06-30 12:38:52.350850105 -0500')) self.assertEqual( (1120153132.3508501, 0), timestamp.unpack_highres_date( 'Thu 2005-06-30 17:38:52.350850105 +0000')) self.assertEqual( (1120153132.3508501, 7200), timestamp.unpack_highres_date( 'Thu 2005-06-30 19:38:52.350850105 +0200')) self.assertEqual( (1152428738.867522, 19800), timestamp.unpack_highres_date( 'Sun 2006-07-09 12:35:38.867522001 +0530')) def test_random(self): t = time.time() o = local_time_offset() t2, o2 = timestamp.unpack_highres_date( timestamp.format_highres_date(t, o)) self.assertEqual(t, t2) self.assertEqual(o, o2) t -= 24 * 3600 * 365 * 2 # Start 2 years ago o = -12 * 3600 for count in range(500): t += random.random() * 24 * 3600 * 30 try: time.gmtime(t + o) except (OverflowError, ValueError): # We've reached the maximum for time_t on this platform break if time.localtime(t).tm_year > 9998: # strptime doesn't always understand years with more than 4 # digits. break # Add 1 wrap around from [-12, 12] o = ((o / 3600 + 13) % 25 - 12) * 3600 date = timestamp.format_highres_date(t, o) t2, o2 = timestamp.unpack_highres_date(date) self.assertEqual( t, t2, 'Failed on date %r, %s,%s diff:%s' % (date, t, o, t2 - t)) self.assertEqual( o, o2, 'Failed on date %r, %s,%s diff:%s' % (date, t, o, t2 - t))