summaryrefslogtreecommitdiff
path: root/test/test_timeout.rb
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_timeout.rb')
-rw-r--r--test/test_timeout.rb156
1 files changed, 149 insertions, 7 deletions
diff --git a/test/test_timeout.rb b/test/test_timeout.rb
index 74b65f119d..e900b10cd7 100644
--- a/test/test_timeout.rb
+++ b/test/test_timeout.rb
@@ -4,12 +4,45 @@ require 'timeout'
class TestTimeout < Test::Unit::TestCase
+ def test_work_is_done_in_same_thread_as_caller
+ assert_equal Thread.current, Timeout.timeout(10){ Thread.current }
+ end
+
+ def test_work_is_done_in_same_fiber_as_caller
+ require 'fiber' # needed for ruby 3.0 and lower
+ assert_equal Fiber.current, Timeout.timeout(10){ Fiber.current }
+ end
+
def test_non_timing_out_code_is_successful
assert_nothing_raised do
assert_equal :ok, Timeout.timeout(1){ :ok }
end
end
+ def test_allows_zero_seconds
+ assert_nothing_raised do
+ assert_equal :ok, Timeout.timeout(0){:ok}
+ end
+ end
+
+ def test_allows_nil_seconds
+ assert_nothing_raised do
+ assert_equal :ok, Timeout.timeout(nil){:ok}
+ end
+ end
+
+ def test_included
+ c = Class.new do
+ include Timeout
+ def test
+ timeout(1) { :ok }
+ end
+ end
+ assert_nothing_raised do
+ assert_equal :ok, c.new.test
+ end
+ end
+
def test_yield_param
assert_equal [5, :ok], Timeout.timeout(5){|s| [s, :ok] }
end
@@ -29,24 +62,87 @@ class TestTimeout < Test::Unit::TestCase
end
end
+ def test_nested_timeout
+ a = nil
+ assert_raise(Timeout::Error) do
+ Timeout.timeout(0.1) {
+ Timeout.timeout(1) {
+ nil while true
+ }
+ a = 1
+ }
+ end
+ assert_nil a
+ end
+
+ class MyNewErrorOuter < StandardError; end
+ class MyNewErrorInner < StandardError; end
+
+ # DOES NOT fail with
+ # - raise new(message) if exc.equal?(e)
+ # + raise new(message) if exc.class == e.class
+ def test_nested_timeout_error_identity
+ begin
+ Timeout.timeout(0.1, MyNewErrorOuter) {
+ Timeout.timeout(1, MyNewErrorInner) {
+ nil while true
+ }
+ }
+ rescue => e
+ assert e.class == MyNewErrorOuter
+ end
+ end
+
+ # DOES fail with
+ # - raise new(message) if exc.equal?(e)
+ # + raise new(message) if exc.class == e.class
+ def test_nested_timeout_which_error_bubbles_up
+ raised_exception = nil
+ begin
+ Timeout.timeout(0.1) {
+ Timeout.timeout(1) {
+ raise Timeout::ExitException.new("inner message")
+ }
+ }
+ rescue Exception => e
+ raised_exception = e
+ end
+
+ assert_equal 'inner message', raised_exception.message
+ end
+
def test_cannot_convert_into_time_interval
bug3168 = '[ruby-dev:41010]'
def (n = Object.new).zero?; false; end
assert_raise(TypeError, bug3168) {Timeout.timeout(n) { sleep 0.1 }}
end
- def test_skip_rescue
- bug8730 = '[Bug #8730]'
+ def test_skip_rescue_standarderror
+ e = nil
+ assert_raise_with_message(Timeout::Error, /execution expired/) do
+ Timeout.timeout 0.01 do
+ begin
+ sleep 3
+ rescue => e
+ flunk "should not see any exception but saw #{e.inspect}"
+ end
+ end
+ end
+ end
+
+ def test_raises_exception_internally
e = nil
- assert_raise_with_message(Timeout::Error, /execution expired/, bug8730) do
+ assert_raise_with_message(Timeout::Error, /execution expired/) do
Timeout.timeout 0.01 do
begin
sleep 3
- rescue Exception => e
+ rescue Exception => exc
+ e = exc
+ raise
end
end
end
- assert_nil(e, bug8730)
+ assert_equal Timeout::ExitException, e.class
end
def test_rescue_exit
@@ -114,11 +210,11 @@ class TestTimeout < Test::Unit::TestCase
bug11344 = '[ruby-dev:49179] [Bug #11344]'
ok = false
assert_raise(Timeout::Error) {
- Thread.handle_interrupt(Timeout::Error => :never) {
+ Thread.handle_interrupt(Timeout::ExitException => :never) {
Timeout.timeout(0.01) {
sleep 0.2
ok = true
- Thread.handle_interrupt(Timeout::Error => :on_blocking) {
+ Thread.handle_interrupt(Timeout::ExitException => :on_blocking) {
sleep 0.2
}
}
@@ -126,4 +222,50 @@ class TestTimeout < Test::Unit::TestCase
}
assert(ok, bug11344)
end
+
+ def test_fork
+ omit 'fork not supported' unless Process.respond_to?(:fork)
+ r, w = IO.pipe
+ pid = fork do
+ r.close
+ begin
+ r = Timeout.timeout(0.01) { sleep 5 }
+ w.write r.inspect
+ rescue Timeout::Error
+ w.write 'timeout'
+ ensure
+ w.close
+ end
+ end
+ w.close
+ Process.wait pid
+ assert_equal 'timeout', r.read
+ r.close
+ end
+
+ def test_threadgroup
+ assert_separately(%w[-rtimeout], <<-'end;')
+ tg = ThreadGroup.new
+ thr = Thread.new do
+ tg.add(Thread.current)
+ Timeout.timeout(10){}
+ end
+ thr.join
+ assert_equal [].to_s, tg.list.to_s
+ end;
+ end
+
+ # https://github.com/ruby/timeout/issues/24
+ def test_handling_enclosed_threadgroup
+ assert_separately(%w[-rtimeout], <<-'end;')
+ Thread.new {
+ t = Thread.current
+ group = ThreadGroup.new
+ group.add(t)
+ group.enclose
+
+ assert_equal 42, Timeout.timeout(1) { 42 }
+ }.join
+ end;
+ end
end