Fix memory leaks in the ctypes test suite, reported by valgrind, by

free()ing the memory we allocate.
This commit is contained in:
Thomas Heller 2006-05-05 18:43:24 +00:00
parent 21a929f5ab
commit 748f6fbf2c
1 changed files with 8 additions and 0 deletions

View File

@ -39,16 +39,19 @@ class SlicesTestCase(unittest.TestCase):
dll = CDLL(_ctypes_test.__file__)
dll.my_strdup.restype = POINTER(c_char)
dll.my_free.restype = None
res = dll.my_strdup(s)
self.failUnlessEqual(res[:len(s)], s)
import operator
self.assertRaises(TypeError, operator.setslice,
res, 0, 5, u"abcde")
dll.free(res)
dll.my_strdup.restype = POINTER(c_byte)
res = dll.my_strdup(s)
self.failUnlessEqual(res[:len(s)-1], range(ord("a"), ord("z")+1))
dll.free(res)
def test_char_array(self):
s = "abcdefghijklmnopqrstuvwxyz\0"
@ -68,12 +71,14 @@ class SlicesTestCase(unittest.TestCase):
dll = CDLL(_ctypes_test.__file__)
dll.my_wcsdup.restype = POINTER(c_wchar)
dll.my_wcsdup.argtypes = POINTER(c_wchar),
dll.my_free.restype = None
res = dll.my_wcsdup(s)
self.failUnlessEqual(res[:len(s)], s)
import operator
self.assertRaises(TypeError, operator.setslice,
res, 0, 5, u"abcde")
dll.free(res)
if sizeof(c_wchar) == sizeof(c_short):
dll.my_wcsdup.restype = POINTER(c_short)
@ -81,8 +86,11 @@ class SlicesTestCase(unittest.TestCase):
dll.my_wcsdup.restype = POINTER(c_int)
elif sizeof(c_wchar) == sizeof(c_long):
dll.my_wcsdup.restype = POINTER(c_long)
else:
return
res = dll.my_wcsdup(s)
self.failUnlessEqual(res[:len(s)-1], range(ord("a"), ord("z")+1))
dll.free(res)
################################################################