ben.carrasco/prowler
1
0 Fork 0
mirror of https://github.com/prowler-cloud/prowler.git synced 2026-11-02 01:15:31 +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 FagoagaandSergio Garcia authored and GitHub committed 2024-03-13 18:15:24 +01:00
1 parent 2176fff8c3
commit bda5d62c72
362 files changed
+5690 -6022
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/providers/aws/services/workspaces/workspaces_service_test.py
+12 -15
View File
@@ -4,10 +4,7 @@ from uuid import uuid4
import botocore
from prowler.providers.aws.services.workspaces.workspaces_service import WorkSpaces
from tests.providers.aws.audit_info_utils import (
AWS_REGION_EU_WEST_1,
set_mocked_aws_audit_info,
)
from tests.providers.aws.utils import AWS_REGION_EU_WEST_1, set_mocked_aws_provider
workspace_id = str(uuid4())
@@ -35,8 +32,8 @@ def mock_make_api_call(self, operation_name, kwarg):
return make_api_call(self, operation_name, kwarg)
def mock_generate_regional_clients(service, audit_info):
regional_client = audit_info.audit_session.client(
def mock_generate_regional_clients(provider, service):
regional_client = provider._session.current_session.client(
service, region_name=AWS_REGION_EU_WEST_1
)
regional_client.region = AWS_REGION_EU_WEST_1
@@ -45,33 +42,33 @@ def mock_generate_regional_clients(service, audit_info):
@patch("botocore.client.BaseClient._make_api_call", new=mock_make_api_call)
@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_WorkSpaces_Service:
# Test WorkSpaces Service
def test_service(self):
audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(audit_info)
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(aws_provider)
assert workspaces.service == "workspaces"
# Test WorkSpaces client
def test_client(self):
audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(audit_info)
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(aws_provider)
for reg_client in workspaces.regional_clients.values():
assert reg_client.__class__.__name__ == "WorkSpaces"
# Test WorkSpaces session
def test__get_session__(self):
audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(audit_info)
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(aws_provider)
assert workspaces.session.__class__.__name__ == "Session"
# Test WorkSpaces describe workspaces
def test__describe_workspaces__(self):
audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(audit_info)
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
workspaces = WorkSpaces(aws_provider)
assert len(workspaces.workspaces) == 1
assert workspaces.workspaces[0].id == workspace_id
assert workspaces.workspaces[0].region == AWS_REGION_EU_WEST_1
tests/providers/aws/services/workspaces/workspaces_volume_encryption_enabled/workspaces_volume_encryption_enabled_test.py
+1 -4
View File
@@ -3,10 +3,7 @@ from unittest import mock
from uuid import uuid4
from prowler.providers.aws.services.workspaces.workspaces_service import WorkSpace
from tests.providers.aws.audit_info_utils import (
AWS_ACCOUNT_NUMBER,
AWS_REGION_EU_WEST_1,
)
from tests.providers.aws.utils import AWS_ACCOUNT_NUMBER, AWS_REGION_EU_WEST_1
WORKSPACE_ID = str(uuid4())
WORKSPACE_ARN = f"arn:aws:workspaces:{AWS_REGION_EU_WEST_1}:{AWS_ACCOUNT_NUMBER}:workspace/{WORKSPACE_ID}"
tests/providers/aws/services/workspaces/workspaces_vpc_2private_1public_subnets_nat/workspaces_vpc_2private_1public_subnets_nat_test.py
+22 -22
View File
@@ -6,10 +6,10 @@ from moto import mock_aws
from prowler.providers.aws.services.vpc.vpc_service import VPC
from prowler.providers.aws.services.workspaces.workspaces_service import WorkSpace
from tests.providers.aws.audit_info_utils import (
from tests.providers.aws.utils import (
AWS_ACCOUNT_NUMBER,
AWS_REGION_EU_WEST_1,
set_mocked_aws_audit_info,
set_mocked_aws_provider,
)
WORKSPACE_ID = str(uuid4())
@@ -52,15 +52,15 @@ class Test_workspaces_vpc_2private_1public_subnets_nat:
)
)
current_audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
"prowler.providers.common.common.get_global_provider",
return_value=aws_provider,
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.vpc_client",
new=VPC(current_audit_info),
new=VPC(aws_provider),
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.workspaces_client",
@@ -120,15 +120,15 @@ class Test_workspaces_vpc_2private_1public_subnets_nat:
)
)
current_audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
"prowler.providers.common.common.get_global_provider",
return_value=aws_provider,
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.vpc_client",
new=VPC(current_audit_info),
new=VPC(aws_provider),
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.workspaces_client",
@@ -205,15 +205,15 @@ class Test_workspaces_vpc_2private_1public_subnets_nat:
)
)
current_audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
"prowler.providers.common.common.get_global_provider",
return_value=aws_provider,
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.vpc_client",
new=VPC(current_audit_info),
new=VPC(aws_provider),
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.workspaces_client",
@@ -309,15 +309,15 @@ class Test_workspaces_vpc_2private_1public_subnets_nat:
)
)
current_audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
"prowler.providers.common.common.get_global_provider",
return_value=aws_provider,
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.vpc_client",
new=VPC(current_audit_info),
new=VPC(aws_provider),
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.workspaces_client",
@@ -421,15 +421,15 @@ class Test_workspaces_vpc_2private_1public_subnets_nat:
)
)
current_audit_info = set_mocked_aws_audit_info([AWS_REGION_EU_WEST_1])
aws_provider = set_mocked_aws_provider([AWS_REGION_EU_WEST_1])
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
"prowler.providers.common.common.get_global_provider",
return_value=aws_provider,
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.vpc_client",
new=VPC(current_audit_info),
new=VPC(aws_provider),
):
with mock.patch(
"prowler.providers.aws.services.workspaces.workspaces_vpc_2private_1public_subnets_nat.workspaces_vpc_2private_1public_subnets_nat.workspaces_client",