1 files changed, 89 insertions, 0 deletions
diff --git a/test/webrick/test_httpresponse.rb b/test/webrick/test_httpresponse.rb
index d5d5552796..aae4973b4d 100644
--- a/test/webrick/test_httpresponse.rb
+++ b/test/webrick/test_httpresponse.rb
@@ -1,5 +1,6 @@
 require "webrick"
 require "minitest/autorun"
+require "stringio"
 
 module WEBrick
   class TestHTTPResponse < MiniTest::Unit::TestCase
@@ -45,5 +46,93 @@ module WEBrick
 
       assert_equal 0, logger.messages.length
     end
+
+    def test_send_body_io
+      body_r, body_w = IO.pipe
+
+      body_w.write 'hello'
+      body_w.close
+
+      @res.body = body_r
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal 'hello', r.read
+    end
+
+    def test_send_body_string
+      @res.body = 'hello'
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal 'hello', r.read
+    end
+
+    def test_send_body_string_io
+      @res.body = StringIO.new 'hello'
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal 'hello', r.read
+    end
+
+    def test_send_body_io_chunked
+      @res.chunked = true
+
+      body_r, body_w = IO.pipe
+
+      body_w.write 'hello'
+      body_w.close
+
+      @res.body = body_r
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal "5\r\nhello\r\n0\r\n\r\n", r.read
+    end
+
+    def test_send_body_string_chunked
+      @res.chunked = true
+
+      @res.body = 'hello'
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal "5\r\nhello\r\n0\r\n\r\n", r.read
+    end
+
+    def test_send_body_string_io_chunked
+      @res.chunked = true
+
+      @res.body = StringIO.new 'hello'
+
+      r, w = IO.pipe
+
+      @res.send_body w
+
+      w.close
+
+      assert_equal "5\r\nhello\r\n0\r\n\r\n", r.read
+    end
   end
 end