# Copyright (C) 2007, 2009, 2011 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.assertEquals(
            (1120153132.3508501, -18000),
            timestamp.unpack_highres_date('Thu 2005-06-30 12:38:52.350850105 -0500'))
        self.assertEquals(
            (1120153132.3508501, 0),
            timestamp.unpack_highres_date('Thu 2005-06-30 17:38:52.350850105 +0000'))
        self.assertEquals(
            (1120153132.3508501, 7200),
            timestamp.unpack_highres_date('Thu 2005-06-30 19:38:52.350850105 +0200'))
        self.assertEquals(
            (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.assertEquals(t, t2)
        self.assertEquals(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.assertEquals(t, t2,
                'Failed on date %r, %s,%s diff:%s' % (date, t, o, t2-t))
            self.assertEquals(o, o2,
                'Failed on date %r, %s,%s diff:%s' % (date, t, o, t2-t))
