From 74aa11b3f2261ccd40f1d1498a3b416f25a65aa4 Mon Sep 17 00:00:00 2001
From: Takeshi KOMIYA <i.tkomiya@gmail.com>
Date: Thu, 12 Sep 2024 03:35:37 +0900
Subject: [PATCH] stdlib: Add types for CSV::Table#each

---
 stdlib/csv/0/csv.rbs          |  6 +++---
 test/stdlib/CSV_Table_test.rb | 24 ++++++++++++++++++++++++
 2 files changed, 27 insertions(+), 3 deletions(-)
 create mode 100644 test/stdlib/CSV_Table_test.rb

diff --git a/stdlib/csv/0/csv.rbs b/stdlib/csv/0/csv.rbs
index 63ede413e..07bda9ad9 100644
--- a/stdlib/csv/0/csv.rbs
+++ b/stdlib/csv/0/csv.rbs
@@ -3002,7 +3002,7 @@ end
 #     table['Name'] # => ["Foo", "Bar", "Baz"]
 #
 class CSV::Table[out Elem] < Object
-  include Enumerable[untyped]
+  include Enumerable[Elem]
   extend Forwardable
 
   # <!--
@@ -3633,8 +3633,8 @@ class CSV::Table[out Elem] < Object
   # Returns a new Enumerator if no block is given:
   #     table.each # => #<Enumerator: #<CSV::Table mode:col row_count:4>:each>
   #
-  def each: () -> Enumerator[untyped, self]
-          | () { (untyped) -> void } -> self
+  def each: () -> Enumerator[Elem, self]
+          | () { (Elem) -> void } -> self
           | () { (*untyped) -> void } -> self
 
   def empty?: (*untyped args) { (*untyped) -> untyped } -> untyped
diff --git a/test/stdlib/CSV_Table_test.rb b/test/stdlib/CSV_Table_test.rb
new file mode 100644
index 000000000..c36d412bf
--- /dev/null
+++ b/test/stdlib/CSV_Table_test.rb
@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative "test_helper"
+require "csv"
+
+class CSV::TableInstanceTest < Test::Unit::TestCase
+  include TestHelper
+
+  library 'csv'
+  testing "CSV::Table[CSV::Row]"
+
+  def test_each
+    table = CSV::Table.new([
+      CSV::Row.new([], []),
+      CSV::Row.new([], []),
+      CSV::Row.new([], [])
+    ])
+
+    assert_send_type "() -> Enumerator[CSV::Row, void]",
+                     table, :each
+    assert_send_type "() { (CSV::Row) -> void } -> CSV::Table",
+                     table, :each do end
+  end
+end