diff options
Diffstat (limited to 'trunk/test/webrick/utils.rb')
-rw-r--r-- | trunk/test/webrick/utils.rb | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/trunk/test/webrick/utils.rb b/trunk/test/webrick/utils.rb new file mode 100644 index 0000000000..39cadcff54 --- /dev/null +++ b/trunk/test/webrick/utils.rb @@ -0,0 +1,56 @@ +begin + loadpath = $:.dup + $:.replace($: | [File.expand_path("../ruby", File.dirname(__FILE__))]) + require 'envutil' +ensure + $:.replace(loadpath) +end +require "webrick" +begin + require "webrick/https" +rescue LoadError +end +require "webrick/httpproxy" + +module TestWEBrick + NullWriter = Object.new + def NullWriter.<<(msg) + puts msg if $DEBUG + return self + end + + RubyBin = "\"#{EnvUtil.rubybin}\"" + RubyBin << " \"-I#{File.expand_path("../..", File.dirname(__FILE__))}/lib\"" + RubyBin << " \"-I#{File.dirname(EnvUtil.rubybin)}/.ext/common\"" + RubyBin << " \"-I#{File.dirname(EnvUtil.rubybin)}/.ext/#{RUBY_PLATFORM}\"" + + module_function + + def start_server(klass, config={}, &block) + server = klass.new({ + :BindAddress => "127.0.0.1", :Port => 0, + :ShutdownSocketWithoutClose =>true, + :ServerType => Thread, + :Logger => WEBrick::Log.new(NullWriter), + :AccessLog => [[NullWriter, ""]] + }.update(config)) + begin + server.start + addr = server.listeners[0].addr + block.yield([server, addr[3], addr[1]]) + ensure + server.shutdown + until server.status == :Stop + sleep 0.1 + end + end + end + + def start_httpserver(config={}, &block) + start_server(WEBrick::HTTPServer, config, &block) + end + + def start_httpproxy(config={}, &block) + start_server(WEBrick::HTTPProxyServer, config, &block) + end +end |