Sha256: 28b65076acffe88aa37e92c4f00804cd93a5fcacd274a9346057889e421f3133
Contents?: true
Size: 1.8 KB
Versions: 42
Compression:
Stored size: 1.8 KB
Contents
# frozen_string_literal: true
require "drb"
require "drb/unix" unless Gem.win_platform?
module ActiveSupport
module Testing
class Parallelization # :nodoc:
class Server
include DRb::DRbUndumped
def initialize
@queue = Queue.new
@active_workers = Concurrent::Map.new
@in_flight = Concurrent::Map.new
end
def record(reporter, result)
raise DRb::DRbConnError if result.is_a?(DRb::DRbUnknown)
@in_flight.delete([result.klass, result.name])
reporter.synchronize do
reporter.record(result)
end
end
def <<(o)
o[2] = DRbObject.new(o[2]) if o
@queue << o
end
def pop
if test = @queue.pop
@in_flight[[test[0].to_s, test[1]]] = test
test
end
end
def start_worker(worker_id)
@active_workers[worker_id] = true
end
def stop_worker(worker_id)
@active_workers.delete(worker_id)
end
def active_workers?
@active_workers.size > 0
end
def shutdown
# Wait for initial queue to drain
while @queue.length != 0
sleep 0.1
end
@queue.close
# Wait until all workers have finished
while active_workers?
sleep 0.1
end
@in_flight.values.each do |(klass, name, reporter)|
result = Minitest::Result.from(klass.new(name))
error = RuntimeError.new("result not reported")
error.set_backtrace([""])
result.failures << Minitest::UnexpectedError.new(error)
reporter.synchronize do
reporter.record(result)
end
end
end
end
end
end
end
Version data entries
42 entries across 42 versions & 7 rubygems