* lib/csv.rb: Enhance each() to support Enumerator.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@31740 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

ChangeLog

@@ -1,7 +1,11 @@
+Thu May 27 00:34:07 2011  James Edward Gray II  <jeg2@ruby-lang.org>
+
+    * lib/csv.rb: Enhance each() to support Enumerator.
+
 Thu May 26 10:32:11 2011  James Edward Gray II  <jeg2@ruby-lang.org>
 
-	* lib/csv.rb: Documentation improvements from Ysiad Ferreiras.
-	  [Ruby 1.9 - Bug #4785]
+    * lib/csv.rb: Documentation improvements from Ysiad Ferreiras.
+      [Ruby 1.9 - Bug #4785]
 
 Thu May 26 15:42:02 2011  Cezary Baginski <cezary.baginski@gmail.com>
 

lib/csv.rb

@@ -1787,8 +1787,12 @@ class CSV
   # The data source must be open for reading.
   #
   def each
-    while row = shift
-      yield row
+    if block_given?
+      while row = shift
+        yield row
+      end
+    else
+      to_enum
     end
   end
 

test/csv/test_interface.rb

@@ -112,6 +112,14 @@ class TestCSV::Interface < TestCSV
       assert_equal(nil, csv.shift)
     end
   end
+  
+  def test_enumerators_are_supported
+    CSV.open(@path, col_sep: "\t", row_sep: "\r\n") do |csv|
+      enum = csv.each
+      assert_instance_of(Enumerator, enum)
+      assert_equal(@expected.shift, enum.next)
+    end
+  end
 
   ### Test Write Interface ###