зеркало из https://github.com/github/ruby.git
117 строки
2.6 KiB
Ruby
117 строки
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
begin
|
|
require "socket"
|
|
require "test/unit"
|
|
rescue LoadError
|
|
end
|
|
|
|
|
|
class TestSocket_UDPSocket < Test::Unit::TestCase
|
|
def test_open
|
|
assert_nothing_raised { UDPSocket.open {} }
|
|
assert_nothing_raised { UDPSocket.open(Socket::AF_INET) {} }
|
|
assert_nothing_raised { UDPSocket.open("AF_INET") {} }
|
|
assert_nothing_raised { UDPSocket.open(:AF_INET) {} }
|
|
end
|
|
|
|
def test_inspect
|
|
UDPSocket.open() {|sock|
|
|
assert_match(/AF_INET\b/, sock.inspect)
|
|
}
|
|
if Socket.const_defined?(:AF_INET6)
|
|
begin
|
|
UDPSocket.open(Socket::AF_INET6) {|sock|
|
|
assert_match(/AF_INET6\b/, sock.inspect)
|
|
}
|
|
rescue Errno::EAFNOSUPPORT
|
|
skip 'AF_INET6 not supported by kernel'
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_connect
|
|
s = UDPSocket.new
|
|
host = Object.new
|
|
class << host; self end.send(:define_method, :to_str) {
|
|
s.close
|
|
"127.0.0.1"
|
|
}
|
|
assert_raise(IOError, "[ruby-dev:25045]") {
|
|
s.connect(host, 1)
|
|
}
|
|
end
|
|
|
|
def test_bind
|
|
s = UDPSocket.new
|
|
host = Object.new
|
|
class << host; self end.send(:define_method, :to_str) {
|
|
s.close
|
|
"127.0.0.1"
|
|
}
|
|
assert_raise(IOError, "[ruby-dev:25057]") {
|
|
s.bind(host, 2000)
|
|
}
|
|
ensure
|
|
s.close if s && !s.closed?
|
|
end
|
|
|
|
def test_bind_addrinuse
|
|
host = "127.0.0.1"
|
|
|
|
in_use = UDPSocket.new
|
|
in_use.bind(host, 0)
|
|
port = in_use.addr[1]
|
|
|
|
s = UDPSocket.new
|
|
|
|
e = assert_raise(Errno::EADDRINUSE) do
|
|
s.bind(host, port)
|
|
end
|
|
|
|
assert_match "bind(2) for \"#{host}\" port #{port}", e.message
|
|
ensure
|
|
in_use.close if in_use
|
|
s.close if s
|
|
end
|
|
|
|
def test_send_too_long
|
|
u = UDPSocket.new
|
|
|
|
e = assert_raise(Errno::EMSGSIZE) do
|
|
u.send "\0" * 100_000, 0, "127.0.0.1", 7 # echo
|
|
end
|
|
|
|
assert_match 'for "127.0.0.1" port 7', e.message
|
|
ensure
|
|
u.close if u
|
|
end
|
|
|
|
def test_bind_no_memory_leak
|
|
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
|
|
s = UDPSocket.new
|
|
s.close
|
|
end;
|
|
100_000.times {begin s.bind("127.0.0.1", 1) rescue IOError; end}
|
|
end;
|
|
end
|
|
|
|
def test_connect_no_memory_leak
|
|
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
|
|
s = UDPSocket.new
|
|
s.close
|
|
end;
|
|
100_000.times {begin s.connect("127.0.0.1", 1) rescue IOError; end}
|
|
end;
|
|
end
|
|
|
|
def test_send_no_memory_leak
|
|
assert_no_memory_leak(["-rsocket"], <<-"end;", <<-"end;", rss: true)
|
|
s = UDPSocket.new
|
|
s.close
|
|
end;
|
|
100_000.times {begin s.send("\0"*100, 0, "127.0.0.1", 1) rescue IOError; end}
|
|
end;
|
|
end
|
|
end if defined?(UDPSocket)
|