Merge pull request #2220 from ParadoxV5/enumerator-chain

Add type for `Enumerator::Chain`

Author: soutaro

core/enumerator.rbs

@@ -613,7 +613,18 @@ end
 # This type of objects can be created by Enumerable#chain and Enumerator#+.
 #
 class Enumerator::Chain[out Elem] < Enumerator[Elem, void]
-  include Enumerable[Elem]
+  # <!--
+  #   rdoc-file=enumerator.c
+  #   - Enumerator::Chain.new(*enums) -> enum
+  # -->
+  # Generates a new enumerator object that iterates over the elements of given
+  # enumerable objects in sequence.
+  #
+  #     e = Enumerator::Chain.new(1..3, [4, 5])
+  #     e.to_a #=> [1, 2, 3, 4, 5]
+  #     e.size #=> 5
+  #
+  def initialize: (*_Each[Elem] enums) -> void
 
   # <!--
   #   rdoc-file=enumerator.c
@@ -626,5 +637,6 @@ class Enumerator::Chain[out Elem] < Enumerator[Elem, void]
   #
   # If no block is given, returns an enumerator.
   #
-  def each: () { (Elem) -> void } -> void
+  def each: () { (Elem) -> void } -> self
+          | () -> Enumerator[Elem, self]
 end

test/stdlib/Enumerator_test.rb

@@ -88,3 +88,26 @@ def test_to_proc
     end.next
   end
 end
+
+class EnumeratorChainInstanceTest < Test::Unit::TestCase
+  include TestHelper
+
+  testing "::Enumerator::Chain[::Integer]"
+
+  def test_each
+    enum = Enumerator::Chain.new 1..3, [4, 5]
+    assert_send_type "() { (Integer) -> nil } -> Enumerator::Chain[Integer]",
+                     enum, :each do end
+  end
+end
+
+class EnumeratorChainSingletonTest < Test::Unit::TestCase
+  include TestHelper
+
+  testing "singleton(::Enumerator::Chain)"
+
+  def test_class_new
+    assert_send_type "(Range[Integer], Array[Integer]) -> Enumerator::Chain[Integer]",
+                     Enumerator::Chain, :new, 1..3, [4, 5]
+  end
+end