/
runner.rb
76 lines (68 loc) · 2.23 KB
/
runner.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
module RSpec
module Core
class Runner
# Register an at_exit hook that runs the suite.
def self.autorun
return if autorun_disabled? || installed_at_exit? || running_in_drb?
at_exit { exit run(ARGV, $stderr, $stdout).to_i unless $! }
@installed_at_exit = true
end
AT_EXIT_HOOK_BACKTRACE_LINE = "#{__FILE__}:#{__LINE__ - 2}:in `autorun'"
def self.disable_autorun!
@autorun_disabled = true
end
def self.autorun_disabled?
@autorun_disabled ||= false
end
def self.installed_at_exit?
@installed_at_exit ||= false
end
def self.running_in_drb?
defined?(DRb) &&
(DRb.current_server rescue false) &&
DRb.current_server.uri =~ /druby\:\/\/127.0.0.1\:/
end
def self.trap_interrupt
trap('INT') do
exit!(1) if RSpec.wants_to_quit
RSpec.wants_to_quit = true
STDERR.puts "\nExiting... Interrupt again to exit immediately."
end
end
# Run a suite of RSpec examples.
#
# This is used internally by RSpec to run a suite, but is available
# for use by any other automation tool.
#
# If you want to run this multiple times in the same process, and you
# want files like spec_helper.rb to be reloaded, be sure to load `load`
# instead of `require`.
#
# #### Parameters
# * +args+ - an array of command-line-supported arguments
# * +err+ - error stream (Default: $stderr)
# * +out+ - output stream (Default: $stdout)
#
# #### Returns
# * +Fixnum+ - exit status code (0/1)
def self.run(args, err=$stderr, out=$stdout)
trap_interrupt
options = ConfigurationOptions.new(args)
options.parse_options
if options.options[:drb]
require 'rspec/core/drb_command_line'
begin
DRbCommandLine.new(options).run(err, out)
rescue DRb::DRbConnError
err.puts "No DRb server is running. Running in local process instead ..."
CommandLine.new(options).run(err, out)
end
else
CommandLine.new(options).run(err, out)
end
ensure
RSpec.reset
end
end
end
end