From 50d11882e19126be7b520e24c04d3238070b43f6 Mon Sep 17 00:00:00 2001
From: akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Wed, 19 Aug 2009 17:32:56 +0000
Subject: * enumerator.c (next_init): don't clear feedvalue.

git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@24588 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/ruby/test_enumerator.rb | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

(limited to 'test/ruby')

diff --git a/test/ruby/test_enumerator.rb b/test/ruby/test_enumerator.rb
index 333dac79c..e70a783a0 100644
--- a/test/ruby/test_enumerator.rb
+++ b/test/ruby/test_enumerator.rb
@@ -266,6 +266,40 @@ class TestEnumerator < Test::Unit::TestCase
     assert_raise(TypeError) { e.feed 2 }
   end
 
+  def test_feed_before_first_next
+    o = Object.new
+    def o.each(ary)
+      ary << yield
+      ary << yield
+      ary << yield
+    end
+    ary = []
+    e = o.to_enum(:each, ary)
+    e.feed 1
+    e.next
+    e.next
+    assert_equal([1], ary)
+  end
+
+  def test_rewind_clear_feed
+    o = Object.new
+    def o.each(ary)
+      ary << yield
+      ary << yield
+      ary << yield
+    end
+    ary = []
+    e = o.to_enum(:each, ary)
+    e.next
+    e.feed 1
+    e.next
+    e.feed 2
+    e.rewind
+    e.next
+    e.next
+    assert_equal([1,nil], ary)
+  end
+
   def test_feed_yielder
     x = nil
     e = Enumerator.new {|y| x = y.yield; 10 }
-- 
cgit