1
0
Fork 0
mirror of https://github.com/puma/puma.git synced 2022-11-09 13:48:40 -05:00
puma--puma/test/test_request_progress.rb

98 lines
2.4 KiB
Ruby
Raw Normal View History

# 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.
require 'test/testhelp'
class UploadBeginHandler < Mongrel::HttpHandler
attr_reader :request_began, :request_progressed, :request_processed
def initialize
@request_notify = true
end
def reset
@request_began = false
@request_progressed = false
@request_processed = false
end
def request_begins(params)
@request_began = true
end
def request_progress(params,len,total)
@request_progressed = true
end
def process(request, response)
@request_processed = true
response.start do |head,body|
body.write("test")
end
end
end
class RequestProgressTest < Test::Unit::TestCase
def setup
@server = Mongrel::HttpServer.new("127.0.0.1", 9998)
@handler = UploadBeginHandler.new
@server.register("/upload", @handler)
@server.run
end
def teardown
@server.stop(true)
end
def test_begin_end_progress
Net::HTTP.get("localhost", "/upload", 9998)
assert @handler.request_began
assert @handler.request_progressed
assert @handler.request_processed
end
def call_and_assert_handlers_in_turn(handlers)
# reset all handlers
handlers.each { |h| h.reset }
# make the call
Net::HTTP.get("localhost", "/upload", 9998)
# assert that each one was fired
handlers.each { |h|
assert h.request_began && h.request_progressed && h.request_processed,
"Callbacks NOT fired for #{h}"
}
end
def test_more_than_one_begin_end_progress
handlers = [@handler]
second = UploadBeginHandler.new
@server.register("/upload", second)
handlers << second
call_and_assert_handlers_in_turn(handlers)
# check three handlers
third = UploadBeginHandler.new
@server.register("/upload", third)
handlers << third
call_and_assert_handlers_in_turn(handlers)
# remove handlers to make sure they've all gone away
@server.unregister("/upload")
handlers.each { |h| h.reset }
Net::HTTP.get("localhost", "/upload", 9998)
handlers.each { |h|
assert !h.request_began && !h.request_progressed && !h.request_processed
}
# re-register upload to the state before this test
@server.register("/upload", @handler)
end
end