diff --git a/test/test_timeout.rb b/test/test_timeout.rb index 9194008..fead81f 100644 --- a/test/test_timeout.rb +++ b/test/test_timeout.rb @@ -427,9 +427,9 @@ def test_timeout_in_trap_handler rd, wr = IO.pipe - signal = Signal.list["USR1"] ? :USR1 : :TERM + signal = :TERM - trap(signal) do + original_handler = trap(signal) do begin Timeout.timeout(0.1) do sleep 1 @@ -444,9 +444,13 @@ def test_timeout_in_trap_handler end end - Process.kill signal, Process.pid + begin + Process.kill signal, Process.pid - assert_equal "OK", rd.read - rd.close + assert_equal "OK", rd.read + rd.close + ensure + trap(signal, original_handler) + end end end