aboutsummaryrefslogtreecommitdiffstats
path: root/spec/rubyspec/library/socket/basicsocket/recv_spec.rb
blob: 5891bf9c877f1a4e15a0aaf1d3af01aeb19411cb (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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
# -*- encoding: binary -*-
require File.expand_path('../../../../spec_helper', __FILE__)
require File.expand_path('../../fixtures/classes', __FILE__)

describe "BasicSocket#recv" do

  before :each do
    @server = TCPServer.new('127.0.0.1', 0)
    @port = @server.addr[1]
  end

  after :each do
    @server.close unless @server.closed?
    ScratchPad.clear
  end

  it "receives a specified number of bytes of a message from another socket"  do
    t = Thread.new do
      client = @server.accept
      ScratchPad.record client.recv(10)
      client.recv(1) # this recv is important
      client.close
    end
    Thread.pass while t.status and t.status != "sleep"
    t.status.should_not be_nil

    socket = TCPSocket.new('127.0.0.1', @port)
    socket.send('hello', 0)
    socket.close

    t.join
    ScratchPad.recorded.should == 'hello'
  end

  platform_is_not :solaris do
    it "accepts flags to specify unusual receiving behaviour" do
      t = Thread.new do
        client = @server.accept

        # in-band data (TCP), doesn't receive the flag.
        ScratchPad.record client.recv(10)

        # this recv is important (TODO: explain)
        client.recv(10)
        client.close
      end
      Thread.pass while t.status and t.status != "sleep"
      t.status.should_not be_nil

      socket = TCPSocket.new('127.0.0.1', @port)
      socket.send('helloU', Socket::MSG_OOB)
      socket.shutdown(1)
      t.join
      socket.close
      ScratchPad.recorded.should == 'hello'
    end
  end

  it "gets lines delimited with a custom separator"  do
    t = Thread.new do
      client = @server.accept
      ScratchPad.record client.gets("\377")

      # this call is important (TODO: explain)
      client.gets(nil)
      client.close
    end
    Thread.pass while t.status and t.status != "sleep"
    t.status.should_not be_nil

    socket = TCPSocket.new('127.0.0.1', @port)
    socket.write("firstline\377secondline\377")
    socket.close

    t.join
    ScratchPad.recorded.should == "firstline\377"
  end

  ruby_version_is "2.3" do
    it "allows an output buffer as third argument" do
      socket = TCPSocket.new('127.0.0.1', @port)
      socket.write("data")

      client = @server.accept
      buf = "foo"
      begin
        client.recv(4, 0, buf)
      ensure
        client.close
      end
      buf.should == "data"

      socket.close
    end
  end
end