本文整理汇总了Python中unittest.mock.MagicMock.headers方法的典型用法代码示例。如果您正苦于以下问题:Python MagicMock.headers方法的具体用法?Python MagicMock.headers怎么用?Python MagicMock.headers使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类unittest.mock.MagicMock
的用法示例。
在下文中一共展示了MagicMock.headers方法的8个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_prefer_name_from_headers
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_prefer_name_from_headers():
response = MagicMock()
response.headers = {}
response.url = 'http://example.com/abc.mp3'
assert get_file_name(response) == 'abc.mp3'
response.headers = {'Content-Disposition': 'attachment;filename="def.mp3"'}
assert get_file_name(response) == 'def.mp3'
示例2: test_get_file_name_when_content_disposition_filename_is_missing
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_get_file_name_when_content_disposition_filename_is_missing():
# Example of a feed with such files:
# http://feeds.wnyc.org/radiolab?format=xml
response = MagicMock()
response.headers = {'Content-Disposition': 'attachment'}
response.url = 'http://feeds.wnyc.org/~r/radiolab/~5/KYQG_JtkTYM/radiolab_podcast16cellmates.mp3'
assert get_file_name(response) == 'radiolab_podcast16cellmates.mp3'
示例3: test_SetsResponseStatusCodeAndResponseDataAndLocationheader
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_SetsResponseStatusCodeAndResponseDataAndLocationheader(self):
mock_response = MagicMock()
mock_response.status = 200
mock_response.headers = { "Location": "the location" }
sut = responseBuilder.ResponseBuilder()
sut._read_response = MagicMock(return_value="response data")
expected = "response data", 200, "the location"
actual = sut.process_response(mock_response, True)
self.assertEqual(expected, actual)
示例4: build
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def build(self):
m = MagicMock()
m.text = self._text
m.url = self._url
m.status_code = self._status_code
m.headers = CaseInsensitiveDict({
"content-type": self._content_type
})
m.content = self._content
m.encoding = self._encoding
m.elapsed = datetime.timedelta(seconds=self._seconds, microseconds=self._microseconds)
m.json.return_value = self._json
return m
示例5: test_query_success
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_query_success(loop, vm):
response = MagicMock()
response.status = 200
response.headers = {'CONTENT-TYPE': 'application/json'}
async def read():
return b'{"c": false}'
response.read.side_effect = read
vm._session.request = AsyncioMagicMock(return_value=response)
data = loop.run_until_complete(asyncio.ensure_future(vm.query("POST", "test", data={"a": True}, params={"b": 1})))
vm._session.request.assert_called_with('POST',
'http://docker/v1.25/test',
data='{"a": true}',
headers={'content-type': 'application/json'},
params={'b': 1},
timeout=300)
assert data == {"c": False}
示例6: test_query_success
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_query_success(loop, vm):
response = MagicMock()
response.status = 200
response.headers = {'CONTENT-TYPE': 'application/json'}
@asyncio.coroutine
def read():
return b'{"c": false}'
response.read.side_effect = read
with asyncio_patch("aiohttp.request", return_value=response) as mock:
data = loop.run_until_complete(asyncio.async(vm.query("POST", "test", data={"a": True}, params={"b": 1})))
mock.assert_called_with('POST',
'http://docker/test',
connector=vm._connector,
data='{"a": true}',
headers={'content-type': 'application/json'},
params={'b': 1})
assert data == {"c": False}
示例7: make_response
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def make_response(self, json_string, status=200, headers=None):
response = MagicMock()
response.status_code = status
response.headers = headers or dict()
response.content = json_string
return ResponseWrapper(response)
示例8: test_get_file_name_from_header
# 需要导入模块: from unittest.mock import MagicMock [as 别名]
# 或者: from unittest.mock.MagicMock import headers [as 别名]
def test_get_file_name_from_header():
# Example of a feed with such files: http://feeds.podtrac.com/fNBYjm-45_h8
response = MagicMock()
response.headers = {
'Content-Disposition': 'attachment;filename="2015-11-10.mp3"'}
assert get_file_name(response) == '2015-11-10.mp3'