如何重置factory_girl序列？

Question

假设我有一个项目工厂

Factory.define :project do |p|
  p.sequence(:title)    { |n| "project #{n} title"                  }
  p.sequence(:subtitle) { |n| "project #{n} subtitle"               }
  p.sequence(:image)    { |n| "../images/content/projects/#{n}.jpg" }
  p.sequence(:date)     { |n| n.weeks.ago.to_date                   }
end

并且我正在创建项目实例

Factory.build :project
Factory.build :project

此时，下次执行 Factory.build(:project) 时，我将收到一个标题设置为“项目 3 标题”的项目实例等等。并不奇怪。

现在说我希望在这个范围内重置我的计数器。比如：

Factory.build :project #=> Project 3
Factory.reset :project #=> project factory counter gets reseted
Factory.build :project #=> A new instance of project 1

实现这一目标的最佳方法是什么？

我目前使用以下版本：

factory_girl (1.3.1) 工厂女孩轨道 (1.0)

Answer 1

只需在您的 before/after 回调中调用 FactoryGirl.reload 即可。这在 FactoryGirl 代码库中定义为：

module FactoryGirl
  def self.reload
    self.factories.clear
    self.sequences.clear
    self.traits.clear
    self.find_definitions
  end
end

由于某种原因，调用 FactoryGirl.sequences.clear 是不够的。进行完全重新加载可能会产生一些开销，但是当我尝试使用/不使用回调时，我的测试需要大约 30 秒才能运行。因此，开销不足以影响我的工作流程。

Answer 2

有一个名为 sequence_by_name 的类方法可以按名称获取序列，然后您可以调用 rewind 它将重置为 1。

FactoryBot.sequence_by_name(:order).rewind

或者如果您想重置全部。

FactoryBot.rewind_sequences

这是 github 上文件的链接

Answer 3

经过我对源码的摸索，终于找到了解决方案。如果您使用的是factory_girl 1.3.2（这是我撰写本文时的最新版本），您可以将以下代码添加到factories.rb文件的顶部：

class Factory  
  def self.reset_sequences
    Factory.factories.each do |name, factory|
      factory.sequences.each do |name, sequence|
        sequence.reset
      end
    end
  end
  
  def sequences
    @sequences
  end
  
  def sequence(name, &block)
    s = Sequence.new(&block)
    
    @sequences ||= {}
    @sequences[name] = s
    
    add_attribute(name) { s.next }
  end
  
  def reset_sequence(name)
    @sequences[name].reset
  end
  
  class Sequence
    def reset
      @value = 0
    end
  end
end

然后，在Cucumber的env.rb中，只需添加：

After do
  Factory.reset_sequences
end

我假设如果您在 rspec 测试中遇到同样的问题，您可以在 :each 方法之后使用 rspecs 。

目前，这种方法只考虑工厂内定义的序列，例如：

Factory.define :specialty do |f|
  f.sequence(:title) { |n| "Test Specialty #{n}"}
  f.sequence(:permalink) { |n| "permalink#{n}" }
end

我还没有编写代码来处理：Factory.sequence...

Answer 4

对于谷歌搜索的人：无需进一步扩展，只需

FactoryGirl.create :user
#=> User id: 1, name: "user_1"
FactoryGirl.create :user
#=> User id: 2, name: "user_2"

DatabaseCleaner.clean_with :truncation #wiping out database with truncation
FactoryGirl.reload

FactoryGirl.create :user
#=> User id: 1, name: "user_1"

上为我工作

* factory_girl (4.3.0)
* factory_girl_rails (4.3.0)

在https://stackoverflow.com/ FactoryGirl.reload a/16048658

Answer 5

根据 ThoughBot 这里< /a>，需要重置测试之间的顺序是一种反模式。

总结：

如果您有这样的事情：

FactoryGirl.define do
  factory :category do
    sequence(:name) {|n| "Category #{n}" }
  end
end

您的测试应该如下所示：

Scenario: Create a post under a category
   Given a category exists with a name of "My Category"
   And I am signed in as an admin
   When I go to create a new post
   And I select "My Category" from "Categories"
   And I press "Create"
   And I go to view all posts
   Then I should see a post with the category "My Category"

不是这样：

Scenario: Create a post under a category
  Given a category exists
  And I am signed in as an admin
  When I go to create a new post
  And I select "Category 1" from "Categories"
  And I press "Create"
  And I go to view all posts
  Then I should see a post with the category "Category 1"

Answer 6

必须确保序列从 1 到 8，然后重新启动到 1，依此类推。像这样实现：

class FGCustomSequence
  def initialize(max)
    @marker, @max = 1, max
  end
  def next
    @marker = (@marker >= @max ? 1 : (@marker + 1))
  end
  def peek
    @marker.to_s
  end
end

FactoryGirl.define do
  factory :image do
    sequence(:picture, FGCustomSequence.new(8)) { |n| "image#{n.to_s}.png" }
  end
end

doc 说“该值只需要支持#next方法。”但为了让 CustomSequence 对象继续运行，它也需要支持 #peek 方法。最后我不知道这会工作多久，因为它有点侵入 FactoryGirl 的内部，当他们做出改变时，这可能无法正常工作

Answer 7

没有内置的方法来重置序列，请参阅此处的源代码：

http://github.com/thoughtbot/factory_girl/blob/master/lib/factory_girl/sequence.rb

但是，有些人已经破解/猴子修补了此功能。这是一个示例：

http://www.pmamediagroup.com/2009/05/smarter-工厂女孩测序/

Answer 8

要重置特定序列，您可以尝试

# spec/factories/schedule_positions.rb
FactoryGirl.define do
  sequence :position do |n| 
    n
  end

  factory :schedule_position do
    position
    position_date Date.today
    ...
  end
end

# spec/models/schedule_position.rb
require 'spec_helper'

describe SchedulePosition do
  describe "Reposition" do
    before(:each) do
      nullify_position
      FactoryGirl.create_list(:schedule_position, 10)
    end
  end

  protected

  def nullify_position
    position = FactoryGirl.sequences.find(:position)
    position.instance_variable_set :@value, FactoryGirl::Sequence::EnumeratorAdapter.new(1)
  end
end

Answer 9

如果您使用 Cucumber，则可以将其添加到步骤定义中：

Given(/^I reload FactoryGirl/) do
  FactoryGirl.reload
end

然后在需要时调用它。