about summary refs log tree commit homepage
path: root/test/unit/test_request_progress.rb
blob: a100426e55de0c42117afb52d979bb0cdb43d9b5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# 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/test_helper'

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
    @port = process_based_port
    redirect_test_io do
      @server = Mongrel::HttpServer.new("127.0.0.1", @port)
    end
    @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", @port)
    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", @port)

    # 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", @port)
    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