From 79fcc7bce04210873ceab2c5bdba3c8c48e1e467 Mon Sep 17 00:00:00 2001
From: luisramos0 <luisramos0@gmail.com>
Date: Mon, 5 Nov 2018 19:23:09 +0000
Subject: [PATCH] Fix user balance calculator spec by setting a target_shipment
 on line item of the order being tested

---
 spec/factories.rb                                      |  2 +-
 .../open_food_network/user_balance_calculator_spec.rb  | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/spec/factories.rb b/spec/factories.rb
index b7129ee5f6..d4e3c6fa7c 100644
--- a/spec/factories.rb
+++ b/spec/factories.rb
@@ -286,7 +286,7 @@ FactoryBot.define do
 
     after(:create) do |order|
       p = create(:simple_product, :distributors => [order.distributor])
-      FactoryBot.create(:line_item, :order => order, :product => p)
+      FactoryBot.create(:line_item_with_shipment, :order => order, :product => p)
       order.reload
     end
   end
diff --git a/spec/lib/open_food_network/user_balance_calculator_spec.rb b/spec/lib/open_food_network/user_balance_calculator_spec.rb
index f4bb92a668..6f2d7cfcde 100644
--- a/spec/lib/open_food_network/user_balance_calculator_spec.rb
+++ b/spec/lib/open_food_network/user_balance_calculator_spec.rb
@@ -11,17 +11,17 @@ module OpenFoodNetwork
 
       let!(:o1) { create(:order_with_totals_and_distribution,
                           user: user1, distributor: hub1,
-                          completed_at: 1.day.ago) 
+                          completed_at: 1.day.ago)
       } #total=10
       let!(:o2) { create(:order_with_totals_and_distribution,
                           user: user1, distributor: hub1,
-                          completed_at: 1.day.ago) 
+                          completed_at: 1.day.ago)
       } #total=10
       let!(:p1) { create(:payment, order: o1, amount: 15.00,
-                          state: "completed") 
+                          state: "completed")
       }
       let!(:p2) { create(:payment, order: o2, amount: 2.00,
-                          state: "completed") 
+                          state: "completed")
       }
 
       it "finds the correct balance for this email and enterprise" do
@@ -32,7 +32,7 @@ module OpenFoodNetwork
         let!(:hub2) { create(:distributor_enterprise) }
         let!(:o3) { create(:order_with_totals_and_distribution,
                             user: user1, distributor: hub2,
-                            completed_at: 1.day.ago) 
+                            completed_at: 1.day.ago)
         } #total=10
         let!(:p3) { create(:payment, order: o3, amount: 15.00,
                             state: "completed")