From 8533714cb29a7955faeebfbcd4549867a717e2ed Mon Sep 17 00:00:00 2001
From: Pepe Fagoaga <pepe@verica.io>
Date: Tue, 24 Oct 2023 16:29:18 +0200
Subject: [PATCH] tests: remove tests folder after execution (#2962)

---
 tests/providers/common/common_outputs_test.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/providers/common/common_outputs_test.py b/tests/providers/common/common_outputs_test.py
index d09a03d6..a756b723 100644
--- a/tests/providers/common/common_outputs_test.py
+++ b/tests/providers/common/common_outputs_test.py
@@ -1,4 +1,5 @@
 from argparse import Namespace
+from os import rmdir
 
 from boto3 import session
 from mock import patch
@@ -109,6 +110,9 @@ class Test_Common_Output_Options:
         assert output_options.verbose
         assert output_options.output_filename == arguments.output_filename
 
+        # Delete testing directory
+        rmdir(arguments.output_directory)
+
     def test_set_provider_output_options_gcp(self):
         #  Set the cloud provider
         provider = "gcp"
@@ -137,6 +141,9 @@ class Test_Common_Output_Options:
         assert output_options.verbose
         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):
         #  Set the cloud provider
         provider = "aws"
@@ -172,6 +179,9 @@ class Test_Common_Output_Options:
             == f"prowler-output-{AWS_ACCOUNT_NUMBER}-{DATETIME}"
         )
 
+        # Delete testing directory
+        rmdir(arguments.output_directory)
+
     def test_set_provider_output_options_azure_domain(self):
         #  Set the cloud provider
         provider = "azure"
@@ -209,6 +219,9 @@ class Test_Common_Output_Options:
             == 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):
         #  Set the cloud provider
         provider = "azure"
@@ -247,6 +260,9 @@ class Test_Common_Output_Options:
             == f"prowler-output-{'-'.join(tenants)}-{DATETIME}"
         )
 
+        # Delete testing directory
+        rmdir(arguments.output_directory)
+
     def test_azure_get_assessment_summary(self):
         # Mock Azure Audit Info
         audit_info = self.set_mocked_azure_audit_info()