adfoster-r7
117 lines
2.7 KiB
Ruby
117 lines
2.7 KiB
Ruby
require 'rex/stopwatch'
|
|
|
|
module Msf
|
|
module ModuleTest
|
|
attr_accessor :tests
|
|
attr_accessor :failures
|
|
attr_accessor :skipped
|
|
|
|
class SkipTestError < ::Exception
|
|
end
|
|
|
|
def initialize(info = {})
|
|
@tests = 0
|
|
@failures = 0
|
|
@skipped = 0
|
|
super
|
|
end
|
|
|
|
def run_all_tests
|
|
tests = self.methods.select { |m| m.to_s =~ /^test_/ }
|
|
tests.each { |test_method|
|
|
begin
|
|
self.send(test_method)
|
|
rescue SkipTestError => e
|
|
# If the entire def is skipped, increment tests and skip count
|
|
@tests += 1
|
|
@skipped += 1
|
|
print_status("SKIPPED: def #{test_method} (#{e.message})")
|
|
end
|
|
}
|
|
end
|
|
|
|
def skip(msg = "No reason given")
|
|
raise SkipTestError, msg
|
|
end
|
|
|
|
def it(msg = "", &block)
|
|
@current_it_msg = msg
|
|
@tests += 1
|
|
begin
|
|
result = block.call
|
|
unless result
|
|
@failures += 1
|
|
print_error("FAILED: #{error}") if error
|
|
@current_it_msg = nil
|
|
print_error("FAILED: #{msg}")
|
|
return
|
|
end
|
|
rescue SkipTestError => e
|
|
@skipped += 1
|
|
@current_it_msg = nil
|
|
print_status("SKIPPED: #{msg} (#{e.message})")
|
|
rescue ::Exception => e
|
|
@failures += 1
|
|
print_error("FAILED: #{msg}")
|
|
print_error("Exception: #{e.class}: #{e}")
|
|
dlog("Exception in testing - #{msg}")
|
|
dlog("Call stack: #{e.backtrace.join("\n")}")
|
|
return
|
|
ensure
|
|
@current_it_msg = nil
|
|
end
|
|
|
|
print_good("#{msg}")
|
|
end
|
|
|
|
def pending(msg = "", &block)
|
|
print_status("PENDING: #{msg}")
|
|
end
|
|
|
|
# @return [Integer] The number of tests that have passed
|
|
def passed
|
|
@tests - @failures
|
|
end
|
|
|
|
# When printing to console, additionally prepend the current test name
|
|
[
|
|
:print,
|
|
:print_line,
|
|
:print_status,
|
|
:print_good,
|
|
|
|
:print_warning,
|
|
:print_error,
|
|
:print_bad,
|
|
].each do |method|
|
|
define_method(method) do |msg|
|
|
super(@current_it_msg ? "[#{@current_it_msg}] #{msg}" : msg)
|
|
end
|
|
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}")
|
|
|
|
@tests = 0
|
|
@failures = 0
|
|
@skipped = 0
|
|
|
|
_res, elapsed_time = Rex::Stopwatch.elapsed_time do
|
|
run_all_tests
|
|
end
|
|
|
|
vprint_status("Testing complete in #{elapsed_time.round(2)} seconds")
|
|
status = "Passed: #{passed}; Failed: #{@failures}; Skipped: #{@skipped}"
|
|
if @failures > 0
|
|
print_error(status)
|
|
else
|
|
print_status(status)
|
|
end
|
|
end
|
|
end
|
|
end
|