1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/spec/ruby/library/socket/ipsocket/recvfrom_spec.rb
eregon b53cf149ad Update to ruby/spec@9be7c7e
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@64180 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2018-08-03 16:19:40 +00:00

123 lines
2.8 KiB
Ruby

require_relative '../spec_helper'
require_relative '../fixtures/classes'
describe "Socket::IPSocket#recvfrom" do
before :each do
@server = TCPServer.new("127.0.0.1", 0)
@port = @server.addr[1]
@client = TCPSocket.new("127.0.0.1", @port)
end
after :each do
@server.close unless @server.closed?
@client.close unless @client.closed?
end
it "reads data from the connection" do
data = nil
t = Thread.new do
client = @server.accept
begin
data = client.recvfrom(6)
ensure
client.close
end
end
@client.send('hello', 0)
@client.shutdown rescue nil
# shutdown may raise Errno::ENOTCONN when sent data is pending.
t.join
data.first.should == 'hello'
end
it "reads up to len bytes" do
data = nil
t = Thread.new do
client = @server.accept
begin
data = client.recvfrom(3)
ensure
client.close
end
end
@client.send('hello', 0)
@client.shutdown rescue nil
t.join
data.first.should == 'hel'
end
it "returns an array with the data and connection info" do
data = nil
t = Thread.new do
client = @server.accept
data = client.recvfrom(3)
client.close
end
@client.send('hello', 0)
@client.shutdown rescue nil
t.join
data.size.should == 2
data.first.should == "hel"
# This does not apply to every platform, dependant on recvfrom(2)
# data.last.should == nil
end
end
describe 'Socket::IPSocket#recvfrom' do
SocketSpecs.each_ip_protocol do |family, ip_address, family_name|
before do
@server = UDPSocket.new(family)
@client = UDPSocket.new(family)
@server.bind(ip_address, 0)
@client.connect(ip_address, @server.connect_address.ip_port)
@hostname = Socket.getaddrinfo(ip_address, nil)[0][2]
end
after do
@client.close
@server.close
end
it 'returns an Array containing up to N bytes and address information' do
@client.write('hello')
port = @client.local_address.ip_port
ret = @server.recvfrom(2)
ret.should == ['he', [family_name, port, @hostname, ip_address]]
end
it 'allows specifying of flags when receiving data' do
@client.write('hello')
@server.recvfrom(2, Socket::MSG_PEEK)[0].should == 'he'
@server.recvfrom(2)[0].should == 'he'
end
describe 'using reverse lookups' do
before do
@server.do_not_reverse_lookup = false
@hostname = Socket.getaddrinfo(ip_address, nil, 0, 0, 0, 0, true)[0][2]
end
it 'includes the hostname in the address Array' do
@client.write('hello')
port = @client.local_address.ip_port
ret = @server.recvfrom(2)
ret.should == ['he', [family_name, port, @hostname, ip_address]]
end
end
end
end