ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-10 14:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(): Refresh credentials when assuming role (#1636)

Browse Source
This commit is contained in:
Nacho Rivera
2023-01-04 08:48:00 +01:00
committed by GitHub
parent 9b8c80b74d
commit f0db63da35
3 changed files with 29 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/providers/aws/aws_provider_test.py
View File

@@ -2,7 +2,11 @@ import boto3
import sure # noqa
from moto import mock_iam, mock_sts
from prowler.providers.aws.aws_provider import assume_role, generate_regional_clients
from prowler.providers.aws.aws_provider import (
AWS_Provider,
assume_role,
generate_regional_clients,
)
from prowler.providers.aws.lib.audit_info.models import AWS_Assume_Role, AWS_Audit_Info
ACCOUNT_ID = 123456789012
@@ -55,7 +59,10 @@ class Test_AWS_Provider:
)
# Call assume_role
assume_role_response = assume_role(audit_info)
aws_provider = AWS_Provider(audit_info)
assume_role_response = assume_role(
aws_provider.aws_session, aws_provider.role_info
)
# Recover credentials for the assume role operation
credentials = assume_role_response["Credentials"]
# Test the response