class Tryouts::Drill

def add_dream(d); @dreams << d; end

def add_dreams(*d); @dreams += d; end

def initialize(name, dtype, *args, &drill)
  @name, @dtype, @drill = name, dtype, drill
  @dreams = []
  if @dtype == :cli
    @sergeant = Tryouts::Drill::Sergeant::CLI.new *args
  elsif @dtype == :api
    default_output = drill.nil? ? args.shift : nil
    @sergeant = Tryouts::Drill::Sergeant::API.new default_output
    @dreams << Tryouts::Drill::Dream.new(*args) unless args.empty?
  else
    raise NoSergeant, "Weird drill sergeant: #{@dtype}"
  end
  # For CLI drills, a block takes precedence over inline args. 
  # A block will contain multiple shell commands (see Rye::Box#batch)
  drill_args = [] if dtype == :cli && drill.is_a?(Proc)
  @reality = Tryouts::Drill::Reality.new
end

def run(context=nil)
  begin
    print Tryouts::DRILL_MSG % @name
    @reality = @sergeant.run @drill, context
    # Store the stash from the drill block
    @reality.stash = context.stash if context.respond_to? :stash
    # If the drill block returned true we assume success if there's no dream
    if @dreams.empty? && @reality.output == true
      @dreams << Tryouts::Drill::Dream.new
      @dreams.first.output = true
    end
  rescue => ex
    @reality.ecode, @reality.etype = -2, ex.class
    @reality.error, @reality.trace = ex.message, ex.backtrace
  end  
  self.success?
end

def success?
  @dreams.each { |d| return false unless d == @reality }
  true
end