2006-06-30 16:42:12 -04:00
|
|
|
# Copyright (c) 2005 Zed A. Shaw
|
|
|
|
|
# You can redistribute it and/or modify it under the same terms as Ruby.
|
|
|
|
|
#
|
|
|
|
|
# Additional work donated by contributors. See http://mongrel.rubyforge.org/attributions.html
|
|
|
|
|
# for more information.
|
2006-05-23 09:07:05 -04:00
|
|
|
|
|
|
|
|
require 'test/unit'
|
|
|
|
|
require 'net/http'
|
|
|
|
|
require 'mongrel'
|
|
|
|
|
require 'timeout'
|
2006-05-30 19:36:48 -04:00
|
|
|
require File.dirname(__FILE__) + "/testhelp.rb"
|
2006-05-23 09:07:05 -04:00
|
|
|
|
|
|
|
|
class SimpleHandler < Mongrel::HttpHandler
|
|
|
|
|
def process(request, response)
|
|
|
|
|
response.start do |head,out|
|
|
|
|
|
head["Content-Type"] = "text/html"
|
|
|
|
|
results = "<html><body>Your request:<br /><pre>#{request.params.to_yaml}</pre><a href=\"/files\">View the files.</a></body></html>"
|
|
|
|
|
out << results
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
class DumbHandler < Mongrel::HttpHandler
|
|
|
|
|
def process(request, response)
|
|
|
|
|
response.start do |head,out|
|
|
|
|
|
head["Content-Type"] = "text/html"
|
|
|
|
|
out.write("test")
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def check_status(results, expecting)
|
|
|
|
|
results.each do |res|
|
|
|
|
|
assert(res.kind_of?(expecting), "Didn't get #{expecting}, got: #{res.class}")
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
class HandlersTest < Test::Unit::TestCase
|
|
|
|
|
|
|
|
|
|
def setup
|
|
|
|
|
stats = Mongrel::StatisticsFilter.new(:sample_rate => 1)
|
|
|
|
|
|
|
|
|
|
@config = Mongrel::Configurator.new :host => '127.0.0.1', :port => 9998 do
|
|
|
|
|
listener do
|
|
|
|
|
uri "/", :handler => SimpleHandler.new
|
|
|
|
|
uri "/", :handler => stats
|
|
|
|
|
uri "/404", :handler => Mongrel::Error404Handler.new("Not found")
|
2006-10-24 19:22:16 -04:00
|
|
|
uri "/dumb", :handler => Mongrel::DeflateFilter.new
|
2006-05-23 09:07:05 -04:00
|
|
|
uri "/dumb", :handler => DumbHandler.new, :in_front => true
|
|
|
|
|
uri "/files", :handler => Mongrel::DirHandler.new("doc")
|
|
|
|
|
uri "/files_nodir", :handler => Mongrel::DirHandler.new("doc",listing_allowed=false, index_html="none")
|
|
|
|
|
uri "/status", :handler => Mongrel::StatusHandler.new(:stats_filter => stats)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
@config.run
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def teardown
|
|
|
|
|
@config.stop
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def test_more_web_server
|
|
|
|
|
res = hit([ "http://localhost:9998/test",
|
|
|
|
|
"http://localhost:9998/dumb",
|
|
|
|
|
"http://localhost:9998/404",
|
|
|
|
|
"http://localhost:9998/files/rdoc/index.html",
|
|
|
|
|
"http://localhost:9998/files/rdoc/nothere.html",
|
|
|
|
|
"http://localhost:9998/files/rdoc/",
|
|
|
|
|
"http://localhost:9998/files_nodir/rdoc/",
|
|
|
|
|
"http://localhost:9998/status",
|
|
|
|
|
])
|
|
|
|
|
|
|
|
|
|
check_status res, String
|
|
|
|
|
end
|
|
|
|
|
|
2006-10-24 19:22:16 -04:00
|
|
|
def test_deflate
|
|
|
|
|
Net::HTTP.start("localhost", 9998) do |h|
|
|
|
|
|
# test that no accept-encoding returns a non-deflated response
|
|
|
|
|
req = h.get("/dumb")
|
|
|
|
|
assert(
|
2007-09-24 12:37:26 -04:00
|
|
|
!req['Content-Encoding'] ||
|
|
|
|
|
!req['Content-Encoding'].include?('deflate'))
|
2006-10-24 19:22:16 -04:00
|
|
|
assert_equal "test", req.body
|
|
|
|
|
|
|
|
|
|
req = h.get("/dumb", {"Accept-Encoding" => "deflate"})
|
|
|
|
|
# -MAX_WBITS stops zlib from looking for a zlib header
|
|
|
|
|
inflater = Zlib::Inflate.new(-Zlib::MAX_WBITS)
|
2007-09-24 12:37:26 -04:00
|
|
|
assert req['Content-Encoding'].include?('deflate')
|
2006-10-24 19:22:16 -04:00
|
|
|
assert_equal "test", inflater.inflate(req.body)
|
|
|
|
|
end
|
2006-05-23 09:07:05 -04:00
|
|
|
end
|
|
|
|
|
|
2006-05-30 19:36:48 -04:00
|
|
|
# TODO: find out why this fails on win32 but nowhere else
|
2006-11-25 01:13:45 -05:00
|
|
|
#def test_posting_fails_dirhandler
|
|
|
|
|
# req = Net::HTTP::Post.new("http://localhost:9998/files/rdoc/")
|
|
|
|
|
# req.set_form_data({'from'=>'2005-01-01', 'to'=>'2005-03-31'}, ';')
|
|
|
|
|
# res = hit [["http://localhost:9998/files/rdoc/",req]]
|
|
|
|
|
# check_status res, Net::HTTPNotFound
|
|
|
|
|
#end
|
2006-05-23 09:07:05 -04:00
|
|
|
|
|
|
|
|
def test_unregister
|
|
|
|
|
@config.listeners["127.0.0.1:9998"].unregister("/")
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
