diff --git a/spec/models/enterprise_spec.rb b/spec/models/enterprise_spec.rb index 930358a350..313148e599 100644 --- a/spec/models/enterprise_spec.rb +++ b/spec/models/enterprise_spec.rb @@ -599,4 +599,19 @@ describe Enterprise do end end end + + describe "#plus_relatives_and_oc_producers" do + it "does not find non-produders " do + supplier = create(:supplier_enterprise) + distributor = create(:distributor_enterprise, is_primary_producer: false) + product = create(:product) + order_cycle = create( + :simple_order_cycle, + suppliers: [supplier], + distributors: [distributor], + variants: [product.master] + ) + expect(distributor.plus_relatives_and_oc_producers(order_cycle)).to eq([supplier]) + end + end end