98 lines
2.4 KiB
Ruby
98 lines
2.4 KiB
Ruby
begin
|
|
require "socket"
|
|
require "test/unit"
|
|
rescue LoadError
|
|
end
|
|
|
|
class TestBasicSocket < Test::Unit::TestCase
|
|
def inet_stream
|
|
sock = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
|
|
yield sock
|
|
ensure
|
|
assert_raise(IOError) {sock.close}
|
|
end
|
|
|
|
def test_getsockopt
|
|
inet_stream do |s|
|
|
n = s.getsockopt(Socket::SOL_SOCKET, Socket::SO_TYPE)
|
|
assert_equal([Socket::SOCK_STREAM].pack("i"), n)
|
|
n = s.getsockopt(Socket::SOL_SOCKET, Socket::SO_ERROR)
|
|
assert_equal([0].pack("i"), n)
|
|
val = Object.new
|
|
class << val; self end.send!(:define_method, :to_int) {
|
|
s.close
|
|
Socket::SO_TYPE
|
|
}
|
|
assert_raise(IOError) {
|
|
n = s.getsockopt(Socket::SOL_SOCKET, val)
|
|
}
|
|
end
|
|
end
|
|
|
|
def test_setsockopt # [ruby-dev:25039]
|
|
s = nil
|
|
linger = [0, 0].pack("ii")
|
|
|
|
val = Object.new
|
|
class << val; self end.send!(:define_method, :to_str) {
|
|
s.close
|
|
linger
|
|
}
|
|
inet_stream do |s|
|
|
assert_equal(0, s.setsockopt(Socket::SOL_SOCKET, Socket::SO_LINGER, linger))
|
|
|
|
assert_raise(IOError) {
|
|
s.setsockopt(Socket::SOL_SOCKET, Socket::SO_LINGER, val)
|
|
}
|
|
end
|
|
|
|
val = Object.new
|
|
class << val; self end.send!(:define_method, :to_int) {
|
|
s.close
|
|
Socket::SO_LINGER
|
|
}
|
|
inet_stream do |s|
|
|
assert_raise(IOError) {
|
|
s.setsockopt(Socket::SOL_SOCKET, val, linger)
|
|
}
|
|
end
|
|
end
|
|
|
|
def test_listen
|
|
s = nil
|
|
log = Object.new
|
|
class << log; self end.send!(:define_method, :to_int) {
|
|
s.close
|
|
2
|
|
}
|
|
inet_stream do |s|
|
|
assert_raise(IOError) {
|
|
s.listen(log)
|
|
}
|
|
end
|
|
end
|
|
end if defined?(Socket)
|
|
|
|
class TestSocket < Test::Unit::TestCase
|
|
def test_unpack_sockaddr
|
|
sockaddr_in = Socket.sockaddr_in(80, "")
|
|
assert_raise(ArgumentError) { Socket.unpack_sockaddr_un(sockaddr_in) }
|
|
sockaddr_un = Socket.sockaddr_un("/tmp/s")
|
|
assert_raise(ArgumentError) { Socket.unpack_sockaddr_in(sockaddr_un) }
|
|
end if Socket.respond_to?(:sockaddr_un)
|
|
|
|
def test_sysaccept
|
|
serv = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
|
|
serv.bind(Socket.sockaddr_in(0, "127.0.0.1"))
|
|
serv.listen 5
|
|
c = Socket.new(Socket::AF_INET, Socket::SOCK_STREAM, 0)
|
|
c.connect(serv.getsockname)
|
|
fd, peeraddr = serv.sysaccept
|
|
assert_equal(c.getsockname, peeraddr)
|
|
ensure
|
|
serv.close if serv
|
|
c.close if c
|
|
IO.for_fd(fd).close if fd
|
|
end
|
|
end if defined?(Socket)
|