-
-
Notifications
You must be signed in to change notification settings - Fork 763
/
runner.rb
91 lines (78 loc) · 2.81 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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
require 'drb/drb'
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?
@installed_at_exit = true
# The order here is important, the at_exit handler must be installed before anyone else gets a chance
# to install their own, that way we can be assured that our exit will be last to run (at_exit stacks).
# see https://github.com/jnicklas/capybara/issues/178
exit_code = nil
at_exit { exit exit_code if exit_code && exit_code != 0 }
exit_code = run(ARGV, $stderr, $stdout).to_i
end
AT_EXIT_HOOK_BACKTRACE_LINE = "#{__FILE__}:#{__LINE__ - 2}:in `autorun'"
def self.disable_autorun!
@autorun_disabled = true
end
def self.autorun_disabled? # :nodoc:
@autorun_disabled ||= false
end
def self.installed_at_exit? # :nodoc:
@installed_at_exit ||= false
end
def self.running_in_drb? # :nodoc:
(DRb.current_server rescue false) &&
DRb.current_server.uri =~ /druby\:\/\/127.0.0.1\:/
end
def self.trap_interrupt # :nodoc:
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]
begin
run_over_drb(options, err, out)
rescue DRb::DRbConnError
err.puts "No DRb server is running. Running in local process instead ..."
run_in_process(options, err, out)
end
else
run_in_process(options, err, out)
end
ensure
RSpec.reset
end
def self.run_over_drb(options, err, out) # :nodoc:
DRbCommandLine.new(options).run(err, out)
end
def self.run_in_process(options, err, out) # :nodoc:
CommandLine.new(options, RSpec::configuration, RSpec::world).run(err, out)
end
end
end
end