/Lib/ctypes/test/test_funcptr.py
http://unladen-swallow.googlecode.com/ · Python · 127 lines · 80 code · 31 blank · 16 comment · 6 complexity · 910303d64b2b5585e8ede93434ff7ef6 MD5 · raw file
- import os, unittest
- from ctypes import *
- try:
- WINFUNCTYPE
- except NameError:
- # fake to enable this test on Linux
- WINFUNCTYPE = CFUNCTYPE
- import _ctypes_test
- lib = CDLL(_ctypes_test.__file__)
- class CFuncPtrTestCase(unittest.TestCase):
- def test_basic(self):
- X = WINFUNCTYPE(c_int, c_int, c_int)
- def func(*args):
- return len(args)
- x = X(func)
- self.failUnlessEqual(x.restype, c_int)
- self.failUnlessEqual(x.argtypes, (c_int, c_int))
- self.failUnlessEqual(sizeof(x), sizeof(c_voidp))
- self.failUnlessEqual(sizeof(X), sizeof(c_voidp))
- def test_first(self):
- StdCallback = WINFUNCTYPE(c_int, c_int, c_int)
- CdeclCallback = CFUNCTYPE(c_int, c_int, c_int)
- def func(a, b):
- return a + b
- s = StdCallback(func)
- c = CdeclCallback(func)
- self.failUnlessEqual(s(1, 2), 3)
- self.failUnlessEqual(c(1, 2), 3)
- # The following no longer raises a TypeError - it is now
- # possible, as in C, to call cdecl functions with more parameters.
- #self.assertRaises(TypeError, c, 1, 2, 3)
- self.failUnlessEqual(c(1, 2, 3, 4, 5, 6), 3)
- if not WINFUNCTYPE is CFUNCTYPE and os.name != "ce":
- self.assertRaises(TypeError, s, 1, 2, 3)
- def test_structures(self):
- WNDPROC = WINFUNCTYPE(c_long, c_int, c_int, c_int, c_int)
- def wndproc(hwnd, msg, wParam, lParam):
- return hwnd + msg + wParam + lParam
- HINSTANCE = c_int
- HICON = c_int
- HCURSOR = c_int
- LPCTSTR = c_char_p
- class WNDCLASS(Structure):
- _fields_ = [("style", c_uint),
- ("lpfnWndProc", WNDPROC),
- ("cbClsExtra", c_int),
- ("cbWndExtra", c_int),
- ("hInstance", HINSTANCE),
- ("hIcon", HICON),
- ("hCursor", HCURSOR),
- ("lpszMenuName", LPCTSTR),
- ("lpszClassName", LPCTSTR)]
- wndclass = WNDCLASS()
- wndclass.lpfnWndProc = WNDPROC(wndproc)
- WNDPROC_2 = WINFUNCTYPE(c_long, c_int, c_int, c_int, c_int)
- # This is no longer true, now that WINFUNCTYPE caches created types internally.
- ## # CFuncPtr subclasses are compared by identity, so this raises a TypeError:
- ## self.assertRaises(TypeError, setattr, wndclass,
- ## "lpfnWndProc", WNDPROC_2(wndproc))
- # instead:
- self.failUnless(WNDPROC is WNDPROC_2)
- # 'wndclass.lpfnWndProc' leaks 94 references. Why?
- self.failUnlessEqual(wndclass.lpfnWndProc(1, 2, 3, 4), 10)
- f = wndclass.lpfnWndProc
- del wndclass
- del wndproc
- self.failUnlessEqual(f(10, 11, 12, 13), 46)
- def test_dllfunctions(self):
- def NoNullHandle(value):
- if not value:
- raise WinError()
- return value
- strchr = lib.my_strchr
- strchr.restype = c_char_p
- strchr.argtypes = (c_char_p, c_char)
- self.failUnlessEqual(strchr("abcdefghi", "b"), "bcdefghi")
- self.failUnlessEqual(strchr("abcdefghi", "x"), None)
- strtok = lib.my_strtok
- strtok.restype = c_char_p
- # Neither of this does work: strtok changes the buffer it is passed
- ## strtok.argtypes = (c_char_p, c_char_p)
- ## strtok.argtypes = (c_string, c_char_p)
- def c_string(init):
- size = len(init) + 1
- return (c_char*size)(*init)
- s = "a\nb\nc"
- b = c_string(s)
- ## b = (c_char * (len(s)+1))()
- ## b.value = s
- ## b = c_string(s)
- self.failUnlessEqual(strtok(b, "\n"), "a")
- self.failUnlessEqual(strtok(None, "\n"), "b")
- self.failUnlessEqual(strtok(None, "\n"), "c")
- self.failUnlessEqual(strtok(None, "\n"), None)
- if __name__ == '__main__':
- unittest.main()