Add ability to remove specific stubs

src/spectator/mocks/double.cr

@@ -103,6 +103,11 @@ module Spectator
       @stubs.unshift(stub)
     end
 
+    protected def _spectator_remove_stub(stub : Stub) : Nil
+      Log.debug { "Removing stub #{stub} from #{_spectator_stubbed_name}" }
+      @stubs.delete(stub)
+    end
+
     protected def _spectator_clear_stubs : Nil
       Log.debug { "Clearing stubs for #{_spectator_stubbed_name}" }
       @stubs.clear

src/spectator/mocks/mock.cr

@@ -50,6 +50,10 @@ module Spectator
             end
           {% end %}
 
+          def _spectator_remove_stub(stub : ::Spectator::Stub) : Nil
+            @_spectator_stubs.try &.delete(stub)
+          end
+
           def _spectator_clear_stubs : Nil
             @_spectator_stubs = nil
           end
@@ -139,6 +143,10 @@ module Spectator
             entry.stubs
           end
 
+          def _spectator_remove_stub(stub : ::Spectator::Stub) : Nil
+            @@_spectator_mock_registry[self]?.try &.stubs.delete(stub)
+          end
+
           def _spectator_clear_stubs : Nil
             @@_spectator_mock_registry.delete(self)
           end

src/spectator/mocks/mocked.cr

@@ -26,6 +26,10 @@ module Spectator
       _spectator_stubs.unshift(stub)
     end
 
+    def _spectator_remove_stub(stub : Stub) : Nil
+      _spectator_stubs.delete(stub)
+    end
+
     def _spectator_clear_stubs : Nil
       _spectator_stubs.clear
     end

src/spectator/mocks/reference_mock_registry.cr

@@ -25,6 +25,12 @@ module Spectator
       @entries[key]
     end
 
+    # Retrieves all stubs defined for a mocked object or nil if the object isn't mocked yet.
+    def []?(object : Reference)
+      key = Box.box(object)
+      @entries[key]?
+    end
+
     # Retrieves all stubs defined for a mocked object.
     #
     # Yields to the block on the first retrieval.

src/spectator/mocks/stubbable.cr

@@ -28,6 +28,9 @@ module Spectator
     # Defines a stub to change the behavior of a method.
     abstract def _spectator_define_stub(stub : Stub) : Nil
 
+    # Removes a specific, previously defined stub.
+    abstract def _spectator_remove_stub(stub : Stub) : Nil
+
     # Clears all previously defined stubs.
     abstract def _spectator_clear_stubs : Nil
 

src/spectator/mocks/stubbed_type.cr

@@ -20,6 +20,10 @@ module Spectator
       _spectator_stubs.unshift(stub)
     end
 
+    def _spectator_remove_stub(stub : Stub) : Nil
+      _spectator_stubs.delete(stub)
+    end
+
     def _spectator_clear_stubs : Nil
       _spectator_stubs.clear
     end

src/spectator/mocks/value_mock_registry.cr

@@ -29,6 +29,12 @@ module Spectator
       @entries[key]
     end
 
+    # Retrieves all stubs defined for a mocked object or nil if the object isn't mocked yet.
+    def []?(object : T)
+      key = value_bytes(object)
+      @entries[key]?
+    end
+
     # Retrieves all stubs defined for a mocked object.
     #
     # Yields to the block on the first retrieval.