From 9da5a7e79dcaed9bf233f6a9a5df54c727f60c77 Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@oriontransfer.co.nz>
Date: Sat, 17 Dec 2022 20:43:21 +1300
Subject: [PATCH] Add tests for `Queue#pop` with fiber scheduler. (#6953)

---
 test/fiber/test_queue.rb | 42 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 test/fiber/test_queue.rb

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