From 29c9ad602d54d4e766b6af86668c6811398fb4c4 Mon Sep 17 00:00:00 2001
From: Pepe Fagoaga <pepe@verica.io>
Date: Fri, 14 Apr 2023 14:29:19 +0200
Subject: [PATCH] fix(test): Mock audit into in Macie (#2213)

---
 .../aws/services/macie/macie_service_test.py  | 36 +++++++++++++++----
 1 file changed, 30 insertions(+), 6 deletions(-)

diff --git a/tests/providers/aws/services/macie/macie_service_test.py b/tests/providers/aws/services/macie/macie_service_test.py
index 42520c10..bf1ab4fe 100644
--- a/tests/providers/aws/services/macie/macie_service_test.py
+++ b/tests/providers/aws/services/macie/macie_service_test.py
@@ -2,12 +2,14 @@ import datetime
 from unittest.mock import patch
 
 import botocore
+from boto3 import session
 
-from prowler.providers.aws.lib.audit_info.audit_info import current_audit_info
+from prowler.providers.aws.lib.audit_info.models import AWS_Audit_Info
 from prowler.providers.aws.services.macie.macie_service import Macie, Session
 
 # Mock Test Region
 AWS_REGION = "eu-west-1"
+AWS_ACCOUNT_NUMBER = "123456789012"
 
 # Mocking Macie2 Calls
 make_api_call = botocore.client.BaseClient._make_api_call
@@ -45,25 +47,47 @@ def mock_generate_regional_clients(service, audit_info):
     new=mock_generate_regional_clients,
 )
 class Test_Macie_Service:
+    def set_mocked_audit_info(self):
+        audit_info = AWS_Audit_Info(
+            session_config=None,
+            original_session=None,
+            audit_session=session.Session(
+                profile_name=None,
+                botocore_session=None,
+            ),
+            audited_account=AWS_ACCOUNT_NUMBER,
+            audited_user_id=None,
+            audited_partition="aws",
+            audited_identity_arn=None,
+            profile=None,
+            profile_region=None,
+            credentials=None,
+            assumed_role_info=None,
+            audited_regions=["us-east-1", "eu-west-1"],
+            organizations_metadata=None,
+            audit_resources=None,
+        )
+
+        return audit_info
+
     # Test Macie Client
     def test__get_client__(self):
-        macie = Macie(current_audit_info)
+        macie = Macie(self.set_mocked_audit_info())
         assert macie.regional_clients[AWS_REGION].__class__.__name__ == "Macie2"
 
     # Test Macie Session
     def test__get_session__(self):
-        macie = Macie(current_audit_info)
+        macie = Macie(self.set_mocked_audit_info())
         assert macie.session.__class__.__name__ == "Session"
 
     # Test Macie Service
     def test__get_service__(self):
-        macie = Macie(current_audit_info)
+        macie = Macie(self.set_mocked_audit_info())
         assert macie.service == "macie2"
 
     def test__get_macie_session__(self):
         # Set partition for the service
-        current_audit_info.audited_partition = "aws"
-        macie = Macie(current_audit_info)
+        macie = Macie(self.set_mocked_audit_info())
         macie.sessions = [
             Session(
                 status="ENABLED",