586f74b929
Replace MIT with the full GNU GPLv3 text, update license metadata in pyproject.toml (+ trove classifiers) and PKGBUILD, and add SPDX-License-Identifier headers to all Python modules and shell scripts. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
52 lines
1.7 KiB
Python
Executable file
52 lines
1.7 KiB
Python
Executable file
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
import unittest
|
|
|
|
from enodia_sentinel import netutil
|
|
|
|
|
|
class TestPublicIP(unittest.TestCase):
|
|
def test_public(self):
|
|
self.assertTrue(netutil.is_public_ip("8.8.8.8"))
|
|
self.assertTrue(netutil.is_public_ip("1.1.1.1"))
|
|
|
|
def test_private_and_special(self):
|
|
for addr in ("10.0.0.5", "192.168.1.1", "172.16.0.1",
|
|
"127.0.0.1", "169.254.1.1", "100.64.0.1", "::1"):
|
|
self.assertFalse(netutil.is_public_ip(addr), addr)
|
|
|
|
def test_zone_and_brackets(self):
|
|
self.assertFalse(netutil.is_public_ip("10.2.0.2%proton0"))
|
|
self.assertFalse(netutil.is_public_ip("[::1]"))
|
|
|
|
def test_garbage(self):
|
|
self.assertFalse(netutil.is_public_ip("not-an-ip"))
|
|
self.assertFalse(netutil.is_public_ip(""))
|
|
|
|
|
|
class TestCidr(unittest.TestCase):
|
|
def test_in_and_out(self):
|
|
cidrs = ["203.0.113.0/24", "10.0.0.0/8"]
|
|
self.assertTrue(netutil.ip_in_cidrs("203.0.113.55", cidrs))
|
|
self.assertTrue(netutil.ip_in_cidrs("10.9.9.9", cidrs))
|
|
self.assertFalse(netutil.ip_in_cidrs("8.8.8.8", cidrs))
|
|
|
|
def test_empty_list(self):
|
|
self.assertFalse(netutil.ip_in_cidrs("8.8.8.8", []))
|
|
|
|
|
|
class TestSplitHostPort(unittest.TestCase):
|
|
def test_ipv4(self):
|
|
self.assertEqual(netutil.split_host_port("1.2.3.4:443"), ("1.2.3.4", "443"))
|
|
|
|
def test_ipv6_bracket(self):
|
|
self.assertEqual(netutil.split_host_port("[2001:db8::1]:22"),
|
|
("2001:db8::1", "22"))
|
|
|
|
def test_zone(self):
|
|
host, port = netutil.split_host_port("10.2.0.2%proton0:61877")
|
|
self.assertEqual(host, "10.2.0.2%proton0")
|
|
self.assertEqual(port, "61877")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|