summaryrefslogtreecommitdiff
path: root/trunk/test/webrick/utils.rb
diff options
context:
space:
mode:
Diffstat (limited to 'trunk/test/webrick/utils.rb')
-rw-r--r--trunk/test/webrick/utils.rb56
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