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
|
require 'thread'
pid_file = ARGV.shift
scenario = ARGV.shift
ruby_exe = ARGV.shift
# We must do this first otherwise there will be a race with the process that
# creates this process and the TERM signal below could go to that process
# instead, which will likely abort the specs process.
Process.setsid if scenario
signaled = false
mutex = Mutex.new
Signal.trap(:TERM) do
if mutex.try_lock
unless signaled
signaled = true
STDOUT.puts "signaled"
STDOUT.flush
end
end
end
File.open(pid_file, "wb") { |f| f.puts Process.pid }
if scenario
# We are sending a signal to ourselves or the process group
process = "Process.getpgid(Process.pid)"
case scenario
when "self"
signal = %["SIGTERM"]
process = "0"
when "group_numeric"
signal = %[-Signal.list["TERM"]]
when "group_short_string"
signal = %["-TERM"]
when "group_full_string"
signal = %["-SIGTERM"]
end
cmd = %[#{ruby_exe} -e 'Process.kill(#{signal}, #{process})']
Thread.new { system cmd }.join
end
sleep 0.1 until signaled
|