mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
050a895439
* Wake up join list within thread EC context. * Consume items from join list so that they are not re-executed. If `rb_fiber_scheduler_unblock` raises an exception, it can result in a segfault if `rb_threadptr_join_list_wakeup` is not within a valid EC. This change moves `rb_threadptr_join_list_wakeup` into the thread's top level EC which initially caused an infinite loop because on exception will retry. We explicitly remove items from the thread's join list to avoid this situation. * Verify the required scheduler interface. * Test several scheduler hooks methods with broken `unblock` implementation.
68 lines
1.2 KiB
Ruby
68 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
require 'test/unit'
|
|
require_relative 'scheduler'
|
|
|
|
class TestFiberSleep < Test::Unit::TestCase
|
|
ITEMS = [0, 1, 2, 3, 4]
|
|
|
|
def test_sleep
|
|
items = []
|
|
|
|
thread = Thread.new do
|
|
scheduler = Scheduler.new
|
|
Fiber.set_scheduler scheduler
|
|
|
|
5.times do |i|
|
|
Fiber.schedule do
|
|
assert_operator sleep(i/100.0), :>=, 0
|
|
items << i
|
|
end
|
|
end
|
|
|
|
# Should be 5 fibers waiting:
|
|
assert_equal scheduler.waiting.size, 5
|
|
end
|
|
|
|
thread.join
|
|
|
|
assert_equal ITEMS, items
|
|
end
|
|
|
|
def test_sleep_returns_seconds_slept
|
|
seconds = nil
|
|
|
|
thread = Thread.new do
|
|
scheduler = Scheduler.new
|
|
Fiber.set_scheduler scheduler
|
|
Fiber.schedule do
|
|
seconds = sleep(2)
|
|
end
|
|
end
|
|
|
|
thread.join
|
|
|
|
assert_operator seconds, :>=, 2, "actual: %p" % seconds
|
|
end
|
|
|
|
def test_broken_sleep
|
|
thread = Thread.new do
|
|
Thread.current.report_on_exception = false
|
|
|
|
scheduler = Scheduler.new
|
|
|
|
def scheduler.kernel_sleep(duration = nil)
|
|
raise "Broken sleep!"
|
|
end
|
|
|
|
Fiber.set_scheduler scheduler
|
|
|
|
Fiber.schedule do
|
|
sleep 0
|
|
end
|
|
end
|
|
|
|
assert_raise(RuntimeError) do
|
|
thread.join
|
|
end
|
|
end
|
|
end
|