diff --git a/lib/sus/raise_exception.rb b/lib/sus/raise_exception.rb index ffed00e..183c0c7 100644 --- a/lib/sus/raise_exception.rb +++ b/lib/sus/raise_exception.rb @@ -12,17 +12,12 @@ def call(assertions, subject) # Didn't throw any exception, so the expectation failed: assertions.assert(false, self) - rescue => exception - # Did we throw the right kind of exception? - if exception.is_a?(@exception_class) - # Did it have the right message? - if @message - assertions.assert(@message === exception.message) - else - assertions.assert(true, self) - end + rescue @exception_class => exception + # Did it have the right message? + if @message + assertions.assert(@message === exception.message) else - raise + assertions.assert(true, self) end end end