summaryrefslogtreecommitdiff
path: root/spec/ruby/core/argf/seek_spec.rb
blob: c983c0cb5a99bee23927f0e45ab2912c5f5fa5a9 (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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
require_relative '../../spec_helper'

describe "ARGF.seek" 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

  it "sets the absolute position relative to beginning of file" do
    argf [@file1_name, @file2_name] do
      @argf.seek 2
      @argf.gets.should == @file1.first[2..-1]
      @argf.seek @file1.first.size
      @argf.gets.should == @file1.last
      @argf.seek 0, IO::SEEK_END
      @argf.gets.should == @file2.first
    end
  end

  it "sets the position relative to current position in file" do
    argf [@file1_name, @file2_name] do
      @argf.seek(0, IO::SEEK_CUR)
      @argf.gets.should == @file1.first
      @argf.seek(-@file1.first.size+2, IO::SEEK_CUR)
      @argf.gets.should == @file1.first[2..-1]
      @argf.seek(1, IO::SEEK_CUR)
      @argf.gets.should == @file1.last[1..-1]
      @argf.seek(3, IO::SEEK_CUR)
      @argf.gets.should == @file2.first
      @argf.seek(@file1.last.size, IO::SEEK_CUR)
      @argf.gets.should == nil
    end
  end

  it "sets the absolute position relative to end of file" do
    argf [@file1_name, @file2_name] do
      @argf.seek(-@file1.first.size-@file1.last.size, IO::SEEK_END)
      @argf.gets.should == @file1.first
      @argf.seek(-6, IO::SEEK_END)
      @argf.gets.should == @file1.last[-6..-1]
      @argf.seek(-4, IO::SEEK_END)
      @argf.gets.should == @file1.last[4..-1]
      @argf.gets.should == @file2.first
      @argf.seek(-6, IO::SEEK_END)
      @argf.gets.should == @file2.last[-6..-1]
    end
  end
end

describe "ARGF.seek" do
  before :each do
    @file1_name = fixture __FILE__, "file1.txt"
  end

  it "takes at least one argument (offset)" do
    argf [@file1_name] do
      lambda { @argf.seek }.should raise_error(ArgumentError)
    end
  end
end