# frozen_string_literal: true

require 'spec_helper'

module OrderManagement
  module Stock
    RSpec.describe Prioritizer do
      let(:order) { create(:order_with_line_items, line_items_count: 2) }
      let(:variant1) { order.line_items[0].variant }
      let(:variant2) { order.line_items[1].variant }

      def pack
        package = Package.new(order)
        yield(package) if block_given?
        package
      end

      it 'keeps a single package' do
        package1 = pack do |package|
          package.add variant1, 1, :on_hand
          package.add variant2, 1, :on_hand
        end

        packages = [package1]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages
        expect(packages.size).to eq 1
      end

      it 'removes duplicate packages' do
        package1 = pack do |package|
          package.add variant1, 1, :on_hand
          package.add variant2, 1, :on_hand
        end
        package2 = pack do |package|
          package.add variant1, 1, :on_hand
          package.add variant2, 1, :on_hand
        end

        packages = [package1, package2]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages
        expect(packages.size).to eq 1
      end

      it 'split over 2 packages' do
        package1 = pack do |package|
          package.add variant1, 1, :on_hand
        end
        package2 = pack do |package|
          package.add variant2, 1, :on_hand
        end

        packages = [package1, package2]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages
        expect(packages.size).to eq 2
      end

      it '1st has some, 2nd has remaining' do
        allow(order.line_items[0]).to receive_messages(quantity: 5)
        package1 = pack do |package|
          package.add variant1, 2, :on_hand
        end
        package2 = pack do |package|
          package.add variant1, 5, :on_hand
        end

        packages = [package1, package2]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages
        expect(packages.count).to eq 2
        expect(packages[0].quantity).to eq 2
        expect(packages[1].quantity).to eq 3
      end

      it '1st has backorder, 2nd has some' do
        allow(order.line_items[0]).to receive_messages(quantity: 5)
        package1 = pack do |package|
          package.add variant1, 5, :backordered
        end
        package2 = pack do |package|
          package.add variant1, 2, :on_hand
        end

        packages = [package1, package2]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages

        expect(packages[0].quantity(:backordered)).to eq 3
        expect(packages[1].quantity(:on_hand)).to eq 2
      end

      it '1st has backorder, 2nd has all' do
        allow(order.line_items[0]).to receive_messages(quantity: 5)
        package1 = pack do |package|
          package.add variant1, 3, :backordered
          package.add variant2, 1, :on_hand
        end
        package2 = pack do |package|
          package.add variant1, 5, :on_hand
        end

        packages = [package1, package2]
        prioritizer = Prioritizer.new(order, packages)
        packages = prioritizer.prioritized_packages
        expect(packages[0].quantity(:backordered)).to eq 0
        expect(packages[1].quantity(:on_hand)).to eq 5
      end
    end
  end
end
