dnrops
/
metasploit-framework
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add support for only running user specified test names

This commit is contained in:
adfoster-r7 2023-06-16 10:53:04 +01:00
parent 50832bef97
commit 92c15f8a6a
No known key found for this signature in database
GPG Key ID: 3BD4FA3818818F04
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/lib/module_test.rb
View File

@ -14,12 +14,22 @@ module Msf
@failures = 0
@skipped = 0
super
register_options(
[
OptString.new("TestName", [false, "Run a specified test method name.", nil]),
], self.class
)
end
def run_all_tests
tests = self.methods.select { |m| m.to_s =~ /^test_/ }
tests = datastore['TestName'].present? ? [datastore['TestName'].to_sym] : self.methods.select { |m| m.to_s =~ /^test_/ }
tests.each { |test_method|
begin
unless respond_to?(test_method)
print_error("test method #{test_method} not found")
next
end
self.send(test_method)
rescue SkipTestError => e
# If the entire def is skipped, increment tests and skip count
@ -50,6 +60,7 @@ module Msf
@skipped += 1
@current_it_msg = nil
print_status("SKIPPED: #{msg} (#{e.message})")
return
rescue ::Exception => e
@failures += 1
print_error("FAILED: #{msg}")