aboutsummaryrefslogtreecommitdiffstats
path: root/spec/rubyspec/core/argf/read_nonblock_spec.rb
blob: 8176a206e5d404550a4ef76f147159c7a81f135b (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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
require File.expand_path('../../../spec_helper', __FILE__)
require File.expand_path('../shared/read', __FILE__)

platform_is_not :windows do
  describe 'ARGF.read_nonblock' do
    it_behaves_like :argf_read, :read_nonblock

    before do
      @file1_name = fixture(__FILE__, 'file1.txt')
      @file2_name = fixture(__FILE__, 'file2.txt')

      @file1 = File.read(@file1_name)
      @file2 = File.read(@file2_name)

      @chunk1 = File.read(@file1_name, 4)
      @chunk2 = File.read(@file2_name, 4)
    end

    it 'reads up to the given amount of bytes' do
      argf [@file1_name] do
        @argf.read_nonblock(4).should == @chunk1
      end
    end

    describe 'when using multiple files' do
      it 'reads up to the given amount of bytes from the first file' do
        argf [@file1_name, @file2_name] do
          @argf.read_nonblock(4).should == @chunk1
        end
      end

      it 'returns an empty String when reading after having read the first file in its entirety' do
        argf [@file1_name, @file2_name] do
          @argf.read_nonblock(File.size(@file1_name)).should == @file1
          @argf.read_nonblock(4).should == ''
        end
      end
    end

    it 'reads up to the given bytes from STDIN' do
      stdin = ruby_exe('print ARGF.read_nonblock(4)', :args => "< #{@file1_name}")

      stdin.should == @chunk1
    end

    it 'reads up to the given bytes from a file when a file and STDIN are present' do
      stdin = ruby_exe("print ARGF.read_nonblock(4)", :args => "#{@file1_name} - < #{@file2_name}")

      stdin.should == @chunk1
    end

    context "with STDIN" do
      before do
        @r, @w = IO.pipe
        @stdin = $stdin
        $stdin = @r
      end

      after do
        $stdin = @stdin
        @w.close
        @r.close unless @r.closed?
      end

      it 'raises IO::EAGAINWaitReadable when empty' do
        argf ['-'] do
          lambda {
            @argf.read_nonblock(4)
          }.should raise_error(IO::EAGAINWaitReadable)
        end
      end

      ruby_version_is "2.3" do
        it 'returns :wait_readable when the :exception is set to false' do
          argf ['-'] do
            @argf.read_nonblock(4, nil, exception: false).should == :wait_readable
          end
        end
      end
    end
  end
end