FactoryGirl 中的 find_or_initialize_by

Question

我想知道 FactoryGirl 中是否有find_or_initialize_by的等价物来解决以下问题：

目标是该模型使用具有相同国家/地区的两个表。 我不想使用国家/地区的序列（正如我在电子邮件中发现的那样）。

Country 有一个唯一性约束，但我的主要问题是当我调用一次 FactoryGirl.create(:click) 时，它会创建两次相同的 Country 记录

因此，验证在测试中失败。

规格：

# models/click_spec.rb
describe Click do
    it "should have a valid constructor" do
        FactoryGirl.create(:click).should be_valid
    end
end

工厂：

# factories/countries.rb
FactoryGirl.define do
    factory :country do
        name "United States"
        slug "us"
    end
end

# factories/offers.rb
FactoryGirl.define do
    factory :offer do
        association :country, factory: :country
        # Other columns
    end
end

# factories/users.rb
FactoryGirl.define do
    factory :user do
        association :country, factory: :country
        # Other columns
    end
end

# factories/clicks.rb
FactoryGirl.define do
    factory :click do
        association :offer, factory: :offer
        association :user, factory: :user
        # Other columns
    end
end

模型：

class Country < ActiveRecord::Base
    validates :name, :slug,
    presence: true,
    uniqueness: { case_sensitive: false }

    validates :slug,
    length: { is: 2 }

end

Answer 1

您应该能够通过使用initialize_with来完成这项工作：

FactoryGirl.define do
  factory :country do
    name "United States"
    slug "us"
    initialize_with { Country.find_or_create_by_name(name) }
  end
end

这将始终使用同一个国家。 您可能希望嵌套工厂以允许其他工厂使用不同的名称：

FactoryGirl.define do
  factory :country do
    initialize_with { Country.find_or_create_by_name(name) }
    factory :united_states do
      name "United States"
      slug "us"
    end
  end
end

Answer 2

我也遇到了类似的问题，我的应用程序的Country模型也是如此。 这就是我所做的。

为确保 FactoryBot 的build和create仍能正常运行，我们应该只覆盖to_create的逻辑，执行以下操作：

factory :country do
  to_create do |instance|
    instance.id = Country.create_with(name: instance.name).find_or_create_by(slug: instance.slug).id
    instance.reload
  end

  name { "United States" }
  slug { "us" }
end

查询解释：

Country
.create_with(name: instance.name) # if not found, create with this `name` (and `slug` defined below)
.find_or_create_by(slug: instance.slug) # find by primary key `slug'

这确保build保持其“构建/初始化对象”的默认行为，并且不执行任何数据库读取或写入，因此它总是很快。 仅覆盖create逻辑以获取现有记录（如果存在），而不是尝试始终创建新记录。

最初发布在https://stackoverflow.com/a/55235861/3956879 上。

看看我的文章解释了这一点。