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
| | #!ruby
# Copyright (C) unicorn hackers <unicorn-public@80x24.org>
# License: GPL-3.0+ <https://www.gnu.org/licenses/gpl-3.0.txt>
# this goes for t/integration.t We'll try to put as many tests
# in here as possible to avoid startup overhead of Ruby.
$orig_rack_200 = nil
def tweak_status_code
$orig_rack_200 = Rack::Utils::HTTP_STATUS_CODES[200]
Rack::Utils::HTTP_STATUS_CODES[200] = "HI"
[ 200, {}, [] ]
end
def restore_status_code
$orig_rack_200 or return [ 500, {}, [] ]
Rack::Utils::HTTP_STATUS_CODES[200] = $orig_rack_200
[ 200, {}, [] ]
end
class WriteOnClose
def each(&block)
@callback = block
end
def close
@callback.call "7\r\nGoodbye\r\n0\r\n\r\n"
end
end
def write_on_close
[ 200, { 'transfer-encoding' => 'chunked' }, WriteOnClose.new ]
end
def env_dump(env)
require 'json'
h = {}
env.each do |k,v|
case v
when String, Integer, true, false; h[k] = v
else
case k
when 'rack.version', 'rack.after_reply'; h[k] = v
end
end
end
h.to_json
end
run(lambda do |env|
case env['REQUEST_METHOD']
when 'GET'
case env['PATH_INFO']
when '/rack-2-newline-headers'; [ 200, { 'X-R2' => "a\nb\nc" }, [] ]
when '/rack-3-array-headers'; [ 200, { 'x-r3' => %w(a b c) }, [] ]
when '/nil-header-value'; [ 200, { 'X-Nil' => nil }, [] ]
when '/unknown-status-pass-through'; [ '666 I AM THE BEAST', {}, [] ]
when '/env_dump'; [ 200, {}, [ env_dump(env) ] ]
when '/write_on_close'; write_on_close
when '/pid'; [ 200, {}, [ "#$$\n" ] ]
end # case PATH_INFO (GET)
when 'POST'
case env['PATH_INFO']
when '/tweak-status-code'; tweak_status_code
when '/restore-status-code'; restore_status_code
end # case PATH_INFO (POST)
# ...
when 'PUT'
# ...
end # case REQUEST_METHOD
end) # run
|