DRY up the proc extension.

benaskins · Nov 12, 2008 · f97bf75 · f97bf75
1 parent f73737c
commit f97bf75
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/lib/cucumber/core_ext/proc.rb b/lib/cucumber/core_ext/proc.rb
@@ -10,10 +10,9 @@ module CallIn
 
       def call_in(obj, *args)
         obj.extend(mod)
-        a = arity == -1 ? 0 : arity
+        if self != StepMother::PENDING && args.length != arity2
-        if self != StepMother::PENDING && args.length != a
           # We have to manually raise when the block has arity -1 (no pipes)
-          raise ArityMismatchError.new("expected #{arity == -1 ? 0 : arity} block argument(s), got #{args.length}")
+          raise ArityMismatchError.new("expected #{arity2} block argument(s), got #{args.length}")
         else
           obj.__send__(meth, *args)
         end