metasploit-gs/test/lib/module_test.rb

module Msf
  module ModuleTest
    attr_accessor :tests
    attr_accessor :failures

    class SkipTestError < ::Exception
    end

    def initialize(info = {})
      @tests = 0
      @failures = 0
      super
    end

    def run_all_tests
      tests = self.methods.select { |m| m.to_s =~ /^test_/ }
      tests.each { |test_method|
        self.send(test_method)
      }
    end

    def skip(msg = "No reason given")
      raise SkipTestError, msg
    end

    def it(msg = "", &block)
      @tests += 1
      begin
        result = block.call
        unless result
          print_error("FAILED: #{msg}")
          print_error("FAILED: #{error}") if error
          @failures += 1
          return
        end
      rescue SkipTestError => e
        print_status("SKIPPED: #{msg} (#{e.message})")
      rescue ::Exception => e
        print_error("FAILED: #{msg}")
        print_error("Exception: #{e.class}: #{e}")
        dlog("Exception in testing - #{msg}")
        dlog("Call stack: #{e.backtrace.join("\n")}")
        return
      end

      print_good("#{msg}")
    end

    def pending(msg = "", &block)
      print_status("PENDING: #{msg}")
    end
  end

  module ModuleTest::PostTest
    include ModuleTest
    def run
      print_status("Running against session #{datastore["SESSION"]}")
      print_status("Session type is #{session.type} and platform is #{session.platform}")

      t = Time.now
      @tests = 0; @failures = 0
      run_all_tests

      vprint_status("Testing complete in #{Time.now - t}")
      if (@failures > 0)
        print_error("Passed: #{@tests - @failures}; Failed: #{@failures}")
      else
        print_status("Passed: #{@tests - @failures}; Failed: #{@failures}")
      end
    end
  end
end