def test_valid_pack(self):
"""Test packing valid enum values."""
field = ('a', 'b', SimpleEnum) # signed byte: -128, 127
out = ElementEnum.valid(field)
self.assertTrue(out)
elem = ElementEnum(field)
test_values = [
({
'a': 0
}, b'\x00'),
({
'a': 2
}, b'\x02'),
({
'a': SimpleEnum.one
}, b'\x01'),
({
'a': SimpleEnum.two
}, b'\x02'),
]
for (in_val, out_val) in test_values:
with self.subTest((out_val, in_val)): # pylint: disable=no-member
ret = elem.pack(in_val)
self.assertEqual(ret, out_val)
def test_invalid_pack(self):
"""Test packing invalid enum values."""
field = ('a', 'b', SimpleEnum) # signed byte: -128, 127
out = ElementEnum.valid(field)
self.assertTrue(out)
elem = ElementEnum(field)
test_values = [
{'a': -1},
{'a': 3},
{'a': 'three'},
{'a': 'ONE'},
]
msg = '{} is not a valid {}'
for val in test_values:
with self.subTest(val): # pylint: disable=no-member
with self.assertRaises(ValueError) as cm:
elem.pack(val)
self.assertEqual(str(cm.exception), msg.format(val['a'], 'SimpleEnum'))
def test_valid_pack(self):
"""Test packing valid enum values."""
field = ('a', 'b', SimpleEnum) # signed byte: -128, 127
out = ElementEnum.valid(field)
self.assertTrue(out)
elem = ElementEnum(field)
test_values = [
({'a': 0}, b'\x00'),
({'a': 2}, b'\x02'),
({'a': SimpleEnum.one}, b'\x01'),
({'a': SimpleEnum.two}, b'\x02'),
({'a': 'two'}, b'\x02'),
]
for (in_val, out_val) in test_values:
with self.subTest((out_val, in_val)): # pylint: disable=no-member
ret = elem.pack(in_val)
self.assertEqual(ret, out_val)
