summaryrefslogtreecommitdiff
path: root/test/ruby/test_io_timeout.rb
blob: e017395980564bd9f032ae864da9e924bc4cc4ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# frozen_string_literal: false

require 'io/nonblock'

class TestIOTimeout < Test::Unit::TestCase
  def with_pipe
    omit "UNIXSocket is not defined!" unless defined?(UNIXSocket)

    begin
      i, o = UNIXSocket.pair

      yield i, o
    ensure
      i.close
      o.close
    end
  end

  def test_timeout_attribute
    with_pipe do |i, o|
      assert_nil i.timeout

      i.timeout = 10
      assert_equal 10, i.timeout
      assert_nil o.timeout

      o.timeout = 20
      assert_equal 20, o.timeout
      assert_equal 10, i.timeout
    end
  end

  def test_timeout_read_exception
    with_pipe do |i, o|
      i.timeout = 0.0001

      assert_raise(IO::TimeoutError) {i.read}
    end
  end

  def test_timeout_gets_exception
    with_pipe do |i, o|
      i.timeout = 0.0001

      assert_raise(IO::TimeoutError) {i.gets}
    end
  end

  def test_timeout_puts
    with_pipe do |i, o|
      i.timeout = 0.0001
      o.puts("Hello World")
      o.close

      assert_equal "Hello World", i.gets.chomp
    end
  end
end