class TestFileDataWriter(unittest.TestCase):
def setUp(self):
self._dir = os.path.dirname(inspect.getfile(self.__class__))
def _file(self, filename):
return os.path.join(self._dir, "files_data_writer", filename)
def test_write_one_byte(self):
self.writer = FileDataWriter(self._file('txt_one_byte'))
self.writer.write(bytearray([0]))
self.writer.close()
with open(self._file('txt_one_byte'), "r") as file_handle:
self.assertEqual(file_handle.read(1), '\x00')
self.assertEqual(file_handle.read(1), '')
def test_write_five_bytes(self):
self.writer = FileDataWriter(self._file('txt_5_bytes'))
self.writer.write(bytearray([1, 2, 3, 4, 5]))
self.writer.close()
with open(self._file('txt_5_bytes'), "r") as file_handle:
self.assertEqual(file_handle.read(1), '\x01')
self.assertEqual(file_handle.read(1), '\x02')
self.assertEqual(file_handle.read(1), '\x03')
self.assertEqual(file_handle.read(1), '\x04')
self.assertEqual(file_handle.read(1), '\x05')
self.assertEqual(file_handle.read(1), '')
def test_write_from_empty_file(self):
self.writer = FileDataWriter(self._file('txt_empty'))
self.writer.write(bytearray())
self.writer.close()
with open(self._file('txt_empty'), "r") as file_handle:
self.assertEqual(file_handle.read(1), '')
class MyTestCase(unittest.TestCase):
def setUp(self):
self.reader = None
self._dir = os.path.dirname(inspect.getfile(self.__class__))
def _file(self, filename):
return os.path.join(self._dir, "data_files", filename)
def tearDown(self):
if self.reader is not None:
self.reader.close()
def test_one_byte(self):
myfile = self._file('txt_one_byte')
self.writer = FileDataWriter(myfile)
self.writer.write(bytearray([1]))
self.writer.close()
with open(myfile, "rb") as file_handle:
self.assertEqual(file_handle.read(1), '\x01')
self.assertEqual(file_handle.read(1), '')
self.reader = FileDataReader(myfile)
stream = self.reader.read(1)
self.assertEqual(stream[0], '\x01')
def test_readinto_one_byte(self):
myfile = self._file('txt_one_byte')
self.writer = FileDataWriter(myfile)
self.writer.write(bytearray([5]))
self.writer.close()
with open(myfile, "rb") as file_handle:
self.assertEqual(file_handle.read(1), '\x05')
self.assertEqual(file_handle.read(1), '')
self.reader = FileDataReader(myfile)
ba = bytearray(1)
self.reader.readinto(ba)
self.assertEqual(len(ba), 1)
self.assertEqual(ba[0], 5)
def test_read_five_bytes(self):
myfile = self._file('txt_5_bytes')
self.writer = FileDataWriter(myfile)
self.writer.write(bytearray([1, 2, 3, 4, 5]))
self.writer.close()
with open(myfile, "rb") as file_handle:
self.assertEqual(file_handle.read(1), '\x01')
self.assertEqual(file_handle.read(1), '\x02')
self.assertEqual(file_handle.read(1), '\x03')
self.assertEqual(file_handle.read(1), '\x04')
self.assertEqual(file_handle.read(1), '\x05')
self.assertEqual(file_handle.read(1), '')
self.reader = FileDataReader(myfile)
stream = self.reader.read(5)
self.assertEqual(len(stream), 5)
self.assertEqual(stream[0], '\x01')
self.assertEqual(stream[1], '\x02')
self.assertEqual(stream[2], '\x03')
self.assertEqual(stream[3], '\x04')
self.assertEqual(stream[4], '\x05')
def test_read_from_empty_file(self):
myfile = self._file('txt_empty')
self.writer = FileDataWriter(myfile)
self.writer.write(bytearray())
self.writer.close()
with open(myfile, "rb") as file_handle:
self.assertEqual(file_handle.read(1), '')
self.reader = FileDataReader(myfile)
stream = self.reader.read(1)
self.assertEqual(len(stream), 0)
def test_read_truncate(self):
myfile = self._file('txt_one_byte_from_multiple_bytes')
self.writer = FileDataWriter(myfile)
self.writer.write(bytearray([0xF0, 0xA4, 0xAD, 0xA2]))
self.writer.close()
with open(myfile, "rb") as file_handle:
self.assertEqual(file_handle.read(1), '\xf0')
self.assertEqual(file_handle.read(1), '\xA4')
self.assertEqual(file_handle.read(1), '\xAD')
self.assertEqual(file_handle.read(1), '\xA2')
self.assertEqual(file_handle.read(1), '')
self.reader = FileDataReader(myfile)
stream = self.reader.read(2)
self.assertEqual(len(stream), 2)
self.assertEqual(stream[0], '\xf0')
self.assertEqual(stream[1], '\xA4')