def test_wrong_get_remaining(self):
self.assertEqual(get_remaining(None), None)
self.assertRaises(TypeError, get_remaining, object)
self.assertEqual(get_remaining(b'\x80\x80'), -1)
self.assertEqual(get_remaining(b'\x00\x80'), -1)
self.assertEqual(get_remaining(b'\x80\x01\x00\x00'), -1)
self.assertRaises(TypeError, get_remaining, None, exception=True)
self.assertRaises(TypeError, get_remaining, object, exception=True)
self.assertRaises(struct.error, get_remaining, b'\x80\x80', exception=True)
self.assertRaises(struct.error, get_remaining, b'\x00\x80', exception=True)
self.assertRaises(struct.error, get_remaining, b'\x00\x01\x00\x00', exception=True)
self.assertRaises(struct.error, get_remaining, b'\x80\x01\x00', exception=True)
def test_get_remaining(self):
self.assertEqual(get_remaining(b'\x00', exception=False), 0)
self.assertEqual(get_remaining(b'\x7f', exception=False), 127)
self.assertEqual(get_remaining(b'\x80\x01', exception=False), 128)
self.assertEqual(get_remaining(b'\xff\x7f', exception=False), 16383)
self.assertEqual(get_remaining(b'\x80\x80\x01', exception=False), 16384)
self.assertEqual(get_remaining(b'\xff\xff\x7f', exception=False), 2097151)
self.assertEqual(get_remaining(b'\x80\x80\x80\x01', exception=False), 2097152)
self.assertEqual(get_remaining(b'\xff\xff\xff\x7f', exception=False), 268435455)