Add flat rate calculator traits to related factories

This commit is contained in:
Kristina Lim
2019-01-11 22:13:21 +08:00
parent 1f72eb9f68
commit 06ba5182b4

View File

@@ -1,12 +1,23 @@
attach_per_item_trait = proc do
trait :per_item do
transient { amount 1 }
calculator { build(:calculator_per_item, preferred_amount: amount) }
FactoryBot.define do
factory :calculator_flat_rate, class: Spree::Calculator::FlatRate do
preferred_amount { generate(:calculator_amount) }
end
end
FactoryBot.modify do
factory :payment_method, &attach_per_item_trait
factory :shipping_method, &attach_per_item_trait
factory :enterprise_fee, &attach_per_item_trait
attach_calculator_traits = proc do
trait :flat_rate do
transient { amount 1 }
calculator { build(:calculator_flat_rate, preferred_amount: amount) }
end
trait :per_item do
transient { amount 1 }
calculator { build(:calculator_per_item, preferred_amount: amount) }
end
end
factory :payment_method, &attach_calculator_traits
factory :shipping_method, &attach_calculator_traits
factory :enterprise_fee, &attach_calculator_traits
end