feat(only_logs): New logging flag to only show execution logs (#1708)

This commit is contained in:
Pepe Fagoaga
2023-01-17 10:13:09 +01:00
committed by GitHub
parent 0cbe80d2ab
commit e7796268b5
11 changed files with 160 additions and 76 deletions

View File

@@ -7,7 +7,7 @@ from prowler.lib.cli.parser import ProwlerArgumentParser
prowler_command = "prowler"
class Test_Outputs:
class Test_Parser:
# Init parser
def setup_method(self):
self.parser = ProwlerArgumentParser()
@@ -25,9 +25,10 @@ class Test_Outputs:
assert not parsed.output_filename
assert "output" in parsed.output_directory
assert not parsed.verbose
assert parsed.no_banner
assert not parsed.no_banner
assert parsed.log_level == "CRITICAL"
assert not parsed.log_file
assert not parsed.only_logs
assert not parsed.checks
assert not parsed.checks_file
assert not parsed.services
@@ -67,9 +68,10 @@ class Test_Outputs:
assert not parsed.output_filename
assert "output" in parsed.output_directory
assert not parsed.verbose
assert parsed.no_banner
assert not parsed.no_banner
assert parsed.log_level == "CRITICAL"
assert not parsed.log_file
assert not parsed.only_logs
assert not parsed.checks
assert not parsed.checks_file
assert not parsed.services
@@ -201,12 +203,18 @@ class Test_Outputs:
def test_root_parser_no_banner_short(self):
command = [prowler_command, "-b"]
parsed = self.parser.parse(command)
assert not parsed.no_banner
assert parsed.no_banner
def test_root_parser_no_banner_long(self):
command = [prowler_command, "--no-banner"]
parsed = self.parser.parse(command)
assert not parsed.no_banner
assert parsed.no_banner
def test_logging_parser_only_logs_set(self):
command = [prowler_command, "--only-logs"]
parsed = self.parser.parse(command)
assert parsed.only_logs
assert parsed.no_banner
def test_logging_parser_log_level_default(self):
log_level = "CRITICAL"

View File

@@ -66,6 +66,7 @@ class Test_Common_Output_Options:
arguments.output_filename = "output_test_filename"
arguments.security_hub = True
arguments.shodan = "test-api-key"
arguments.only_logs = False
audit_info = self.set_mocked_aws_audit_info()
allowlist_file = ""
@@ -94,6 +95,7 @@ class Test_Common_Output_Options:
arguments.verbose = True
arguments.security_hub = True
arguments.shodan = "test-api-key"
arguments.only_logs = False
# Mock AWS Audit Info
audit_info = self.set_mocked_aws_audit_info()
@@ -125,6 +127,7 @@ class Test_Common_Output_Options:
arguments.output_modes = ["html", "csv", "json"]
arguments.output_directory = "output_test_directory"
arguments.verbose = True
arguments.only_logs = False
# Mock Azure Audit Info
audit_info = self.set_mocked_azure_audit_info()
@@ -159,6 +162,7 @@ class Test_Common_Output_Options:
arguments.output_modes = ["html", "csv", "json"]
arguments.output_directory = "output_test_directory"
arguments.verbose = True
arguments.only_logs = False
# Mock Azure Audit Info
audit_info = self.set_mocked_azure_audit_info()