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
| | 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
Kgio.accept_class = nil
assert_equal Kgio::Socket, Kgio.accept_class
end
def test_tcp_socket
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]
socks = []
Kgio.accept_class = Kgio::TCPSocket
socks << TCPSocket.new(@host, @port)
assert_instance_of Kgio::TCPSocket, @srv.kgio_accept
socks << TCPSocket.new(@host, @port)
IO.select([@srv])
assert_instance_of Kgio::TCPSocket, @srv.kgio_tryaccept
Kgio.accept_class = nil
socks << TCPSocket.new(@host, @port)
assert_instance_of Kgio::Socket, @srv.kgio_accept
socks << TCPSocket.new(@host, @port)
IO.select([@srv])
assert_instance_of Kgio::Socket, @srv.kgio_tryaccept
Kgio.accept_class = Kgio::UNIXSocket
socks << TCPSocket.new(@host, @port)
assert_instance_of Kgio::UNIXSocket, @srv.kgio_accept
socks << TCPSocket.new(@host, @port)
IO.select([@srv])
assert_instance_of Kgio::UNIXSocket, @srv.kgio_tryaccept
socks << TCPSocket.new(@host, @port)
assert_instance_of FooSocket, @srv.kgio_accept(FooSocket)
socks << TCPSocket.new(@host, @port)
IO.select([@srv])
assert_instance_of FooSocket, @srv.kgio_tryaccept(FooSocket)
socks.each(&:close)
end
end
|