diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_auto_chunk.rb | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/test/test_auto_chunk.rb b/test/test_auto_chunk.rb index a97fe26..e1847f6 100644 --- a/test/test_auto_chunk.rb +++ b/test/test_auto_chunk.rb @@ -18,7 +18,12 @@ class TestAutoChunk < Testcase app = Rack::Builder.new do use Rack::ContentType, "text/plain" run(lambda do |env| - [ 200, {}, %w(a b c) ] + case env['PATH_INFO'] + when '/204' + [ 204, {}, [] ] + else + [ 200, {}, %w(a b c) ] + end end) end app(:rack, app) { listen "#{host}:#{port}" } @@ -50,6 +55,21 @@ class TestAutoChunk < Testcase assert_equal 200, res.code.to_i assert_equal 'abc', res.body end + + s = TCPSocket.new(host, port) + s.write("GET /204 HTTP/1.1\r\nHost: example.com\r\n\r\n") + buf = s.readpartial(1024) + assert_match %r{\r\n\r\n\z}, buf + refute_match %r{^Transfer-Encoding}i, buf + assert_match %r{^Connection: keep-alive\r\n}, buf + assert_nil IO.select([s], nil, nil, 1), 'connection persists..' + + # maek sure another on the same connection works + s.write("GET / HTTP/1.1\r\nHost: example.com\r\n\r\n") + buf = s.readpartial(1024) + assert_match %r{\AHTTP/1\.1 200}, buf + assert_match(%r{^Transfer-Encoding: chunked\r\n}, buf) + s.close ensure quit_wait(pid) end |