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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
$:.unshift File.dirname(__FILE__) + "/../lib"
$:.unshift File.dirname(__FILE__) + "/mocks"
require 'test/unit'
require 'stringio'
require 'fcgi_handler'
RAILS_ROOT = File.dirname(__FILE__) if !defined?(RAILS_ROOT)
class RailsFCGIHandler
attr_reader :exit_code
attr_reader :reloaded
attr_accessor :thread
attr_reader :gc_runs
def trap(signal, handler, &block)
handler ||= block
(@signal_handlers ||= Hash.new)[signal] = handler
end
def exit(code=0)
@exit_code = code
(thread || Thread.current).exit
end
def send_signal(which)
@signal_handlers[which].call(which)
end
def restore!
@reloaded = true
end
def reload!
@reloaded = true
end
alias_method :old_run_gc!, :run_gc!
def run_gc!
@gc_runs ||= 0
@gc_runs += 1
old_run_gc!
end
end
class RailsFCGIHandlerTest < Test::Unit::TestCase
def setup
@log = StringIO.new
@handler = RailsFCGIHandler.new(@log)
FCGI.time_to_sleep = nil
FCGI.raise_exception = nil
Dispatcher.time_to_sleep = nil
Dispatcher.raise_exception = nil
end
def test_uninterrupted_processing
@handler.process!
assert_nil @handler.exit_code
assert_nil @handler.when_ready
end
def test_interrupted_via_HUP_when_not_in_request
FCGI.time_to_sleep = 1
@handler.thread = Thread.new { @handler.process! }
sleep 0.1 # let the thread get started
@handler.send_signal("HUP")
@handler.thread.join
assert_nil @handler.exit_code
assert_equal :reload, @handler.when_ready
assert @handler.reloaded
end
def test_interrupted_via_HUP_when_in_request
Dispatcher.time_to_sleep = 1
@handler.thread = Thread.new { @handler.process! }
sleep 0.1 # let the thread get started
@handler.send_signal("HUP")
@handler.thread.join
assert_nil @handler.exit_code
assert_equal :reload, @handler.when_ready
assert @handler.reloaded
end
def test_interrupted_via_USR1_when_not_in_request
FCGI.time_to_sleep = 1
@handler.thread = Thread.new { @handler.process! }
sleep 0.1 # let the thread get started
@handler.send_signal("USR1")
@handler.thread.join
assert_nil @handler.exit_code
assert_equal :exit, @handler.when_ready
end
def test_interrupted_via_USR1_when_in_request
Dispatcher.time_to_sleep = 1
@handler.thread = Thread.new { @handler.process! }
sleep 0.1 # let the thread get started
@handler.send_signal("USR1")
@handler.thread.join
assert_nil @handler.exit_code
assert_equal :exit, @handler.when_ready
end
def test_interrupted_via_TERM
Dispatcher.time_to_sleep = 1
@handler.thread = Thread.new { @handler.process! }
sleep 0.1 # let the thread get started
@handler.send_signal("TERM")
@handler.thread.join
assert_equal 0, @handler.exit_code
assert_nil @handler.when_ready
end
%w(RuntimeError SignalException).each do |exception|
define_method("test_#{exception}_in_fcgi") do
FCGI.raise_exception = Object.const_get(exception)
@handler.process!
assert_match %r{Dispatcher failed to catch}, @log.string
case exception
when "RuntimeError"
assert_match %r{almost killed}, @log.string
when "SignalException"
assert_match %r{^killed}, @log.string
end
end
define_method("test_#{exception}_in_dispatcher") do
Dispatcher.raise_exception = Object.const_get(exception)
@handler.process!
assert_match %r{Dispatcher failed to catch}, @log.string
case exception
when "RuntimeError"
assert_no_match %r{killed}, @log.string
when "SignalException"
assert_match %r{^killed}, @log.string
end
end
end
end
class RailsFCGIHandlerPeriodicGCTest < Test::Unit::TestCase
def setup
@log = StringIO.new
FCGI.time_to_sleep = nil
FCGI.raise_exception = nil
FCGI.each_cgi_count = nil
Dispatcher.time_to_sleep = nil
Dispatcher.raise_exception = nil
Dispatcher.dispatch_hook = nil
end
def teardown
FCGI.each_cgi_count = nil
Dispatcher.dispatch_hook = nil
GC.enable
end
def test_normal_gc
@handler = RailsFCGIHandler.new(@log)
assert_nil @handler.gc_request_period
# When GC is enabled, GC.disable disables and returns false.
assert_equal false, GC.disable
end
def test_periodic_gc
Dispatcher.dispatch_hook = lambda do |cgi|
# When GC is disabled, GC.enable enables and returns true.
assert_equal true, GC.enable
GC.disable
end
@handler = RailsFCGIHandler.new(@log, 10)
assert_equal 10, @handler.gc_request_period
FCGI.each_cgi_count = 1
@handler.process!
assert_equal 1, @handler.gc_runs
FCGI.each_cgi_count = 10
@handler.process!
assert_equal 3, @handler.gc_runs
FCGI.each_cgi_count = 25
@handler.process!
assert_equal 6, @handler.gc_runs
assert_nil @handler.exit_code
assert_nil @handler.when_ready
end
end
|
