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
|
# frozen_string_literal: true
require 'test/unit'
require 'socket'
class TestSocketOption < Test::Unit::TestCase
def test_new
data = [1].pack("i")
sockopt = Socket::Option.new(:INET, :SOCKET, :KEEPALIVE, data)
assert_equal(Socket::AF_INET, sockopt.family)
assert_equal(Socket::SOL_SOCKET, sockopt.level)
assert_equal(Socket::SO_KEEPALIVE, sockopt.optname)
assert_equal(Socket::SO_KEEPALIVE, sockopt.optname)
assert_equal(data, sockopt.data)
end
def test_bool
opt = Socket::Option.bool(:INET, :SOCKET, :KEEPALIVE, true)
assert_equal(1, opt.int)
opt = Socket::Option.bool(:INET, :SOCKET, :KEEPALIVE, false)
assert_equal(0, opt.int)
opt = Socket::Option.int(:INET, :SOCKET, :KEEPALIVE, 0)
assert_equal(false, opt.bool)
opt = Socket::Option.int(:INET, :SOCKET, :KEEPALIVE, 1)
assert_equal(true, opt.bool)
opt = Socket::Option.int(:INET, :SOCKET, :KEEPALIVE, 2)
assert_equal(true, opt.bool)
begin
Socket.open(:INET, :STREAM) {|s|
s.setsockopt(Socket::Option.bool(:INET, :SOCKET, :KEEPALIVE, true))
assert_equal(true, s.getsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE).bool)
s.setsockopt(Socket::Option.bool(:INET, :SOCKET, :KEEPALIVE, false))
assert_equal(false, s.getsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE).bool)
}
rescue TypeError
if /aix/ =~ RUBY_PLATFORM
skip "Known bug in getsockopt(2) on AIX"
end
raise $!
end
end
def test_ipv4_multicast_loop
sockopt = Socket::Option.ipv4_multicast_loop(128)
assert_equal('#<Socket::Option: INET IP MULTICAST_LOOP 128>', sockopt.inspect)
assert_equal(Socket::AF_INET, sockopt.family)
assert_equal(Socket::IPPROTO_IP, sockopt.level)
assert_equal(Socket::IP_MULTICAST_LOOP, sockopt.optname)
assert_equal(128, sockopt.ipv4_multicast_loop)
end
def test_ipv4_multicast_loop_size
expected_size = Socket.open(:INET, :DGRAM) {|s|
s.getsockopt(:IP, :MULTICAST_LOOP).to_s.bytesize
}
assert_equal(expected_size, Socket::Option.ipv4_multicast_loop(0).to_s.bytesize)
end
def test_ipv4_multicast_ttl
sockopt = Socket::Option.ipv4_multicast_ttl(128)
assert_equal('#<Socket::Option: INET IP MULTICAST_TTL 128>', sockopt.inspect)
assert_equal(Socket::AF_INET, sockopt.family)
assert_equal(Socket::IPPROTO_IP, sockopt.level)
assert_equal(Socket::IP_MULTICAST_TTL, sockopt.optname)
assert_equal(128, sockopt.ipv4_multicast_ttl)
end
def test_ipv4_multicast_ttl_size
expected_size = Socket.open(:INET, :DGRAM) {|s|
s.getsockopt(:IP, :MULTICAST_TTL).to_s.bytesize
}
assert_equal(expected_size, Socket::Option.ipv4_multicast_ttl(0).to_s.bytesize)
end
def test_unpack
sockopt = Socket::Option.new(:INET, :SOCKET, :KEEPALIVE, [1].pack("i"))
assert_equal([1], sockopt.unpack("i"))
assert_equal([1], sockopt.data.unpack("i"))
end
end
|