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

require 'spec_helper'

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

    FactoryGirl.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(FactoryGirl.create(:user).name).to eq 'persisted!'
    expect(FactoryGirl.create(:post).name).to eq 'persisted!'
  end

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

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

  it 'handles inline trait override' do
    expect(FactoryGirl.create(:child_user, :override_to_create).name).to eq 'override'
    expect(FactoryGirl.create(:child_post, :override_to_create).name).to eq 'override'
  end

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

    FactoryGirl.define do
      factory :company
    end

    expect(FactoryGirl.create(:company).name).to eq 'persisted!'
    expect(FactoryGirl.create(:company, :override_to_create).name).to eq 'override'
  end
end

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

    FactoryGirl.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(FactoryGirl.create(:user)).to be_new_record
    expect(FactoryGirl.create(:post)).to be_new_record
  end

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

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

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