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
| | # Copyright (c) 2005 Zed A. Shaw
# You can redistribute it and/or modify it under the same terms as Ruby.
#
# Additional work donated by contributors. See http://mongrel.rubyforge.org/attributions.html
# for more information.
require 'test/test_helper'
include Unicorn
class TestHandler
def call(env)
# response.socket.write("HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\nhello!\n")
[200, { 'Content-Type' => 'text/plain' }, ['hello!\n']]
end
end
class WebServerTest < Test::Unit::TestCase
def setup
@valid_request = "GET / HTTP/1.1\r\nHost: www.zedshaw.com\r\nContent-Type: text/plain\r\n\r\n"
@port = unused_port
@tester = TestHandler.new
redirect_test_io do
@server = HttpServer.new(@tester, :listeners => [ "127.0.0.1:#{@port}" ] )
@server.start
end
end
def teardown
redirect_test_io do
@server.stop(true)
end
end
def test_broken_app
teardown
port = unused_port
app = lambda { |env| raise RuntimeError, "hello" }
# [200, {}, []] }
redirect_test_io do
@server = HttpServer.new(app, :listeners => [ "127.0.0.1:#{port}"] )
@server.start
end
sock = nil
assert_nothing_raised do
sock = TCPSocket.new('127.0.0.1', port)
sock.syswrite("GET / HTTP/1.0\r\n\r\n")
end
assert_match %r{\AHTTP/1.[01] 500\b}, sock.sysread(4096)
assert_nothing_raised { sock.close }
end
def test_simple_server
results = hit(["http://localhost:#{@port}/test"])
assert_equal 'hello!\n', results[0], "Handler didn't really run"
end
def do_test(string, chunk, close_after=nil, shutdown_delay=0)
# Do not use instance variables here, because it needs to be thread safe
socket = TCPSocket.new("127.0.0.1", @port);
request = StringIO.new(string)
chunks_out = 0
while data = request.read(chunk)
chunks_out += socket.write(data)
socket.flush
sleep 0.2
if close_after and chunks_out > close_after
socket.close
sleep 1
end
end
sleep(shutdown_delay)
socket.write(" ") # Some platforms only raise the exception on attempted write
socket.flush
end
def test_trickle_attack
do_test(@valid_request, 3)
end
def test_close_client
assert_raises IOError do
do_test(@valid_request, 10, 20)
end
end
def test_bad_client
redirect_test_io do
do_test("GET /test HTTP/BAD", 3)
end
end
def test_bad_client_400
sock = nil
assert_nothing_raised do
sock = TCPSocket.new('127.0.0.1', @port)
sock.syswrite("GET / HTTP/1.0\r\nHost: foo\rbar\r\n\r\n")
end
assert_match %r{\AHTTP/1.[01] 400\b}, sock.sysread(4096)
assert_nothing_raised { sock.close }
end
def test_header_is_too_long
redirect_test_io do
long = "GET /test HTTP/1.1\r\n" + ("X-Big: stuff\r\n" * 15000) + "\r\n"
assert_raises Errno::ECONNRESET, Errno::EPIPE, Errno::ECONNABORTED, Errno::EINVAL, IOError do
do_test(long, long.length/2, 10)
end
end
end
def test_file_streamed_request
body = "a" * (Unicorn::Const::MAX_BODY * 2)
long = "GET /test HTTP/1.1\r\nContent-length: #{body.length}\r\n\r\n" + body
do_test(long, Unicorn::Const::CHUNK_SIZE * 2 -400)
end
end
|