51 lines
1.3 KiB
Ruby
51 lines
1.3 KiB
Ruby
describe "modifying inherited factories with traits" do
|
|
before do
|
|
define_model("User", gender: :string, admin: :boolean, age: :integer)
|
|
FactoryBot.define do
|
|
factory :user do
|
|
trait(:female) { gender { "Female" } }
|
|
trait(:male) { gender { "Male" } }
|
|
|
|
trait(:young_admin) do
|
|
admin { true }
|
|
age { 17 }
|
|
end
|
|
|
|
female
|
|
young_admin
|
|
|
|
factory :female_user do
|
|
gender { "Female" }
|
|
age { 25 }
|
|
end
|
|
|
|
factory :male_user do
|
|
gender { "Male" }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
it "returns the correct value for overridden attributes from traits" do
|
|
expect(FactoryBot.build(:male_user).gender).to eq "Male"
|
|
end
|
|
|
|
it "returns the correct value for overridden attributes from traits defining multiple attributes" do
|
|
expect(FactoryBot.build(:female_user).gender).to eq "Female"
|
|
expect(FactoryBot.build(:female_user).age).to eq 25
|
|
expect(FactoryBot.build(:female_user).admin).to eq true
|
|
end
|
|
|
|
it "allows modification of attributes created via traits" do
|
|
FactoryBot.modify do
|
|
factory :male_user do
|
|
age { 20 }
|
|
end
|
|
end
|
|
|
|
expect(FactoryBot.build(:male_user).gender).to eq "Male"
|
|
expect(FactoryBot.build(:male_user).age).to eq 20
|
|
expect(FactoryBot.build(:male_user).admin).to eq true
|
|
end
|
|
end
|