# -*- encoding: binary -*- require 'test/unit' require 'tempfile' require 'raindrops' require 'socket' require 'pp' $stderr.sync = $stdout.sync = true class TestTCP_Info < Test::Unit::TestCase TEST_ADDR = ENV['UNICORN_TEST_ADDR'] || '127.0.0.1' # Linux kernel commit 5ee3afba88f5a79d0bff07ddd87af45919259f91 TCP_INFO_useful_listenq = `uname -r`.strip >= '2.6.24' def test_tcp_server_unacked return if RUBY_PLATFORM !~ /linux/ # unacked not implemented on others... s = TCPServer.new(TEST_ADDR, 0) rv = Raindrops::TCP_Info.new s c = TCPSocket.new TEST_ADDR, s.addr[1] tmp = Raindrops::TCP_Info.new s TCP_INFO_useful_listenq and assert_equal 1, tmp.unacked assert_equal 0, rv.unacked a = s.accept tmp = Raindrops::TCP_Info.new s assert_equal 0, tmp.unacked before = tmp.object_id tmp.get!(s) assert_equal before, tmp.object_id ensure [ c, a, s ].compact.each(&:close) end def test_accessors s = TCPServer.new TEST_ADDR, 0 tmp = Raindrops::TCP_Info.new s tcp_info_methods = tmp.methods - Object.new.methods assert tcp_info_methods.size >= 32 tcp_info_methods.each do |m| next if m.to_sym == :get! next if ! tmp.respond_to?(m) val = tmp.__send__ m assert_kind_of Integer, val assert val >= 0 end assert tmp.respond_to?(:state), 'every OS knows about TCP state, right?' ensure s.close end def test_tcp_server_delayed delay = 0.010 delay_ms = (delay * 1000).to_i s = TCPServer.new(TEST_ADDR, 0) c = TCPSocket.new TEST_ADDR, s.addr[1] c.syswrite "." sleep(delay * 1.2) a = s.accept i = Raindrops::TCP_Info.new(a) assert i.last_data_recv >= delay_ms, "#{i.last_data_recv} < #{delay_ms}" ensure c.close if c a.close if a s.close end def test_tcp_server_state_closed s = TCPServer.new(TEST_ADDR, 0) c = TCPSocket.new(TEST_ADDR, s.addr[1]) i = Raindrops::TCP_Info.allocate a = s.accept i.get!(a) state = i.state if Raindrops.const_defined?(:TCP) assert_equal state, Raindrops::TCP[:ESTABLISHED] end c = c.close sleep(0.01) # wait for kernel to update state i.get!(a) assert_not_equal state, i.state ensure s.close if s c.close if c a.close if a end end if defined? Raindrops::TCP_Info