Function set_mocked_aws_provider

tests/providers/aws/utils.py:97–169

(
    audited_regions: list[str] = [AWS_REGION_US_EAST_1],
    audited_account: str = AWS_ACCOUNT_NUMBER,
    audited_account_arn: str = AWS_ACCOUNT_ARN,
    audited_partition: str = AWS_COMMERCIAL_PARTITION,
    expected_checks: list[str] = [],
    profile_region: str = None,
    audit_config: dict = {},
    fixer_config: dict = {},
    mutelist: dict = None,
    scan_unused_services: bool = True,
    audit_session: session.Session = session.Session(
        profile_name=None,
        botocore_session=None,
    ),
    original_session: session.Session = None,
    enabled_regions: set = None,
    arguments: Namespace = Namespace(),
    status: list[str] = [],
    create_default_organization: bool = True,
)

# This here causes to call this function mocking the AWS calls
@mock_aws
def set_mocked_aws_provider(
    audited_regions: list[str] = [AWS_REGION_US_EAST_1],
    audited_account: str = AWS_ACCOUNT_NUMBER,
    audited_account_arn: str = AWS_ACCOUNT_ARN,
    audited_partition: str = AWS_COMMERCIAL_PARTITION,
    expected_checks: list[str] = [],
    profile_region: str = None,
    audit_config: dict = {},
    fixer_config: dict = {},
    mutelist: dict = None,
    scan_unused_services: bool = True,
    audit_session: session.Session = session.Session(
        profile_name=None,
        botocore_session=None,
    ),
    original_session: session.Session = None,
    enabled_regions: set = None,
    arguments: Namespace = Namespace(),
    status: list[str] = [],
    create_default_organization: bool = True,
) -> AwsProvider:
    if audited_regions is None:
        raise ValueError(
            "audited_regions is None, which means all 36 regions will be used. "
            "Pass an explicit list of regions instead."
        )

    if create_default_organization:
        # Create default AWS Organization
        create_default_aws_organization()

    # Default arguments
    arguments = set_default_provider_arguments(arguments, status)

    # AWS Provider
    provider = AwsProvider()

    # Mock Session
    session_config = AwsProvider.set_session_config(None)
    provider._session.session_config = session_config
    provider._session.original_session = original_session
    provider._session.current_session = audit_session
    audit_session._session.set_default_client_config(session_config)
    # Mock Identity
    provider._identity.account = audited_account
    provider._identity.account_arn = audited_account_arn
    provider._identity.user_id = None
    provider._identity.partition = audited_partition
    provider._identity.identity_arn = None
    provider._identity.profile = None
    provider._identity.profile_region = profile_region
    provider._identity.audited_regions = audited_regions
    # Mock Configiration
    provider._scan_unused_services = scan_unused_services
    provider._enabled_regions = (
        enabled_regions
        if enabled_regions is not None
        else (set(audited_regions) if audited_regions else None)