require 'minitest/autorun'
require_relative '../lib/3scale_time_range'

class GranulateTest < Minitest::Test

  def setup
    @gran = TimeRange.new(
      DateTime.parse("2012-10-09 07:23"), DateTime.parse("2014-02-05 13:45")
    ).granulate
  end

  def test_granulates_by_year
    assert_equal @gran.years, [
      TimeRange.new(
        DateTime.parse("2013-01-01").beginning_of_year,
        DateTime.parse("2013-12-31").end_of_year)
    ]
  end

  def test_granulates_by_month
    assert_equal @gran.months, [
        TimeRange.new(
          DateTime.parse("2012-11-01").beginning_of_month,
          DateTime.parse("2012-12-31").end_of_month),

        TimeRange.new(
          DateTime.parse("2014-01-01").beginning_of_month,
          DateTime.parse("2014-01-31").end_of_month)
      ]
  end

  def test_granulates_by_day
    assert_equal @gran.days, [
        TimeRange.new(
          DateTime.parse("2012-10-10").beginning_of_day,
          DateTime.parse("2012-10-31").end_of_day),

        TimeRange.new(
          DateTime.parse("2014-02-01").beginning_of_day,
          DateTime.parse("2014-02-04").end_of_day)
      ]
  end

  def test_exposes_information_on_not_granulated_ranges
    assert_equal @gran.rest, [
        TimeRange.new(
          DateTime.parse("2012-10-09 07:23"),
          DateTime.parse("2012-10-09").end_of_day),

        TimeRange.new(
          DateTime.parse("2014-02-05").beginning_of_day,
          DateTime.parse("2014-02-05 13:45"))
      ]
  end

  def test_works_also_for_short_ranges
    @gran = TimeRange.new(
      DateTime.parse("2012-10-09 07:23"), DateTime.parse("2012-10-09 13:45")
    ).granulate

    assert_equal @gran.rest, [
        TimeRange.new(
          DateTime.parse("2012-10-09 07:23"),
          DateTime.parse("2012-10-09 13:45"))
      ]
  end

  def test_granulation_is_cached_in_time_range
    range = TimeRange.new(
      DateTime.parse("2012-10-09 07:23"), DateTime.parse("2012-10-09 13:45")
    )

    assert_equal range.granulate.object_id, range.granulate.object_id
  end

  def test_properly_parses_open_ended_ranges_1
    gran = (DateTime.parse("2011-01-01")...DateTime.parse("2012-01-01")).
      to_time_range.granulate

    assert_equal 1, gran.years.size
    assert_equal 0, gran.months.size
    assert_equal 0, gran.days.size
    assert_equal 0, gran.rest.size

    refute gran.years.first.exclude_end?
  end

  def test_properly_parses_open_ended_ranges_2
    gran = (DateTime.parse("2011-01-01")...DateTime.parse("2011-12-31").end_of_year).
      to_time_range.granulate

    assert_equal 0, gran.years.size
    assert_equal 1, gran.months.size
    assert_equal 1, gran.days.size
    assert_equal 1, gran.rest.size

    refute gran.months.first.exclude_end?
    refute gran.days.first.exclude_end?
    assert gran.rest.first.exclude_end?
  end

  def test_properly_parses_close_ended_ranges_1
    gran = (DateTime.parse("2011-01-01")..DateTime.parse("2012-01-01")).
      to_time_range.granulate

    assert_equal 1, gran.years.size
    assert_equal 0, gran.months.size
    assert_equal 0, gran.days.size
    assert_equal 1, gran.rest.size

    refute gran.rest.first.exclude_end?
  end

  def test_properly_parses_close_ended_ranges_2
    gran = (DateTime.parse("2011-01-01")..DateTime.parse("2011-12-31").end_of_year).
      to_time_range.granulate

    assert_equal 1, gran.years.size
    assert_equal 0, gran.months.size
    assert_equal 0, gran.days.size
    assert_equal 0, gran.rest.size

    refute gran.years.first.exclude_end?
  end

  def test_properly_handles_Time_UTC_ends_of_months
    d = DateTime.parse('2010-01-01')
    gran = (d.beginning_of_month.to_time.utc..d.end_of_month.to_time.utc).to_time_range.granulate

    assert_equal 0, gran.years.size
    assert_equal 1, gran.months.size
    assert_equal 0, gran.days.size
    assert_equal 0, gran.rest.size
  end

end