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_valid_unpack(self): """Test unpacking valid enum values.""" field = ('a', 'b', SimpleEnum) # signed byte: -128, 127 out = ElementEnum.valid(field) self.assertTrue(out) elem = ElementEnum(field) test_values = [ (SimpleEnum.zero, b'\x00'), (SimpleEnum.one, b'\x01'), (SimpleEnum.two, b'\x02'), ] for (out_val, in_val) in test_values: with self.subTest((out_val, in_val)): # pylint: disable=no-member (ret, unused) = elem.unpack({}, in_val) self.assertEqual(unused, b'') 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_invalid_unpack(self): """Test unpacking invalid enum values.""" field = ('a', 'b', SimpleEnum) # signed byte: -128, 127 out = ElementEnum.valid(field) self.assertTrue(out) elem = ElementEnum(field) test_values = [ (b'\xFF', -1), (b'\x03', 3), (b'\x7F', 127), (b'\x10', 16), (b'\x80', -128), ] # msg = '{} is not a valid {}' for (in_val, out_val) in test_values: with self.subTest((in_val, out_val)): # pylint: disable=no-member # with self.assertRaises(ValueError) as cm: with self.assertRaises(ValueError): elem.unpack({}, in_val)