initial commit + release

everything shou^Wmight be working...

1 files changed, 152 insertions, 0 deletions

diff --git a/test/lib_read_write.rb b/test/lib_read_write.rb
new file mode 100644
index 0000000..7fd24d5
--- /dev/null
+++ b/test/lib_read_write.rb
@@ -0,0 +1,152 @@
+require 'test/unit'
+require 'io/nonblock'
+$-w = true
+require 'kgio'
+
+module LibReadWriteTest
+  def teardown
+    assert_nothing_raised do
+      @rd.close unless @rd.closed?
+      @wr.close unless @wr.closed?
+    end
+    assert_nothing_raised do
+      Kgio.wait_readable = Kgio.wait_writable = nil
+    end
+  end
+
+  def test_read_eof
+    @wr.close
+    assert_raises(EOFError) { @rd.kgio_read 5 }
+  end
+
+  def test_write_closed
+    @rd.close
+    assert_raises(Errno::EPIPE) { @wr.kgio_write "HI" }
+  end
+
+  def test_write_conv
+    assert_equal nil, @wr.kgio_write(10)
+    assert_equal "10", @rd.kgio_read(2)
+  end
+
+  def test_read_empty
+    assert_equal Kgio::WaitReadable, @rd.kgio_read(1)
+  end
+
+  def test_read_too_much
+    assert_equal nil, @wr.kgio_write("hi")
+    assert_equal "hi", @rd.kgio_read(4)
+  end
+
+  def test_read_short
+    assert_equal nil, @wr.kgio_write("hi")
+    assert_equal "h", @rd.kgio_read(1)
+    assert_equal "i", @rd.kgio_read(1)
+  end
+
+  def test_read_extra_buf
+    tmp = ""
+    tmp_object_id = tmp.object_id
+    assert_equal nil, @wr.kgio_write("hi")
+    rv = @rd.kgio_read(2, tmp)
+    assert_equal "hi", rv
+    assert_equal rv.object_id, tmp.object_id
+    assert_equal tmp_object_id, rv.object_id
+  end
+
+  def test_write_return_wait_writable
+    tmp = []
+    tmp << @wr.kgio_write("HI") until tmp[-1] == Kgio::WaitWritable
+    assert_equal Kgio::WaitWritable, tmp.pop
+    assert tmp.size > 0
+    penultimate = tmp.pop
+    assert(penultimate == "I" || penultimate == nil)
+    assert tmp.size > 0
+    tmp.each { |count| assert_equal nil, count }
+  end
+
+  def test_monster_write
+    buf = "." * 1024 * 1024 * 10
+    rv = @wr.kgio_write(buf)
+    assert_kind_of String, rv
+    assert rv.size < buf.size
+    assert_equal(buf, (rv + @rd.read(buf.size - rv.size)))
+  end
+
+  def test_wait_readable_ruby_default
+    def @rd.ruby
+      raise RuntimeError, "Hello"
+    end
+    assert_nothing_raised { Kgio.wait_readable = :ruby }
+    elapsed = 0
+    foo = nil
+    t0 = Time.now
+    thr = Thread.new { sleep 1; @wr.write "HELLO" }
+    assert_nothing_raised do
+      foo = @rd.kgio_read(5)
+      elapsed = Time.now - t0
+    end
+    assert elapsed >= 1.0
+    assert_equal "HELLO", foo
+    thr.join
+    assert_equal 5, thr.value
+  end
+
+  def test_wait_writable_ruby_default
+    def @wr.ruby
+      raise RuntimeError, "Hello"
+    end
+    buf = "." * 512
+    nr = 0
+    begin
+      nr += @wr.write_nonblock(buf)
+    rescue Errno::EAGAIN
+      break
+    end while true
+    assert_nothing_raised { Kgio.wait_writable = :ruby }
+    elapsed = 0
+    foo = nil
+    t0 = Time.now
+    thr = Thread.new { sleep 1; @rd.readpartial(nr) }
+    assert_nothing_raised do
+      foo = @wr.kgio_write("HELLO")
+      elapsed = Time.now - t0
+    end
+    assert_nil foo
+    assert elapsed >= 1.0
+    assert(String === foo || foo == nil)
+    assert_kind_of String, thr.value
+  end
+
+  def test_wait_readable_method
+    def @rd.moo
+      defined?(@z) ? raise(RuntimeError, "Hello") : @z = "HI"
+    end
+    assert_nothing_raised { Kgio.wait_readable = :moo }
+    foo = nil
+    begin
+      foo = @rd.kgio_read(5)
+      assert false
+    rescue RuntimeError => e
+      assert_equal("Hello", e.message)
+    end
+    assert_equal "HI", @rd.instance_variable_get(:@z)
+    assert_nil foo
+  end
+
+  def test_wait_writable_method
+    def @wr.moo
+      defined?(@z) ? raise(RuntimeError, "Hello") : @z = "HI"
+    end
+    assert_nothing_raised { Kgio.wait_writable = :moo }
+    n = []
+    begin
+      loop { n << @wr.kgio_write("HIHIHIHIHIHI") }
+      assert false
+    rescue RuntimeError => e
+      assert_equal("Hello", e.message)
+    end
+    assert n.size > 0
+    assert_equal "HI", @wr.instance_variable_get(:@z)
+  end
+end