Changeset 178:37f36ea93838 for wokkel/test/test_generic.py

Timestamp:

Jan 12, 2013, 4:40:38 PM (9 years ago)

Author:

Ralph Meijer <ralphm@…>

Branch:

default

Message:

Properly encode IDN domain names for SRV lookups.

Before Twisted 12.3.0, the SRV lookup done for outgoing
connections succeeded if passed a unicode string with all-ASCII
code points. A recent change made the DNS code more strict, and
only byte string are accepted as domain name. See
​http://twistedmatrix.com/trac/ticket/6245 for details.

This change makes sure domain names are encoded to their ASCII
Compatible Equivalent (ACE) version before passing the resulting
byte string to twisted.names.srvconnect.SRVConnector, as per
RFC 3490.

Note that while connecting to servers with an IDN domain name
now works properly, authentication using MD5-Digest SASL mechanism
will fail until ​http://twistedmatrix.com/trac/ticket/5066 has been
resolved.

Fixes: #77.

File:

• wokkel/test/test_generic.py (modified) (1 diff)

wokkel/test/test_generic.py

@@ -269 +269 @@
         """
         self.assertIdentical(None, self.request.timeout)
+
+
+
+class PrepareIDNNameTests(unittest.TestCase):
+    """
+    Tests for L{wokkel.generic.prepareIDNName}.
+    """
+
+    def test_bytestring(self):
+        """
+        An ASCII-encoded byte string is left as-is.
+        """
+        name = b"example.com"
+        result = generic.prepareIDNName(name)
+        self.assertEqual(b"example.com", result)
+
+
+    def test_unicode(self):
+        """
+        A unicode all-ASCII name is converted to an ASCII byte string.
+        """
+        name = u"example.com"
+        result = generic.prepareIDNName(name)
+        self.assertEqual(b"example.com", result)
+
+
+    def test_unicodeNonASCII(self):
+        """
+        A unicode with non-ASCII is converted to its ACE equivalent.
+        """
+        name = u"\u00e9chec.example.com"
+        result = generic.prepareIDNName(name)
+        self.assertEqual(b"xn--chec-9oa.example.com", result)
+
+
+    def test_unicodeHalfwidthIdeographicFullStop(self):
+        """
+        Exotic dots in unicode names are converted to Full Stop.
+        """
+        name = u"\u00e9chec.example\uff61com"
+        result = generic.prepareIDNName(name)
+        self.assertEqual(b"xn--chec-9oa.example.com", result)
+
+
+    def test_unicodeTrailingDot(self):
+        """
+        Unicode names with trailing dots retain the trailing dot.
+
+        L{encodings.idna.ToASCII} doesn't allow the empty string as the input,
+        hence the implementation needs to strip a trailing dot, and re-add it
+        after encoding the labels.
+        """
+        name = u"example.com."
+        result = generic.prepareIDNName(name)
+        self.assertEqual(b"example.com.", result)