def test_download_to_stream_source_header(self, mock_get):
mock_get.return_value.status_code = 200
mock_get.return_value.content = b'xxx'
mock_get.return_value.headers = {'data': 'test'}
response = HttpResponse('test', content_type='text/javascript')
download_to_stream('fake_url', response)
self.assertEqual(response['data'], 'test')
def test_download_to_stream_source_error_io(self, mock_write, mock_get):
mock_get.return_value.status_code = 200
mock_get.return_value.content = b'xxx'
with self.assertRaises(IOError):
download_to_stream(
'fake_url', HttpResponse('test',
content_type='text/javascript'))
def test_headers_can_be_specified_for_download(self, get_mocked):
# Required to specified language for example
download_to_stream('http://google.com',
open(os.devnull),
silent=True,
headers={'Accept-language': 'fr'})
get_mocked.assert_called_with('http://google.com',
headers={'Accept-language': 'fr'})
def test_headers_can_be_specified_for_download(self, get_mocked):
# Required to specified language for example
get_mocked.return_value.status_code = 200
get_mocked.return_value.content = "x"
download_to_stream('http://google.com',
open(os.devnull, 'w'),
silent=True,
headers={'Accept-language': 'fr'})
get_mocked.assert_called_with('http://google.com',
headers={'Accept-language': 'fr'})
def test_download_to_stream_source_is_None(self, mock_get):
def itfails(url, headers):
raise requests.exceptions.ConnectionError()
mock_get.side_effect = itfails
source = download_to_stream('fake_url', 'fake_stream', silent=True)
self.assertIsNone(source)
def test_headers_can_be_specified_for_download(self, get_mocked):
# Required to specified language for example
get_mocked.return_value.status_code = 200
get_mocked.return_value.content = "x"
download_to_stream('http://google.com', open(os.devnull, 'w'), silent=True, headers={'Accept-language': 'fr'})
get_mocked.assert_called_with('http://google.com', headers={'Accept-language': 'fr'})