require 'test/unit'
require 'io/nonblock'
$-w = true
require 'kgio'

class TestAcceptClass < Test::Unit::TestCase
  class FooSocket < Kgio::Socket
  end

  def setup
    assert_equal Kgio::Socket, Kgio.accept_class
  end

  def teardown
    assert_nothing_raised { Kgio.accept_class = nil }
    assert_equal Kgio::Socket, Kgio.accept_class
  end

  def test_tcp_socket
    assert_nothing_raised { Kgio.accept_class = Kgio::TCPSocket }
    assert_equal Kgio::TCPSocket, Kgio.accept_class
  end

  def test_invalid
    assert_raises(TypeError) { Kgio.accept_class = TCPSocket }
    assert_equal Kgio::Socket, Kgio.accept_class
  end

  def test_accepted_class
    @host = ENV["TEST_HOST"] || '127.0.0.1'
    @srv = Kgio::TCPServer.new(@host, 0)
    @port = @srv.addr[1]

    assert_nothing_raised { Kgio.accept_class = Kgio::TCPSocket }
    client = TCPSocket.new(@host, @port)
    assert_instance_of Kgio::TCPSocket, @srv.kgio_accept
    client = TCPSocket.new(@host, @port)
    IO.select([@srv])
    assert_instance_of Kgio::TCPSocket, @srv.kgio_tryaccept

    assert_nothing_raised { Kgio.accept_class = nil }
    client = TCPSocket.new(@host, @port)
    assert_instance_of Kgio::Socket, @srv.kgio_accept
    client = TCPSocket.new(@host, @port)
    IO.select([@srv])
    assert_instance_of Kgio::Socket, @srv.kgio_tryaccept

    assert_nothing_raised { Kgio.accept_class = Kgio::UNIXSocket }
    client = TCPSocket.new(@host, @port)
    assert_instance_of Kgio::UNIXSocket, @srv.kgio_accept
    client = TCPSocket.new(@host, @port)
    IO.select([@srv])
    assert_instance_of Kgio::UNIXSocket, @srv.kgio_tryaccept

    client = TCPSocket.new(@host, @port)
    assert_instance_of FooSocket, @srv.kgio_accept(FooSocket)

    client = TCPSocket.new(@host, @port)
    IO.select([@srv])
    assert_instance_of FooSocket, @srv.kgio_tryaccept(FooSocket)
  end
end
