1 files changed, 105 insertions, 0 deletions
diff --git a/spec/ruby/library/stringio/shared/each.rb b/spec/ruby/library/stringio/shared/each.rb
new file mode 100644
index 0000000000..0fde23634e
--- /dev/null
+++ b/spec/ruby/library/stringio/shared/each.rb
@@ -0,0 +1,105 @@
+describe :stringio_each_separator, shared: true do
+  before :each do
+    @io = StringIO.new("a b c d e\n1 2 3 4 5")
+  end
+
+  it "uses the passed argument as the line separator" do
+    seen = []
+    @io.send(@method, " ") {|s| seen << s}
+    seen.should == ["a ", "b ", "c ", "d ", "e\n1 ", "2 ", "3 ", "4 ", "5"]
+  end
+
+  it "does not change $_" do
+    $_ = "test"
+    @io.send(@method, " ") { |s| s}
+    $_.should == "test"
+  end
+
+  it "returns self" do
+    @io.send(@method) {|l| l }.should equal(@io)
+  end
+
+  it "tries to convert the passed separator to a String using #to_str" do
+    obj = mock("to_str")
+    obj.stub!(:to_str).and_return(" ")
+
+    seen = []
+    @io.send(@method, obj) { |l| seen << l }
+    seen.should == ["a ", "b ", "c ", "d ", "e\n1 ", "2 ", "3 ", "4 ", "5"]
+  end
+
+  it "yields self's content starting from the current position when the passed separator is nil" do
+    seen = []
+    io = StringIO.new("1 2 1 2 1 2")
+    io.pos = 2
+    io.send(@method, nil) {|s| seen << s}
+    seen.should == ["2 1 2 1 2"]
+  end
+
+  it "yields each paragraph when passed an empty String as separator" do
+    seen = []
+    io = StringIO.new("para1\n\npara2\n\n\npara3")
+    io.send(@method, "") {|s| seen << s}
+    seen.should == ["para1\n\n", "para2\n\n", "para3"]
+  end
+end
+
+describe :stringio_each_no_arguments, shared: true do
+  before :each do
+    @io = StringIO.new("a b c d e\n1 2 3 4 5")
+  end
+
+  it "yields each line to the passed block" do
+    seen = []
+    @io.send(@method) {|s| seen << s }
+    seen.should == ["a b c d e\n", "1 2 3 4 5"]
+  end
+
+  it "yields each line starting from the current position" do
+    seen = []
+    @io.pos = 4
+    @io.send(@method) {|s| seen << s }
+    seen.should == ["c d e\n", "1 2 3 4 5"]
+  end
+
+  it "does not change $_" do
+    $_ = "test"
+    @io.send(@method) { |s| s}
+    $_.should == "test"
+  end
+
+  it "uses $/ as the default line separator" do
+    seen = []
+    begin
+      old_rs, $/ = $/, " "
+      @io.send(@method) {|s| seen << s }
+      seen.should eql(["a ", "b ", "c ", "d ", "e\n1 ", "2 ", "3 ", "4 ", "5"])
+    ensure
+      $/ = old_rs
+    end
+  end
+
+  it "returns self" do
+    @io.send(@method) {|l| l }.should equal(@io)
+  end
+
+  it "returns an Enumerator when passed no block" do
+    enum = @io.send(@method)
+    enum.instance_of?(Enumerator).should be_true
+
+    seen = []
+    enum.each { |b| seen << b }
+    seen.should == ["a b c d e\n", "1 2 3 4 5"]
+  end
+end
+
+describe :stringio_each_not_readable, shared: true do
+  it "raises an IOError" do
+    io = StringIO.new("a b c d e", "w")
+    lambda { io.send(@method) { |b| b } }.should raise_error(IOError)
+
+    io = StringIO.new("a b c d e")
+    io.close_read
+    lambda { io.send(@method) { |b| b } }.should raise_error(IOError)
+  end
+end