summaryrefslogtreecommitdiff
path: root/spec/mspec/lib/mspec/runner/actions/timeout.rb
blob: 543b7366d7af2110ce2a69c0f6e4da7732a811fb (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
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
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 :after, 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
              if @current_state
                STDERR.puts "\nExample took longer than the configured timeout of #{@timeout}s:"
                STDERR.puts "#{@current_state.description}"
              else
                STDERR.puts "\nSome code outside an example took longer than the configured timeout of #{@timeout}s"
              end
              STDERR.flush

              show_backtraces
              exit 2
            end
          end
        end
      end
    end
  end

  def before(state = nil)
    time = now
    @queue << -> do
      @current_state = state
      @started = time
    end
  end

  def after(state = nil)
    @queue << -> do
      @current_state = nil
    end
  end

  def finish
    @thread.kill
    @thread.join
  end

  private def show_backtraces
    if RUBY_ENGINE == 'truffleruby'
      STDERR.puts 'Java stacktraces:'
      Process.kill :SIGQUIT, Process.pid
      sleep 1
    end

    STDERR.puts "\nRuby backtraces:"
    if defined?(Truffle::Debug.show_backtraces)
      Truffle::Debug.show_backtraces
    else
      Thread.list.each do |thread|
        unless thread == Thread.current
          STDERR.puts thread.inspect, thread.backtrace, ''
        end
      end
    end
  end
end