From 79b86f535d5be5a8b7927bcaedeaf0d8921ab7c6 Mon Sep 17 00:00:00 2001
From: Matt-Yorkley <9029026+Matt-Yorkley@users.noreply.github.com>
Date: Tue, 19 Jan 2021 13:16:39 +0000
Subject: [PATCH] Move class definitions in adjustment migration

---
 .../20201219120055_add_order_to_adjustments.rb  | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/db/migrate/20201219120055_add_order_to_adjustments.rb b/db/migrate/20201219120055_add_order_to_adjustments.rb
index 85d3048fc3..9748a4372d 100644
--- a/db/migrate/20201219120055_add_order_to_adjustments.rb
+++ b/db/migrate/20201219120055_add_order_to_adjustments.rb
@@ -1,13 +1,14 @@
 class AddOrderToAdjustments < ActiveRecord::Migration
-  def up
-    class Spree::Adjustment < ActiveRecord::Base
-      belongs_to :adjustable, polymorphic: true
-      belongs_to :order, class_name: "Spree::Order"
-    end
-    class Spree::LineItem < ActiveRecord::Base
-      belongs_to :order, class_name: "Spree::Order"
-    end
+  class Spree::Adjustment < ActiveRecord::Base
+    belongs_to :adjustable, polymorphic: true
+    belongs_to :order, class_name: "Spree::Order"
+  end
 
+  class Spree::LineItem < ActiveRecord::Base
+    belongs_to :order, class_name: "Spree::Order"
+  end
+
+  def up
     add_column :spree_adjustments, :order_id, :integer
 
     # Ensure migration can use the new column