Add a test for protocol.checkIPv6Address() and IPv6 sample to test_check_local

Author: g1itch

src/tests/test_protocol.py

@@ -25,6 +25,16 @@ def test_checkIPv4Address(self):
         self.assertFalse(  # 8.8.8.8
             protocol.checkIPv4Address(b'\x08\x08\x08\x08', token, True))
 
+    def test_checkIPv6Address(self):
+        """Check the results of protocol.checkIPv6Address()"""
+        test_ip = '2001:db8::ff00:42:8329'
+        self.assertEqual(
+            'test', protocol.checkIPv6Address(
+                protocol.encodeHost(test_ip), 'test'))
+        self.assertFalse(
+            protocol.checkIPv6Address(
+                protocol.encodeHost(test_ip), 'test', True))
+
     def test_check_local(self):
         """Check the logic of TCPConnection.local"""
         self.assertTrue(
@@ -35,6 +45,8 @@ def test_check_local(self):
             protocol.checkIPAddress(protocol.encodeHost('10.42.43.1'), True))
         self.assertTrue(
             protocol.checkIPAddress(protocol.encodeHost('172.31.255.2'), True))
+        self.assertFalse(protocol.checkIPAddress(
+            protocol.encodeHost('2001:db8::ff00:42:8329'), True))
 
         globalhost = protocol.encodeHost('8.8.8.8')
         self.assertFalse(protocol.checkIPAddress(globalhost, True))