mirror of
https://github.com/QB64Official/qb64.git
synced 2024-09-19 14:25:15 +00:00
75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
import unittest
|
|
from ctypes import *
|
|
|
|
import _ctypes_test
|
|
|
|
lib = CDLL(_ctypes_test.__file__)
|
|
|
|
class StringPtrTestCase(unittest.TestCase):
|
|
|
|
def test__POINTER_c_char(self):
|
|
class X(Structure):
|
|
_fields_ = [("str", POINTER(c_char))]
|
|
x = X()
|
|
|
|
# NULL pointer access
|
|
self.assertRaises(ValueError, getattr, x.str, "contents")
|
|
b = c_buffer("Hello, World")
|
|
from sys import getrefcount as grc
|
|
self.assertEqual(grc(b), 2)
|
|
x.str = b
|
|
self.assertEqual(grc(b), 3)
|
|
|
|
# POINTER(c_char) and Python string is NOT compatible
|
|
# POINTER(c_char) and c_buffer() is compatible
|
|
for i in range(len(b)):
|
|
self.assertEqual(b[i], x.str[i])
|
|
|
|
self.assertRaises(TypeError, setattr, x, "str", "Hello, World")
|
|
|
|
def test__c_char_p(self):
|
|
class X(Structure):
|
|
_fields_ = [("str", c_char_p)]
|
|
x = X()
|
|
|
|
# c_char_p and Python string is compatible
|
|
# c_char_p and c_buffer is NOT compatible
|
|
self.assertEqual(x.str, None)
|
|
x.str = "Hello, World"
|
|
self.assertEqual(x.str, "Hello, World")
|
|
b = c_buffer("Hello, World")
|
|
self.assertRaises(TypeError, setattr, x, "str", b)
|
|
|
|
|
|
def test_functions(self):
|
|
strchr = lib.my_strchr
|
|
strchr.restype = c_char_p
|
|
|
|
# c_char_p and Python string is compatible
|
|
# c_char_p and c_buffer are now compatible
|
|
strchr.argtypes = c_char_p, c_char
|
|
self.assertEqual(strchr("abcdef", "c"), "cdef")
|
|
self.assertEqual(strchr(c_buffer("abcdef"), "c"), "cdef")
|
|
|
|
# POINTER(c_char) and Python string is NOT compatible
|
|
# POINTER(c_char) and c_buffer() is compatible
|
|
strchr.argtypes = POINTER(c_char), c_char
|
|
buf = c_buffer("abcdef")
|
|
self.assertEqual(strchr(buf, "c"), "cdef")
|
|
self.assertEqual(strchr("abcdef", "c"), "cdef")
|
|
|
|
# XXX These calls are dangerous, because the first argument
|
|
# to strchr is no longer valid after the function returns!
|
|
# So we must keep a reference to buf separately
|
|
|
|
strchr.restype = POINTER(c_char)
|
|
buf = c_buffer("abcdef")
|
|
r = strchr(buf, "c")
|
|
x = r[0], r[1], r[2], r[3], r[4]
|
|
self.assertEqual(x, ("c", "d", "e", "f", "\000"))
|
|
del buf
|
|
# x1 will NOT be the same as x, usually:
|
|
x1 = r[0], r[1], r[2], r[3], r[4]
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|