tests: remove tests folder after execution (#2962)

This commit is contained in:
Pepe Fagoaga
2023-10-24 16:29:18 +02:00
committed by GitHub
parent b822c19d2c
commit 8533714cb2

View File

@@ -1,4 +1,5 @@
from argparse import Namespace from argparse import Namespace
from os import rmdir
from boto3 import session from boto3 import session
from mock import patch from mock import patch
@@ -109,6 +110,9 @@ class Test_Common_Output_Options:
assert output_options.verbose assert output_options.verbose
assert output_options.output_filename == arguments.output_filename assert output_options.output_filename == arguments.output_filename
# Delete testing directory
rmdir(arguments.output_directory)
def test_set_provider_output_options_gcp(self): def test_set_provider_output_options_gcp(self):
# Set the cloud provider # Set the cloud provider
provider = "gcp" provider = "gcp"
@@ -137,6 +141,9 @@ class Test_Common_Output_Options:
assert output_options.verbose assert output_options.verbose
assert output_options.output_filename == arguments.output_filename assert output_options.output_filename == arguments.output_filename
# Delete testing directory
rmdir(arguments.output_directory)
def test_set_provider_output_options_aws_no_output_filename(self): def test_set_provider_output_options_aws_no_output_filename(self):
# Set the cloud provider # Set the cloud provider
provider = "aws" provider = "aws"
@@ -172,6 +179,9 @@ class Test_Common_Output_Options:
== f"prowler-output-{AWS_ACCOUNT_NUMBER}-{DATETIME}" == f"prowler-output-{AWS_ACCOUNT_NUMBER}-{DATETIME}"
) )
# Delete testing directory
rmdir(arguments.output_directory)
def test_set_provider_output_options_azure_domain(self): def test_set_provider_output_options_azure_domain(self):
# Set the cloud provider # Set the cloud provider
provider = "azure" provider = "azure"
@@ -209,6 +219,9 @@ class Test_Common_Output_Options:
== f"prowler-output-{audit_info.identity.domain}-{DATETIME}" == f"prowler-output-{audit_info.identity.domain}-{DATETIME}"
) )
# Delete testing directory
rmdir(arguments.output_directory)
def test_set_provider_output_options_azure_tenant_ids(self): def test_set_provider_output_options_azure_tenant_ids(self):
# Set the cloud provider # Set the cloud provider
provider = "azure" provider = "azure"
@@ -247,6 +260,9 @@ class Test_Common_Output_Options:
== f"prowler-output-{'-'.join(tenants)}-{DATETIME}" == f"prowler-output-{'-'.join(tenants)}-{DATETIME}"
) )
# Delete testing directory
rmdir(arguments.output_directory)
def test_azure_get_assessment_summary(self): def test_azure_get_assessment_summary(self):
# Mock Azure Audit Info # Mock Azure Audit Info
audit_info = self.set_mocked_azure_audit_info() audit_info = self.set_mocked_azure_audit_info()