thoughtbot--factory_bot/spec/acceptance/modify_inherited_spec.rb

require "spec_helper"

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