FactoryGirl - override association with trait?

Question

Let's say I have the following ActiveRecord models:

class Car
  belongs_to :driver
end

class Driver
  # Has attribute :name
  has_one :car
end

And I define a couple of factories using these models:

FactoryGirl.define do
  factory :car do
    association :driver

    trait :fast_car do
      association :driver, :fast
    end
  end
end

FactoryGirl.define do
  factory :driver do
    name 'Jason'

    trait :fast do
      name 'Mario'
    end
  end
end

When I execute the following code:

car = FactoryGirl.create(:car, :fast_car)

I would expect car.driver.name to equal Mario, but instead it equals Jason. This leads me to believe that you can't use traits to override associations for factories. Is this true? If so, what would be the proper way to override the associated Driver for a fast car?

Firewarden

Answer 1

Fortunately, you can. You need to specify the factory keyword for an association with an array, where the first element is the name of the factory that you want to use for the association and the rest elements are the factory's traits:

FactoryGirl.define do
  factory :car do
    association :driver

    trait :fast_car do
      association :driver, factory: [:driver, :fast]
    end
  end
end

FactoryGirl.define do
  factory :driver do
    name 'Jason'

    trait :fast do
      name 'Mario'
    end
  end
end

Broomfield