mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
ab70e53ac3
Added a Net::HTTP::Patch class which expects a message body and response body. It recycles the post method into a patch one, that will send the encoded representation to the server. Summarizing, a new class has been created, the post method extracted into send_entity, including a new argument, which defines which class to use (Post or Patch) and finally a patch method was created. [ruby-core:30426] Patched by Guilherme Silveira <guilherme.silveira AT caelum.com.br> git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@28119 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
109 lines
2.5 KiB
Ruby
109 lines
2.5 KiB
Ruby
require 'webrick'
|
|
begin
|
|
require "webrick/https"
|
|
rescue LoadError
|
|
# SSL features cannot be tested
|
|
end
|
|
require 'webrick/httpservlet/abstract'
|
|
|
|
module TestNetHTTPUtils
|
|
def start(&block)
|
|
new().start(&block)
|
|
end
|
|
|
|
def new
|
|
klass = Net::HTTP::Proxy(config('proxy_host'), config('proxy_port'))
|
|
http = klass.new(config('host'), config('port'))
|
|
http.set_debug_output logfile()
|
|
http
|
|
end
|
|
|
|
def config(key)
|
|
self.class::CONFIG[key]
|
|
end
|
|
|
|
def logfile
|
|
$DEBUG ? $stderr : NullWriter.new
|
|
end
|
|
|
|
def setup
|
|
spawn_server
|
|
end
|
|
|
|
def teardown
|
|
@server.shutdown
|
|
until @server.status == :Stop
|
|
sleep 0.1
|
|
end
|
|
# resume global state
|
|
Net::HTTP.version_1_2
|
|
end
|
|
|
|
def spawn_server
|
|
server_config = {
|
|
:BindAddress => config('host'),
|
|
:Port => config('port'),
|
|
:Logger => WEBrick::Log.new(NullWriter.new),
|
|
:AccessLog => [],
|
|
:ShutdownSocketWithoutClose => true,
|
|
:ServerType => Thread,
|
|
}
|
|
server_config[:OutputBufferSize] = 4 if config('chunked')
|
|
if defined?(OpenSSL) and config('ssl_enable')
|
|
server_config.update({
|
|
:SSLEnable => true,
|
|
:SSLCertificate => config('ssl_certificate'),
|
|
:SSLPrivateKey => config('ssl_private_key'),
|
|
})
|
|
end
|
|
@server = WEBrick::HTTPServer.new(server_config)
|
|
@server.mount('/', Servlet, config('chunked'))
|
|
@server.start
|
|
n_try_max = 5
|
|
begin
|
|
TCPSocket.open(config('host'), config('port')).close
|
|
rescue Errno::ECONNREFUSED
|
|
sleep 0.2
|
|
n_try_max -= 1
|
|
raise 'cannot spawn server; give up' if n_try_max < 0
|
|
retry
|
|
end
|
|
end
|
|
|
|
$test_net_http = nil
|
|
$test_net_http_data = (0...256).to_a.map {|i| i.chr }.join('') * 64
|
|
$test_net_http_data.force_encoding("ASCII-8BIT")
|
|
$test_net_http_data_type = 'application/octet-stream'
|
|
|
|
class Servlet < WEBrick::HTTPServlet::AbstractServlet
|
|
def initialize(this, chunked = false)
|
|
@chunked = chunked
|
|
end
|
|
|
|
def do_GET(req, res)
|
|
res['Content-Type'] = $test_net_http_data_type
|
|
res.body = $test_net_http_data
|
|
res.chunked = @chunked
|
|
end
|
|
|
|
# echo server
|
|
def do_POST(req, res)
|
|
res['Content-Type'] = req['Content-Type']
|
|
res.body = req.body
|
|
res.chunked = @chunked
|
|
end
|
|
|
|
def do_PATCH(req, res)
|
|
res['Content-Type'] = req['Content-Type']
|
|
res.body = req.body
|
|
res.chunked = @chunked
|
|
end
|
|
end
|
|
|
|
class NullWriter
|
|
def <<(s) end
|
|
def puts(*args) end
|
|
def print(*args) end
|
|
def printf(*args) end
|
|
end
|
|
end
|