require 'assert/result'
require 'assert/result_set'

module Assert
  class Test

    # a Test is some code/method to run in the scope of a Context.  After a
    # a test runs, it should have some assertions which are its results.

    attr_reader :name, :code, :context_class
    attr_accessor :results

    def initialize(name, context_class, code = nil, &block)
      @context_class = context_class
      @name = name_from_context(name)
      @code = (code || block)
      @results = ResultSet.new
    end

    def run(view=nil)
      @results.view = view
      begin
        run_scope = @context_class.new(self)
        run_setup(run_scope)
        if @code.kind_of?(::Proc)
          run_scope.instance_eval(&@code)
        elsif run_scope.respond_to?(@code.to_s)
          run_scope.send(@code.to_s)
        end
      rescue Result::TestSkipped => err
        @results << Result::Skip.new(self.name, err)
      rescue Exception => err
        @results << Result::Error.new(self.name, err)
      ensure
        begin
          run_teardown(run_scope) if run_scope
        rescue Exception => teardown_err
          @results << Result::Error.new(self.name, teardown_err)
        end
      end
      @results.view = nil
      @results
    end

    def run_setup(scope)
      @context_class.all_setup_blocks.each do |setup|
        scope.instance_eval(&setup)
      end
    end

    def run_teardown(scope)
      @context_class.all_teardown_blocks.each do |teardown|
        scope.instance_eval(&teardown)
      end
    end

    Assert::Result.types.each do |name, klass|
      define_method "#{name}_results" do
        @results.select{|r| r.kind_of?(klass) }
      end
    end

    def result_count(type=nil)
      if Assert::Result.types.include?(type)
        self.send("#{type}_results").size
      else
        @results.size
      end
    end

    def <=>(other_test)
      self.name <=> other_test.name
    end

    def inspect
      attributes_string = ([ :name, :context_class, :results ].collect do |attr|
        "@#{attr}=#{self.send(attr).inspect}"
      end).join(" ")
      "#<#{self.class} #{attributes_string}>"
    end

    protected

    def name_from_context(name)
      name = name.gsub(/^test:\s+should/, "should")
      [ @context_class.full_description, name ].compact.join(" ")
    end

  end
end