class Tryouts::Drill::Sergeant::Benchmark

def initialize(reps=nil)
  @reps = (1..1000000).include?(reps) ? reps : 5
  @stats = Tryouts::Stats.new
end

def run(block, context, &inline)
  # A Proc object takes precedence over an inline block. 
  runtime = (block.nil? ? inline : block)
  response = Tryouts::Drill::Reality.new
  
  if runtime.nil?
    raise "We need a block to benchmark"
  else
    begin
      
      @reps.times do
        run = ::Benchmark.realtime &runtime
        @stats.sample run
      end
      
      # We add the output after we run the block so that
      # that it'll remain nil if an exception was raised
      response.output = @stats
      
    rescue => e
      puts e.message, e.backtrace if Tryouts.verbose > 2
      response.etype = e.class
      response.error = e.message
      response.trace = e.backtrace
    end
  end
  response
end