[pypy-svn] r51297 - pypy/dist/pypy/lib/app_test/ctypes

[fijal at codespeak.net]

Author: fijal
Date: Tue Feb  5 16:56:32 2008
New Revision: 51297

Added:
   pypy/dist/pypy/lib/app_test/ctypes/test_unicode.py   (contents, props changed)
Log:
Port this test, not working.


Added: pypy/dist/pypy/lib/app_test/ctypes/test_unicode.py
==============================================================================
--- (empty file)
+++ pypy/dist/pypy/lib/app_test/ctypes/test_unicode.py	Tue Feb  5 16:56:32 2008
@@ -0,0 +1,114 @@
+# coding: latin-1
+import ctypes
+import py
+py.test.skip("Unsupported")
+
+try:
+    ctypes.c_wchar
+except AttributeError:
+    pass
+else:
+    def setup_module(mod):
+        import conftest
+        dll = ctypes.CDLL(str(conftest.sofile))
+        mod.wcslen = dll.my_wcslen
+        mod.wcslen.argtypes = [ctypes.c_wchar_p]
+        mod.func = dll._testfunc_p_p
+
+    class TestUnicode:
+        def setup_method(self, method):
+            self.prev_conv_mode = ctypes.set_conversion_mode("ascii", "strict")
+
+        def teardown_method(self, method):
+            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
+            assert wcslen(u"abc") == 3
+            assert wcslen(u"ab\u2070") == 3
+            # string args are converted
+            assert wcslen("abc") == 3
+            raises(ctypes.ArgumentError, wcslen, "abä")
+
+        def test_ascii_replace(self):
+            ctypes.set_conversion_mode("ascii", "replace")
+            assert wcslen(u"abc") == 3
+            assert wcslen(u"ab\u2070") == 3
+            assert wcslen("abc") == 3
+            assert wcslen("abä") == 3
+
+        def test_ascii_ignore(self):
+            ctypes.set_conversion_mode("ascii", "ignore")
+            assert wcslen(u"abc") == 3
+            assert wcslen(u"ab\u2070") == 3
+            # ignore error mode skips non-ascii characters
+            assert wcslen("abc") == 3
+            assert wcslen("äöüß") == 0
+
+        def test_latin1_strict(self):
+            ctypes.set_conversion_mode("latin-1", "strict")
+            assert wcslen(u"abc") == 3
+            assert wcslen(u"ab\u2070") == 3
+            assert wcslen("abc") == 3
+            assert wcslen("äöüß") == 4
+
+        def test_buffers(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            buf = ctypes.create_unicode_buffer("abc")
+            assert len(buf) == 3+1
+
+            ctypes.set_conversion_mode("ascii", "replace")
+            buf = ctypes.create_unicode_buffer("abäöü")
+            assert buf[:] == u"ab\uFFFD\uFFFD\uFFFD\0"
+
+            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..
+            assert buf[:] == u"ab\0\0\0\0"
+
+    class TestString(TestUnicode):
+        def setup_method(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_method(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")
+            assert func("abc") == "abc"
+            assert func(u"abc") == "abc"
+            raises(ctypes.ArgumentError, func, u"abä")
+
+        def test_ascii_ignore(self):
+            ctypes.set_conversion_mode("ascii", "ignore")
+            assert func("abc") == "abc"
+            assert func(u"abc") == "abc"
+            assert func(u"äöüß") == ""
+
+        def test_ascii_replace(self):
+            ctypes.set_conversion_mode("ascii", "replace")
+            assert func("abc") == "abc"
+            assert func(u"abc") == "abc"
+            assert func(u"äöüß") == "????"
+
+        def test_buffers(self):
+            ctypes.set_conversion_mode("ascii", "strict")
+            buf = ctypes.create_string_buffer(u"abc")
+            assert len(buf) == 3+1
+
+            ctypes.set_conversion_mode("ascii", "replace")
+            buf = ctypes.create_string_buffer(u"abäöü")
+            assert buf[:] == "ab???\0"
+
+            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..
+            assert buf[:] == "ab\0\0\0\0"
+
+if __name__ == '__main__':
+    unittest.main()