Make sure a block always exists

thoughtbot · Nov 21, 2011 · 2bf5cea · 2bf5cea
1 parent c46c090
commit 2bf5cea
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/factory_girl/factory.rb b/lib/factory_girl/factory.rb
@@ -34,6 +34,8 @@ def default_strategy #:nodoc:
     end
 
     def run(proxy_class, overrides, &block) #:nodoc:
+      block ||= lambda {|result| result }
+
       runner_options = {
         :attributes  => attributes,
         :callbacks   => callbacks,
@@ -42,9 +44,7 @@ def run(proxy_class, overrides, &block) #:nodoc:
         :proxy_class => proxy_class
       }
 
-      result = Runner.new(runner_options).run(overrides)
-
-      block ? block.call(result) : result
+      block[Runner.new(runner_options).run(overrides)]
     end
 
     def human_names