class ActiveSupport::Testing::Parallelization

def self.after_fork_hook(&blk)
  @@after_fork_hooks << blk
end

def self.run_cleanup_hook(&blk)
  @@run_cleanup_hooks << blk
end

def <<(work)
  @queue_server << work
end

def initialize(worker_count)
  @worker_count = worker_count
  @queue_server = Server.new
  @worker_pool = []
  @url = DRb.start_service("drbunix:", @queue_server).uri
end

def shutdown
  @queue_server.shutdown
  @worker_pool.each { |pid| Process.waitpid pid }
end

def size
  @worker_count
end

def start
  @worker_pool = @worker_count.times.map do |worker|
    Worker.new(worker, @url).start
  end
end