test/test_tcp_server.rb

   1 require 'test/unit'
   2 require 'io/nonblock'
   3 $-w = true
   4 require 'kgio'
   5
   6 class TestKgioTCPServer < Test::Unit::TestCase
   7
   8   def setup
   9     @host = ENV["TEST_HOST"] || '127.0.0.1'
  10     @srv = Kgio::TCPServer.new(@host, 0)
  11     @port = @srv.addr[1]
  12   end
  13
  14   def teardown
  15     @srv.close unless @srv.closed?
  16     Kgio.accept_cloexec = true
  17     Kgio.accept_nonblock = false
  18   end
  19
  20   def test_accept
  21     a = TCPSocket.new(@host, @port)
  22     b = @srv.kgio_accept
  23     assert_kind_of Kgio::Socket, b
  24     assert_equal @host, b.kgio_addr
  25   end
  26
  27   def test_accept_nonblock
  28     @srv.nonblock = true
  29     assert_equal nil, @srv.kgio_accept
  30   end
  31 end