* lib/webrick/httpresponse.rb (WEBrick::HTTPResponse#send_body_io):

  use readpartial to get data even if the response is streaming data and
  each data is smaller than @buffer_size.
  patched by yu nobuoka. [ruby-dev:45471] [Bug #6230]

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@35253 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 64 insertions, 0 deletions

diff --git a/test/webrick/test_httpserver.rb b/test/webrick/test_httpserver.rb
index 526d72c381..bbfef53611 100644
--- a/test/webrick/test_httpserver.rb
+++ b/test/webrick/test_httpserver.rb
@@ -258,6 +258,70 @@ class TestWEBrickHTTPServer < Test::Unit::TestCase
     assert_equal(stopped, 1)
   end
 
+  # This class is needed by test_response_io_with_chunked_set method
+  class EventManagerForChunkedResponseTest
+    def initialize
+      @listeners = []
+    end
+    def add_listener( &block )
+      @listeners << block
+    end
+    def raise_str_event( str )
+      @listeners.each{ |e| e.call( :str, str ) }
+    end
+    def raise_close_event()
+      @listeners.each{ |e| e.call( :cls ) }
+    end
+  end
+  def test_response_io_with_chunked_set
+    evt_man = EventManagerForChunkedResponseTest.new
+    t = Thread.new do
+      begin
+        config = {
+          :ServerName => "localhost"
+        }
+        TestWEBrick.start_httpserver(config) do |server, addr, port, log|
+          body_strs = [ 'aaaaaa', 'bb', 'cccc' ]
+          server.mount_proc( "/", ->( req, res ){
+            # Test for setting chunked...
+            res.chunked = true
+            r,w = IO.pipe
+            evt_man.add_listener do |type,str|
+              type == :cls ? ( w.close ) : ( w << str )
+            end
+            res.body = r
+          } )
+          Thread.pass while server.status != :Running
+          http = Net::HTTP.new(addr, port)
+          req  = Net::HTTP::Get.new("/")
+          http.request(req) do |res|
+            i = 0
+            evt_man.raise_str_event( body_strs[i] )
+            res.read_body do |s|
+              assert_equal( body_strs[i], s )
+              i += 1
+              if i < body_strs.length
+                evt_man.raise_str_event( body_strs[i] )
+              else
+                evt_man.raise_close_event()
+              end
+            end
+            assert_equal( body_strs.length, i )
+          end
+        end
+      rescue => err
+        flunk( 'exception raised in thread: ' + err.to_s )
+      end
+    end
+    if t.join( 3 ).nil?
+      evt_man.raise_close_event()
+      flunk( 'timeout' )
+      if t.join( 1 ).nil?
+        Thread.kill t
+      end
+    end
+  end
+
   def test_response_io_without_chunked_set
     config = {
       :ServerName => "localhost"