Allow setting id for build_stubbed objects

Closes #451
thoughtbot · Nov 7, 2012 · e85c008 · e85c008
1 parent fc14baf
commit e85c008
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/factory_girl/strategy/stub.rb b/lib/factory_girl/strategy/stub.rb
@@ -21,7 +21,7 @@ def next_id
       end
 
       def stub_database_interaction_on_result(result_instance)
-        result_instance.id = next_id
+        result_instance.id ||= next_id
 
         result_instance.instance_eval do
           def persisted?

diff --git a/spec/acceptance/build_stubbed_spec.rb b/spec/acceptance/build_stubbed_spec.rb
@@ -134,3 +134,17 @@
     expect { build_stubbed(:thing_without_timestamp, :created_at => Time.now) }.to raise_error(NoMethodError, /created_at=/)
   end
 end
+
+describe 'defaulting `id`' do
+  before do
+    define_model('Post')
+
+    FactoryGirl.define do
+      factory :post
+    end
+  end
+
+  it 'allows overriding id' do
+    FactoryGirl.build_stubbed(:post, id: 12).id.should eq 12
+  end
+end