~bzr-pqm/bzr/bzr.dev

# 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 bzrlib import (

    tests,

    timestamp,

    )

from bzrlib.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 xrange(500):

            t += random.random()*24*3600*30

            o = ((o/3600 + 13) % 25 - 12)*3600 # Add 1 wrap around from [-12, 12]

            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))