Contents

module Riot
  class Assertion < RunnableBlock
    class << self
      def macros; @@macros ||= {}; end

      def register_macro(name, assertion_macro, expect_exception=false)
        macros[name.to_s] = assertion_macro.new
      end
    end

    def initialize(description, &definition)
      super
      @expectings, @expectation_block = [], false, nil
      @macro = AssertionMacro.default
    end

    def run(situation)
      @expectings << situation.evaluate(&@expectation_block) if @expectation_block
      actual = situation.evaluate(&definition)
      @macro.evaluate((@macro.expects_exception? ? nil : actual), *@expectings)
    rescue Exception => e
      @macro.expects_exception? ? @macro.evaluate(e, *@expectings) : @macro.error(e)
    end
  private
    def enhance_with_macro(name, *expectings, &expectation_block)
      @macro, @expectings, @expectation_block = self.class.macros[name.to_s], expectings, expectation_block
      raise(NoMethodError, name) unless @macro
      self
    end
    alias :method_missing :enhance_with_macro
  end # Assertion
end # Riot

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
riot-0.10.12.pre	lib/riot/assertion.rb