47 lines
1.1 KiB
Ruby
47 lines
1.1 KiB
Ruby
|
RSpec::Matchers.define :abort_execution do
|
||
|
|
match do |code_block|
|
||
|
|
@captured_stderr = StringIO.new
|
||
|
|
original_stderr = $stderr
|
||
|
|
$stderr = @captured_stderr
|
||
|
|
|
||
|
|
code_block.call
|
||
|
|
|
||
|
|
false
|
||
|
|
rescue SystemExit => e
|
||
|
|
captured = @captured_stderr.string.chomp
|
||
|
|
@actual_exit_code = e.status
|
||
|
|
break false unless e.status == 1
|
||
|
|
|
||
|
|
if @message
|
||
|
|
if @message.is_a? String
|
||
|
|
@message == captured
|
||
|
|
elsif @message.is_a? Regexp
|
||
|
|
@message.match?(captured)
|
||
|
|
else
|
||
|
|
raise ArgumentError, 'with_message must be either a String or a Regular Expression'
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
ensure
|
||
|
|
$stderr = original_stderr
|
||
|
|
end
|
||
|
|
|
||
|
|
chain :with_message do |message|
|
||
|
|
@message = message
|
||
|
|
end
|
||
|
|
|
||
|
|
failure_message do |block|
|
||
|
|
unless @actual_exit_code
|
||
|
|
break "expected #{block} to abort with '#{@message}' but didnt call abort."
|
||
|
|
end
|
||
|
|
|
||
|
|
if @actual_exit_code != 1
|
||
|
|
break "expected #{block} to abort with: '#{@message}' but exited with success instead."
|
||
|
|
end
|
||
|
|
|
||
|
|
"expected #{block} to abort with: '#{@message}' \n but received: '#{@captured_stderr.string.chomp}' instead."
|
||
|
|
end
|
||
|
|
|
||
|
|
supports_block_expectations
|
||
|
|
end
|