97 lines
2.1 KiB
Ruby
97 lines
2.1 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,
|
|
}
|
|
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)
|
|
@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 do_GET(req, res)
|
|
res['Content-Type'] = $test_net_http_data_type
|
|
res.body = $test_net_http_data
|
|
end
|
|
|
|
# echo server
|
|
def do_POST(req, res)
|
|
res['Content-Type'] = req['Content-Type']
|
|
res.body = req.body
|
|
end
|
|
end
|
|
|
|
class NullWriter
|
|
def <<(s) end
|
|
def puts(*args) end
|
|
def print(*args) end
|
|
def printf(*args) end
|
|
end
|
|
end
|