diff options
author | Samuel Williams <samuel.williams@oriontransfer.co.nz> | 2022-12-17 20:43:21 +1300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-12-17 20:43:21 +1300 |
commit | 9da5a7e79dcaed9bf233f6a9a5df54c727f60c77 (patch) | |
tree | 9439c8d7626fb6e5ac5112b89bfe793b1fe7f4c9 /test | |
parent | 297df9240772a14141988aef63400e3988584ced (diff) | |
download | ruby-9da5a7e79dcaed9bf233f6a9a5df54c727f60c77.tar.gz |
Add tests for `Queue#pop` with fiber scheduler. (#6953)
Diffstat (limited to 'test')
-rw-r--r-- | test/fiber/test_queue.rb | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/test/fiber/test_queue.rb b/test/fiber/test_queue.rb new file mode 100644 index 0000000000..10d9540492 --- /dev/null +++ b/test/fiber/test_queue.rb @@ -0,0 +1,42 @@ +# frozen_string_literal: true +require 'test/unit' +require_relative 'scheduler' + +class TestFiberQueue < Test::Unit::TestCase + def test_pop_with_timeout + queue = Thread::Queue.new + result = :unspecified + + Thread.new do + scheduler = Scheduler.new + Fiber.set_scheduler(scheduler) + + Fiber.schedule do + result = queue.pop(timeout: 0.0001) + end + + scheduler.run + end.join + + assert_nil result + end + + def test_pop_with_timeout_and_value + queue = Thread::Queue.new + queue.push(:something) + result = :unspecified + + Thread.new do + scheduler = Scheduler.new + Fiber.set_scheduler(scheduler) + + Fiber.schedule do + result = queue.pop(timeout: 0.0001) + end + + scheduler.run + end.join + + assert_equal :something, result + end +end |