2 files changed, 130 insertions, 0 deletions
diff --git a/test/ruby/test_readpartial.rb b/test/ruby/test_readpartial.rb
new file mode 100644
index 0000000000..526425dc57
--- /dev/null
+++ b/test/ruby/test_readpartial.rb
@@ -0,0 +1,72 @@
+require 'test/unit'
+require 'timeout'
+require 'fcntl'
+
+class TestReadPartial < Test::Unit::TestCase
+  def make_pipe
+    r, w = IO.pipe
+    begin
+      yield r, w
+    ensure
+      r.close unless r.closed?
+      w.close unless w.closed?
+    end
+  end
+
+  def pipe
+    make_pipe {|r, w|
+      yield r, w
+    }
+    return unless defined?(Fcntl::F_SETFL)
+    return unless defined?(Fcntl::F_GETFL)
+    return unless defined?(Fcntl::O_NONBLOCK)
+    make_pipe {|r, w|
+      r.fcntl(Fcntl::F_SETFL, r.fcntl(Fcntl::F_GETFL) | Fcntl::O_NONBLOCK)
+      yield r, w
+    }
+  end
+
+  def test_length_zero
+    pipe {|r, w|
+      assert_equal('', r.readpartial(0))
+    }
+  end
+
+  def test_closed_pipe
+    pipe {|r, w|
+      w << 'abc'
+      w.close
+      assert_equal('ab', r.readpartial(2))
+      assert_equal('c', r.readpartial(2))
+      assert_raises(EOFError) { r.readpartial(2) }
+      assert_raises(EOFError) { r.readpartial(2) }
+    }
+  end
+
+  def test_open_pipe
+    pipe {|r, w|
+      w << 'abc'
+      assert_equal('ab', r.readpartial(2))
+      assert_equal('c', r.readpartial(2))
+      assert_raises(TimeoutError) {
+        timeout(0.1) { r.readpartial(2) }
+      }
+    }
+  end
+
+  def test_with_stdio
+    pipe {|r, w|
+      w << "abc\ndef\n"
+      assert_equal("abc\n", r.gets)
+      w << "ghi\n"
+      assert_equal("de", r.readpartial(2))
+      assert_equal("f\n", r.readpartial(4096))
+      assert_equal("ghi\n", r.readpartial(4096))
+      assert_raises(TimeoutError) {
+        timeout(0.1) { r.readpartial(2) }
+      }
+    }
+  end
+
+end
+
diff --git a/test/zlib/test_zlib.rb b/test/zlib/test_zlib.rb
new file mode 100644
index 0000000000..5a646cf8de
--- /dev/null
+++ b/test/zlib/test_zlib.rb
@@ -0,0 +1,58 @@
+require 'test/unit/testsuite'
+require 'test/unit/testcase'
+require 'stringio'
+
+begin
+  require 'zlib'
+rescue LoadError
+end
+
+if defined? Zlib
+  class TestZlibGzipReader < Test::Unit::TestCase
+    D0 = "\037\213\010\000S`\017A\000\003\003\000\000\000\000\000\000\000\000\000"
+    def test_read0
+      assert_equal("", Zlib::GzipReader.new(StringIO.new(D0)).read(0))
+    end
+
+    def test_ungetc # [ruby-dev:24060]
+      s = ""
+      w = Zlib::GzipWriter.new(StringIO.new(s))
+      w << (1...1000).to_a.inspect
+      w.close
+      r = Zlib::GzipReader.new(StringIO.new(s))
+      r.read(100)
+      r.ungetc ?a
+      assert_nothing_raised {
+        r.read(100)
+        r.read
+        r.close
+      }
+    end
+
+    def test_ungetc_paragraph # [ruby-dev:24065]
+      s = ""
+      w = Zlib::GzipWriter.new(StringIO.new(s))
+      w << "abc"
+      w.close
+      r = Zlib::GzipReader.new(StringIO.new(s))
+      $/ = ""
+      r.ungetc ?\n
+      assert_equal("abc", r.gets)
+      assert_nothing_raised {
+        r.read
+        r.close
+      }
+    end
+  end
+
+  class TestZlibGzipWriter < Test::Unit::TestCase
+    def test_invalid_new
+      # [ruby-dev:23228]
+      assert_raise(NoMethodError) { Zlib::GzipWriter.new(nil).close }
+      # [ruby-dev:23344]
+      assert_raise(NoMethodError) { Zlib::GzipWriter.new(true).close }
+      assert_raise(NoMethodError) { Zlib::GzipWriter.new(0).close }
+      assert_raise(NoMethodError) { Zlib::GzipWriter.new(:hoge).close }
+    end
+  end
+end