Merge pull request #9088 from mkllnk/9024-setup-script

Update setup script for removed callbacks

lib/tasks/sample_data/order_factory.rb

@@ -30,14 +30,19 @@ module SampleData
       order.save
 
       log "- complete order - shipped"
+      create_shipped_order
+    end
+
+    private
+
+    def create_shipped_order
       order = create_complete_order
+      order.payments.first.amount = order.total
       order.payments.first.capture!
       order.save
       order.shipment.reload.ship!
     end
 
-    private
-
     def create_cart_order
       order = create_order
       order.save

spec/lib/tasks/sample_data_rake_spec.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require 'rake'
+
+describe 'sample_data.rake' do
+  before(:all) do
+    Rake.application.rake_require 'tasks/sample_data'
+    Rake::Task.define_task(:environment)
+  end
+
+  before do
+    # Create seed data required by the sample data.
+    create(:user)
+    DefaultStockLocation.find_or_create
+    DefaultShippingCategory.find_or_create
+  end
+
+  it "creates some sample data to play with" do
+    Rake.application.invoke_task "ofn:sample_data"
+
+    expect(EnterpriseGroup.count).to eq 1
+    expect(Customer.count).to eq 2
+    expect(Spree::Order.count).to eq 5
+  end
+end