def test_int_handler_basic(self):
with self.assertRaises(AttributeError):
self.state.attribute
handler = int_handler(self.state, 'attribute')
self.assertEqual(self.state.attribute, 0)
handler(b'\x00\x00\x00\x01')
self.assertEqual(self.state.attribute, 1)
def test_int_handler_basic(self):
with self.assertRaises(AttributeError):
self.state.attribute
handler = int_handler(self.state, "attribute")
self.assertEqual(self.state.attribute, 0)
handler(b"\x00\x00\x00\x01")
self.assertEqual(self.state.attribute, 1)
def test_int_handler_enum(self):
class TestWrapper(enum.Enum):
ValOne = 1
ValTwo = 2
handler = int_handler(self.state, 'attribute', wrapper=TestWrapper)
handler(b'\x00\x00\x00\x02')
self.assertEqual(self.state.attribute, TestWrapper.ValTwo)
def test_int_handler_enum(self):
class TestWrapper(enum.Enum):
ValOne = 1
ValTwo = 2
handler = int_handler(self.state, 'attribute', wrapper=TestWrapper)
handler(b'\x00\x00\x00\x02')
self.assertEqual(self.state.attribute, TestWrapper.ValTwo)
def test_int_handler_wrapper(self):
wrapper = lambda x: x * x
handler = int_handler(self.state, 'attribute', wrapper=wrapper)
handler(b'\x00\x00\x00\x04')
self.assertEqual(self.state.attribute, 0x10)
def test_int_handler_shift(self):
handler = int_handler(self.state, 'attribute', shift=8)
handler(b'\xF0\xFF\xFF\x0F')
self.assertEqual(self.state.attribute, 0xFFF0FFFF)
def test_int_handler_mask(self):
handler = int_handler(self.state, 'attribute', mask=0xFF00)
handler(b'\x00\x00\xFF\xFF')
self.assertEqual(self.state.attribute, 0xFF00)
def test_int_handler_enum(self):
handler = int_handler(self.state, 'attribute', wrapper=TestEnum)
handler(b'\x00\x00\x00\x02')
self.assertEqual(self.state.attribute, TestEnum.ValTwo)
def test_int_handler_enum(self):
handler = int_handler(self.state, "attribute", wrapper=TestEnum)
handler(b"\x00\x00\x00\x02")
self.assertEqual(self.state.attribute, TestEnum.ValTwo)
def test_int_handler_shift(self):
handler = int_handler(self.state, 'attribute', shift=8)
handler(b'\xF0\xFF\xFF\x0F')
self.assertEqual(self.state.attribute, 0xFFF0FFFF)
def test_int_handler_wrapper(self):
wrapper = lambda x: x * x
handler = int_handler(self.state, "attribute", wrapper=wrapper)
handler(b"\x00\x00\x00\x04")
self.assertEqual(self.state.attribute, 0x10)
def test_int_handler_shift(self):
handler = int_handler(self.state, "attribute", shift=8)
handler(b"\xF0\xFF\xFF\x0F")
self.assertEqual(self.state.attribute, 0xFFF0FFFF)
def test_int_handler_mask(self):
handler = int_handler(self.state, "attribute", mask=0xFF00)
handler(b"\x00\x00\xFF\xFF")
self.assertEqual(self.state.attribute, 0xFF00)
def test_int_handler_default(self):
handler = int_handler(self.state, 'attribute', default=7)
self.assertEqual(self.state.attribute, 7)
handler(b'\x00\x00\x00\x08')
self.assertEqual(self.state.attribute, 8)
def test_int_handler_default(self):
handler = int_handler(self.state, 'attribute', default=7)
self.assertEqual(self.state.attribute, 7)
handler(b'\x00\x00\x00\x08')
self.assertEqual(self.state.attribute, 8)
def test_int_handler_default(self):
handler = int_handler(self.state, "attribute", default=7)
self.assertEqual(self.state.attribute, 7)
handler(b"\x00\x00\x00\x08")
self.assertEqual(self.state.attribute, 8)
def test_int_handler_mask(self):
handler = int_handler(self.state, 'attribute', mask=0xFF00)
handler(b'\x00\x00\xFF\xFF')
self.assertEqual(self.state.attribute, 0xFF00)
