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)