Add with stub modifier

src/spectator/mocks/exception_stub.cr

@@ -11,6 +11,12 @@ module Spectator
       raise @exception
     end
 
+    # Returns a new stub with constrained arguments.
+    def with(*args, **kwargs)
+      constraint = Arguments.new(args, kwargs)
+      self.class.new(method, @exception, constraint, location)
+    end
+
     # Creates the stub.
     def initialize(method : Symbol, @exception : Exception, constraint : AbstractArguments? = nil, location : Location? = nil)
       super(method, constraint, location)

src/spectator/mocks/multi_value_stub.cr

@@ -15,6 +15,12 @@ module Spectator
       end
     end
 
+    # Returns a new stub with constrained arguments.
+    def with(*args, **kwargs)
+      constraint = Arguments.new(args, kwargs)
+      self.class.new(method, @values, constraint, location)
+    end
+
     # Creates the stub.
     def initialize(method : Symbol, @values : Array(T), constraint : AbstractArguments? = nil, location : Location? = nil)
       super(method, constraint, location)

src/spectator/mocks/null_stub.cr

@@ -7,5 +7,11 @@ module Spectator
     # Invokes the stubbed implementation.
     def call(call : MethodCall) : Nil
     end
+
+    # Returns a new stub with constrained arguments.
+    def with(*args, **kwargs)
+      constraint = Arguments.new(args, kwargs)
+      self.class.new(method, constraint, location)
+    end
   end
 end

src/spectator/mocks/proc_stub.cr

@@ -10,6 +10,12 @@ module Spectator
       @proc.call(call.arguments)
     end
 
+    # Returns a new stub with constrained arguments.
+    def with(*args, **kwargs)
+      constraint = Arguments.new(args, kwargs)
+      self.class.new(method, @proc, constraint, location)
+    end
+
     # Creates the stub.
     def initialize(method : Symbol, @proc : Proc(AbstractArguments, T), constraint : AbstractArguments? = nil, location : Location? = nil)
       super(method, constraint, location)
@@ -20,4 +26,12 @@ module Spectator
       initialize(method, block, constraint, location)
     end
   end
+
+  module StubModifiers
+    # Returns a new stub with an argument constraint.
+    def with(*args, **kwargs, &block : AbstractArguments -> T) forall T
+      constraint = Arguments.new(args, kwargs)
+      ProcStub(T).new(method, block, constraint, location)
+    end
+  end
 end

src/spectator/mocks/stub_modifiers.cr

@@ -1,5 +1,7 @@
 module Spectator
   # Mixin intended for `Stub` to return new, modified stubs.
   module StubModifiers
+    # Returns a new stub of the same type with constrained arguments.
+    abstract def with(*args, **kwargs)
   end
 end

src/spectator/mocks/value_stub.cr

@@ -11,6 +11,12 @@ module Spectator
       @value
     end
 
+    # Returns a new stub with constrained arguments.
+    def with(*args, **kwargs)
+      constraint = Arguments.new(args, kwargs)
+      self.class.new(method, @value, constraint, location)
+    end
+
     # Creates the stub.
     def initialize(method : Symbol, @value : T, constraint : AbstractArguments? = nil, location : Location? = nil)
       super(method, constraint, location)