Permalink
Fetching contributors…
Cannot retrieve contributors at this time
81 lines (63 sloc) 2 KB
describe 'global initialize_with' do
before do
define_class('User') do
attr_accessor :name
def initialize(name)
@name = name
end
end
define_class('Post') do
attr_reader :name
def initialize(name)
@name = name
end
end
FactoryBot.define do
initialize_with { new("initialize_with") }
trait :with_initialize_with do
initialize_with { new("trait initialize_with") }
end
factory :user do
factory :child_user
factory :child_user_with_trait do
with_initialize_with
end
end
factory :post do
factory :child_post
factory :child_post_with_trait do
with_initialize_with
end
end
end
end
it 'handles base initialize_with' do
expect(FactoryBot.build(:user).name).to eq 'initialize_with'
expect(FactoryBot.build(:post).name).to eq 'initialize_with'
end
it 'handles child initialize_with' do
expect(FactoryBot.build(:child_user).name).to eq 'initialize_with'
expect(FactoryBot.build(:child_post).name).to eq 'initialize_with'
end
it 'handles child initialize_with with trait' do
expect(FactoryBot.build(:child_user_with_trait).name).to eq 'trait initialize_with'
expect(FactoryBot.build(:child_post_with_trait).name).to eq 'trait initialize_with'
end
it 'handles inline trait override' do
expect(FactoryBot.build(:child_user, :with_initialize_with).name).to eq 'trait initialize_with'
expect(FactoryBot.build(:child_post, :with_initialize_with).name).to eq 'trait initialize_with'
end
it 'uses initialize_with globally across FactoryBot.define' do
define_class('Company') do
attr_reader :name
def initialize(name)
@name = name
end
end
FactoryBot.define do
factory :company
end
expect(FactoryBot.build(:company).name).to eq 'initialize_with'
expect(FactoryBot.build(:company, :with_initialize_with).name).to eq 'trait initialize_with'
end
end