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])
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])
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)
