def test_authorize_security_group_ingress_with_cidr_ip(aws_client):
#arrange
client = MagicMock()
aws_client.return_value = client
requested_security_group_id='sg-123456789abc'
cidr_ip='0.0.0.0/0'
ip_protocol='tcp'
from_port=0
to_port=80
#act
authorize_security_group_ingress(
requested_security_group_id=requested_security_group_id,
ip_protocol=ip_protocol,
from_port=from_port,
to_port=to_port,
cidr_ip=cidr_ip
)
#assert
client.authorize_security_group_ingress.assert_called_with(
GroupId=requested_security_group_id,
IpPermissions=[
{
'IpProtocol': ip_protocol,
'FromPort': from_port,
'ToPort': to_port,
'IpRanges': [{'CidrIp': cidr_ip}],
'UserIdGroupPairs': [{}]
}
]
)
def test_authorize_security_group_ingress_with_cidr_ip(aws_client):
# arrange
client = MagicMock()
aws_client.return_value = client
requested_security_group_id = "sg-123456789abc"
cidr_ip = "0.0.0.0/0"
ip_protocol = "tcp"
from_port = 0
to_port = 80
# act
authorize_security_group_ingress(
requested_security_group_id=requested_security_group_id,
ip_protocol=ip_protocol,
from_port=from_port,
to_port=to_port,
cidr_ip=cidr_ip,
)
# assert
client.authorize_security_group_ingress.assert_called_with(
GroupId=requested_security_group_id,
IpPermissions=[{
"IpProtocol": ip_protocol,
"FromPort": from_port,
"ToPort": to_port,
"IpRanges": [{
"CidrIp": cidr_ip
}],
"UserIdGroupPairs": [{}],
}],
)