def setUp(self):
super(TestChunkDownloadHandler, self).setUp()
self.chunks = [
{'url': 'http://1.2.3.4:6000/AAAA', 'pos': '0', 'size': 32},
{'url': 'http://1.2.3.4:6001/BBBB', 'pos': '0', 'size': 32}]
self.size = 32
self.offset = 0
self.handler = ChunkDownloadHandler(
self.chunks, self.size, self.offset)
class TestChunkDownloadHandler(unittest.TestCase):
def setUp(self):
super(TestChunkDownloadHandler, self).setUp()
self.chunks = [
{'url': 'http://1.2.3.4:6000/AAAA', 'pos': '0', 'size': 32},
{'url': 'http://1.2.3.4:6001/BBBB', 'pos': '0', 'size': 32}]
self.size = 32
self.offset = 0
self.handler = ChunkDownloadHandler(
self.chunks, self.size, self.offset)
@patch('oiopy.object_storage.close_source')
@patch('oiopy.object_storage.ChunkDownloadHandler._get_chunk_source')
def test_get_stream(self, mock_chunk_source, mock_close):
parts = ('foo', 'bar')
source = TestSource(parts)
mock_chunk_source.return_value = source
stream = self.handler.get_stream()
d = ''.join(stream)
self.assertTrue(d, 'foobar')
mock_close.assert_called_once_with(source)
@patch('oiopy.object_storage.close_source')
def test_make_stream(self, mock_close):
source = TestSource(('foo', 'bar'))
stream = self.handler._make_stream(source)
d = ''.join(stream)
self.assertTrue(d, 'foobar')
mock_close.assert_called_once_with(source)
@patch('oiopy.object_storage.close_source')
def test_make_stream_timeout(self, mock_close):
h = self.handler
source = TestSource(('foo', None))
source2 = TestSource(('bar',))
h._fast_forward = Mock()
stream = h._make_stream(source)
with patch.object(h, '_get_chunk_source', lambda: source2):
parts = list(stream)
self.assertEqual(parts, ['foo', 'bar'])
h._fast_forward.assert_called_once_with(len('foo'))
mock_close.assert_any_call(source)
mock_close.assert_any_call(source2)
self.assertEqual(mock_close.call_count, 2)
@patch('oiopy.object_storage.close_source')
def test_make_stream_timeout_resume_failure(self, mock_close):
h = self.handler
source = TestSource(('foo', None))
h._fast_forward = Mock()
stream = h._make_stream(source)
with patch.object(h, '_get_chunk_source', lambda: None):
self.assertEqual('foo', next(stream))
with self.assertRaises(ChunkReadTimeout):
next(stream)
h._fast_forward.assert_called_once_with(len('foo'))
mock_close.assert_called_once_with(source)
def test_get_chunk_source(self):
with set_http_connect(200, body='foobar'):
source = self.handler._get_chunk_source()
self.assertEqual('foobar', source.read())
def test_get_chunk_source_resume_timeout(self):
h = self.handler
with set_http_connect(Timeout(), 200, body='foobar'):
source = h._get_chunk_source()
self.assertEqual('foobar', source.read())
self.assertEqual(h.failed_chunks, [self.chunks[0]])
def test_get_chunk_source_resume_404(self):
h = self.handler
with set_http_connect(404, 200, body='foobar'):
source = h._get_chunk_source()
self.assertEqual('foobar', source.read())
self.assertEqual(h.failed_chunks, [self.chunks[0]])