summaryrefslogtreecommitdiff
path: root/spec/ruby/library/socket/basicsocket/read_nonblock_spec.rb
blob: ea5e65da5c93f78ffef393383f4d80ad279863ec (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
require_relative '../spec_helper'
require_relative '../fixtures/classes'

describe "BasicSocket#read_nonblock" do
  SocketSpecs.each_ip_protocol do |family, ip_address|
    before :each do
      @r = Socket.new(family, :DGRAM)
      @w = Socket.new(family, :DGRAM)

      @r.bind(Socket.pack_sockaddr_in(0, ip_address))
      @w.send("aaa", 0, @r.getsockname)
    end

    after :each do
      @r.close unless @r.closed?
      @w.close unless @w.closed?
    end

    it "receives data after it's ready" do
      IO.select([@r], nil, nil, 2)
      @r.read_nonblock(5).should == "aaa"
    end

    platform_is_not :windows do
      it 'returned data is binary encoded regardless of the external encoding' do
        IO.select([@r], nil, nil, 2)
        @r.read_nonblock(1).encoding.should == Encoding::BINARY

        @w.send("bbb", 0, @r.getsockname)
        @r.set_encoding(Encoding::ISO_8859_1)
        IO.select([@r], nil, nil, 2)
        buffer = @r.read_nonblock(3)
        buffer.should == "bbb"
        buffer.encoding.should == Encoding::BINARY
      end
    end

    it 'replaces the content of the provided buffer without changing its encoding' do
      buffer = "initial data".dup.force_encoding(Encoding::UTF_8)

      IO.select([@r], nil, nil, 2)
      @r.read_nonblock(3, buffer)
      buffer.should == "aaa"
      buffer.encoding.should == Encoding::UTF_8

      @w.send("bbb", 0, @r.getsockname)
      @r.set_encoding(Encoding::ISO_8859_1)
      IO.select([@r], nil, nil, 2)
      @r.read_nonblock(3, buffer)
      buffer.should == "bbb"
      buffer.encoding.should == Encoding::UTF_8
    end

    platform_is :linux do
      it 'does not set the IO in nonblock mode' do
        require 'io/nonblock'
        @r.nonblock = false
        IO.select([@r], nil, nil, 2)
        @r.read_nonblock(3).should == "aaa"
        @r.should_not.nonblock?
      end
    end

    platform_is_not :linux, :windows do
      it 'sets the IO in nonblock mode' do
        require 'io/nonblock'
        @r.nonblock = false
        IO.select([@r], nil, nil, 2)
        @r.read_nonblock(3).should == "aaa"
        @r.should.nonblock?
      end
    end
  end
end