2 files changed, 79 insertions, 0 deletions
diff --git a/spec/ruby/core/enumerator/shared/each.rb b/spec/ruby/core/enumerator/shared/each.rb
new file mode 100644
index 0000000000..18ca773207
--- /dev/null
+++ b/spec/ruby/core/enumerator/shared/each.rb
@@ -0,0 +1,46 @@
+# #each passes source-yielded values to the block by ordinary block arity
+# (rb_yield_values2 semantics in CRuby), unlike the Enumerable collection methods
+# which pack them via rb_enum_values_pack() (see enumerable/shared/value_packing.rb).
+describe :enum_each, shared: true do
+  # @object must be set to a Proc that wraps an Enumerator into the receiver
+  # under test (e.g. -> e { e } for Enumerator#each, -> e { e.lazy } for Lazy#each).
+  describe "with a source that yields multiple values" do
+    before :each do
+      @enum = @object.call(Enumerator.new { |y| y.yield 1, 2; y.yield 3, 4 })
+    end
+
+    it "yields the first value to a single-argument block" do
+      collected = []
+      @enum.each { |x| collected << x }
+      collected.should == [1, 3]
+    end
+
+    it "yields each value to a multi-argument block" do
+      collected = []
+      @enum.each { |x, y| collected << [x, y] }
+      collected.should == [[1, 2], [3, 4]]
+    end
+
+    it "gathers the values for a splat block" do
+      collected = []
+      @enum.each { |*args| collected << args }
+      collected.should == [[1, 2], [3, 4]]
+    end
+  end
+
+  describe "with a source that yields a single value" do
+    it "yields the value to a single-argument block" do
+      collected = []
+      @object.call(Enumerator.new { |y| y.yield 7; y.yield 8 }).each { |x| collected << x }
+      collected.should == [7, 8]
+    end
+  end
+
+  describe "with a source that yields no value" do
+    it "yields nil to a single-argument block" do
+      collected = []
+      @object.call(Enumerator.new { |y| y.yield; y.yield }).each { |x| collected << x }
+      collected.should == [nil, nil]
+    end
+  end
+end
diff --git a/spec/ruby/core/enumerator/shared/with_index.rb b/spec/ruby/core/enumerator/shared/with_index.rb
new file mode 100644
index 0000000000..0992397e95
--- /dev/null
+++ b/spec/ruby/core/enumerator/shared/with_index.rb
@@ -0,0 +1,33 @@
+require_relative '../../../spec_helper'
+
+describe :enum_with_index, shared: true do
+
+  require_relative '../fixtures/classes'
+
+  before :each do
+    @origin = [1, 2, 3, 4]
+    @enum = @origin.to_enum
+  end
+
+  it "passes each element and its index to block" do
+    a = []
+    @enum.send(@method) { |o, i| a << [o, i] }
+    a.should == [[1, 0], [2, 1], [3, 2], [4, 3]]
+  end
+
+  it "returns the object being enumerated when given a block" do
+    @enum.send(@method) { |o, i| :glark }.should.equal?(@origin)
+  end
+
+  it "binds splat arguments properly" do
+    acc = []
+    @enum.send(@method) { |*b| c,d = b; acc << c; acc << d }
+    [1, 0, 2, 1, 3, 2, 4, 3].should == acc
+  end
+
+  it "returns an enumerator if no block is supplied" do
+    ewi = @enum.send(@method)
+    ewi.should.instance_of?(Enumerator)
+    ewi.to_a.should == [[1, 0], [2, 1], [3, 2], [4, 3]]
+  end
+end