ben.carrasco/prowler
1
0
Fork 0
mirror of https://github.com/prowler-cloud/prowler.git synced 2026-07-24 21:11:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(aws): Replace audit_info for provider (#3521)

Browse Source 
Co-authored-by: Sergio Garcia <38561120+sergargar@users.noreply.github.com>
This commit is contained in:
Pepe Fagoaga
2024-03-13 18:15:24 +01:00
committed by GitHub
parent 2176fff8c3
commit bda5d62c72
362 changed files with 5690 additions and 6022 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/providers/aws/services/awslambda/awslambda_service_test.py
+10 -10
View File
@@ -10,11 +10,11 @@ from boto3 import client, resource
from moto import mock_aws
from prowler.providers.aws.services.awslambda.awslambda_service import AuthType, Lambda
from tests.providers.aws.audit_info_utils import (
from tests.providers.aws.utils import (
AWS_ACCOUNT_NUMBER,
AWS_REGION_EU_WEST_1,
AWS_REGION_US_EAST_1,
set_mocked_aws_audit_info,
set_mocked_aws_provider,
)
LAMBDA_FUNCTION_CODE = """def lambda_handler(event, context):
@@ -47,11 +47,11 @@ def mock_request_get(_):
# Mock generate_regional_clients()
def mock_generate_regional_clients(service, audit_info):
regional_client_eu_west_1 = audit_info.audit_session.client(
def mock_generate_regional_clients(provider, service):
regional_client_eu_west_1 = provider.session.current_session.client(
service, region_name=AWS_REGION_EU_WEST_1
)
regional_client_us_east_1 = audit_info.audit_session.client(
regional_client_us_east_1 = provider.session.current_session.client(
service, region_name=AWS_REGION_US_EAST_1
)
regional_client_eu_west_1.region = AWS_REGION_EU_WEST_1
@@ -63,13 +63,13 @@ def mock_generate_regional_clients(service, audit_info):
@patch(
"prowler.providers.aws.lib.service.service.generate_regional_clients",
"prowler.providers.aws.aws_provider.AwsProvider.generate_regional_clients",
new=mock_generate_regional_clients,
)
class Test_Lambda_Service:
# Test Lambda Client
def test__get_client__(self):
awslambda = Lambda(set_mocked_aws_audit_info([AWS_REGION_US_EAST_1]))
awslambda = Lambda(set_mocked_aws_provider([AWS_REGION_US_EAST_1]))
assert (
awslambda.regional_clients[AWS_REGION_EU_WEST_1].__class__.__name__
== "Lambda"
@@ -77,12 +77,12 @@ class Test_Lambda_Service:
# Test Lambda Session
def test__get_session__(self):
awslambda = Lambda(set_mocked_aws_audit_info([AWS_REGION_US_EAST_1]))
awslambda = Lambda(set_mocked_aws_provider([AWS_REGION_US_EAST_1]))
assert awslambda.session.__class__.__name__ == "Session"
# Test Lambda Service
def test__get_service__(self):
awslambda = Lambda(set_mocked_aws_audit_info([AWS_REGION_US_EAST_1]))
awslambda = Lambda(set_mocked_aws_provider([AWS_REGION_US_EAST_1]))
assert awslambda.service == "lambda"
@mock_aws
@@ -192,7 +192,7 @@ class Test_Lambda_Service:
new=mock_request_get,
):
awslambda = Lambda(
set_mocked_aws_audit_info(audited_regions=[AWS_REGION_US_EAST_1])
set_mocked_aws_provider(audited_regions=[AWS_REGION_US_EAST_1])
)
assert awslambda.functions
assert len(awslambda.functions) == 2