initial commit

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