summaryrefslogtreecommitdiff
path: root/spec/ruby/core/argf/rewind_spec.rb
blob: fcb2e81a3e24e676e12b265f0640f8894cf24989 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
require_relative '../../spec_helper'

describe "ARGF.rewind" do
  before :each do
    @file1_name = fixture __FILE__, "file1.txt"
    @file2_name = fixture __FILE__, "file2.txt"

    @file1 = File.readlines @file1_name
    @file2 = File.readlines @file2_name
  end

  # NOTE: this test assumes that fixtures files have two lines each
  it "goes back to beginning of current file" do
    argf [@file1_name, @file2_name] do
      @argf.gets
      @argf.rewind
      @argf.gets.should == @file1.first

      @argf.gets # finish reading file1

      @argf.gets
      @argf.rewind
      @argf.gets.should == @file2.first
    end
  end

  it "resets ARGF.lineno to 0" do
    script = fixture __FILE__, "rewind.rb"
    out = ruby_exe(script, args: [@file1_name, @file2_name])
    out.should == "0\n1\n0\n"
  end

  it "raises an ArgumentError when end of stream reached" do
    argf [@file1_name, @file2_name] do
      @argf.read
      lambda { @argf.rewind }.should raise_error(ArgumentError)
    end
  end
end