Handle injecting stubbed abstract method

2022-03-19 12:33:09 -06:00 · 2022-03-19 12:33:09 -06:00 · 25f546245a
parent 9091b8d027
commit 25f546245a
1 changed files with 6 additions and 2 deletions
--- a/src/spectator/mocks/stubbable.cr
+++ b/src/spectator/mocks/stubbable.cr
@ -270,8 +270,12 @@ module Spectator

    # Utility for defining a stubbed method and a fallback.
    private macro inject_stub(method)
-      {{method}}
-      stub {{method}}
+      {% if method.abstract? %}
+        abstract_stub {{method}}
+      {% else %}
+        {{method}}
+        stub {{method}}
+      {% end %}
    end

    # Redefines all methods on a type to conditionally respond to messages.