本文整理汇总了Python中requests.Response.iter_content方法的典型用法代码示例。如果您正苦于以下问题:Python Response.iter_content方法的具体用法?Python Response.iter_content怎么用?Python Response.iter_content使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类requests.Response
的用法示例。
在下文中一共展示了Response.iter_content方法的4个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_normal_content_encoding
# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import iter_content [as 别名]
def test_normal_content_encoding(self):
URL = 'http://pulpproject.org/primary.xml'
req = DownloadRequest(URL, StringIO())
response = Response()
response.status_code = httplib.OK
response.iter_content = mock.MagicMock(return_value=['abc'], spec_set=response.iter_content)
session = threaded.build_session(self.config)
session.get = mock.MagicMock(return_value=response, spec_set=session.get)
report = self.downloader._fetch(req, session)
self.assertEqual(report.state, DOWNLOAD_SUCCEEDED)
self.assertEqual(report.bytes_downloaded, 3)
# passing "None" for headers lets the requests library add whatever
# headers it thinks are appropriate.
session.get.assert_called_once_with(URL, headers=None)
示例2: test_normal_content_encoding
# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import iter_content [as 别名]
def test_normal_content_encoding(self):
URL = 'http://fakeurl/primary.xml'
req = DownloadRequest(URL, StringIO())
response = Response()
response.status_code = httplib.OK
response.iter_content = mock.MagicMock(return_value=['abc'], spec_set=response.iter_content)
self.session.get = mock.MagicMock(return_value=response, spec_set=self.session.get)
report = self.downloader._fetch(req)
self.assertEqual(report.state, report.DOWNLOAD_SUCCEEDED)
self.assertEqual(report.bytes_downloaded, 3)
# passing "None" for headers lets the requests library add whatever
# headers it thinks are appropriate.
self.session.get.assert_called_once_with(
URL, headers={}, timeout=(self.config.connect_timeout, self.config.read_timeout), verify=True)
示例3: _stream_to_file
# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import iter_content [as 别名]
def _stream_to_file(self, file: Any, max_size: int,
req: requests.Response) -> None:
# start streaming download
finished = False
try:
size = 0
for chunk in req.iter_content(1024):
file.write(chunk)
size += len(chunk)
if size > max_size:
msg = 'Downloaded archive is bigger than the ' \
'allowed %i bytes' % max_size
raise MaximumDownloadSizeExceededException(msg)
finished = True
finally:
# in case any errors occurred, get rid of the file
file.close()
if not finished:
os.remove(file.name)
示例4: test_summary
# 需要导入模块: from requests import Response [as 别名]
# 或者: from requests.Response import iter_content [as 别名]
def test_summary(mock_request):
# Given
url = 'http://www.google.com'
mock_response = Response()
mock_response.headers.get = MagicMock(return_value = 'html')
mock_response.url = url
mock_response.encoding = 'UTF-8'
mock_response.consumed = False
mock_response.raw = MagicMock()
mock_response.iter_content = lambda s: ['<html><head><title>Test Title</head><body></body></html>']
mock_request.return_value = mock_response
# When
summ = Summary(url)
# summ._html = '<html><head><title>Test Title</head><body></body></html>'
summ.extract()
# Then
# mock_response.raw.close.assert_called_with()
assert summ.title == 'Test Title'