You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.4 KiB
75 lines
2.4 KiB
6 years ago
|
import unittest
|
||
|
from ctypes import *
|
||
|
|
||
|
class StructFieldsTestCase(unittest.TestCase):
|
||
|
# Structure/Union classes must get 'finalized' sooner or
|
||
|
# later, when one of these things happen:
|
||
|
#
|
||
|
# 1. _fields_ is set.
|
||
|
# 2. An instance is created.
|
||
|
# 3. The type is used as field of another Structure/Union.
|
||
|
# 4. The type is subclassed
|
||
|
#
|
||
|
# When they are finalized, assigning _fields_ is no longer allowed.
|
||
|
|
||
|
def test_1_A(self):
|
||
|
class X(Structure):
|
||
|
pass
|
||
|
self.assertEqual(sizeof(X), 0) # not finalized
|
||
|
X._fields_ = [] # finalized
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
|
||
|
def test_1_B(self):
|
||
|
class X(Structure):
|
||
|
_fields_ = [] # finalized
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
|
||
|
def test_2(self):
|
||
|
class X(Structure):
|
||
|
pass
|
||
|
X()
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
|
||
|
def test_3(self):
|
||
|
class X(Structure):
|
||
|
pass
|
||
|
class Y(Structure):
|
||
|
_fields_ = [("x", X)] # finalizes X
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
|
||
|
def test_4(self):
|
||
|
class X(Structure):
|
||
|
pass
|
||
|
class Y(X):
|
||
|
pass
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
Y._fields_ = []
|
||
|
self.assertRaises(AttributeError, setattr, X, "_fields_", [])
|
||
|
|
||
|
# __set__ and __get__ should raise a TypeError in case their self
|
||
|
# argument is not a ctype instance.
|
||
|
def test___set__(self):
|
||
|
class MyCStruct(Structure):
|
||
|
_fields_ = (("field", c_int),)
|
||
|
self.assertRaises(TypeError,
|
||
|
MyCStruct.field.__set__, 'wrong type self', 42)
|
||
|
|
||
|
class MyCUnion(Union):
|
||
|
_fields_ = (("field", c_int),)
|
||
|
self.assertRaises(TypeError,
|
||
|
MyCUnion.field.__set__, 'wrong type self', 42)
|
||
|
|
||
|
def test___get__(self):
|
||
|
class MyCStruct(Structure):
|
||
|
_fields_ = (("field", c_int),)
|
||
|
self.assertRaises(TypeError,
|
||
|
MyCStruct.field.__get__, 'wrong type self', 42)
|
||
|
|
||
|
class MyCUnion(Union):
|
||
|
_fields_ = (("field", c_int),)
|
||
|
self.assertRaises(TypeError,
|
||
|
MyCUnion.field.__get__, 'wrong type self', 42)
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
unittest.main()
|