113 lines
2.6 KiB
Ruby
113 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
# For easier debugging set `PUMA_DEBUG=1`
|
|
|
|
describe Gitlab::Cluster::Mixins::PumaCluster do
|
|
PUMA_STARTUP_TIMEOUT = 30
|
|
|
|
context 'when running Puma in Cluster-mode' do
|
|
using RSpec::Parameterized::TableSyntax
|
|
|
|
where(:signal, :exitstatus, :termsig) do
|
|
# executes phased restart block
|
|
:USR1 | 140 | nil
|
|
:USR2 | 140 | nil
|
|
:INT | 140 | nil
|
|
:HUP | 140 | nil
|
|
|
|
# does not execute phased restart block
|
|
:TERM | nil | 15
|
|
end
|
|
|
|
with_them do
|
|
it 'properly handles process lifecycle' do
|
|
with_puma(workers: 1) do |pid|
|
|
Process.kill(signal, pid)
|
|
|
|
child_pid, child_status = Process.wait2(pid)
|
|
expect(child_pid).to eq(pid)
|
|
expect(child_status.exitstatus).to eq(exitstatus)
|
|
expect(child_status.termsig).to eq(termsig)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def with_puma(workers:, timeout: PUMA_STARTUP_TIMEOUT)
|
|
with_puma_config(workers: workers) do |puma_rb|
|
|
cmdline = [
|
|
"bundle", "exec", "puma",
|
|
"-C", puma_rb,
|
|
"-I", Rails.root.to_s
|
|
]
|
|
|
|
IO.popen(cmdline) do |process|
|
|
# wait for process to start:
|
|
# [2123] * Listening on tcp://127.0.0.1:0
|
|
wait_for_output(process, /Listening on/, timeout: timeout)
|
|
consume_output(process)
|
|
|
|
yield(process.pid)
|
|
ensure
|
|
begin
|
|
Process.kill(:KILL, process.pid)
|
|
rescue Errno::ESRCH
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def with_puma_config(workers:)
|
|
Dir.mktmpdir do |dir|
|
|
File.write "#{dir}/puma.rb", <<-EOF
|
|
require './lib/gitlab/cluster/lifecycle_events'
|
|
require './lib/gitlab/cluster/mixins/puma_cluster'
|
|
|
|
workers #{workers}
|
|
bind "tcp://127.0.0.1:0"
|
|
preload_app!
|
|
|
|
app -> (env) { [404, {}, ['']] }
|
|
|
|
Puma::Cluster.prepend(#{described_class})
|
|
|
|
mutex = Mutex.new
|
|
|
|
Gitlab::Cluster::LifecycleEvents.on_before_blackout_period do
|
|
mutex.synchronize do
|
|
exit(140)
|
|
end
|
|
end
|
|
|
|
# redirect stderr to stdout
|
|
$stderr.reopen($stdout)
|
|
EOF
|
|
|
|
yield("#{dir}/puma.rb")
|
|
end
|
|
end
|
|
|
|
def wait_for_output(process, output, timeout:)
|
|
Timeout.timeout(timeout) do
|
|
loop do
|
|
line = process.readline
|
|
puts "PUMA_DEBUG: #{line}" if ENV['PUMA_DEBUG']
|
|
break if line =~ output
|
|
end
|
|
end
|
|
end
|
|
|
|
def consume_output(process)
|
|
Thread.new do
|
|
loop do
|
|
line = process.readline
|
|
puts "PUMA_DEBUG: #{line}" if ENV['PUMA_DEBUG']
|
|
end
|
|
rescue
|
|
end
|
|
end
|
|
end
|