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

@@ -1,7 +1,7 @@
 
-require 'muack/definition'
+require 'muack/error'
 
 module Muack
   class Modifier < Struct.new(:mock, :defi)
     # Public API
     def with_any_args
@@ -10,13 +10,24 @@
     end
 
     # Public API
     def returns val=nil, &block
       defi.block = block || lambda{ val }
+      self
     end
 
     # Public API
+    def proxy
+      defi.proxy = true
+      self
+    end
+
+    # Public API
     def times number
+      if mock.__mock_class == Stub
+        raise StubHasNoTimes.new(object, defi, number)
+      end
+
       if number >= 1
         (number - 1).times{ mock.__mock_defis_push(defi) }
       elsif number == 0
         mock.__mock_defis_pop(defi)
       else