diff options
| author | Hiroshi SHIBATA <[email protected]> | 2024-07-09 17:44:25 +0900 |
|---|---|---|
| committer | git <[email protected]> | 2024-07-10 23:06:06 +0000 |
| commit | c7eb9ac6f96edec97332b473432261490b48ea26 () | |
| tree | 2ed87f6c7de133fb23e5c13c23bef57d0ca48bc3 /test/net/http/utils.rb | |
| parent | 0ee3960685e283d8e75149a8777eb0109d41509a (diff) | |
[ruby/net-http] Rewrite WEBrick server with TCPServer and OpenSSL::SSL::SSLServer
https://.com/ruby/net-http/commit/b01bcf6d7f
| -rw-r--r-- | test/net/http/utils.rb | 351 |
1 files changed, 285 insertions, 66 deletions
@@ -1,13 +1,220 @@ # frozen_string_literal: false -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 @@ -35,89 +242,101 @@ module TestNetHTTPUtils def teardown if @server @server.shutdown - @server_thread.join - WEBrick::Utils::TimeoutHandler.terminate end @log_tester.call(@log) if @log_tester - # resume global state Net::HTTP.version_1_2 end def spawn_server @log = [] - @log_tester = lambda {|log| assert_equal([], log ) } @config = self.class::CONFIG - server_config = { - :BindAddress => config('host'), - :Port => 0, - :Logger => WEBrick::Log.new(@log, WEBrick::BasicLog::WARN), - :AccessLog => [], - :ServerType => Thread, - } - server_config[:OutputBufferSize] = 4 if config('chunked') - server_config[:RequestTimeout] = config('RequestTimeout') if config('RequestTimeout') - if defined?(OpenSSL) and config('ssl_enable') - server_config.update({ - :SSLEnable => true, - :SSLCertificate => config('ssl_certificate'), - :SSLPrivateKey => config('ssl_private_key'), - :SSLTmpDhCallback => config('ssl_tmp_dh_callback'), - }) - end - @server = WEBrick::HTTPServer.new(server_config) - @server.mount('/', Servlet, config('chunked')) - @server_thread = @server.start - @config['port'] = @server[:Port] - 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) - if req['Accept'] != '*/*' - res['Content-Type'] = req['Accept'] else - res['Content-Type'] = $test_net_http_data_type end - res.body = $test_net_http_data - res.chunked = @chunked end - # echo server - def do_POST(req, res) - res['Content-Type'] = req['Content-Type'] - res['X-request-uri'] = req.request_uri.to_s - res.body = req.body - res.chunked = @chunked end - def do_(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 def self.clean_http_proxy_env orig = { - 'http_proxy' => ENV['http_proxy'], 'http_proxy_user' => ENV['http_proxy_user'], 'http_proxy_pass' => ENV['http_proxy_pass'], - 'no_proxy' => ENV['no_proxy'], } orig.each_key do |key| |