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
| | # frozen_string_literal: false
begin
require 'aggregate'
have_aggregate = true
rescue LoadError => e
warn "W: #{e} skipping #{__FILE__}"
end
require 'test/unit'
require 'raindrops'
require 'io/wait'
class TestLastDataRecv < Test::Unit::TestCase
def setup
Raindrops::Aggregate::LastDataRecv.default_aggregate = []
end
def teardown
Raindrops::Aggregate::LastDataRecv.default_aggregate = nil
end
def test_accept_nonblock_agg
s = Socket.new(:INET, :STREAM, 0)
s.listen(128)
addr = s.connect_address
s.extend(Raindrops::Aggregate::LastDataRecv)
s.raindrops_aggregate = []
c = Socket.new(:INET, :STREAM, 0)
c.connect(addr)
c.write '.' # for TCP_DEFER_ACCEPT
client, ai = s.accept_nonblock(exception: false)
assert client.kind_of?(Socket)
assert ai.kind_of?(Addrinfo)
assert_equal 1, s.raindrops_aggregate.size
assert s.raindrops_aggregate[0].instance_of?(Integer)
client, ai = s.accept_nonblock(exception: false)
assert_equal :wait_readable, client
assert_nil ai
assert_equal 1, s.raindrops_aggregate.size
assert_raise(IO::WaitReadable) { s.accept_nonblock }
end
def test_accept_nonblock_one
s = TCPServer.new('127.0.0.1', 0)
s.extend(Raindrops::Aggregate::LastDataRecv)
s.raindrops_aggregate = []
addr = s.addr
c = TCPSocket.new(addr[3], addr[1])
c.write '.' # for TCP_DEFER_ACCEPT
client = s.accept_nonblock(exception: false)
assert client.kind_of?(TCPSocket)
assert_equal 1, s.raindrops_aggregate.size
assert s.raindrops_aggregate[0].instance_of?(Integer)
client = s.accept_nonblock(exception: false)
assert_equal :wait_readable, client
assert_equal 1, s.raindrops_aggregate.size
assert_raise(IO::WaitReadable) { s.accept_nonblock }
end
end if RUBY_PLATFORM =~ /linux/ && have_aggregate
|