Add type for Enumerator::Chain

Author: ParadoxV5

core/enumerator.rbs

@@ -613,7 +613,22 @@ end
 # This type of objects can be created by Enumerable#chain and Enumerator#+.
 #
 class Enumerator::Chain[out Elem] < Enumerator[Elem, void]
-  include Enumerable[Elem]
+  # * `Rubocop: `self` type is not allowed in this context
+  # * {Enumerator::Chain#each} without block doesn't return `self`, unlike {Enumerator#each}.
+  #include Enumerator::_Each[Enum, self] # Workaround: def each:
+
+  # <!--
+  #   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 +641,10 @@ class Enumerator::Chain[out Elem] < Enumerator[Elem, void]
   #
   # If no block is given, returns an enumerator.
   #
-  def each: () { (Elem) -> void } -> void
+  def each: () { (Enum) -> void } -> self
+          | () -> Enumerator[E, self]
+
+  # wrong argument type chain (expected enumerator) (TypeError)
+  def with_index: (?Integer) ?{ (?) -> untyped } -> bot
+  def each_with_index: () ?{ (?) -> untyped } -> bot
 end

test/stdlib/Enumerator_test.rb

@@ -88,3 +88,22 @@ def test_to_proc
     end.next
   end
 end
+
+class EnumeratorChainTest < Test::Unit::TestCase
+  include TestHelper
+
+  testing "::Enumerator::Chain[::Integer]"
+
+  def test_class_new
+    assert_send_type "(*_Each[Integer] enums) -> Enumerator::Chain[Integer]",
+                     Enumerator::Chain, :new, 1..3, [4, 5]
+  end
+
+  def test_each
+    enum = Enumerator::Chain.new 1..3, [4, 5]
+    assert_send_type "() { (Integer) -> nil } -> Enumerator::Chain[Integer]",
+                     enum, :each do end
+    assert_send_type "() -> Enumerator[Integer, Enumerator::Chain[Integer]]",
+                     enum, :each
+  end
+end