summaryrefslogtreecommitdiff
path: root/test/webrick/utils.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/webrick/utils.rb')
-rw-r--r--test/webrick/utils.rb36
1 files changed, 36 insertions, 0 deletions
diff --git a/test/webrick/utils.rb b/test/webrick/utils.rb
new file mode 100644
index 00000000000..50ffd759aaa
--- /dev/null
+++ b/test/webrick/utils.rb
@@ -0,0 +1,36 @@
+require "webrick"
+require "webrick/httpproxy"
+
+module TestWEBrick
+ NullWriter = Object.new
+ def NullWriter.<<(msg)
+ puts msg if $DEBUG
+ return self
+ end
+
+ module_function
+
+ def start_server(klass, config={}, &block)
+ server = klass.new({
+ :BindAddress => "127.0.0.1", :Port => 0,
+ :Logger => WEBrick::Log.new(NullWriter),
+ :AccessLog => [[NullWriter, ""]]
+ }.update(config))
+ begin
+ thread = Thread.start{ server.start }
+ addr = server.listeners[0].addr
+ block.call([server, addr[3], addr[1]])
+ ensure
+ server.stop
+ thread.join
+ 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