diff options
Diffstat (limited to 'plugins')
-rw-r--r-- | plugins/tests/ip_in_subnet.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/plugins/tests/ip_in_subnet.py b/plugins/tests/ip_in_subnet.py new file mode 100644 index 0000000..0b5a4a6 --- /dev/null +++ b/plugins/tests/ip_in_subnet.py @@ -0,0 +1,16 @@ +import ipaddress + +def test_ip_in_subnet(ip, cidr): + return ipaddress.ip_address(ip) in ipaddress.ip_network(cidr) + +def test_subnet_contains_ip(cidr, ip): + return ipaddress.ip_address(ip) in ipaddress.ip_network(cidr) + +class TestModule(object): + test_map = { + 'in_subnet': test_ip_in_subnet, + 'contains_ip': test_subnet_contains_ip, + } + + def tests(self): + return self.test_map |