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
92
93
94
95
96
97
98
99
100
101
102
103
104
|
# frozen_string_literal: true
require "test_helper"
class PumaWorkerKillerTest < Test::Unit::TestCase
def test_starts
port = 0 # http://stackoverflow.com/questions/200484/how-do-you-find-a-free-tcp-server-port-using-ruby
command = "#{RbConfig.ruby} /usr/bin/puma #{fixture_path.join("default.ru")} -t 1:1 -w 2 --preload --debug -p #{port}"
options = {wait_for: "booted", timeout: 5, env: {"PUMA_FREQUENCY" => 1}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "PumaWorkerKiller")
end
end
def test_without_preload
port = 0 # http://stackoverflow.com/questions/200484/how-do-you-find-a-free-tcp-server-port-using-ruby
command = "#{RbConfig.ruby} /usr/bin/puma #{fixture_path.join("default.ru")} -t 1:1 -w 2 --debug -p #{port} -C #{fixture_path.join("config/puma_worker_killer_start.rb")}"
options = {wait_for: "booted", timeout: 10, env: {"PUMA_FREQUENCY" => 1}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "PumaWorkerKiller")
end
end
def test_kills_large_app
file = fixture_path.join("big.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 2 --preload --debug -p #{port}"
options = {wait_for: "booted", timeout: 5, env: {"PUMA_FREQUENCY" => 1, "PUMA_RAM" => 1}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "Out of memory")
end
end
def test_pre_term
file = fixture_path.join("pre_term.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 2 --preload --debug -p #{port}"
options = {wait_for: "booted", timeout: 5, env: {"PUMA_FREQUENCY" => 1, "PUMA_RAM" => 1}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "Out of memory")
assert_contains(spawn, "About to terminate worker:") # defined in pre_term.ru
end
end
def test_on_calculation
file = fixture_path.join("on_calculation.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 2 --preload --debug -p #{port}"
options = {wait_for: "booted", timeout: 5, env: {"PUMA_FREQUENCY" => 1, "PUMA_RAM" => 1}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "Out of memory")
assert_contains(spawn, "Current memory footprint:") # defined in on_calculate.ru
end
end
def assert_contains(spawn, string)
assert spawn.wait(string), "Expected logs to contain '#{string}' but it did not, contents: #{spawn.log.read}"
end
def test_rolling_restart
file = fixture_path.join("rolling_restart.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 2 --preload --debug -p #{port}"
puts command.inspect
options = {wait_for: "booted", timeout: 15, env: {}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "Rolling Restart")
end
end
def test_rolling_restart_worker_kill_check
file = fixture_path.join("rolling_restart.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 1 --preload --debug -p #{port}"
puts command.inspect
options = {wait_for: "booted", timeout: 120, env: {}}
WaitForIt.new(command, options) do |spawn|
# at least 2 matches for TERM (so we set a timeout value longer - 120sec)
spawn.wait!(/TERM.*TERM/m)
term_ids = spawn.log.read.scan(/TERM to pid (\d*)/)
assert term_ids.sort == term_ids.uniq.sort
end
end
def test_rolling_pre_term
file = fixture_path.join("rolling_pre_term.ru")
port = 0
command = "#{RbConfig.ruby} /usr/bin/puma #{file} -t 1:1 -w 2 --preload --debug -p #{port}"
puts command.inspect
options = {wait_for: "booted", timeout: 15, env: {}}
WaitForIt.new(command, options) do |spawn|
assert_contains(spawn, "Rolling Restart")
assert_contains(spawn, "About to terminate (rolling) worker:") # defined in rolling_pre_term.ru
end
end
end
|
