def test_chunkwise_transfer_with_exception(self): """Check if we see the exception raised during reading.""" data = self.random_string(1024) fobj = FailingStringIO(data) iterator = file_transfer.chunks(fobj, 256) self.assertRaises(FailingStringIO.expected_exception, iterator.next)
def test_chunkwise_transfer_with_remainder(self): """Check if we get three chunks with 256 Bytes and one with 253.""" data = self.random_string(1021) fobj = StringIO.StringIO(data) chunks = list(file_transfer.chunks(fobj, 256)) self.assertEqual(len(chunks), 4) self.assertEqual(chunks[0], data[:256]) self.assertEqual(chunks[1], data[256:512]) self.assertEqual(chunks[2], data[512:768]) self.assertEqual(chunks[3], data[768:1021])