blob: c85bf49ad3cc1aed0dd43c96c6799367d5cdc2e2 (
plain)
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
|
class TimeoutAction
def initialize(timeout)
@timeout = timeout
@queue = Queue.new
@started = now
end
def register
MSpec.register :start, self
MSpec.register :before, self
MSpec.register :finish, self
end
private def now
Process.clock_gettime(Process::CLOCK_MONOTONIC)
end
private def fetch_item
@queue.pop(true)
rescue ThreadError
nil
end
def start
@thread = Thread.new do
loop do
if action = fetch_item
action.call
else
wakeup_at = @started + @timeout
left = wakeup_at - now
sleep left if left > 0
Thread.pass # Let the main thread run
if @queue.empty?
elapsed = now - @started
if elapsed > @timeout
STDERR.puts "\n#{@current_state.description}"
STDERR.flush
abort "Example took longer than the configured timeout of #{@timeout}s"
end
end
end
end
end
end
def before(state = nil)
time = now
@queue << -> do
@current_state = state
@started = time
end
end
def finish
@thread.kill
@thread.join
end
end
|