require 'timezone'
require 'timezone/zone'
require 'test/unit'
require "mocha/setup"
require 'timecop'

class TimezoneTest < Test::Unit::TestCase
  def test_valid_timezone
    assert_nothing_raised do
      Timezone::Zone.new :zone => 'Australia/Sydney'
    end
  end

  def test_nil_timezone
    assert_raise Timezone::Error::NilZone do
      Timezone::Zone.new :zone => nil
    end
  end

  def test_invalid_timezone
    assert_raise Timezone::Error::InvalidZone do
      Timezone::Zone.new :zone => 'Foo/Bar'
    end
  end

  def test_timezone_list
    list = Timezone::Zone.list "Australia/Sydney", "America/Chicago"
    assert list.is_a?(Array)
    assert list.count == 2
    assert list.first.is_a?(Hash)
    assert list.first[:zone] == "Australia/Sydney"
  end

  def test_timezone_list
    Timecop.freeze(Time.new(2012,2,2,0,0,0)) do
      assert !Timezone::Zone.list('EST5EDT').first[:dst]
    end
    Timecop.freeze(Time.new(2013,6,6,0,0,0)) do
      assert Timezone::Zone.list('EST5EDT').first[:dst]
    end
  end

  def test_timezone_custom_list_order
    Timezone::Configure.order_list_by = :title
    Timezone::Configure.replace "America/Chicago", with: "Chicago"
    list = Timezone::Zone.list "Australia/Sydney", "America/Chicago"
    assert list.first[:title] == "Australia/Sydney"
    assert list.last[:title] == "Chicago"
    assert list.last[:zone] == "America/Chicago"
  end

  def test_timezone_default_list
    Timezone::Configure.default_for_list = "America/Chicago", "Australia/Sydney"
    list = Timezone::Zone.list
    assert list.count == 2
    assert list.first.has_value? "Australia/Sydney"
  end

  def test_timezone_names
    zones = Timezone::Zone.names
    assert zones.is_a?(Array)
    assert zones.count > 0
    assert zones.include? "Australia/Sydney"
  end

  def time_timezone_equivalence
    gmt = Timezone::Zone.new :zone => 'GMT'
    australia = Timezone::Zone.new :zone => 'Australia/Sydney'
    assert gmt == gmt
    assert gmt <= australia
    assert gmt < australia
  end

  def test_getting_utc_offset
    assert_equal 36000, Timezone::Zone.new(:zone => 'Australia/Sydney').utc_offset(Time.parse('2011-06-05'))
    assert_equal -25200, Timezone::Zone.new(:zone => 'America/Los_Angeles').utc_offset(Time.parse('2011-06-05'))
    assert_equal 20700, Timezone::Zone.new(:zone => 'Asia/Kathmandu').utc_offset(Time.parse('2011-06-05'))

    assert_equal -18000, Timezone::Zone.new(:zone => 'America/New_York').utc_offset(Time.parse('2011-01-11'))
    assert_equal -14400, Timezone::Zone.new(:zone => 'America/New_York').utc_offset(Time.parse('2011-06-11'))
  end

  def test_loading_GMT_timezone
    timezone = Timezone::Zone.new :zone => 'GMT'
    assert_equal Time.now.utc.to_i, timezone.time(Time.now).to_i
  end

  def test_loading_historical_time
    timezone = Timezone::Zone.new :zone => 'America/Los_Angeles'
    local = Time.strptime('2011-02-11T13:20:00Z', '%Y-%m-%dT%H:%M:%SZ')
    utc = Time.strptime('2011-02-11T21:20:00Z', '%Y-%m-%dT%H:%M:%SZ')
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  # http://www.timeanddate.com/worldclock/clockchange.html?n=137&year=2011
  def test_historical_time_change_in_la_forward
    timezone = Timezone::Zone.new :zone => 'America/Los_Angeles'
    local = Time.strptime('2011-03-13T01:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('2011-03-13T09:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i

    timezone = Timezone::Zone.new :zone => 'America/Los_Angeles'
    local = Time.strptime('2011-03-13T03:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('2011-03-13T10:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  # http://www.timeanddate.com/worldclock/clockchange.html?n=137&year=2011
  def test_historical_time_change_in_la_backward
    timezone = Timezone::Zone.new :zone => 'America/Los_Angeles'
    local = Time.strptime('2011-11-06T01:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('2011-11-06T08:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i

    timezone = Timezone::Zone.new :zone => 'America/Los_Angeles'
    local = Time.strptime('2011-11-06T01:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('2011-11-06T09:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  # http://www.timeanddate.com/worldclock/clockchange.html?n=2364&year=1940
  def test_historical_time_change_in_hebron
    timezone = Timezone::Zone.new :zone => 'Asia/Hebron'
    local = Time.strptime('1940-05-31T23:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('1940-05-31T21:59:59 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i

    timezone = Timezone::Zone.new :zone => 'Asia/Hebron'
    local = Time.strptime('1940-06-01T01:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    utc = Time.strptime('1940-05-31T22:00:00 UTC', '%Y-%m-%dT%H:%M:%S %Z')
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  def test_loading_half_hour_timezone
    timezone = Timezone::Zone.new :zone => 'Asia/Kathmandu'
    utc = Time.utc(2011, 1, 4, 3, 51, 29)
    local = Time.utc(2011, 1, 4, 9, 36, 29)
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  class HTTPTestClient
    class << self ; attr_accessor :body ; end

    Response = Struct.new(:body) do
      def code ; '200' ; end
    end

    def initialize(protocol, host)
    end

    def get(url)
      HTTPTestClient::Response.new(self.class.body)
    end
  end

  def test_using_lat_lon_coordinates
    mock_path = File.expand_path(File.join(File.dirname(__FILE__), 'mocks'))
    HTTPTestClient.body = File.open(mock_path + '/lat_lon_coords.txt').read

    Timezone::Configure.begin do |c|
      c.http_client = HTTPTestClient
      c.username = 'timezone'
    end

    timezone = Timezone::Zone.new :latlon => [-34.92771808058, 138.477041423321]
    assert_equal 'Australia/Adelaide', timezone.zone
  end

  def test_api_limit_read_lat_lon_coordinates
    mock_path = File.expand_path(File.join(File.dirname(__FILE__), 'mocks'))
    HTTPTestClient.body = File.open(mock_path + '/api_limit_reached.txt').read

    Timezone::Configure.begin do |c|
      c.http_client = HTTPTestClient
      c.username = 'timezone'
    end

    assert_raise Timezone::Error::GeoNames, 'api limit reached' do
      Timezone::Zone.new :latlon => [-34.92771808058, 138.477041423321]
    end
  end

  def test_australian_timezone_with_dst
    timezone = Timezone::Zone.new :zone => 'Australia/Adelaide'
    utc = Time.utc(2010, 12, 23, 19, 37, 15)
    local = Time.utc(2010, 12, 24, 6, 7, 15)
    assert_equal local.to_i, timezone.time(utc).to_i
  end

  def test_configure_url_default
    assert_equal 'api.geonames.org', Timezone::Configure.url
  end

  def test_configure_url_custom
    Timezone::Configure.begin { |c| c.url = 'www.newtimezoneserver.com' }
    assert_equal 'www.newtimezoneserver.com', Timezone::Configure.url
    # clean up url after test
    Timezone::Configure.begin { |c| c.url = nil }
  end

  def test_utc_offset_without_dst
    timezone = Timezone::Zone.new :zone => 'Europe/Helsinki'
    # just before DST starts
    utc = Time.utc(2012, 3, 25, 0, 59, 59)
    assert_equal timezone.utc_offset(utc), 7200
    # on the second DST ends
    utc = Time.utc(2012, 10, 28, 1, 0, 0)
    assert_equal timezone.utc_offset(utc), 7200
  end

  def test_utc_offset_with_dst
    timezone = Timezone::Zone.new :zone => 'Europe/Helsinki'
    # on the second DST starts
    utc = Time.utc(2012, 3, 25, 1, 0, 0)
    assert_equal timezone.utc_offset(utc), 10800
    # right before DST end
    utc = Time.utc(2012, 10, 28, 0, 59, 59)
    assert_equal timezone.utc_offset(utc), 10800
  end

  def test_utc_offset_without_timestamps
    File.open(File.join(File.dirname(__FILE__),"data/Helsinki_rules_without_timestamps.json")) do |f|
      rules = JSON.parse(f.read)
      timezone = Timezone::Zone.new :zone => 'Europe/Helsinki'
      # TODO [panthomakos] Resolve this using a stub.
      timezone.instance_variable_set(:@rules, rules)
      utc = Time.utc(2012, 3, 25, 0, 59, 59)
      assert_equal timezone.utc_offset(utc), 7200
      utc = Time.utc(2012, 3, 25, 1, 0, 0)
      assert_equal timezone.utc_offset(utc), 10800
    end
  end

  def test_active_support_timezone
    timezone = Timezone::Zone.new(:zone => 'Australia/Adelaide')
    assert_equal 'Australia/Adelaide', timezone.zone
    assert_equal 'Adelaide', timezone.active_support_time_zone
  end
end