1 files changed, 32 insertions, 0 deletions
diff --git a/spec/ruby/shared/enumerator/with_index.rb b/spec/ruby/shared/enumerator/with_index.rb
new file mode 100644
index 0000000000..37048a7b52
--- /dev/null
+++ b/spec/ruby/shared/enumerator/with_index.rb
@@ -0,0 +1,32 @@
+require File.expand_path('../../../spec_helper', __FILE__)
+
+describe :enum_with_index, shared: true do
+
+  require File.expand_path('../../../fixtures/enumerator/classes', __FILE__)
+
+  before :each do
+    @enum = [1, 2, 3, 4].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
+    [1, 2, 3, 4].should == @enum.send(@method) { |o, i| :glark }
+  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 be_an_instance_of(Enumerator)
+    ewi.to_a.should == [[1, 0], [2, 1], [3, 2], [4, 3]]
+  end
+end