81 lines
2.0 KiB
Ruby
81 lines
2.0 KiB
Ruby
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
|