Add tests for `Queue#pop` with fiber scheduler. (#6953)

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