mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
c46824d094
I suspect that sometimes on CI the last thread is prempted before eaching the exit hook causing the test to flake. I can't find a good way to force it to run.
78 lines
2.3 KiB
Ruby
78 lines
2.3 KiB
Ruby
# frozen_string_literal: false
|
|
require 'envutil'
|
|
|
|
class TestThreadInstrumentation < Test::Unit::TestCase
|
|
def setup
|
|
pend("TODO: No windows support yet") if /mswin|mingw|bccwin/ =~ RUBY_PLATFORM
|
|
end
|
|
|
|
THREADS_COUNT = 3
|
|
|
|
def test_thread_instrumentation
|
|
require '-test-/thread/instrumentation'
|
|
Bug::ThreadInstrumentation.reset_counters
|
|
Bug::ThreadInstrumentation::register_callback
|
|
|
|
begin
|
|
threads = threaded_cpu_work
|
|
assert_equal [false] * THREADS_COUNT, threads.map(&:status)
|
|
counters = Bug::ThreadInstrumentation.counters
|
|
counters.each do |c|
|
|
assert_predicate c, :nonzero?, "Call counters: #{counters.inspect}"
|
|
end
|
|
|
|
assert_equal THREADS_COUNT, counters.first
|
|
assert_in_delta THREADS_COUNT, counters.last, 1 # It's possible that a thread didn't execute its EXIT hook yet.
|
|
ensure
|
|
Bug::ThreadInstrumentation::unregister_callback
|
|
end
|
|
end
|
|
|
|
def test_thread_instrumentation_fork_safe
|
|
skip "No fork()" unless Process.respond_to?(:fork)
|
|
|
|
require '-test-/thread/instrumentation'
|
|
Bug::ThreadInstrumentation::register_callback
|
|
|
|
read_pipe, write_pipe = IO.pipe
|
|
|
|
begin
|
|
pid = fork do
|
|
Bug::ThreadInstrumentation.reset_counters
|
|
threads = threaded_cpu_work
|
|
write_pipe.write(Marshal.dump(threads.map(&:status)))
|
|
write_pipe.write(Marshal.dump(Bug::ThreadInstrumentation.counters))
|
|
write_pipe.close
|
|
exit!(0)
|
|
end
|
|
write_pipe.close
|
|
_, status = Process.wait2(pid)
|
|
assert_predicate status, :success?
|
|
|
|
thread_statuses = Marshal.load(read_pipe)
|
|
assert_equal [false] * THREADS_COUNT, thread_statuses
|
|
|
|
counters = Marshal.load(read_pipe)
|
|
read_pipe.close
|
|
counters.each do |c|
|
|
assert_predicate c, :nonzero?, "Call counters: #{counters.inspect}"
|
|
end
|
|
|
|
assert_equal THREADS_COUNT, counters.first
|
|
assert_in_delta THREADS_COUNT, counters.last, 1 # It's possible that a thread didn't execute its EXIT hook yet.
|
|
ensure
|
|
Bug::ThreadInstrumentation::unregister_callback
|
|
end
|
|
end
|
|
|
|
def test_thread_instrumentation_unregister
|
|
require '-test-/thread/instrumentation'
|
|
assert Bug::ThreadInstrumentation::register_and_unregister_callbacks
|
|
end
|
|
|
|
private
|
|
|
|
def threaded_cpu_work
|
|
THREADS_COUNT.times.map { Thread.new { 100.times { |i| i + i } } }.each(&:join)
|
|
end
|
|
end
|