https://git.ruby-lang.org/ruby.git/commit/?id=1f0e0dfb22

From 1f0e0dfb228fd14b3f6687539ba274ba6a2d1643 Mon Sep 17 00:00:00 2001
From: Nobuyoshi Nakada <nobu@r...>
Date: Fri, 12 Feb 2021 12:21:49 +0900
Subject: Thread::Queue.new should accept an Enumerable [Feature #17327]

Enumerable implements #to_a but not #to_array.
---
 spec/ruby/core/queue/initialize_spec.rb | 14 ++++----------
 test/ruby/test_thread_queue.rb          | 22 ++++++++++++++++++++++
 thread_sync.c                           |  2 +-
 3 files changed, 27 insertions(+), 11 deletions(-)

diff --git a/spec/ruby/core/queue/initialize_spec.rb b/spec/ruby/core/queue/initialize_spec.rb
index 5cecfe5..83c7e59 100644
--- a/spec/ruby/core/queue/initialize_spec.rb
+++ b/spec/ruby/core/queue/initialize_spec.rb
@@ -18,9 +18,9 @@ describe "Queue#initialize" do https://github.com/ruby/ruby/blob/trunk/spec/ruby/core/queue/initialize_spec.rb#L18
       q.should.empty?
     end
 
-    it "uses #to_ary on the provided Enumerable" do
+    it "uses #to_a on the provided Enumerable" do
       enumerable = MockObject.new('mock-enumerable')
-      enumerable.should_receive(:to_ary).and_return([1, 2, 3])
+      enumerable.should_receive(:to_a).and_return([1, 2, 3])
       q = Queue.new(enumerable)
       q.size.should == 3
       q.should_not.empty?
@@ -30,15 +30,9 @@ describe "Queue#initialize" do https://github.com/ruby/ruby/blob/trunk/spec/ruby/core/queue/initialize_spec.rb#L30
       q.should.empty?
     end
 
-    it "raises if the provided Enumerable does not respond to #to_ary" do
+    it "raises if the provided Enumerable does not respond to #to_a" do
       enumerable = MockObject.new('mock-enumerable')
-      -> { Queue.new(enumerable) }.should raise_error(TypeError, "no implicit conversion of MockObject into Array")
-    end
-
-    it "raises if the provided Enumerable #to_ary does not return an Array" do
-      enumerable = MockObject.new('mock-enumerable')
-      enumerable.should_receive(:to_ary).and_return(14)
-      -> { Queue.new(enumerable) }.should raise_error(TypeError, "can't convert MockObject to Array (MockObject#to_ary gives Integer)")
+      -> { Queue.new(enumerable) }.should raise_error(TypeError, "can't convert MockObject into Array")
     end
   end
 end
diff --git a/test/ruby/test_thread_queue.rb b/test/ruby/test_thread_queue.rb
index b0fa878..6185abf 100644
--- a/test/ruby/test_thread_queue.rb
+++ b/test/ruby/test_thread_queue.rb
@@ -54,6 +54,28 @@ class TestThreadQueue < Test::Unit::TestCase https://github.com/ruby/ruby/blob/trunk/test/ruby/test_thread_queue.rb#L54
     assert_equal 0, to_workers.size
   end
 
+  def test_queue_initialize
+    e = Class.new do
+      include Enumerable
+      def initialize(list) @list = list end
+      def each(&block) @list.each(&block) end
+    end
+
+    all_assertions_foreach(nil,
+                           [Array, "Array"],
+                           [e, "Enumerable"],
+                           [Struct.new(:to_a), "Array-like"],
+                           ) do |a, type|
+      q = Queue.new(a.new([1,2,3]))
+      assert_equal(3, q.size, type)
+      assert_not_predicate(q, :empty?, type)
+      assert_equal(1, q.pop, type)
+      assert_equal(2, q.pop, type)
+      assert_equal(3, q.pop, type)
+      assert_predicate(q, :empty?, type)
+    end
+  end
+
   def test_sized_queue_initialize
     q = SizedQueue.new(1)
     assert_equal 1, q.max
diff --git a/thread_sync.c b/thread_sync.c
index 131ace2..94e6340 100644
--- a/thread_sync.c
+++ b/thread_sync.c
@@ -858,7 +858,7 @@ rb_queue_initialize(int argc, VALUE *argv, VALUE self) https://github.com/ruby/ruby/blob/trunk/thread_sync.c#L858
     list_head_init(queue_waitq(q));
     rb_scan_args(argc, argv, "01", &initial);
     if (argc == 1) {
-        rb_ary_concat(q->que, rb_convert_type(initial, T_ARRAY, "Array", "to_ary"));
+        rb_ary_concat(q->que, rb_to_array(initial));
     }
     return self;
 }
-- 
cgit v1.1


--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/