# File activesupport/lib/active_support/testing/parallelization/server.rb, line 14
def initialize
  @queue = Queue.new
  @active_workers = Concurrent::Map.new
  @worker_pids = Concurrent::Map.new
  @in_flight = Concurrent::Map.new
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 32
def <<(o)
  o[2] = DRbObject.new(o[2]) if o
  @queue << o
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 64
def active_workers?
  @active_workers.size > 0
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 68
def interrupt
  @queue.clear
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 37
def pop
  if test = @queue.pop
    @in_flight[[test[0].to_s, test[1]]] = test
    test
  end
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 21
def record(reporter, result)
  raise DRb::DRbConnError if result.is_a?(DRb::DRbUnknown)

  @in_flight.delete([result.klass, result.name])

  reporter.synchronize do
    reporter.prerecord(PrerecordResultClass.new(result.klass), result.name)
    reporter.record(result)
  end
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 54
def remove_dead_workers(dead_pids)
  dead_pids.each do |dead_pid|
    worker_id = @worker_pids.key(dead_pid)
    if worker_id
      @active_workers.delete(worker_id)
      @worker_pids.delete(worker_id)
    end
  end
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 72
def shutdown
  # Wait for initial queue to drain
  while @queue.length != 0
    sleep 0.1
  end

  @queue.close

  wait_for_active_workers

  @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
rescue Interrupt
  warn "Interrupted. Exiting..."

  @queue.close

  wait_for_active_workers
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 44
def start_worker(worker_id, worker_pid)
  @active_workers[worker_id] = true
  @worker_pids[worker_id] = worker_pid
end

# File activesupport/lib/active_support/testing/parallelization/server.rb, line 49
def stop_worker(worker_id, worker_pid)
  @active_workers.delete(worker_id)
  @worker_pids.delete(worker_id)
end