From 7fbb5b13fb6587e1465bd3ae1492912d164289c4 Mon Sep 17 00:00:00 2001
From: Andrew Spinks <aspinks@dius.com.au>
Date: Wed, 7 Aug 2013 12:43:31 +1000
Subject: [PATCH] Fix line item cache migration to work across orders in the
 staging servers.

---
 db/migrate/20130805050109_update_line_item_caching.rb | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/db/migrate/20130805050109_update_line_item_caching.rb b/db/migrate/20130805050109_update_line_item_caching.rb
index 1663b8ab4a..42760e7c02 100644
--- a/db/migrate/20130805050109_update_line_item_caching.rb
+++ b/db/migrate/20130805050109_update_line_item_caching.rb
@@ -2,11 +2,17 @@ class UpdateLineItemCaching < ActiveRecord::Migration
 
   class SpreeLineItem < ActiveRecord::Base
     belongs_to :shipping_method, class_name: 'Spree::ShippingMethod'
+    belongs_to :variant, :class_name => "Spree::Variant"
 
     def itemwise_shipping_cost
       order = OpenStruct.new :line_items => [self]
       shipping_method.compute_amount(order)
     end
+
+    def amount
+      price * quantity
+    end
+    alias total amount
   end