1 files changed, 53 insertions, 0 deletions
diff --git a/tests/utils/ipv6.py b/tests/utils/ipv6.py
new file mode 100644
index 0000000000..1713de82b8
--- /dev/null
+++ b/tests/utils/ipv6.py
@@ -0,0 +1,53 @@
+from __future__ import unicode_literals
+
+from django.utils import unittest
+from django.utils.ipv6 import is_valid_ipv6_address, clean_ipv6_address
+
+class TestUtilsIPv6(unittest.TestCase):
+
+    def test_validates_correct_plain_address(self):
+        self.assertTrue(is_valid_ipv6_address('fe80::223:6cff:fe8a:2e8a'))
+        self.assertTrue(is_valid_ipv6_address('2a02::223:6cff:fe8a:2e8a'))
+        self.assertTrue(is_valid_ipv6_address('1::2:3:4:5:6:7'))
+        self.assertTrue(is_valid_ipv6_address('::'))
+        self.assertTrue(is_valid_ipv6_address('::a'))
+        self.assertTrue(is_valid_ipv6_address('2::'))
+
+    def test_validates_correct_with_v4mapping(self):
+        self.assertTrue(is_valid_ipv6_address('::ffff:254.42.16.14'))
+        self.assertTrue(is_valid_ipv6_address('::ffff:0a0a:0a0a'))
+
+    def test_validates_incorrect_plain_address(self):
+        self.assertFalse(is_valid_ipv6_address('foo'))
+        self.assertFalse(is_valid_ipv6_address('127.0.0.1'))
+        self.assertFalse(is_valid_ipv6_address('12345::'))
+        self.assertFalse(is_valid_ipv6_address('1::2:3::4'))
+        self.assertFalse(is_valid_ipv6_address('1::zzz'))
+        self.assertFalse(is_valid_ipv6_address('1::2:3:4:5:6:7:8'))
+        self.assertFalse(is_valid_ipv6_address('1:2'))
+        self.assertFalse(is_valid_ipv6_address('1:::2'))
+
+    def test_validates_incorrect_with_v4mapping(self):
+        self.assertFalse(is_valid_ipv6_address('::ffff:999.42.16.14'))
+        self.assertFalse(is_valid_ipv6_address('::ffff:zzzz:0a0a'))
+        # The ::1.2.3.4 format used to be valid but was deprecated
+        # in rfc4291 section 2.5.5.1
+        self.assertTrue(is_valid_ipv6_address('::254.42.16.14'))
+        self.assertTrue(is_valid_ipv6_address('::0a0a:0a0a'))
+        self.assertFalse(is_valid_ipv6_address('::999.42.16.14'))
+        self.assertFalse(is_valid_ipv6_address('::zzzz:0a0a'))
+
+    def test_cleanes_plain_address(self):
+        self.assertEqual(clean_ipv6_address('DEAD::0:BEEF'), 'dead::beef')
+        self.assertEqual(clean_ipv6_address('2001:000:a:0000:0:fe:fe:beef'), '2001:0:a::fe:fe:beef')
+        self.assertEqual(clean_ipv6_address('2001::a:0000:0:fe:fe:beef'), '2001:0:a::fe:fe:beef')
+
+    def test_cleanes_with_v4_mapping(self):
+        self.assertEqual(clean_ipv6_address('::ffff:0a0a:0a0a'), '::ffff:10.10.10.10')
+        self.assertEqual(clean_ipv6_address('::ffff:1234:1234'), '::ffff:18.52.18.52')
+        self.assertEqual(clean_ipv6_address('::ffff:18.52.18.52'), '::ffff:18.52.18.52')
+
+    def test_unpacks_ipv4(self):
+        self.assertEqual(clean_ipv6_address('::ffff:0a0a:0a0a', unpack_ipv4=True), '10.10.10.10')
+        self.assertEqual(clean_ipv6_address('::ffff:1234:1234', unpack_ipv4=True), '18.52.18.52')
+        self.assertEqual(clean_ipv6_address('::ffff:18.52.18.52', unpack_ipv4=True), '18.52.18.52')