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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
require File.expand_path('../spec_helper', __FILE__)
describe Coolio::TCPSocket do
let :loop do
Coolio::Loop.new
end
before :each do
@echo = TCPServer.new("127.0.0.1", 0)
@host = @echo.addr[3]
@port = @echo.addr[1]
@running = true
@echo_thread = Thread.new do
socks = [@echo]
begin
serv socks
ensure
socks.each do |s|
s.close
end
end
Thread.pass
end
end
def serv(socks)
while @running
selected = select(socks, [], [], 0.1)
next if selected.nil?
selected[0].each do |s|
if s == @echo
socks.push s.accept
next
end
begin
unless s.eof?
s.write(s.read_nonblock 1)
end
rescue SystemCallError, EOFError, IOError, SocketError
end
end
end
end
def shutdown
if @running
@running = false
@echo_thread.join
end
end
after :each do
shutdown
end
context "#close" do
it "detaches all watchers on #close before loop#run" do
client = Coolio::TCPSocket.connect(@host, @port)
loop.attach client
client.close
expect(loop.watchers.size).to eq 0
end
end
context "#on_connect" do
class OnConnect < Cool.io::TCPSocket
attr :connected
def on_connect
@connected = true
end
end
it "connected client called on_connect" do
begin
c = OnConnect.connect(@host, @port)
loop.attach c
loop.run_once
expect(c.connected).to eq true
ensure
c.close
end
end
end
context "#on_connect_failed" do
class OnConnectFailed < Cool.io::TCPSocket
attr :connect_failed
def on_connect_failed
@connect_failed = true
end
end
it "try to connect dead host" do
serv = TCPServer.new(0)
dead_host = serv.addr[3]
dead_port = serv.addr[1]
serv.close
c = OnConnectFailed.connect(dead_host, dead_port)
loop.attach c
loop.run_once # on_connect_failed
expect(c.connect_failed).to eq true
end
end
context "#on_close" do
class Closed < StandardError; end
class OnClose < Cool.io::TCPSocket
def on_close
raise Closed
end
end
let :client do
OnClose.connect(@host, @port)
end
before :each do
loop.attach client
loop.run_once # on_connect
client.write "0"
end
it "disconnect from client" do
expect { client.close }.to raise_error(Closed)
end
it "disconnect from server" do
shutdown
expect { loop.run }.to raise_error(Closed)
end
end
context "#on_read" do
class Finished < StandardError; end
class OnRead < Cool.io::TCPSocket
attr :read_data, :times
def on_connect
@read_data = ""
@times = 0
end
def on_read(data)
@read_data += data
@times += 1
if @times < 5
write "#{@times}"
else
close
raise Finished
end
end
end
it "receive 5 times" do
c = OnRead.connect(@host, @port)
loop.attach c
loop.run_once # on_connect
c.write "0"
expect { loop.run }.to raise_error(Finished)
expect(c.times).to eq 5
expect(c.read_data).to eq "01234"
end
end
context "#on_write_complete" do
class WriteComplete < StandardError; end
class OnWriteComplete < Cool.io::TCPSocket
attr :called
def on_write_complete
@called = true
close
raise WriteComplete
end
end
it "on_write_complete is called" do
c = OnWriteComplete.connect(@host, @port)
loop.attach c
loop.run_once # on_connect
c.write "aaa"
expect { loop.run }.to raise_error(WriteComplete)
end
end
end
|
