Improve timeout tests.

1 files changed, 20 insertions, 1 deletions

diff --git a/test/fiber/test_timeout.rb b/test/fiber/test_timeout.rb
index 127e4b0084..c17092b8e6 100644
--- a/test/fiber/test_timeout.rb
+++ b/test/fiber/test_timeout.rb
@@ -14,7 +14,7 @@ class TestFiberTimeout < Test::Unit::TestCase
 
       Fiber.schedule do
         begin
-          Timeout.timeout(0.01) do
+          Timeout.timeout(0.001) do
             sleep(1)
           end
         rescue
@@ -27,4 +27,23 @@ class TestFiberTimeout < Test::Unit::TestCase
 
     assert_kind_of(Timeout::Error, error)
   end
+
+  MESSAGE = "Hello World"
+
+  def test_timeout_on_main_fiber
+    message = nil
+
+    thread = Thread.new do
+      scheduler = Scheduler.new
+      Fiber.set_scheduler scheduler
+
+      Timeout.timeout(1) do
+        message = MESSAGE
+      end
+    end
+
+    thread.join
+
+    assert_equal MESSAGE, message
+  end
 end