spree_sample 1.3.2 → 1.3.3

data/db/samples/products.rb

@@ -0,0 +1,122 @@
+Spree::Sample.load_sample("tax_categories")
+
+clothing = Spree::TaxCategory.find_by_name!("Clothing")
+
+default_attrs = {
+  :description => Faker::Lorem.paragraph,
+  :available_on => Time.zone.now
+}
+
+products = [
+  { 
+    :name => "Ruby on Rails Tote",
+    :tax_category => clothing,
+    :price => 15.99,
+    :eur_price => 14,
+  },
+  {
+    :name => "Ruby on Rails Bag",
+    :tax_category => clothing,
+    :price => 22.99,
+    :eur_price => 19,
+  },
+  {
+    :name => "Ruby on Rails Baseball Jersey",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Ruby on Rails Jr. Spaghetti",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+
+  },
+  {
+    :name => "Ruby on Rails Ringer T-Shirt",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Ruby Baseball Jersey",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Apache Baseball Jersey",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Spree Baseball Jersey",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Spree Jr. Spaghetti",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Spree Ringer T-Shirt",
+    :tax_category => clothing,
+    :price => 19.99,
+    :eur_price => 16
+  },
+  {
+    :name => "Spree Tote",
+    :tax_category => clothing,
+    :price => 15.99,
+    :eur_price => 14,
+  },
+  {
+    :name => "Spree Bag",
+    :tax_category => clothing,
+    :price => 22.99,
+    :eur_price => 19
+  },
+  {
+    :name => "Ruby on Rails Mug",
+    :price => 13.99,
+    :eur_price => 12
+  },
+  {
+    :name => "Ruby on Rails Stein",
+    :price => 16.99,
+    :eur_price => 14
+  },
+  {
+    :name => "Spree Stein",
+    :price => 16.99,
+    :eur_price => 14,
+  },
+  {
+    :name => "Spree Mug",
+    :price => 13.99,
+    :eur_price => 12
+  }
+]
+
+# Need to call reset_column information here
+# Otherwise running db:migrate db:seed spree_sample:load will fail
+# complaining about cost_currency missing
+Spree::Variant.reset_column_information
+
+products.each do |product_attrs|
+  eur_price = product_attrs.delete(:eur_price)
+  Spree::Config[:currency] = "USD"
+
+  product = Spree::Product.create!(default_attrs.merge(product_attrs), :without_protection => true)
+  Spree::Config[:currency] = "EUR"
+  product.reload
+  product.price = eur_price
+  product.save!
+end
+
+Spree::Config[:currency] = "USD"

data/db/samples/prototypes.rb

@@ -0,0 +1,21 @@
+prototypes = [
+  {
+    :name => "Shirt",
+    :properties => ["Manufacturer", "Brand", "Model", "Shirt Type", "Sleeve Type", "Material", "Fit", "Gender"]
+  },
+  {
+    :name => "Bag",
+    :properties => ["Type", "Size", "Material"]
+  },
+  {
+    :name => "Mugs",
+    :properties => ["Size", "Type"]
+  }
+]
+
+prototypes.each do |prototype_attrs|
+  prototype = Spree::Prototype.create!(:name => prototype_attrs[:name])
+  prototype_attrs[:properties].each do |property|
+    prototype.properties << Spree::Property.find_by_name!(property)
+  end
+end

data/db/samples/shipments.rb

@@ -0,0 +1,18 @@
+first_order = Spree::Order.find_by_number!("R123456789")
+last_order = Spree::Order.find_by_number!("R987654321")
+
+Spree::Shipment.create!({
+  :order => first_order,
+  :number => Array.new(11){rand(11)}.join,
+  :shipping_method => Spree::ShippingMethod.find_by_name!("UPS Ground (USD)"),
+  :address => Spree::Address.first,
+  :state => "pending"
+}, :without_protection => true)
+
+Spree::Shipment.create!({
+  :order => last_order,
+  :number => Array.new(11){rand(11)}.join,
+  :shipping_method => Spree::ShippingMethod.find_by_name!("UPS Ground (USD)"),
+  :address => Spree::Address.first,
+  :state => "pending"
+}, :without_protection => true)

data/db/samples/shipping_categories.rb

@@ -0,0 +1 @@
+Spree::ShippingCategory.create!(:name => "Default Shipping")

data/db/samples/shipping_methods.rb

@@ -0,0 +1,49 @@
+begin
+north_america = Spree::Zone.find_by_name!("North America")
+rescue ActiveRecord::RecordNotFound
+  puts "Couldn't find 'North America' zone. Did you run `rake db:seed` first?"
+  puts "That task will set up the countries, states and zones required for Spree."
+  exit
+end
+
+europe_vat = Spree::Zone.find_by_name!("EU_VAT")
+
+shipping_methods = [
+  {
+    :name => "UPS Ground (USD)",
+    :zone => north_america,
+    :calculator => Spree::Calculator::FlatRate.create!
+  },
+  {
+    :name => "UPS Two Day (USD)",
+    :zone => north_america,
+    :calculator => Spree::Calculator::FlatRate.create!
+  },
+  {
+    :name => "UPS One Day (USD)",
+    :zone => north_america,
+    :calculator => Spree::Calculator::FlatRate.create!
+  },
+  {
+    :name => "UPS Ground (EUR)",
+    :zone => europe_vat,
+    :calculator => Spree::Calculator::FlatRate.create!
+  }
+]
+
+shipping_methods.each do |shipping_method_attrs|
+  Spree::ShippingMethod.create!(shipping_method_attrs, :without_protection => true)
+end
+
+{
+  "UPS Ground (USD)" => [5, "USD"],
+  "UPS Ground (EUR)" => [5, "EUR"],
+  "UPS One Day (USD)" => [15, "USD"],
+  "UPS Two Day (USD)" => [10, "USD"]
+}.each do |shipping_method_name, (price, currency)|
+  shipping_method = Spree::ShippingMethod.find_by_name!(shipping_method_name)
+  shipping_method.calculator.preferred_amount = price
+  shipping_method.calculator.preferred_currency = currency
+  shipping_method.save!
+end
+

data/db/samples/tax_categories.rb

@@ -0,0 +1,2 @@
+Spree::TaxCategory.create!(:name => "Clothing")
+Spree::TaxCategory.create!(:name => "Food")

data/db/samples/tax_rates.rb

@@ -0,0 +1,10 @@
+north_america = Spree::Zone.find_by_name!("North America")
+clothing = Spree::TaxCategory.find_by_name!("Clothing")
+tax_rate = Spree::TaxRate.create({
+  :name => "North America",
+  :zone => north_america, 
+  :amount => 0.05,
+  :tax_category => clothing},
+  :without_protection => true)
+tax_rate.calculator = Spree::Calculator::DefaultTax.create!
+tax_rate.save!

data/db/samples/taxonomies.rb

@@ -0,0 +1,8 @@
+taxonomies = [
+  { :name => "Categories" },
+  { :name => "Brand" }
+]
+
+taxonomies.each do |taxonomy_attrs|
+  Spree::Taxonomy.create!(taxonomy_attrs)
+end

data/db/samples/taxons.rb

@@ -0,0 +1,121 @@
+Spree::Sample.load_sample("taxonomies")
+Spree::Sample.load_sample("products")
+
+categories = Spree::Taxonomy.find_by_name!("Categories")
+brands = Spree::Taxonomy.find_by_name!("Brand")
+
+products = { 
+  :ror_tote => "Ruby on Rails Tote",
+  :ror_bag => "Ruby on Rails Bag",
+  :ror_mug => "Ruby on Rails Mug",
+  :ror_stein => "Ruby on Rails Stein",
+  :ror_baseball_jersey => "Ruby on Rails Baseball Jersey",
+  :ror_jr_spaghetti => "Ruby on Rails Jr. Spaghetti",
+  :ror_ringer => "Ruby on Rails Ringer T-Shirt",
+  :spree_stein => "Spree Stein",
+  :spree_mug => "Spree Mug",
+  :spree_ringer => "Spree Ringer T-Shirt",
+  :spree_baseball_jersey =>  "Spree Baseball Jersey",
+  :spree_tote => "Spree Tote",
+  :spree_bag => "Spree Bag",
+  :spree_jr_spaghetti => "Spree Jr. Spaghetti",
+  :apache_baseball_jersey => "Apache Baseball Jersey",
+  :ruby_baseball_jersey => "Ruby Baseball Jersey",
+}
+
+
+products.each do |key, name|
+  products[key] = Spree::Product.find_by_name!(name)
+end
+
+taxons = [
+  {
+    :name => "Categories",
+    :taxonomy => categories,
+    :position => 0
+  },
+  {
+    :name => "Bags",
+    :taxonomy => categories,
+    :parent => "Categories",
+    :position => 1,
+    :products => [
+      products[:ror_tote],
+      products[:ror_bag],
+      products[:spree_tote],
+      products[:spree_bag]
+    ]
+  },
+  {
+    :name => "Mugs",
+    :taxonomy => categories,
+    :parent => "Categories",
+    :position => 2,
+    :products => [
+      products[:ror_mug],
+      products[:ror_stein],
+      products[:spree_stein],
+      products[:spree_mug]
+    ]
+  },
+  {
+    :name => "Clothing",
+    :taxonomy => categories,
+    :parent => "Categories" 
+  },
+  {
+    :name => "Shirts",
+    :taxonomy => categories,
+    :parent => "Clothing",
+    :position => 0,
+    :products => [
+      products[:ror_jr_spaghetti],
+      products[:spree_jr_spaghetti]
+    ]
+  },
+  {
+    :name => "T-Shirts",
+    :taxonomy => categories,
+    :parent => "Clothing" ,
+    :products => [
+      products[:ror_baseball_jersey],
+      products[:ror_ringer],
+      products[:apache_baseball_jersey],
+      products[:ruby_baseball_jersey],
+      products[:spree_baseball_jersey],
+      products[:spree_ringer]
+    ],
+    :position => 0
+  },
+  {
+    :name => "Brands",
+    :taxonomy => brands
+  },
+  {
+    :name => "Ruby",
+    :taxonomy => brands,
+    :parent => "Brand" 
+  },
+  {
+    :name => "Apache",
+    :taxonomy => brands,
+    :parent => "Brand" 
+  },
+  {
+    :name => "Spree",
+    :taxonomy => brands,
+    :parent => "Brand"
+  },
+  {
+    :name => "Rails",
+    :taxonomy => brands,
+    :parent => "Brand"
+  },
+]
+
+taxons.each do |taxon_attrs|
+  if taxon_attrs[:parent]
+    taxon_attrs[:parent] = Spree::Taxon.find_by_name!(taxon_attrs[:parent])
+  end
+  Spree::Taxon.create!(taxon_attrs, :without_protection => true)
+end

data/db/samples/variants.rb

@@ -0,0 +1,170 @@
+Spree::Sample.load_sample("option_values")
+Spree::Sample.load_sample("products")
+
+ror_baseball_jersey = Spree::Product.find_by_name!("Ruby on Rails Baseball Jersey") 
+ror_tote = Spree::Product.find_by_name!("Ruby on Rails Tote")
+ror_bag = Spree::Product.find_by_name!("Ruby on Rails Bag")
+ror_jr_spaghetti = Spree::Product.find_by_name!("Ruby on Rails Jr. Spaghetti")
+ror_mug = Spree::Product.find_by_name!("Ruby on Rails Mug")
+ror_ringer = Spree::Product.find_by_name!("Ruby on Rails Ringer T-Shirt")
+ror_stein = Spree::Product.find_by_name!("Ruby on Rails Stein")
+spree_baseball_jersey = Spree::Product.find_by_name!("Spree Baseball Jersey")
+spree_stein = Spree::Product.find_by_name!("Spree Stein")
+spree_jr_spaghetti = Spree::Product.find_by_name!("Spree Jr. Spaghetti")
+spree_mug = Spree::Product.find_by_name!("Spree Mug")
+spree_ringer = Spree::Product.find_by_name!("Spree Ringer T-Shirt")
+spree_tote = Spree::Product.find_by_name!("Spree Tote")
+spree_bag = Spree::Product.find_by_name!("Spree Bag")
+ruby_baseball_jersey = Spree::Product.find_by_name!("Ruby Baseball Jersey")
+apache_baseball_jersey = Spree::Product.find_by_name!("Apache Baseball Jersey")
+
+small = Spree::OptionValue.find_by_name!("Small")
+medium = Spree::OptionValue.find_by_name!("Medium")
+large = Spree::OptionValue.find_by_name!("Large")
+extra_large = Spree::OptionValue.find_by_name!("Extra Large")
+
+red = Spree::OptionValue.find_by_name!("Red")
+blue = Spree::OptionValue.find_by_name!("Blue")
+green = Spree::OptionValue.find_by_name!("Green")
+
+default_attrs = {
+  :count_on_hand => 10
+}
+
+variants = [
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [small, red],
+    :sku => "ROR-00001",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [small, blue],
+    :sku => "ROR-00002",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [small, green],
+    :sku => "ROR-00003",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [medium, red],
+    :sku => "ROR-00004",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [medium, blue],
+    :sku => "ROR-00005",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [medium, green],
+    :sku => "ROR-00006",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [large, red],
+    :sku => "ROR-00007",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [large, blue],
+    :sku => "ROR-00008",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [large, green],
+    :sku => "ROR-00009",
+    :cost_price => 17
+  },
+  {
+    :product => ror_baseball_jersey,
+    :option_values => [extra_large, green],
+    :sku => "ROR-00012",
+    :cost_price => 17
+  },
+]
+
+masters = {
+  ror_baseball_jersey => {
+    :sku => "ROR-001",
+    :cost_price => 17,
+  },
+  ror_tote => {
+    :sku => "ROR-00011",
+    :cost_price => 17
+  },
+  ror_bag => {
+    :sku => "ROR-00012",
+    :cost_price => 21
+  },
+  ror_jr_spaghetti => {
+    :sku => "ROR-00013",
+    :cost_price => 17
+  },
+  ror_mug => {
+    :sku => "ROR-00014",
+    :cost_price => 11
+  },
+  ror_ringer => {
+    :sku => "ROR-00015",
+    :cost_price => 17
+  },
+  ror_stein => {
+    :sku => "ROR-00016",
+    :cost_price => 15
+  },
+  apache_baseball_jersey => {
+    :sku => "APC-00001",
+    :cost_price => 17
+  },
+  ruby_baseball_jersey => {
+    :sku => "RUB-00001",
+    :cost_price => 17
+  },
+  spree_baseball_jersey => {
+    :sku => "SPR-00001",
+    :cost_price => 17
+  },
+  spree_stein => {
+    :sku => "SPR-00016",
+    :cost_price => 15
+  },
+  spree_jr_spaghetti => {
+    :sku => "SPR-00013",
+    :cost_price => 17
+  },
+  spree_mug => {
+    :sku => "SPR-00014",
+    :cost_price => 11
+  },
+  spree_ringer => {
+    :sku => "SPR-00015",
+    :cost_price => 17
+  },
+  spree_tote => {
+    :sku => "SPR-00011",
+    :cost_price => 13
+  },
+  spree_bag => {
+    :sku => "SPR-00012",
+    :cost_price => 21
+  }
+}
+
+variants.each do |variant_attrs|
+  Spree::Variant.create!(variant_attrs, :without_protection => true)
+end
+
+masters.each do |product, variant_attrs|
+  product.master.update_attributes!(variant_attrs)
+end

data/db/samples.rb

@@ -0,0 +1,2 @@
+Spree::Sample.load_sample("payment_methods")
+Spree::Sample.load_sample("shipping_categories")

data/lib/spree/sample.rb

@@ -0,0 +1,17 @@
+module Spree
+  module Sample
+    def self.load_sample(file)
+      path = File.expand_path(samples_path + "#{file}.rb")
+      # Check to see if the specified file has been loaded before
+      if !$LOADED_FEATURES.include?(path)
+        require path
+        puts "Loaded #{file.titleize} samples"
+      end
+    end
+
+    private
+      def self.samples_path
+        Pathname.new(File.join(File.dirname(__FILE__), '..', '..', 'db', 'samples'))
+      end
+  end
+end

data/lib/spree_sample.rb

@@ -1,7 +1,31 @@
 require 'spree_core'
+require 'spree/sample'
 
 module SpreeSample
   class Engine < Rails::Engine
     engine_name 'spree_sample'
+
+    # Needs to be here so we can access it inside the tests
+    def self.load_samples
+      Spree::Sample.load_sample("payment_methods")
+      Spree::Sample.load_sample("shipping_categories")
+      Spree::Sample.load_sample("shipping_methods")
+      Spree::Sample.load_sample("tax_categories")
+      Spree::Sample.load_sample("tax_rates")
+
+      Spree::Sample.load_sample("products")
+      Spree::Sample.load_sample("taxons")
+      Spree::Sample.load_sample("option_values")
+      Spree::Sample.load_sample("product_option_types")
+      Spree::Sample.load_sample("product_properties")
+      Spree::Sample.load_sample("prototypes")
+      Spree::Sample.load_sample("variants")
+      Spree::Sample.load_sample("assets")
+
+      Spree::Sample.load_sample("orders")
+      Spree::Sample.load_sample("line_items")
+      Spree::Sample.load_sample("adjustments")
+      Spree::Sample.load_sample("payments")
+    end
   end
 end

data/lib/tasks/sample.rake

@@ -1,11 +1,24 @@
 require 'ffaker'
+require 'pathname'
+require 'spree/sample'
 
 namespace :spree_sample do
   desc 'Loads sample data'
-  task :load do
-    sample_path = File.join(File.dirname(__FILE__), '..', '..', 'db', 'sample')
+  task :load => :environment do
+    if ARGV.include?("db:migrate")
+      puts %Q{
+Please run db:migrate separately from spree_sample:load.
 
-    Rake::Task['db:load_dir'].reenable
-    Rake::Task['db:load_dir'].invoke(sample_path)
+Running db:migrate and spree_sample:load at the same time has been known to
+cause problems where columns may be not available during sample data loading.
+
+Migrations have been run. Please run "rake spree_sample:load" by itself now.
+      }
+      exit(1)
+    end
+
+    SpreeSample::Engine.load_samples
   end
 end
+
+