# 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()