2012-11-05 19:49:57 -05:00
|
|
|
# coding: US-ASCII
|
2015-12-16 00:07:31 -05:00
|
|
|
# frozen_string_literal: false
|
2009-10-30 09:09:03 -04:00
|
|
|
require 'net/http'
|
|
|
|
require 'test/unit'
|
|
|
|
require 'stringio'
|
|
|
|
|
|
|
|
class HTTPResponseTest < Test::Unit::TestCase
|
|
|
|
def test_singleline_header
|
2012-07-19 18:43:38 -04:00
|
|
|
io = dummy_io(<<EOS)
|
2009-10-30 09:09:03 -04:00
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
2012-08-10 01:25:08 -04:00
|
|
|
assert_equal('5', res['content-length'])
|
|
|
|
assert_equal('close', res['connection'])
|
2009-10-30 09:09:03 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_multiline_header
|
2012-07-19 18:43:38 -04:00
|
|
|
io = dummy_io(<<EOS)
|
2009-10-30 09:09:03 -04:00
|
|
|
HTTP/1.1 200 OK
|
|
|
|
X-Foo: XXX
|
|
|
|
YYY
|
|
|
|
X-Bar:
|
|
|
|
XXX
|
|
|
|
\tYYY
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
2012-08-10 01:25:08 -04:00
|
|
|
assert_equal('XXX YYY', res['x-foo'])
|
|
|
|
assert_equal('XXX YYY', res['x-bar'])
|
2009-10-30 09:09:03 -04:00
|
|
|
end
|
|
|
|
|
2012-07-19 18:43:38 -04:00
|
|
|
def test_read_body
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Length: 5
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal 'hello', body
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_read_body_block
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Length: 5
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
|
|
|
|
body = ''
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
res.read_body do |chunk|
|
|
|
|
body << chunk
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal 'hello', body
|
|
|
|
end
|
|
|
|
|
2017-05-22 03:36:00 -04:00
|
|
|
def test_read_body_block_mod
|
2020-06-22 13:45:08 -04:00
|
|
|
# http://ci.rvm.jp/results/trunk-mjit-wait@silicon-docker/3019353
|
2021-01-22 20:52:17 -05:00
|
|
|
if defined?(RubyVM::JIT) ? RubyVM::JIT.enabled? : defined?(RubyVM::MJIT) && RubyVM::MJIT.enabled?
|
2021-01-14 20:54:42 -05:00
|
|
|
skip 'too unstable with --jit-wait, and extending read_timeout did not help it'
|
|
|
|
end
|
2017-05-22 03:36:00 -04:00
|
|
|
IO.pipe do |r, w|
|
|
|
|
buf = 'x' * 1024
|
|
|
|
buf.freeze
|
|
|
|
n = 1024
|
|
|
|
len = n * buf.size
|
|
|
|
th = Thread.new do
|
|
|
|
w.write("HTTP/1.1 200 OK\r\nContent-Length: #{len}\r\n\r\n")
|
|
|
|
n.times { w.write(buf) }
|
|
|
|
:ok
|
|
|
|
end
|
|
|
|
io = Net::BufferedIO.new(r)
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
nr = 0
|
|
|
|
res.reading_body io, true do
|
|
|
|
# should be allowed to modify the chunk given to them:
|
|
|
|
res.read_body do |chunk|
|
|
|
|
nr += chunk.size
|
|
|
|
chunk.clear
|
|
|
|
end
|
|
|
|
end
|
|
|
|
assert_equal len, nr
|
|
|
|
assert_equal :ok, th.value
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-07-19 18:43:38 -04:00
|
|
|
def test_read_body_content_encoding_deflate
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
2019-12-16 09:39:10 -05:00
|
|
|
Content-Length: 13
|
2012-07-19 18:43:38 -04:00
|
|
|
|
2019-12-16 09:39:10 -05:00
|
|
|
x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15
|
2015-06-29 03:14:31 -04:00
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
res.decode_content = true
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal 'hello', body
|
|
|
|
else
|
|
|
|
assert_equal 'deflate', res['content-encoding']
|
2019-12-16 09:39:10 -05:00
|
|
|
assert_equal "x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15", body
|
2015-06-29 03:14:31 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_read_body_content_encoding_deflate_uppercase
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: DEFLATE
|
2019-12-16 09:39:10 -05:00
|
|
|
Content-Length: 13
|
2015-06-29 03:14:31 -04:00
|
|
|
|
2019-12-16 09:39:10 -05:00
|
|
|
x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15
|
2012-07-19 18:43:38 -04:00
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
2013-02-13 20:08:19 -05:00
|
|
|
res.decode_content = true
|
2012-07-19 18:43:38 -04:00
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
2012-08-10 06:53:18 -04:00
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal 'hello', body
|
|
|
|
else
|
2015-07-01 10:52:05 -04:00
|
|
|
assert_equal 'DEFLATE', res['content-encoding']
|
2019-12-16 09:39:10 -05:00
|
|
|
assert_equal "x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15", body
|
2012-08-10 06:53:18 -04:00
|
|
|
end
|
|
|
|
end
|
2012-07-19 18:43:38 -04:00
|
|
|
|
|
|
|
def test_read_body_content_encoding_deflate_chunked
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
|
|
|
Transfer-Encoding: chunked
|
|
|
|
|
2019-12-16 09:39:10 -05:00
|
|
|
6
|
|
|
|
x\x9C\xCBH\xCD\xC9
|
|
|
|
7
|
|
|
|
\xC9\a\x00\x06,\x02\x15
|
2012-07-19 18:43:38 -04:00
|
|
|
0
|
|
|
|
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
2013-02-13 20:08:19 -05:00
|
|
|
res.decode_content = true
|
2012-07-19 18:43:38 -04:00
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
2012-08-10 06:53:18 -04:00
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal 'hello', body
|
|
|
|
else
|
|
|
|
assert_equal 'deflate', res['content-encoding']
|
2019-12-16 09:39:10 -05:00
|
|
|
assert_equal "x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15", body
|
2012-08-10 06:53:18 -04:00
|
|
|
end
|
|
|
|
end
|
2012-07-19 18:43:38 -04:00
|
|
|
|
2013-02-13 20:08:19 -05:00
|
|
|
def test_read_body_content_encoding_deflate_disabled
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
2019-12-16 09:39:10 -05:00
|
|
|
Content-Length: 13
|
2013-02-13 20:08:19 -05:00
|
|
|
|
2019-12-16 09:39:10 -05:00
|
|
|
x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15
|
2013-02-13 20:08:19 -05:00
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
res.decode_content = false # user set accept-encoding in request
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal 'deflate', res['content-encoding'], 'Bug #7831'
|
2019-12-16 09:39:10 -05:00
|
|
|
assert_equal "x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15", body, 'Bug #7381'
|
2013-02-13 20:08:19 -05:00
|
|
|
end
|
|
|
|
|
2012-07-19 18:43:38 -04:00
|
|
|
def test_read_body_content_encoding_deflate_no_length
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
|
|
|
|
2019-12-16 09:39:10 -05:00
|
|
|
x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15
|
2012-07-19 18:43:38 -04:00
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
2013-02-13 20:08:19 -05:00
|
|
|
res.decode_content = true
|
2012-07-19 18:43:38 -04:00
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
2012-08-10 06:53:18 -04:00
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal 'hello', body
|
|
|
|
else
|
|
|
|
assert_equal 'deflate', res['content-encoding']
|
2019-12-16 09:39:10 -05:00
|
|
|
assert_equal "x\x9C\xCBH\xCD\xC9\xC9\a\x00\x06,\x02\x15\r\n", body
|
2012-08-10 06:53:18 -04:00
|
|
|
end
|
|
|
|
end
|
2012-07-19 18:43:38 -04:00
|
|
|
|
|
|
|
def test_read_body_content_encoding_deflate_content_range
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Accept-Ranges: bytes
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: gzip
|
|
|
|
Content-Length: 10
|
|
|
|
Content-Range: bytes 0-9/55
|
|
|
|
|
|
|
|
\x1F\x8B\b\x00\x00\x00\x00\x00\x00\x03
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal "\x1F\x8B\b\x00\x00\x00\x00\x00\x00\x03", body
|
|
|
|
end
|
|
|
|
|
2015-08-01 04:23:32 -04:00
|
|
|
def test_read_body_content_encoding_deflate_empty_body
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
|
|
|
Content-Length: 0
|
|
|
|
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
res.decode_content = true
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal '', body
|
|
|
|
else
|
|
|
|
assert_equal 'deflate', res['content-encoding']
|
|
|
|
assert_equal '', body
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_read_body_content_encoding_deflate_empty_body_no_length
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Encoding: deflate
|
|
|
|
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
res.decode_content = true
|
|
|
|
|
|
|
|
body = nil
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
body = res.read_body
|
|
|
|
end
|
|
|
|
|
|
|
|
if Net::HTTP::HAVE_ZLIB
|
|
|
|
assert_equal nil, res['content-encoding']
|
|
|
|
assert_equal '', body
|
|
|
|
else
|
|
|
|
assert_equal 'deflate', res['content-encoding']
|
|
|
|
assert_equal '', body
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2012-07-19 18:43:38 -04:00
|
|
|
def test_read_body_string
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Connection: close
|
|
|
|
Content-Length: 5
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
|
|
|
|
body = ''
|
|
|
|
|
|
|
|
res.reading_body io, true do
|
|
|
|
res.read_body body
|
|
|
|
end
|
|
|
|
|
|
|
|
assert_equal 'hello', body
|
|
|
|
end
|
|
|
|
|
2012-12-21 15:36:07 -05:00
|
|
|
def test_uri_equals
|
|
|
|
uri = URI 'http://example'
|
|
|
|
|
|
|
|
response = Net::HTTPResponse.new '1.1', 200, 'OK'
|
|
|
|
|
|
|
|
response.uri = nil
|
|
|
|
|
|
|
|
assert_nil response.uri
|
|
|
|
|
|
|
|
response.uri = uri
|
|
|
|
|
|
|
|
assert_equal uri, response.uri
|
2016-02-19 02:45:58 -05:00
|
|
|
assert_not_same uri, response.uri
|
2012-12-21 15:36:07 -05:00
|
|
|
end
|
|
|
|
|
2014-12-23 13:10:19 -05:00
|
|
|
def test_ensure_zero_space_does_not_regress
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200OK
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
2015-06-01 22:18:44 -04:00
|
|
|
assert_raise Net::HTTPBadResponse do
|
2014-12-23 13:10:19 -05:00
|
|
|
Net::HTTPResponse.read_new(io)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_allow_trailing_space_after_status
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200\s
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
assert_equal('1.1', res.http_version)
|
|
|
|
assert_equal('200', res.code)
|
|
|
|
assert_equal('', res.message)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_normal_status_line
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200 OK
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
assert_equal('1.1', res.http_version)
|
|
|
|
assert_equal('200', res.code)
|
|
|
|
assert_equal('OK', res.message)
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_allow_empty_reason_code
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 200
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
assert_equal('1.1', res.http_version)
|
|
|
|
assert_equal('200', res.code)
|
|
|
|
assert_equal(nil, res.message)
|
|
|
|
end
|
|
|
|
|
2016-11-08 15:25:44 -05:00
|
|
|
def test_raises_exception_with_missing_reason
|
|
|
|
io = dummy_io(<<EOS)
|
|
|
|
HTTP/1.1 404
|
|
|
|
Content-Length: 5
|
|
|
|
Connection: close
|
|
|
|
|
|
|
|
hello
|
|
|
|
EOS
|
|
|
|
|
|
|
|
res = Net::HTTPResponse.read_new(io)
|
|
|
|
assert_equal(nil, res.message)
|
2018-06-06 05:01:04 -04:00
|
|
|
assert_raise Net::HTTPClientException do
|
2016-11-08 15:25:44 -05:00
|
|
|
res.error!
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2018-06-24 05:31:09 -04:00
|
|
|
def test_read_code_type
|
|
|
|
res = Net::HTTPUnknownResponse.new('1.0', '???', 'test response')
|
|
|
|
assert_equal Net::HTTPUnknownResponse, res.code_type
|
|
|
|
|
|
|
|
res = Net::HTTPInformation.new('1.0', '1xx', 'test response')
|
|
|
|
assert_equal Net::HTTPInformation, res.code_type
|
|
|
|
|
|
|
|
res = Net::HTTPSuccess.new('1.0', '2xx', 'test response')
|
|
|
|
assert_equal Net::HTTPSuccess, res.code_type
|
|
|
|
|
|
|
|
res = Net::HTTPRedirection.new('1.0', '3xx', 'test response')
|
|
|
|
assert_equal Net::HTTPRedirection, res.code_type
|
|
|
|
|
|
|
|
res = Net::HTTPClientError.new('1.0', '4xx', 'test response')
|
|
|
|
assert_equal Net::HTTPClientError, res.code_type
|
|
|
|
|
|
|
|
res = Net::HTTPServerError.new('1.0', '5xx', 'test response')
|
|
|
|
assert_equal Net::HTTPServerError, res.code_type
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_inspect_response
|
|
|
|
res = Net::HTTPUnknownResponse.new('1.0', '???', 'test response')
|
|
|
|
assert_equal '#<Net::HTTPUnknownResponse ??? test response readbody=false>', res.inspect
|
|
|
|
|
|
|
|
res = Net::HTTPUnknownResponse.new('1.0', '???', 'test response')
|
|
|
|
socket = Net::BufferedIO.new(StringIO.new('test body'))
|
|
|
|
res.reading_body(socket, true) {}
|
|
|
|
assert_equal '#<Net::HTTPUnknownResponse ??? test response readbody=true>', res.inspect
|
|
|
|
end
|
|
|
|
|
2009-10-30 09:09:03 -04:00
|
|
|
private
|
|
|
|
|
|
|
|
def dummy_io(str)
|
2012-07-19 18:43:38 -04:00
|
|
|
str = str.gsub(/\n/, "\r\n")
|
|
|
|
|
2009-10-30 09:09:03 -04:00
|
|
|
Net::BufferedIO.new(StringIO.new(str))
|
|
|
|
end
|
|
|
|
end
|