fix(test): Mock audit into in SecurityHub CodeBuild (#2225)

tests/providers/aws/services/codebuild/codebuild_service_test.py

@@ -4,7 +4,6 @@ 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.codebuild.codebuild_service import Codebuild
 
@@ -77,22 +76,22 @@ class Test_Codebuild_Service:
 
     # Test Codebuild Session
     def test__get_session__(self):
-        codebuild = Codebuild(current_audit_info)
+        codebuild = Codebuild(self.set_mocked_audit_info())
         assert codebuild.session.__class__.__name__ == "Session"
 
     # Test Codebuild Service
     def test__get_service__(self):
-        codebuild = Codebuild(current_audit_info)
+        codebuild = Codebuild(self.set_mocked_audit_info())
         assert codebuild.service == "codebuild"
 
     def test__list_projects__(self):
-        codebuild = Codebuild(current_audit_info)
+        codebuild = Codebuild(self.set_mocked_audit_info())
         assert len(codebuild.projects) == 1
         assert codebuild.projects[0].name == "test"
         assert codebuild.projects[0].region == AWS_REGION
 
     def test__list_builds_for_project__(self):
-        codebuild = Codebuild(current_audit_info)
+        codebuild = Codebuild(self.set_mocked_audit_info())
         assert len(codebuild.projects) == 1
         assert codebuild.projects[0].name == "test"
         assert codebuild.projects[0].region == AWS_REGION

tests/providers/aws/services/securityhub/securityhub_service_test.py

@@ -1,13 +1,14 @@
 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.securityhub.securityhub_service import SecurityHub
 
 # Mock Test Region
 AWS_REGION = "eu-west-1"
-
+AWS_ACCOUNT_NUMBER = "123456789012"
 # Mocking Access Analyzer Calls
 make_api_call = botocore.client.BaseClient._make_api_call
 
@@ -57,23 +58,45 @@ def mock_generate_regional_clients(service, audit_info):
     new=mock_generate_regional_clients,
 )
 class Test_SecurityHub_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 SecurityHub Client
     def test__get_client__(self):
-        access_analyzer = SecurityHub(current_audit_info)
+        security_hub = SecurityHub(self.set_mocked_audit_info())
         assert (
-            access_analyzer.regional_clients[AWS_REGION].__class__.__name__
+            security_hub.regional_clients[AWS_REGION].__class__.__name__
             == "SecurityHub"
         )
 
     # Test SecurityHub Session
     def test__get_session__(self):
-        access_analyzer = SecurityHub(current_audit_info)
-        assert access_analyzer.session.__class__.__name__ == "Session"
+        security_hub = SecurityHub(self.set_mocked_audit_info())
+        assert security_hub.session.__class__.__name__ == "Session"
 
     def test__describe_hub__(self):
         # Set partition for the service
-        current_audit_info.audited_partition = "aws"
-        securityhub = SecurityHub(current_audit_info)
+        securityhub = SecurityHub(self.set_mocked_audit_info())
         assert len(securityhub.securityhubs) == 1
         assert (
             securityhub.securityhubs[0].arn