2020-05-14 13:10:55 +03:00
|
|
|
# frozen_string_literal: true
|
|
|
|
require 'test/unit'
|
|
|
|
require_relative 'scheduler'
|
|
|
|
|
2020-05-23 10:43:58 +03:00
|
|
|
class TestFiberEnumerator < Test::Unit::TestCase
|
2020-05-14 13:10:55 +03:00
|
|
|
MESSAGE = "Hello World"
|
|
|
|
|
|
|
|
def test_read_characters
|
2022-01-11 15:17:53 +03:00
|
|
|
omit "UNIXSocket is not defined!" unless defined?(UNIXSocket)
|
2020-05-14 13:10:55 +03:00
|
|
|
|
|
|
|
i, o = UNIXSocket.pair
|
2020-07-16 06:19:47 +03:00
|
|
|
|
2020-05-14 13:10:55 +03:00
|
|
|
message = String.new
|
|
|
|
|
|
|
|
thread = Thread.new do
|
|
|
|
scheduler = Scheduler.new
|
2020-10-16 04:25:58 +03:00
|
|
|
Fiber.set_scheduler scheduler
|
2020-05-14 13:10:55 +03:00
|
|
|
|
|
|
|
e = i.to_enum(:each_char)
|
|
|
|
|
2020-08-20 15:52:36 +03:00
|
|
|
Fiber.schedule do
|
2020-05-14 13:10:55 +03:00
|
|
|
o.write("Hello World")
|
|
|
|
o.close
|
|
|
|
end
|
|
|
|
|
2020-08-20 15:52:36 +03:00
|
|
|
Fiber.schedule do
|
2020-05-14 13:10:55 +03:00
|
|
|
begin
|
|
|
|
while c = e.next
|
|
|
|
message << c
|
|
|
|
end
|
|
|
|
rescue StopIteration
|
|
|
|
# Ignore.
|
|
|
|
end
|
|
|
|
|
|
|
|
i.close
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
thread.join
|
|
|
|
|
|
|
|
assert_equal(MESSAGE, message)
|
2020-07-15 10:36:57 +03:00
|
|
|
assert_predicate(i, :closed?)
|
|
|
|
assert_predicate(o, :closed?)
|
2020-05-14 13:10:55 +03:00
|
|
|
end
|
|
|
|
end
|