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(ec2_instance_..._ssm): mock ssm service and client in all the tests (#2804)

Browse Source
This commit is contained in:
Nacho Rivera
2023-09-05 12:34:02 +02:00
committed by GitHub
parent 03ad61abc6
commit 44b979b4a4
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/providers/aws/services/ec2/ec2_instance_managed_by_ssm/ec2_instance_managed_by_ssm_test.py
View File

@@ -50,9 +50,18 @@ class Test_ec2_instance_managed_by_ssm_test:
current_audit_info = self.set_mocked_audit_info()
ssm_client = mock.MagicMock
ssm_client.managed_instances = {}
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
), mock.patch(
"prowler.providers.aws.services.ssm.ssm_service.SSM",
new=ssm_client,
), mock.patch(
"prowler.providers.aws.services.ssm.ssm_client.ssm_client",
new=ssm_client,
), mock.patch(
"prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client",
new=EC2(current_audit_info),
@@ -80,6 +89,9 @@ class Test_ec2_instance_managed_by_ssm_test:
UserData="This is some user_data",
)[0]
ssm_client = mock.MagicMock
ssm_client.managed_instances = {}
from prowler.providers.aws.services.ec2.ec2_service import EC2
current_audit_info = self.set_mocked_audit_info()
@@ -87,6 +99,12 @@ class Test_ec2_instance_managed_by_ssm_test:
with mock.patch(
"prowler.providers.aws.lib.audit_info.audit_info.current_audit_info",
new=current_audit_info,
), mock.patch(
"prowler.providers.aws.services.ssm.ssm_service.SSM",
new=ssm_client,
), mock.patch(
"prowler.providers.aws.services.ssm.ssm_client.ssm_client",
new=ssm_client,
), mock.patch(
"prowler.providers.aws.services.ec2.ec2_instance_managed_by_ssm.ec2_instance_managed_by_ssm.ec2_client",
new=EC2(current_audit_info),