discourse/lib/autospec/spork_runner.rb

116 lines
2.4 KiB
Ruby
Raw Normal View History

2013-11-01 18:57:50 -04:00
require "drb/drb"
require "autospec/rspec_runner"
2013-10-23 19:06:05 -04:00
module Autospec
2013-11-01 18:57:50 -04:00
class SporkRunner < RspecRunner
2013-10-23 19:06:05 -04:00
def start
if already_running?(pid_file)
puts "autospec appears to be running, it is possible the pid file is old"
puts "if you are sure it is not running, delete #{pid_file}"
return
end
write_pid_file(pid_file, Process.pid)
start_spork
@spork_running = true
end
def running?
2013-11-01 18:57:50 -04:00
# launch a thread that will wait for spork to die
2013-10-23 19:06:05 -04:00
@monitor_thread ||=
Thread.new do
Process.wait(@spork_pid)
@spork_running = false
end
2013-11-01 18:57:50 -04:00
2013-10-23 19:06:05 -04:00
@spork_running
end
2013-11-01 18:57:50 -04:00
def run(specs)
args = ["-r", "#{File.dirname(__FILE__)}/formatter.rb",
"-f", "Autospec::Formatter", specs.split].flatten
spork_service.run(args,$stderr,$stdout)
2013-10-23 19:06:05 -04:00
end
2013-11-01 18:57:50 -04:00
def reload
stop_spork
sleep 1
start_spork
2013-10-23 19:06:05 -04:00
end
def abort
spork_service.abort
end
2013-11-01 18:57:50 -04:00
def stop
2013-10-23 19:06:05 -04:00
stop_spork
end
private
2013-11-01 18:57:50 -04:00
2013-10-23 19:06:05 -04:00
def spork_pid_file
Rails.root + "tmp/pids/spork.pid"
end
def pid_file
Rails.root + "tmp/pids/autospec.pid"
end
def already_running?(pid_file)
if File.exists? pid_file
pid = File.read(pid_file).to_i
Process.getpgid(pid) rescue nil
end
end
2013-11-01 18:57:50 -04:00
def write_pid_file(file, pid)
2013-10-23 19:06:05 -04:00
FileUtils.mkdir_p(Rails.root + "tmp/pids")
File.open(file,'w') do |f|
f.write(pid)
end
end
def spork_running?
spork_service.port rescue nil
end
def spork_service
unless @drb_listener_running
begin
DRb.start_service("druby://127.0.0.1:0")
rescue SocketError, Errno::EADDRNOTAVAIL
DRb.start_service("druby://:0")
end
@drb_listener_running = true
end
@spork_service ||= DRbObject.new_with_uri("druby://127.0.0.1:8989")
end
def start_spork
if already_running?(spork_pid_file)
puts "Killing old orphan spork instance"
stop_spork
sleep 1
end
@spork_pid = Process.spawn({'RAILS_ENV' => 'test'}, "bundle exec spork")
write_pid_file(spork_pid_file, @spork_pid)
running = false
while !running
running = spork_running?
sleep 0.01
end
2013-11-01 18:57:50 -04:00
end
2013-10-23 19:06:05 -04:00
2013-11-01 18:57:50 -04:00
def stop_spork
pid = File.read(spork_pid_file).to_i
Process.kill("SIGTERM", pid) rescue nil
2013-10-23 19:06:05 -04:00
end
2013-11-01 18:57:50 -04:00
2013-10-23 19:06:05 -04:00
end
2013-11-01 18:57:50 -04:00
2013-10-23 19:06:05 -04:00
end