1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/io/wait/test_io_wait.rb
nobu e786aa711d io/wait: add IO#wait_writable method
* ext/io/wait/wait.c (io_wait_writable): this is easier to use than
  IO.select for a single IO object and is immune to the
  limitations/innefficiency of select() on platforms where poll/ppoll
  is available.  patched by Eric Wong.  [Feature #4646]


git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@37785 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2012-11-21 13:40:00 +00:00

108 lines
2 KiB
Ruby

# -*- coding: us-ascii -*-
require 'test/unit'
require 'timeout'
require 'socket'
begin
require 'io/wait'
rescue LoadError
end
class TestIOWait < Test::Unit::TestCase
def setup
if /mswin|mingw/ =~ RUBY_PLATFORM
@r, @w = Socket.pair(Socket::AF_INET, Socket::SOCK_STREAM, 0)
else
@r, @w = IO.pipe
end
end
def teardown
@r.close unless @r.closed?
@w.close unless @w.closed?
end
def test_nread
assert_equal 0, @r.nread
@w.syswrite "."
sleep 0.1
assert_equal 1, @r.nread
end
def test_nread_buffered
@w.syswrite ".\n!"
assert_equal ".\n", @r.read(2)
assert_equal 1, @r.nread
end
def test_ready?
refute @r.ready?, "shouldn't ready, but ready"
@w.syswrite "."
sleep 0.1
assert @r.ready?, "should ready, but not"
end
def test_buffered_ready?
@w.syswrite ".\n!"
assert_equal ".\n", @r.gets
assert @r.ready?
end
def test_wait
assert_nil @r.wait(0)
@w.syswrite "."
sleep 0.1
assert_equal @r, @r.wait(0)
end
def test_wait_buffered
@w.syswrite ".\n!"
assert_equal ".\n", @r.gets
assert_equal true, @r.wait(0)
end
def test_wait_forever
Thread.new { sleep 0.01; @w.syswrite "." }
assert_equal @r, @r.wait
end
def test_wait_eof
Thread.new { sleep 0.01; @w.close }
assert_nil @r.wait
end
def test_wait_writable
assert_equal @w, @w.wait_writable
end
def test_wait_writable_timeout
assert_equal @w, @w.wait_writable(0.001)
written = fill_pipe
assert_nil @w.wait_writable(0.001)
@r.read(written)
assert_equal @w, @w.wait_writable(0.001)
end
def test_wait_writable_EPIPE
fill_pipe
@r.close
assert_equal @w, @w.wait_writable
end
def test_wait_writable_closed
@w.close
assert_raises(IOError) { @w.wait_writable }
end
private
def fill_pipe
written = 0
buf = " " * 4096
begin
written += @w.write_nonblock(buf)
rescue Errno::EAGAIN
return written
end while true
end
end if IO.method_defined?(:wait)