1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
|
# 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
ensure
scheduler.close
end
assert_raise(RuntimeError) do
thread.join
end
end
end
|