from datetime import datetime, date, timedelta
import unittest
import moment

# Helpful strftime() format that imcludes all parts of the date including the time zone.
fmt = "%Y-%m-%d %H:%M:%S %Z"

class TestMoment(unittest.TestCase):

  new_york = [
    # - 1918 -
    [datetime(1918, 3, 31, 6, 59, 59), -1633280401000, "EST", 300, 1, 59],
    [datetime(1918, 3, 31, 7, 0, 0), -1633280400000, "EDT", 240, 3, 0],
    [datetime(1918, 10, 27, 5, 59, 59), -1615140001000, "EDT", 240, 1, 59],
    [datetime(1918, 10, 27, 6, 0, 0), -1615140000000, "EST", 300, 1, 0],
    # - 1979 -
    [datetime(1979, 4, 29, 6, 59, 59), 294217199000, "EST", 300, 1, 59],
    [datetime(1979, 4, 29, 7, 0, 0), 294217200000, "EDT", 240, 3, 0],
    [datetime(1979, 10, 28, 5, 59, 59), 309938399000, "EDT", 240, 1, 59],
    [datetime(1979, 10, 28, 6, 0, 0), 309938400000, "EST", 300, 1, 0],
    # - 2037 -
    [datetime(2037, 3, 8, 6, 59, 59), 2120108399000, "EST", 300, 1, 59],
    [datetime(2037, 3, 8, 7, 0, 0), 2120108400000, "EDT", 240, 3, 0],
    [datetime(2037, 11, 1, 5, 59, 59), 2140667999000, "EDT", 240, 1, 59]
  ]
  new_york_errors = [
    ["America/New_York", "2037-3-8 6:59:59", TypeError],
    ["America/New_York", [2037, 3, 8, 6, 59, 59], TypeError],
    ["America/new_york", datetime(1979, 4, 29, 6, 59, 59), KeyError]
  ]

  los_angeles = [
    # - 1918 -
    # Spanning non-existent hour
    [datetime(1918, 3, 31, 1, 59, 59, 0), -1633269601000, "PST", 480, 1, 59],
    [datetime(1918, 3, 31, 2, 0, 0, 0), -1633273200000, "PST", 480, 1, 0],
    [datetime(1918, 3, 31, 2, 59, 59, 0), -1633269601000, "PST", 480, 1, 59],
    [datetime(1918, 3, 31, 3, 0, 0, 0), -1633269600000, "PDT", 420, 3, 0],
    # Spanning doubly-existent hour
    [datetime(1918, 10, 27, 0, 59, 59, 0), -1615132801000, "PDT", 420, 0, 59],
    [datetime(1918, 10, 27, 1, 0, 0, 0), -1615132800000, "PDT", 420, 1, 0],
    [datetime(1918, 10, 27, 1, 59, 59, 0), -1615129201000, "PDT", 420, 1, 59],
    [datetime(1918, 10, 27, 2, 0, 0, 0), -1615125600000, "PST", 480, 2, 0],
    # - 2008 -
    # Spanning non-existent hour
    [datetime(2008, 3, 9, 1, 59, 59, 0), 1205056799000, "PST", 480, 1, 59],
    [datetime(2008, 3, 9, 2, 0, 0, 0), 1205053200000, "PST", 480, 1, 0],
    [datetime(2008, 3, 9, 2, 59, 59, 0), 1205056799000, "PST", 480, 1, 59],
    [datetime(2008, 3, 9, 3, 0, 0, 0), 1205056800000, "PDT", 420, 3, 0],
    # Spanning doubly-existent hour
    [datetime(2008, 11, 2, 0, 59, 59, 0), 1225612799000, "PDT", 420, 0, 59],
    [datetime(2008, 11, 2, 1, 0, 0, 0), 1225612800000, "PDT", 420, 1, 0],
    [datetime(2008, 11, 2, 1, 59, 59, 0), 1225616399000, "PDT", 420, 1, 59],
    [datetime(2008, 11, 2, 2, 0, 0, 0), 1225620000000, "PST", 480, 2, 0],
    # - 2037 -
    [datetime(2037, 3, 8, 1, 59, 59, 0), 2120119199000, "PST", 480, 1, 59],
    [datetime(2037, 3, 8, 2, 0, 0, 0), 2120115600000, "PST", 480, 1, 0],
    [datetime(2037, 11, 1, 0, 59, 59, 0), 2140675199000, "PDT", 420, 0, 59],
    [datetime(2037, 11, 1, 1, 0, 0, 0), 2140675200000, "PDT", 420, 1, 0],
  ]

  def assertMatches(self, data_entry, moment_obj):
    date, timestamp, abbr, offset, hour, minute = data_entry
    dt        = moment_obj.datetime()
    self.assertEqual(moment_obj.timestamp, timestamp)
    self.assertEqual(moment_obj.zoneAbbr(), abbr)
    self.assertEqual(moment_obj.zoneOffset(), timedelta(minutes=-offset))
    self.assertEqual(dt.hour, hour)
    self.assertEqual(dt.minute, minute)

  # For each UTC date, convert to New York time and compare with expected values
  def test_standard_entry(self):
    name = "America/New_York"
    data = self.new_york
    for entry in data:
      date      = entry[0]
      timestamp = entry[1]
      m   = moment.tz(date).tz(name)
      mts = moment.tz(timestamp, name)
      self.assertMatches(entry, m)
      self.assertMatches(entry, mts)
    error_data = self.new_york_errors
    for entry in error_data:
      name  = entry[0]
      date  = entry[1]
      error = entry[2]
      self.assertRaises(error, moment.tz, date, name)

  # For each Los Angeles date, check that the returned date matches expected values
  def test_zone_entry(self):
    name = "America/Los_Angeles"
    data = self.los_angeles
    for entry in data:
      date      = entry[0]
      timestamp = entry[1]
      m         = moment.tz(date, name)
      self.assertMatches(entry, m)


  def test_zone(self):
    name = "America/New_York"
    tzinfo = moment.tzinfo(name)
    data = self.new_york
    for entry in data:
      date      = entry[0]
      ts        = entry[1]
      abbr      = entry[2]
      offset    = entry[3]
      dt = moment.tz(ts, name).datetime()
      self.assertEqual(dt.tzname(), abbr)
      self.assertEqual(dt.utcoffset(), timedelta(minutes=-offset))

  def test_ts_to_dt(self):
    # Verify that ts_to_dt works as expected.
    value_sec = 1426291200      # 2015-03-14 00:00:00 in UTC

    value_dt_utc = moment.ts_to_dt(value_sec, moment.get_zone('UTC'))
    value_dt_aware = moment.ts_to_dt(value_sec, moment.get_zone('America/New_York'))
    self.assertEqual(value_dt_utc.strftime("%Y-%m-%d %H:%M:%S %Z"), '2015-03-14 00:00:00 UTC')
    self.assertEqual(value_dt_aware.strftime("%Y-%m-%d %H:%M:%S %Z"), '2015-03-13 20:00:00 EDT')

  def test_dst_switches(self):
    # Verify that conversions around DST switches happen correctly. (This is tested in other tests
    # as well, but this test case is more focused and easier to debug.)
    dst_before = -1633280401
    dst_begin  = -1633280400
    dst_end    = -1615140001
    dst_after  = -1615140000

    # Should have no surprises in converting to UTC, since there are not DST dfferences.
    def ts_to_dt_utc(dt):
      return moment.ts_to_dt(dt, moment.get_zone('UTC'))
    self.assertEqual(ts_to_dt_utc(dst_before).strftime(fmt), "1918-03-31 06:59:59 UTC")
    self.assertEqual(ts_to_dt_utc(dst_begin ).strftime(fmt), "1918-03-31 07:00:00 UTC")
    self.assertEqual(ts_to_dt_utc(dst_end   ).strftime(fmt), "1918-10-27 05:59:59 UTC")
    self.assertEqual(ts_to_dt_utc(dst_after ).strftime(fmt), "1918-10-27 06:00:00 UTC")

    # Converting to America/New_York should produce correct jumps.
    def ts_to_dt_nyc(dt):
      return moment.ts_to_dt(dt, moment.get_zone('America/New_York'))
    self.assertEqual(ts_to_dt_nyc(dst_before).strftime(fmt), "1918-03-31 01:59:59 EST")
    self.assertEqual(ts_to_dt_nyc(dst_begin ).strftime(fmt), "1918-03-31 03:00:00 EDT")
    self.assertEqual(ts_to_dt_nyc(dst_end   ).strftime(fmt), "1918-10-27 01:59:59 EDT")
    self.assertEqual(ts_to_dt_nyc(dst_after ).strftime(fmt), "1918-10-27 01:00:00 EST")
    self.assertEqual(ts_to_dt_nyc(dst_after + 3599).strftime(fmt), "1918-10-27 01:59:59 EST")


  def test_tzinfo(self):
    # Verify that tzinfo works correctly.
    ts1 = 294217199000      # In EST
    ts2 = 294217200000      # In EDT (spring forward, we skip ahead by 1 hour)
    utc_dt1 = datetime(1979, 4, 29, 6, 59, 59)
    utc_dt2 = datetime(1979, 4, 29, 7, 0, 0)
    self.assertEqual(moment.tz(ts1).datetime().strftime(fmt), '1979-04-29 06:59:59 UTC')
    self.assertEqual(moment.tz(ts2).datetime().strftime(fmt), '1979-04-29 07:00:00 UTC')

    # Verify that we get correct time zone variation depending on DST status.
    nyc_dt1 = moment.tz(ts1, 'America/New_York').datetime()
    nyc_dt2 = moment.tz(ts2, 'America/New_York').datetime()
    self.assertEqual(nyc_dt1.strftime(fmt), '1979-04-29 01:59:59 EST')
    self.assertEqual(nyc_dt2.strftime(fmt), '1979-04-29 03:00:00 EDT')

    # Make sure we can get timestamps back from these datatimes.
    self.assertEqual(moment.dt_to_ts(nyc_dt1)*1000, ts1)
    self.assertEqual(moment.dt_to_ts(nyc_dt2)*1000, ts2)

    # Verify that the datetime objects we get produce correct time zones in terms of DST when we
    # manipulate them. NOTE: it is a bit unexpected that we add 1hr + 1sec rather than just 1sec,
    # but it seems like that is how Python datetime works. Note that timezone does get switched
    # correctly between EDT and EST.
    self.assertEqual(nyc_dt1 + timedelta(seconds=3601), nyc_dt2)
    self.assertEqual(nyc_dt2 - timedelta(seconds=3601), nyc_dt1)
    self.assertEqual((nyc_dt1 + timedelta(seconds=3601)).strftime(fmt), '1979-04-29 03:00:00 EDT')
    self.assertEqual((nyc_dt2 - timedelta(seconds=3601)).strftime(fmt), '1979-04-29 01:59:59 EST')


  def test_dt_to_ds(self):
    # Verify that dt_to_ts works for both naive and aware datetime objects.
    value_dt = datetime(2015, 3, 14, 0, 0)     # In UTC
    value_sec = 1426291200
    tzla = moment.get_zone('America/Los_Angeles')
    def format_utc(ts):
      return moment.ts_to_dt(ts, moment.get_zone('UTC')).strftime(fmt)

    # Check that a naive datetime is interpreted in UTC.
    self.assertEqual(value_dt.strftime("%Y-%m-%d %H:%M:%S %Z"), '2015-03-14 00:00:00 ')
    self.assertEqual(moment.dt_to_ts(value_dt), value_sec)    # Interpreted in UTC

    # Get an explicit UTC version and make sure that also works.
    value_dt_utc = value_dt.replace(tzinfo=moment.TZ_UTC)
    self.assertEqual(value_dt_utc.strftime(fmt), '2015-03-14 00:00:00 UTC')
    self.assertEqual(moment.dt_to_ts(value_dt_utc), value_sec)

    # Get an aware datetime, and make sure that works too.
    value_dt_aware = moment.ts_to_dt(value_sec, moment.get_zone('America/New_York'))
    self.assertEqual(value_dt_aware.strftime(fmt), '2015-03-13 20:00:00 EDT')
    self.assertEqual(moment.dt_to_ts(value_dt_aware), value_sec)

    # Check that dt_to_ts pays attention to the timezone.
    # If we interpret midnight in LA time, it's a later timestamp.
    self.assertEqual(format_utc(moment.dt_to_ts(value_dt, tzla)), '2015-03-14 07:00:00 UTC')
    # The second argument is ignored if the datetime is aware.
    self.assertEqual(format_utc(moment.dt_to_ts(value_dt_utc, tzla)), '2015-03-14 00:00:00 UTC')
    self.assertEqual(format_utc(moment.dt_to_ts(value_dt_aware, tzla)), '2015-03-14 00:00:00 UTC')

    # If we modify an aware datetime, we may get a new timezone abbreviation.
    value_dt_aware -= timedelta(days=28)
    self.assertEqual(value_dt_aware.strftime(fmt), '2015-02-13 20:00:00 EST')

  def test_date_to_ts(self):
    d = date(2015, 3, 14)
    tzla = moment.get_zone('America/Los_Angeles')
    def format_utc(ts):
      return moment.ts_to_dt(ts, moment.get_zone('UTC')).strftime(fmt)

    self.assertEqual(format_utc(moment.date_to_ts(d)), '2015-03-14 00:00:00 UTC')
    self.assertEqual(format_utc(moment.date_to_ts(d, tzla)), '2015-03-14 07:00:00 UTC')
    self.assertEqual(moment.ts_to_dt(moment.date_to_ts(d, tzla), tzla).strftime(fmt),
                     '2015-03-14 00:00:00 PDT')


  def test_parse_iso(self):
    tzny = moment.get_zone('America/New_York')
    iso = moment.parse_iso
    self.assertEqual(iso('2011-11-11T11:11:11'), 1321009871.000000)
    self.assertEqual(iso('2019-01-22T00:47:39.219071-05:00'), 1548136059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219071-0500'), 1548136059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219071', timezone=tzny), 1548136059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219071'), 1548118059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219071Z'), 1548118059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219071Z', timezone=tzny), 1548118059.219071)
    self.assertEqual(iso('2019-01-22T00:47:39.219'), 1548118059.219)
    self.assertEqual(iso('2019-01-22T00:47:39'), 1548118059)
    self.assertEqual(iso('2019-01-22 00:47:39.219071'), 1548118059.219071)
    self.assertEqual(iso('2019-01-22 00:47:39'), 1548118059)
    self.assertEqual(iso('2019-01-22'), 1548115200)

  def test_parse_iso_date(self):
    tzny = moment.get_zone('America/New_York')
    iso = moment.parse_iso_date
    # Note that time components and time zone do NOT affect the returned timestamp.
    self.assertEqual(iso('2019-01-22'), 1548115200)
    self.assertEqual(iso('2019-01-22T00:47:39.219071'), 1548115200)
    self.assertEqual(iso('2019-01-22 00:47:39Z'), 1548115200)
    self.assertEqual(iso('2019-01-22T00:47:39.219071-05:00'), 1548115200)
    self.assertEqual(iso('2019-01-22T00:47:39.219071+05:00'), 1548115200)

if __name__ == "__main__":
  unittest.main()