FactoryGirl DRY sau khi tạo/xây dựng móc

Question

Tôi muốn khô cạn sau khi tạo/xây dựng móc trong nhà máy của tôi:

FactoryGirl.define do 

    factory :poll do 

    sequence :title do |n| 
     "MyPollTitle#{n}" 
    end 
    sequence :description do |n| 
     "MyPollDescription#{n}" 
    end 
    user 

    factory :poll_with_answers do 

     ignore do 
     answers_count 2 
     end 

     after(:build) do |poll, evaluator| 
     evaluator.answers_count.times do 
      poll.answers << build(:answer, poll: poll) 
     end 
     end 

     after(:create) do |poll, evaluator| 
     evaluator.answers_count.times do 
      poll.answers << create(:answer, poll: poll) 
     end 
     end 
    end 
    end 
end 

Vấn đề tôi đang phải đối mặt là có vẻ như tôi không thể xác định phương pháp trong FG? Ý tưởng làm thế nào để DRY này lên?

Answer 1

Thứ nhất, after(:create) ngầm gọi after(:build), ít nhất là trong các phiên bản gần đây của FactoryGirl:

sau (: xây dựng) - gọi sau khi một nhà máy được xây dựng (thông qua FactoryGirl.build, FactoryGirl.create)

https://github.com/thoughtbot/factory_girl/blob/master/GETTING_STARTED.md#callbacks

Vì vậy, trong trường hợp của bạn, đoạn code sau nên là đủ:

after(:build) do |poll, evaluator| 
    evaluator.answers_count.times do 
    poll.answers << build(:answer, poll: poll) 
    end 
end 

Tuy nhiên, after(:build) không được kích hoạt khi bạn sử dụng build_stubbed() thay vì build(), đó là những gì tôi đã cố gắng làm khi tôi gặp chủ đề này. Để xử lý mã này, hóa ra bạn có thể sử dụng phương thức callback() để gọi cùng một khối cho nhiều phương thức:

factory :user do 
    callback(:after_build, :after_stub) do |user| 
    do_something_with(user) 
    end 
end 

Answer 2

Đây có thể là một thủ thuật rẻ tiền, nhưng bạn có thể tạo ra một lambda trong nhà máy thứ hai:

factory :poll_with_answers do 
    ignore do 
    answers_count 2 
    end 

    make_answers = lambda do |poll, evaluator, method| 
    evaluator.answers_count.times do 
     poll.answers << send(method, :answer, poll: poll) 
    end 
    end 

    after(:build) do |poll, evaluator| 
    make_answers.call poll, evaluator, :build 
    end 

    after(:create) do |poll, evaluator| 
    make_answers.call poll, evaluator, :create 
    end 
end 

Tôi không ở tất cả hài lòng với mô hình này, nhưng ít nhất nó DRY của cụ lên.