test/test_timeout.rb

   1 # frozen_string_literal: false
   2 require 'test/unit'
   3 require 'timeout'
   4
   5 class TestTimeout < Test::Unit::TestCase
   6
   7   def test_non_timing_out_code_is_successful
   8     assert_nothing_raised do
   9       assert_equal :ok, Timeout.timeout(1){ :ok }
  10     end
  11   end
  12
  13   def test_yield_param
  14     assert_equal [5, :ok], Timeout.timeout(5){|s| [s, :ok] }
  15   end
  16
  17   def test_queue
  18     q = Thread::Queue.new
  19     assert_raise(Timeout::Error, "[ruby-dev:32935]") {
  20       Timeout.timeout(0.01) { q.pop }
  21     }
  22   end
  23
  24   def test_timeout
  25     assert_raise(Timeout::Error) do
  26       Timeout.timeout(0.1) {
  27         nil while true
  28       }
  29     end
  30   end
  31
  32   def test_cannot_convert_into_time_interval
  33     bug3168 = '[ruby-dev:41010]'
  34     def (n = Object.new).zero?; false; end
  35     assert_raise(TypeError, bug3168) {Timeout.timeout(n) { sleep 0.1 }}
  36   end
  37
  38   def test_skip_rescue
  39     bug8730 = '[Bug #8730]'
  40     e = nil
  41     assert_raise_with_message(Timeout::Error, /execution expired/, bug8730) do
  42       Timeout.timeout 0.01 do
  43         begin
  44           sleep 3
  45         rescue Exception => e
  46         end
  47       end
  48     end
  49     assert_nil(e, bug8730)
  50   end
  51
  52   def test_rescue_exit
  53     exc = Class.new(RuntimeError)
  54     e = nil
  55     assert_nothing_raised(exc) do
  56       Timeout.timeout 0.01, exc do
  57         begin
  58           sleep 3
  59         rescue exc => e
  60         end
  61       end
  62     end
  63     assert_raise_with_message(exc, 'execution expired') {raise e if e}
  64   end
  65
  66   def test_custom_exception
  67     bug9354 = '[ruby-core:59511] [Bug #9354]'
  68     err = Class.new(StandardError) do
  69       def initialize(msg) super end
  70     end
  71     assert_nothing_raised(ArgumentError, bug9354) do
  72       assert_equal(:ok, Timeout.timeout(100, err) {:ok})
  73     end
  74     assert_raise_with_message(err, 'execution expired') do
  75       Timeout.timeout 0.01, err do
  76         sleep 3
  77       end
  78     end
  79     assert_raise_with_message(err, /connection to ruby-lang.org expired/) do
  80       Timeout.timeout 0.01, err, "connection to ruby-lang.org expired" do
  81         sleep 3
  82       end
  83     end
  84   end
  85
  86   def test_exit_exception
  87     assert_raise_with_message(Timeout::Error, "boon") do
  88       Timeout.timeout(10, Timeout::Error) do
  89         raise Timeout::Error, "boon"
  90       end
  91     end
  92   end
  93
  94   def test_raise_with_message
  95     bug17812 = '[ruby-core:103502] [Bug #17812]: Timeout::Error doesn\'t let two-argument raise() set a new message'
  96     exc = Timeout::Error.new('foo')
  97     assert_raise_with_message(Timeout::Error, 'bar', bug17812) do
  98       raise exc, 'bar'
  99     end
 100   end
 101
 102   def test_enumerator_next
 103     bug9380 = '[ruby-dev:47872] [Bug #9380]: timeout in Enumerator#next'
 104     e = (o=Object.new).to_enum
 105     def o.each
 106       sleep
 107     end
 108     assert_raise_with_message(Timeout::Error, 'execution expired', bug9380) do
 109       Timeout.timeout(0.01) {e.next}
 110     end
 111   end
 112
 113   def test_handle_interrupt
 114     bug11344 = '[ruby-dev:49179] [Bug #11344]'
 115     ok = false
 116     assert_raise(Timeout::Error) {
 117       Thread.handle_interrupt(Timeout::Error => :never) {
 118         Timeout.timeout(0.01) {
 119           sleep 0.2
 120           ok = true
 121           Thread.handle_interrupt(Timeout::Error => :on_blocking) {
 122             sleep 0.2
 123           }
 124         }
 125       }
 126     }
 127     assert(ok, bug11344)
 128   end
 129 end