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
|
# test/prompt_regexp_test.rb
require File.expand_path('../test_helper', __FILE__)
require 'socket'
describe "Reading from socket client" do
before do
@fake_port = 11301
@tube_name = 'tube.to.test'
@fake_server = Thread.start do
server = TCPServer.new(@fake_port)
loop do
IO.select([server])
client = server.accept_nonblock
while line = client.gets
case line
when /list-tubes-watched/i
client.print "OK #{7+@tube_name.size}\r\n---\n- #{@tube_name}\n\r\n"
when /watch #{@tube_name}/i
client.print "WATCHING 1\r\n"
when /reserve/i
client.print "RESERVED 17 25\r\n"
client.print "[first part]"
# Emulate network delay
sleep 0.5
client.print "[second part]\r\n"
else
client.print "ERROR\r\n"
end
end
end
end
slept = 0
while @beanstalk.nil?
begin
@beanstalk = Beaneater.new("localhost:#{@fake_port}")
rescue Beaneater::NotConnected
raise 'Could not connect to fake beanstalkd server' if slept > 1
sleep 0.1
slept += 0.1
end
end
end
it 'should reserve job with full body' do
job = @beanstalk.tubes[@tube_name].reserve
assert_equal '[first part][second part]', job.body
end
after do
@beanstalk.close
@fake_server.kill
end
end
|
