2016-11-22 10:05:49 -05:00
|
|
|
require "test_helper"
|
2011-12-07 16:43:16 -05:00
|
|
|
|
2016-11-22 10:05:49 -05:00
|
|
|
require "puma/cli"
|
|
|
|
require "puma/control_cli"
|
2011-12-07 17:48:41 -05:00
|
|
|
|
2012-11-29 15:05:29 -05:00
|
|
|
# These don't run on travis because they're too fragile
|
|
|
|
|
2016-11-22 10:05:49 -05:00
|
|
|
class TestIntegration < Minitest::Test
|
2011-12-07 16:43:16 -05:00
|
|
|
def setup
|
|
|
|
@state_path = "test/test_puma.state"
|
|
|
|
@bind_path = "test/test_server.sock"
|
|
|
|
@control_path = "test/test_control.sock"
|
2016-02-06 22:00:29 -05:00
|
|
|
@token = "xxyyzz"
|
2012-09-02 23:33:09 -04:00
|
|
|
@tcp_port = 9998
|
|
|
|
|
|
|
|
@server = nil
|
2012-09-06 01:09:42 -04:00
|
|
|
@script = nil
|
2013-07-05 19:54:15 -04:00
|
|
|
|
|
|
|
@wait, @ready = IO.pipe
|
|
|
|
|
|
|
|
@events = Puma::Events.strings
|
|
|
|
@events.on_booted { @ready << "!" }
|
2011-12-07 16:43:16 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def teardown
|
|
|
|
File.unlink @state_path rescue nil
|
|
|
|
File.unlink @bind_path rescue nil
|
|
|
|
File.unlink @control_path rescue nil
|
2012-09-02 23:33:09 -04:00
|
|
|
|
2013-07-05 19:54:15 -04:00
|
|
|
@wait.close
|
|
|
|
@ready.close
|
|
|
|
|
2012-09-02 23:33:09 -04:00
|
|
|
if @server
|
|
|
|
Process.kill "INT", @server.pid
|
2013-06-19 19:16:15 -04:00
|
|
|
begin
|
|
|
|
Process.wait @server.pid
|
|
|
|
rescue Errno::ECHILD
|
|
|
|
end
|
|
|
|
|
2012-09-02 23:33:09 -04:00
|
|
|
@server.close
|
|
|
|
end
|
2012-09-06 01:09:42 -04:00
|
|
|
|
|
|
|
if @script
|
|
|
|
@script.close!
|
|
|
|
end
|
2012-09-02 23:33:09 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def server(opts)
|
|
|
|
core = "#{Gem.ruby} -rubygems -Ilib bin/puma"
|
|
|
|
cmd = "#{core} --restart-cmd '#{core}' -b tcp://127.0.0.1:#{@tcp_port} #{opts}"
|
2012-09-06 01:09:42 -04:00
|
|
|
tf = Tempfile.new "puma-test"
|
|
|
|
tf.puts "exec #{cmd}"
|
|
|
|
tf.close
|
|
|
|
|
|
|
|
@script = tf
|
|
|
|
|
|
|
|
@server = IO.popen("sh #{tf.path}", "r")
|
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
wait_for_server_to_boot
|
2012-09-06 01:09:42 -04:00
|
|
|
|
2012-09-02 23:33:09 -04:00
|
|
|
@server
|
|
|
|
end
|
|
|
|
|
|
|
|
def signal(which)
|
|
|
|
Process.kill which, @server.pid
|
2011-12-07 16:43:16 -05:00
|
|
|
end
|
|
|
|
|
2013-07-05 19:54:15 -04:00
|
|
|
def wait_booted
|
|
|
|
@wait.sysread 1
|
|
|
|
end
|
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
# reuses an existing connection to make sure that works
|
|
|
|
def restart_server(connection)
|
|
|
|
signal :USR2
|
|
|
|
|
|
|
|
sleep 1
|
|
|
|
|
|
|
|
connection.write "GET / HTTP/1.1\r\n\r\n" # trigger it to start by sending a new request
|
|
|
|
|
|
|
|
assert_raises Errno::ECONNRESET do
|
|
|
|
Timeout.timeout(2) do
|
|
|
|
raise Errno::ECONNRESET unless connection.read(2)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
wait_for_server_to_boot
|
|
|
|
end
|
|
|
|
|
|
|
|
def connect
|
|
|
|
s = TCPSocket.new "localhost", @tcp_port
|
|
|
|
s << "GET / HTTP/1.1\r\n\r\n"
|
|
|
|
true until s.gets == "\r\n"
|
|
|
|
s
|
|
|
|
end
|
|
|
|
|
|
|
|
def wait_for_server_to_boot
|
|
|
|
true while @server.gets =~ /Ctrl-C/
|
|
|
|
sleep(Puma.jruby? ? 5 : 1) # TODO: not sure why that is needed ... something deterministic would be better
|
|
|
|
end
|
|
|
|
|
2011-12-07 16:43:16 -05:00
|
|
|
def test_stop_via_pumactl
|
2016-03-06 00:44:53 -05:00
|
|
|
if Puma.jruby? || Puma.windows?
|
2011-12-07 17:58:30 -05:00
|
|
|
assert true
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2016-02-06 22:00:29 -05:00
|
|
|
conf = Puma::Configuration.new do |c|
|
|
|
|
c.quiet
|
|
|
|
c.state_path @state_path
|
|
|
|
c.bind "unix://#{@bind_path}"
|
|
|
|
c.activate_control_app "unix://#{@control_path}", :auth_token => @token
|
|
|
|
c.rackup "test/hello.ru"
|
|
|
|
end
|
|
|
|
|
|
|
|
l = Puma::Launcher.new conf, :events => @events
|
2011-12-07 16:43:16 -05:00
|
|
|
|
2011-12-07 17:48:41 -05:00
|
|
|
t = Thread.new do
|
2016-02-06 22:00:29 -05:00
|
|
|
Thread.current.abort_on_exception = true
|
|
|
|
l.run
|
2011-12-07 16:43:16 -05:00
|
|
|
end
|
|
|
|
|
2013-07-05 19:54:15 -04:00
|
|
|
wait_booted
|
2011-12-07 16:53:36 -05:00
|
|
|
|
|
|
|
s = UNIXSocket.new @bind_path
|
|
|
|
s << "GET / HTTP/1.0\r\n\r\n"
|
|
|
|
assert_equal "Hello World", s.read.split("\r\n").last
|
|
|
|
|
2013-07-05 19:54:15 -04:00
|
|
|
sout = StringIO.new
|
|
|
|
|
2011-12-07 17:48:41 -05:00
|
|
|
ccli = Puma::ControlCLI.new %W!-S #{@state_path} stop!, sout
|
2011-12-07 16:43:16 -05:00
|
|
|
|
2011-12-07 17:48:41 -05:00
|
|
|
ccli.run
|
2011-12-07 16:43:16 -05:00
|
|
|
|
2011-12-07 17:48:41 -05:00
|
|
|
assert_kind_of Thread, t.join(1), "server didn't stop"
|
2011-12-07 16:43:16 -05:00
|
|
|
end
|
2012-09-02 23:33:09 -04:00
|
|
|
|
2016-02-06 22:00:29 -05:00
|
|
|
def test_phased_restart_via_pumactl
|
2016-11-22 10:05:49 -05:00
|
|
|
skip("Too finicky, fails 50% of the time on CI.")
|
|
|
|
|
2016-03-06 00:44:53 -05:00
|
|
|
if Puma.jruby? || Puma.windows?
|
2015-01-15 07:58:16 -05:00
|
|
|
assert true
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2016-02-06 22:00:29 -05:00
|
|
|
conf = Puma::Configuration.new do |c|
|
|
|
|
c.quiet
|
|
|
|
c.state_path @state_path
|
|
|
|
c.bind "unix://#{@bind_path}"
|
|
|
|
c.activate_control_app "unix://#{@control_path}", :auth_token => @token
|
|
|
|
c.workers 2
|
|
|
|
c.worker_shutdown_timeout 1
|
|
|
|
c.rackup "test/hello-stuck.ru"
|
|
|
|
end
|
|
|
|
|
|
|
|
l = Puma::Launcher.new conf, :events => @events
|
|
|
|
|
|
|
|
Thread.abort_on_exception = true
|
2015-01-15 07:58:16 -05:00
|
|
|
|
|
|
|
t = Thread.new do
|
2016-02-06 22:00:29 -05:00
|
|
|
Thread.current.abort_on_exception = true
|
|
|
|
l.run
|
2015-01-15 07:58:16 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
wait_booted
|
|
|
|
|
|
|
|
# Make both workers stuck
|
|
|
|
s1 = UNIXSocket.new @bind_path
|
|
|
|
s1 << "GET / HTTP/1.0\r\n\r\n"
|
|
|
|
|
|
|
|
sout = StringIO.new
|
|
|
|
|
|
|
|
# Phased restart
|
|
|
|
ccli = Puma::ControlCLI.new %W!-S #{@state_path} phased-restart!, sout
|
|
|
|
ccli.run
|
|
|
|
sleep 20
|
|
|
|
@events.stdout.rewind
|
|
|
|
log = @events.stdout.readlines.join("")
|
|
|
|
assert_match(/TERM sent/, log)
|
2016-12-12 14:56:15 -05:00
|
|
|
assert_match(/- Worker \d \(pid: \d+\) booted, phase: 1/, log)
|
2015-01-15 07:58:16 -05:00
|
|
|
|
|
|
|
# Stop
|
|
|
|
ccli = Puma::ControlCLI.new %W!-S #{@state_path} stop!, sout
|
|
|
|
ccli.run
|
|
|
|
|
|
|
|
assert_kind_of Thread, t.join(5), "server didn't stop"
|
|
|
|
end
|
|
|
|
|
2016-12-12 14:56:15 -05:00
|
|
|
def test_kill_unknown_via_pumactl
|
|
|
|
if Puma.jruby? || Puma.windows?
|
|
|
|
assert true
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
|
|
|
# we run ls to get a 'safe' pid to pass off as puma in cli stop
|
|
|
|
# do not want to accidently kill a valid other process
|
|
|
|
io = IO.popen("ls")
|
|
|
|
safe_pid = io.pid
|
|
|
|
Process.wait safe_pid
|
|
|
|
|
|
|
|
sout = StringIO.new
|
|
|
|
|
|
|
|
e = assert_raises SystemExit do
|
|
|
|
ccli = Puma::ControlCLI.new %W!-p #{safe_pid} stop!, sout
|
|
|
|
ccli.run
|
|
|
|
end
|
|
|
|
sout.rewind
|
|
|
|
assert_match(/No pid '\d+' found/, sout.readlines.join(""))
|
|
|
|
assert_equal(1, e.status)
|
|
|
|
end
|
|
|
|
|
2016-02-06 22:00:29 -05:00
|
|
|
def test_restart_closes_keepalive_sockets
|
2012-09-02 23:33:09 -04:00
|
|
|
server("-q test/hello.ru")
|
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
s = connect
|
|
|
|
s.read
|
|
|
|
restart_server(s)
|
2016-03-06 00:44:53 -05:00
|
|
|
|
|
|
|
s = TCPSocket.new "127.0.0.1", @tcp_port
|
2012-09-02 23:33:09 -04:00
|
|
|
s << "GET / HTTP/1.0\r\n\r\n"
|
|
|
|
assert_equal "Hello World", s.read.split("\r\n").last
|
|
|
|
end
|
2012-09-06 01:09:42 -04:00
|
|
|
|
2016-02-06 22:00:29 -05:00
|
|
|
def test_restart_closes_keepalive_sockets_workers
|
2016-03-06 00:44:53 -05:00
|
|
|
if Puma.jruby?
|
|
|
|
assert true
|
|
|
|
return
|
|
|
|
end
|
|
|
|
|
2012-09-10 11:35:50 -04:00
|
|
|
server("-q -w 2 test/hello.ru")
|
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
s = connect
|
|
|
|
s.read
|
|
|
|
restart_server s
|
2012-09-10 11:35:50 -04:00
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
s = TCPSocket.new "localhost", @tcp_port
|
|
|
|
s << "GET / HTTP/1.0\r\n\r\n"
|
|
|
|
assert_equal "Hello World", s.read.split("\r\n").last
|
|
|
|
end
|
2012-09-10 11:35:50 -04:00
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
# It does not share environments between multiple generations, which would break Dotenv
|
|
|
|
def test_restart_restores_environment
|
|
|
|
server("-q test/hello-env.ru")
|
2012-09-10 11:35:50 -04:00
|
|
|
|
2017-04-01 14:12:54 -04:00
|
|
|
s = connect
|
|
|
|
initial_reply = s.read
|
|
|
|
restart_server(s)
|
2012-09-10 11:35:50 -04:00
|
|
|
|
|
|
|
s = TCPSocket.new "localhost", @tcp_port
|
|
|
|
s << "GET / HTTP/1.0\r\n\r\n"
|
2017-04-01 14:12:54 -04:00
|
|
|
new_reply = s.read.split("\r\n").last
|
|
|
|
|
|
|
|
assert_includes initial_reply, "Hello RAND"
|
|
|
|
assert_includes new_reply, "Hello RAND"
|
|
|
|
refute_equal initial_reply, new_reply
|
2012-09-10 11:35:50 -04:00
|
|
|
end
|
2016-03-06 00:44:53 -05:00
|
|
|
end
|