1 files changed, 129 insertions, 0 deletions

diff --git a/lib/python2.7/ctypes/test/test_unicode.py b/lib/python2.7/ctypes/test/test_unicode.py
new file mode 100644
index 0000000..6557479
--- /dev/null
+++ b/lib/python2.7/ctypes/test/test_unicode.py
@@ -0,0 +1,129 @@
+# coding: latin-1
+import unittest
+import ctypes
+
+try:
+    ctypes.c_wchar
+except AttributeError:
+    pass
+else:
+    import _ctypes_test
+    dll = ctypes.CDLL(_ctypes_test.__file__)
+    wcslen = dll.my_wcslen
+    wcslen.argtypes = [ctypes.c_wchar_p]
+
+
+    class UnicodeTestCase(unittest.TestCase):
+        def setUp(self):
+            self.prev_conv_mode = ctypes.set_conversion_mode("ascii", "strict")
+
+        def tearDown(self):
+            ctypes.set_conversion_mode(*self.prev_conv_mode)
+
+        def test_ascii_strict(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            # no conversions take place with unicode arguments
+            self.assertEqual(wcslen(u"abc"), 3)
+            self.assertEqual(wcslen(u"ab\u2070"), 3)
+            # string args are converted
+            self.assertEqual(wcslen("abc"), 3)
+            self.assertRaises(ctypes.ArgumentError, wcslen, "abä")
+
+        def test_ascii_replace(self):
+            ctypes.set_conversion_mode("ascii", "replace")
+            self.assertEqual(wcslen(u"abc"), 3)
+            self.assertEqual(wcslen(u"ab\u2070"), 3)
+            self.assertEqual(wcslen("abc"), 3)
+            self.assertEqual(wcslen("abä"), 3)
+
+        def test_ascii_ignore(self):
+            ctypes.set_conversion_mode("ascii", "ignore")
+            self.assertEqual(wcslen(u"abc"), 3)
+            self.assertEqual(wcslen(u"ab\u2070"), 3)
+            # ignore error mode skips non-ascii characters
+            self.assertEqual(wcslen("abc"), 3)
+            self.assertEqual(wcslen("äöüß"), 0)
+
+        def test_latin1_strict(self):
+            ctypes.set_conversion_mode("latin-1", "strict")
+            self.assertEqual(wcslen(u"abc"), 3)
+            self.assertEqual(wcslen(u"ab\u2070"), 3)
+            self.assertEqual(wcslen("abc"), 3)
+            self.assertEqual(wcslen("äöüß"), 4)
+
+        def test_buffers(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            buf = ctypes.create_unicode_buffer("abc")
+            self.assertEqual(len(buf), 3+1)
+
+            ctypes.set_conversion_mode("ascii", "replace")
+            buf = ctypes.create_unicode_buffer("abäöü")
+            self.assertEqual(buf[:], u"ab\uFFFD\uFFFD\uFFFD\0")
+            self.assertEqual(buf[::], u"ab\uFFFD\uFFFD\uFFFD\0")
+            self.assertEqual(buf[::-1], u"\0\uFFFD\uFFFD\uFFFDba")
+            self.assertEqual(buf[::2], u"a\uFFFD\uFFFD")
+            self.assertEqual(buf[6:5:-1], u"")
+
+            ctypes.set_conversion_mode("ascii", "ignore")
+            buf = ctypes.create_unicode_buffer("abäöü")
+            # is that correct? not sure.  But with 'ignore', you get what you pay for..
+            self.assertEqual(buf[:], u"ab\0\0\0\0")
+            self.assertEqual(buf[::], u"ab\0\0\0\0")
+            self.assertEqual(buf[::-1], u"\0\0\0\0ba")
+            self.assertEqual(buf[::2], u"a\0\0")
+            self.assertEqual(buf[6:5:-1], u"")
+
+    import _ctypes_test
+    func = ctypes.CDLL(_ctypes_test.__file__)._testfunc_p_p
+
+    class StringTestCase(UnicodeTestCase):
+        def setUp(self):
+            self.prev_conv_mode = ctypes.set_conversion_mode("ascii", "strict")
+            func.argtypes = [ctypes.c_char_p]
+            func.restype = ctypes.c_char_p
+
+        def tearDown(self):
+            ctypes.set_conversion_mode(*self.prev_conv_mode)
+            func.argtypes = None
+            func.restype = ctypes.c_int
+
+        def test_ascii_replace(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            self.assertEqual(func("abc"), "abc")
+            self.assertEqual(func(u"abc"), "abc")
+            self.assertRaises(ctypes.ArgumentError, func, u"abä")
+
+        def test_ascii_ignore(self):
+            ctypes.set_conversion_mode("ascii", "ignore")
+            self.assertEqual(func("abc"), "abc")
+            self.assertEqual(func(u"abc"), "abc")
+            self.assertEqual(func(u"äöüß"), "")
+
+        def test_ascii_replace(self):
+            ctypes.set_conversion_mode("ascii", "replace")
+            self.assertEqual(func("abc"), "abc")
+            self.assertEqual(func(u"abc"), "abc")
+            self.assertEqual(func(u"äöüß"), "????")
+
+        def test_buffers(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            buf = ctypes.create_string_buffer(u"abc")
+            self.assertEqual(len(buf), 3+1)
+
+            ctypes.set_conversion_mode("ascii", "replace")
+            buf = ctypes.create_string_buffer(u"abäöü")
+            self.assertEqual(buf[:], "ab???\0")
+            self.assertEqual(buf[::], "ab???\0")
+            self.assertEqual(buf[::-1], "\0???ba")
+            self.assertEqual(buf[::2], "a??")
+            self.assertEqual(buf[6:5:-1], "")
+
+            ctypes.set_conversion_mode("ascii", "ignore")
+            buf = ctypes.create_string_buffer(u"abäöü")
+            # is that correct? not sure.  But with 'ignore', you get what you pay for..
+            self.assertEqual(buf[:], "ab\0\0\0\0")
+            self.assertEqual(buf[::], "ab\0\0\0\0")
+            self.assertEqual(buf[::-1], "\0\0\0\0ba")
+
+if __name__ == '__main__':
+    unittest.main()