openfoodnetwork/spec/services/variants_stock_levels_spec.rb

require 'spec_helper'

describe VariantsStockLevels do
  let(:order) { create(:order) }

  let!(:line_item) do
    create(:line_item, order: order, variant: variant_in_the_order, quantity: 2, max_quantity: 3)
  end
  let!(:variant_in_the_order) { create(:variant) }
  let!(:variant_not_in_the_order) { create(:variant) }

  let(:variant_stock_levels) { VariantsStockLevels.new }

  before do
    variant_in_the_order.on_hand = 4
    variant_not_in_the_order.on_hand = 2
    order.reload
  end

  it "returns a hash with variant id, quantity, max_quantity, on hand and on demand" do
    expect(variant_stock_levels.call(order, [variant_in_the_order.id])).to eq(
      variant_in_the_order.id => { quantity: 2, max_quantity: 3, on_hand: 4, on_demand: false }
    )
  end

  it "includes all line items, even when the variant_id is not specified" do
    expect(variant_stock_levels.call(order, [])).to eq(
      variant_in_the_order.id => { quantity: 2, max_quantity: 3, on_hand: 4, on_demand: false }
    )
  end

  it "includes an empty quantity entry for variants that aren't in the order" do
    variant_ids = [variant_in_the_order.id, variant_not_in_the_order.id]
    expect(variant_stock_levels.call(order, variant_ids)).to eq(
      variant_in_the_order.id => { quantity: 2, max_quantity: 3, on_hand: 4, on_demand: false },
      variant_not_in_the_order.id => { quantity: 0, max_quantity: 0, on_hand: 2, on_demand: false }
    )
  end

  describe "when variant is on_demand" do
    let!(:variant_in_the_order) { create(:variant, on_demand: true) }

    before { variant_in_the_order.on_hand = 0 }

    it "includes the actual on_hand value and on_demand: true" do
      expect(variant_stock_levels.call(order, [variant_in_the_order.id])).to eq(
        variant_in_the_order.id => { quantity: 2, max_quantity: 3, on_hand: 0, on_demand: true }
      )
    end
  end

  describe "when the variant has an override" do
    let!(:distributor) { create(:distributor_enterprise) }
    let(:supplier) { variant_in_the_order.product.supplier }
    let!(:order_cycle) {
      create(:simple_order_cycle, suppliers: [supplier], distributors: [distributor],
                                  variants: [variant_in_the_order, variant_not_in_the_order])
    }
    let!(:variant_override_in_order) {
      create(:variant_override, hub: distributor,
                                variant: variant_in_the_order,
                                count_on_hand: 200)
    }
    let!(:variant_override_not_in_order) {
      create(:variant_override, hub: distributor,
                                variant: variant_not_in_the_order,
                                count_on_hand: 201)
    }

    before do
      order.order_cycle = order_cycle
      order.distributor = distributor
      order.save
    end

    context "when the variant is in the order" do
      it "returns the on_hand value of the override" do
        expect(variant_stock_levels.call(order, [variant_in_the_order.id])).to eq(
          variant_in_the_order.id => {
            quantity: 2, max_quantity: 3, on_hand: 200, on_demand: false
          }
        )
      end
    end

    context "with variants that are not in the order" do
      it "returns the on_hand value of the override" do
        variant_ids = [variant_in_the_order.id, variant_not_in_the_order.id]
        expect(variant_stock_levels.call(order, variant_ids)).to eq(
          variant_in_the_order.id => {
            quantity: 2, max_quantity: 3, on_hand: 200, on_demand: false
          },
          variant_not_in_the_order.id => {
            quantity: 0, max_quantity: 0, on_hand: 201, on_demand: false
          }
        )
      end
    end
  end
end