FactoryBot的随机特性

Question

我想使用FactoryBot随机返回特征：

FactoryBot.define do
  factory :user do

    [:active, inactive].sample

    trait :active do
      active { true }
      ...
    end
    trait :inactive do
      active { false }
      ...
    end
  end
end

要做到这一点：

(1..5).map{ |e| FactoryBot.build(:user) }.map(&:active?)
=> [true, false, false, true, false]

实际上是这样的：

FactoryBot.define do
  factory :user do
    active                    { [true, false].Sample }
    name                      { "name-#{SecureRandom.uuid}" }
    birthday                  { active == true ? rand(18..99).years.ago - rand(0..365).days.ago : nil }
    preferred_contact_method  { active == true ? %w(phone email).sample : nil }
    activated_at              { active == true ? rand(1..200).days.ago : nil }
    contact_phone_number      { preferred_contact_method == "phone" ? "+33XXXXXXXXX" : nil }
    contact_email             { preferred_contact_method == "email" ? "[email protected]" : nil }
  end
end

有可能吗？

Answer 1

在最终的答案中，我发现了一个事实，即trait只执行一次阻止它。我将暂时分享我所使用的代码示例，但如果您测试它，您将只有活跃用户或非活动用户，但不能同时拥有两者。

---

您可以在两个lambda中导出特征的逻辑，然后随机选择一个：

trait_active = lambda do |context|
    context.active { true }
    #...
end
trait_inactive = lambda do |context|
    context.active { false }
    # ...
end

FactoryBot.define do
    factory :user do
        trait :active do
            trait_active.call(self)
        end
        trait :inactive do
            trait_inactive.call(self)
        end
        trait :schrodinger do
            [trait_active, trait_inactive].sample.call(self)
        end
    end
end

lambda中的context属性在这里非常重要，你可以看到更多关于那个in this answer。