lib/muack/modifier.rb in muack-0.5.1 vs lib/muack/modifier.rb in muack-0.5.2

@@ -8,9 +8,14 @@
       defi.args = [WithAnyArgs]
       self
     end
 
     # Public API
+    def returns val=nil, &block
+      defi.block = block || lambda{ val }
+    end
+
+    # Public API
     def times number
       if number >= 1
         (number - 1).times{ mock.__mock_defis_push(defi) }
       elsif number == 0
         mock.__mock_defis_pop(defi)