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

describe "global to_create" do
  before do
    define_model("User", name: :string)
    define_model("Post", name: :string)

    FactoryBot.define do
      to_create { |instance| instance.name = "persisted!" }

      trait :override_to_create do
        to_create { |instance| instance.name = "override" }
      end

      factory :user do
        name { "John Doe" }

        factory :child_user

        factory :child_user_with_trait do
          override_to_create
        end
      end

      factory :post do
        name { "Great title" }

        factory :child_post

        factory :child_post_with_trait do
          override_to_create
        end
      end
    end
  end

  it "handles base to_create" do
    expect(FactoryBot.create(:user).name).to eq "persisted!"
    expect(FactoryBot.create(:post).name).to eq "persisted!"
  end

  it "handles child to_create" do
    expect(FactoryBot.create(:child_user).name).to eq "persisted!"
    expect(FactoryBot.create(:child_post).name).to eq "persisted!"
  end

  it "handles child to_create with trait" do
    expect(FactoryBot.create(:child_user_with_trait).name).to eq "override"
    expect(FactoryBot.create(:child_post_with_trait).name).to eq "override"
  end

  it "handles inline trait override" do
    user = FactoryBot.create(:child_user, :override_to_create)
    post = FactoryBot.create(:child_post, :override_to_create)

    expect(user.name).to eq "override"
    expect(post.name).to eq "override"
  end

  it "uses to_create globally across FactoryBot.define" do
    define_model("Company", name: :string)

    FactoryBot.define do
      factory :company
    end

    company = FactoryBot.create(:company)
    override_company = FactoryBot.create(:company, :override_to_create)

    expect(company.name).to eq "persisted!"
    expect(override_company.name).to eq "override"
  end
end

describe "global skip_create" do
  before do
    define_model("User", name: :string)
    define_model("Post", name: :string)

    FactoryBot.define do
      skip_create

      trait :override_to_create do
        to_create { |instance| instance.name = "override" }
      end

      factory :user do
        name { "John Doe" }

        factory :child_user

        factory :child_user_with_trait do
          override_to_create
        end
      end

      factory :post do
        name { "Great title" }

        factory :child_post

        factory :child_post_with_trait do
          override_to_create
        end
      end
    end
  end

  it "does not persist any record" do
    expect(FactoryBot.create(:user)).to be_new_record
    expect(FactoryBot.create(:post)).to be_new_record
  end

  it "does not persist child records" do
    expect(FactoryBot.create(:child_user)).to be_new_record
    expect(FactoryBot.create(:child_post)).to be_new_record
  end

  it "honors overridden to_create" do
    expect(FactoryBot.create(:child_user_with_trait).name).to eq "override"
    expect(FactoryBot.create(:child_post_with_trait).name).to eq "override"
  end

  it "honors inline trait to_create" do
    expect(FactoryBot.create(:child_user, :override_to_create).name).to eq "override"
    expect(FactoryBot.create(:child_post, :override_to_create).name).to eq "override"
  end
end