from datetime import datetime, timedelta from pytz import timezone from unittest.mock import Mock, patch from parameterized import parameterized, param import dateparser.timezone_parser from dateparser.timezone_parser import pop_tz_offset_from_string, get_local_tz_offset, StaticTzInfo from dateparser import parse from tests import BaseTestCase class TestTZPopping(BaseTestCase): def setUp(self): super().setUp() self.initial_string = self.datetime_string = self.timezone_offset = NotImplemented @parameterized.expand([ param('Sep 03 2014 | 4:32 pm EDT', -4), param('17th October, 2034 @ 01:08 am PDT', -7), param('17th October, 2034 @ 01:08 am (PDT)', -7), param('October 17, 2014 at 7:30 am PST', -8), param('20 Oct 2014 13:08 CET', +1), param('20 Oct 2014 13:08cet', +1), param('Nov 25 2014 | 10:17 pm EST', -5), param('Nov 25 2014 | 10:17 pm +0600', +6), param('Nov 25 2014 | 10:17 pm -0930', -9.5), param('20 Oct 2014 | 05:17 am -1200', -12), param('20 Oct 2014 | 05:17 am +0000', 0), param('20 Oct 2014 | 05:17 am -0000', 0), param('15 May 2004', None), param('Wed Aug 05 12:00:00 EDTERR 2015', None), param('Wed Aug 05 12:00:00 EDT 2015', -4), param('April 10, 2016 at 12:00:00 UTC', 0), param('April 10, 2016 at 12:00:00 MEZ', 1), param('April 10, 2016 at 12:00:00 MESZ', 2), param('April 10, 2016 at 12:00:00 GMT+2', 2), param('April 10, 2016 at 12:00:00 UTC+2:00', 2), param('April 10, 2016 at 12:00:00 GMT+02:00', 2), param('April 10, 2016 at 12:00:00 UTC+5:30', 5.5), param('April 10, 2016 at 12:00:00 GMT+05:30', 5.5), param('April 10, 2016 at 12:00:00 UTC-2', -2), param('April 10, 2016 at 12:00:00 GMT-2:00', -2), param('April 10, 2016 at 12:00:00 UTC-02:00', -2), param('April 10, 2016 at 12:00:00 GMT-9:30', -9.5), param('April 10, 2016 at 12:00:00 UTC-09:30', -9.5), param('Thu, 24 Nov 2016 16:03:00 UT', 0), param('Fri Sep 23 2016 10:34:51 GMT+0800 (CST)', 8), param('Fri Sep 23 2016 10:34:51 GMT+12', 12), param('Fri Sep 23 2016 10:34:51 UTC+13', 13), param('Fri Sep 23 2016 10:34:51 GMT+1245 (CST)', 12.75), param('Fri Sep 23 2016 10:34:51 UTC+1245', 12.75), param('2019-07-17T12:30:00.000-03:30', -3.5), param('2019-07-17T12:30:00.000-02:30', -2.5), ]) def test_extracting_valid_offset(self, initial_string, expected_offset): self.given_string(initial_string) self.when_offset_popped_from_string() self.then_offset_is(expected_offset) @parameterized.expand([ param('Sep 03 2014 | 4:32 pm EDT', 'Sep 03 2014 | 4:32 pm '), param('17th October, 2034 @ 01:08 am PDT', '17th October, 2034 @ 01:08 am '), param('October 17, 2014 at 7:30 am PST', 'October 17, 2014 at 7:30 am '), param('20 Oct 2014 13:08 CET', '20 Oct 2014 13:08 '), param('20 Oct 2014 13:08cet', '20 Oct 2014 13:08'), param('Nov 25 2014 | 10:17 pm EST', 'Nov 25 2014 | 10:17 pm '), param('17th October, 2034 @ 01:08 am +0700', '17th October, 2034 @ 01:08 am '), param('Sep 03 2014 4:32 pm +0630', 'Sep 03 2014 4:32 pm '), ]) def test_timezone_deleted_from_string(self, initial_string, result_string): self.given_string(initial_string) self.when_offset_popped_from_string() self.then_string_modified_to(result_string) def test_string_not_changed_if_no_timezone(self): self.given_string('15 May 2004') self.when_offset_popped_from_string() self.then_string_modified_to('15 May 2004') def given_string(self, string_): self.initial_string = string_ def when_offset_popped_from_string(self): self.datetime_string, timezone_offset = pop_tz_offset_from_string(self.initial_string) if timezone_offset: self.timezone_offset = timezone_offset.utcoffset('') else: self.timezone_offset = timezone_offset def then_string_modified_to(self, expected_string): self.assertEqual(expected_string, self.datetime_string) def then_offset_is(self, expected_offset): delta = timedelta(hours=expected_offset) if expected_offset is not None else None self.assertEqual(delta, self.timezone_offset) class TestLocalTZOffset(BaseTestCase): def setUp(self): super().setUp() self.timezone_offset = NotImplemented @parameterized.expand([ param(utc='2014-08-20 4:32', local='2014-08-20 8:32', offset=+4), param(utc='2052-01-02 11:07', local='2052-01-02 10:07', offset=-1), param(utc='2013-12-31 23:59', local='2014-01-01 00:29', offset=+0.5), param(utc='2011-07-14 11:59', local='2011-07-13 23:59', offset=-12), param(utc='2014-10-18 17:41', local='2014-10-18 17:41', offset=0), ]) def test_timezone_offset_calculation(self, utc, local, offset): self.given_utc_time(utc) self.given_local_time(local) self.when_offset_popped_from_string() self.then_offset_is(offset) def given_utc_time(self, datetime_string): self._given_time(datetime_string, 'utcnow') def given_local_time(self, datetime_string): self._given_time(datetime_string, 'now') def when_offset_popped_from_string(self): self.timezone_offset = get_local_tz_offset() def then_offset_is(self, expected_offset): delta = timedelta(seconds=3600 * expected_offset) if expected_offset is not None else None self.assertEqual(delta, self.timezone_offset) def _given_time(self, datetime_string, getter_name): datetime_cls = dateparser.timezone_parser.datetime if not isinstance(datetime_cls, Mock): datetime_cls = Mock(wraps=datetime) datetime_obj = datetime.strptime(datetime_string, '%Y-%m-%d %H:%M') setattr(datetime_cls, getter_name, Mock(return_value=datetime_obj)) self.add_patch( patch('dateparser.timezone_parser.datetime', new=datetime_cls) ) class TestTimeZoneConversion(BaseTestCase): def setUp(self): super().setUp() self.settings = {} self.parser = parse self.result = NotImplemented @parameterized.expand([ param('2015-12-31 10:04 AM', 'Asia/Karachi', 'UTC', datetime(2015, 12, 31, 5, 4)), param('2015-12-30 10:04 AM', 'Asia/Karachi', '+0200', datetime(2015, 12, 30, 7, 4)), ]) def test_timezone_conversion(self, datestring, from_tz, to_tz, expected): self.given_from_timezone(from_tz) self.given_to_timezone(to_tz) self.when_date_is_parsed(datestring) self.then_date_is(expected) def given_from_timezone(self, timezone): self.settings['TIMEZONE'] = timezone def given_to_timezone(self, timezone): self.settings['TO_TIMEZONE'] = timezone def when_date_is_parsed(self, datestring): self.result = self.parser(datestring, settings=self.settings) def then_date_is(self, date): self.assertEqual(date, self.result) class TestStaticTzInfo(BaseTestCase): def setUp(self): super().setUp() @parameterized.expand([ param(given_date=datetime(2007, 1, 18, tzinfo=timezone('UTC'))), param(given_date=datetime(2003, 3, 31, tzinfo=timezone('US/Arizona'))), param(given_date=datetime(2000, 2, 20, tzinfo=timezone('Pacific/Samoa'))), ]) def test_localize_raises_error_if_date_has_tzinfo(self, given_date): self.timezone_info = StaticTzInfo('UTC\\+00:00', timedelta(0)) self.when_date_is_localized(given_date) self.then_error_was_raised(ValueError, ['Not naive datetime (tzinfo is already set)']) def when_date_is_localized(self, given_date): try: self.localized_date = self.timezone_info.localize(given_date) except Exception as error: self.error = error def then_localized_date_is(self, expected_date, expected_tzname): self.assertEqual(self.localized_date.date(), expected_date.date()) self.assertEqual(self.localized_date.tzname(), expected_tzname)