require 'test_helper'

module Workarea
  module Reports
    class SalesBySkuTest < TestCase
      setup :add_data, :time_travel

      def add_data
        Metrics::SkuByDay.inc(
          key: { sku: 'foo' },
          at: Time.zone.local(2018, 10, 27),
          orders: 1,
          units_sold: 2,
          merchandise: 10.to_m,
          discounts: 0.to_m,
          revenue: 10.to_m
        )

        Metrics::SkuByDay.inc(
          key: { sku: 'foo' },
          at: Time.zone.local(2018, 10, 28),
          orders: 2,
          units_sold: 4,
          merchandise: 20.to_m,
          discounts: -5.to_m,
          revenue: 15.to_m
        )

        Metrics::SkuByDay.inc(
          key: { sku: 'foo' },
          at: Time.zone.local(2018, 10, 29),
          orders: 3,
          units_sold: 6,
          merchandise: 30.to_m,
          discounts: -3.to_m,
          revenue: 27.to_m
        )

        Metrics::SkuByDay.inc(
          key: { sku: 'bar' },
          at: Time.zone.local(2018, 10, 27),
          orders: 2,
          units_sold: 3,
          merchandise: 11.to_m,
          discounts: 0.to_m,
          revenue: 11.to_m
        )

        Metrics::SkuByDay.inc(
          key: { sku: 'bar' },
          at: Time.zone.local(2018, 10, 28),
          orders: 3,
          units_sold: 5,
          merchandise: 21.to_m,
          discounts: -6.to_m,
          revenue: 15.to_m
        )

        Metrics::SkuByDay.inc(
          key: { sku: 'bar' },
          at: Time.zone.local(2018, 10, 29),
          orders: 4,
          units_sold: 7,
          merchandise: 31.to_m,
          discounts: -4.to_m,
          revenue: 27.to_m
        )
      end

      def time_travel
        travel_to Time.zone.local(2018, 10, 30)
      end

      def test_grouping_and_summing
        report = SalesBySku.new
        assert_equal(2, report.results.length)

        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(6, foo['orders'])
        assert_equal(12, foo['units_sold'])
        assert_equal(60, foo['merchandise'])
        assert_equal(-8, foo['discounts'])
        assert_equal(52, foo['revenue'])

        bar = report.results.detect { |r| r['_id'] == 'bar' }
        assert_equal(9, bar['orders'])
        assert_equal(15, bar['units_sold'])
        assert_equal(63, bar['merchandise'])
        assert_equal(-10, bar['discounts'])
        assert_equal(53, bar['revenue'])
      end

      def test_date_ranges
        report = SalesBySku.new
        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(6, foo['orders'])

        report = SalesBySku.new(starts_at: '2018-10-28', ends_at: '2018-10-28')
        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(2, foo['orders'])

        report = SalesBySku.new(starts_at: '2018-10-28', ends_at: '2018-10-29')
        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(5, foo['orders'])

        report = SalesBySku.new(starts_at: '2018-10-28')
        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(5, foo['orders'])

        report = SalesBySku.new(ends_at: '2018-10-28')
        foo = report.results.detect { |r| r['_id'] == 'foo' }
        assert_equal(3, foo['orders'])
      end

      def test_filtering_skus
        report = SalesBySku.new(
          skus: %w(foo),
          starts_at: '2018-09-28',
          ends_at: '2018-10-28'
        )

        assert_equal(1, report.results.size)
        assert_equal('foo', report.results.first['_id'])
      end

      def test_time_zones
        current_zone = Time.zone

        ['UTC', 'Eastern Time (US & Canada)'].each do |zone|
          Time.zone = zone

          Metrics::SkuByDay.delete_all
          Metrics::SkuByDay.inc(key: { sku: 'foo' }, orders: 1, at: Time.zone.local(2018, 11, 15, 23))
          Metrics::SkuByDay.inc(key: { sku: 'foo' }, orders: 2, at: Time.zone.local(2018, 11, 16, 1))
          assert_equal(2, Metrics::SkuByDay.count)

          report = SalesBySku.new(starts_at: '2018-11-15', ends_at: '2018-11-15')
          assert_equal(
            1,
            report.results.detect { |r| r['_id'] == 'foo' }['orders'],
            "Failed on #{zone}"
          )

          report = SalesBySku.new(starts_at: '2018-11-16', ends_at: '2018-11-16')
          assert_equal(
            2,
            report.results.detect { |r| r['_id'] == 'foo' }['orders'],
            "Failed on #{zone}"
          )
        end

      ensure
        Time.zone = current_zone
      end

      def test_sorting
        report = SalesBySku.new(sort_by: 'orders', sort_direction: 'asc')
        assert_equal('foo', report.results.first['_id'])

        report = SalesBySku.new(sort_by: 'orders', sort_direction: 'desc')
        assert_equal('bar', report.results.first['_id'])

        report = SalesBySku.new(sort_by: 'discounts', sort_direction: 'asc')
        assert_equal('bar', report.results.first['_id'])

        report = SalesBySku.new(sort_by: 'discounts', sort_direction: 'desc')
        assert_equal('foo', report.results.first['_id'])
      end
    end
  end
end