Files
prowler/tests/providers/aws/services/ec2/lib/security_groups_test.py
2022-12-13 09:07:55 +01:00

22 lines
606 B
Python

import pytest
from prowler.providers.aws.services.ec2.lib.security_groups import _is_cidr_public
class Test_security_groups:
def test__is_cidr_public_Public_IP(self):
cidr = "0.0.0.0/0"
assert _is_cidr_public(cidr)
def test__is_cidr_public_Private_IP(self):
cidr = "10.0.0.0/8"
assert not _is_cidr_public(cidr)
def test__is_cidr_public_Bad_Private_IP(self):
cidr = "10.0.0.0/0"
with pytest.raises(ValueError) as ex:
_is_cidr_public(cidr)
assert ex.type == ValueError
assert ex.match(f"{cidr} has host bits set")