def test_read_byte(self):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([1, 2, 3, 4, 5])))
self.assertEqual(reader.read_byte(),1)
self.assertEqual(reader.read_byte(),2)
self.assertEqual(reader.read_byte(),3)
self.assertEqual(reader.read_byte(),4)
self.assertEqual(reader.read_byte(),5)
with self.assertRaises(EOFError):
reader.read_byte()
def test_read_long_not_enough_bytes(self):
with self.assertRaises(EOFError):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([0xF1, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7])))
reader.read_long()
def test_read_long(self):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([0xF1, 0xF2, 0xF3, 0xF4, 0xF5, 0xF6, 0xF7, 0xF8])))
self.assertEqual(reader.read_long(),0xF8F7F6F5F4F3F2F1)
def test_read_int_not_enough_bytes(self):
with self.assertRaises(EOFError):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([0xF1, 0xF2, 0xF3])))
reader.read_int()
def test_read_int(self):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([0xF1, 0xF2, 0xF3, 0xF4])))
self.assertEqual(reader.read_int(),0xF4F3F2F1)
def test_read_bytes_error(self):
with self.assertRaises(EOFError):
reader = LittleEndianDataReaderByteReader(MyDataReader(bytearray([1, 2, 3, 4, 5])))
reader.read_bytes(6)
def test_read_bytes(self):
ba = bytearray([1, 2, 3, 4, 5])
reader = LittleEndianDataReaderByteReader(MyDataReader(ba))
stream = reader.read_bytes(5)
for i in range(5):
self.assertEqual(stream[i],ba[i])
