| 1234567891011121314151617181920212223242526272829 |
- def testNonChunkedDownload(self):
- bytes_http = object()
- http = object()
- download_stream = six.StringIO()
- download = transfer.Download.FromStream(download_stream, total_size=52)
- download.bytes_http = bytes_http
- base_url = 'https://part.one/'
- with mock.patch.object(http_wrapper, 'MakeRequest',
- autospec=True) as make_request:
- make_request.return_value = http_wrapper.Response(
- info={
- 'content-range': 'bytes 0-51/52',
- 'status': http_client.OK,
- },
- content=string.ascii_lowercase * 2,
- request_url=base_url,
- )
- request = http_wrapper.Request(url='https://part.one/')
- download.InitializeDownload(request, http=http)
- self.assertEqual(1, make_request.call_count)
- received_request = make_request.call_args[0][1]
- self.assertEqual(base_url, received_request.url)
- self.assertRangeAndContentRangeCompatible(
- received_request, make_request.return_value)
- download_stream.seek(0)
- self.assertEqual(string.ascii_lowercase * 2,
- download_stream.getvalue())
|
