我有测试:
class MyTests(TestCase):
def setUp(self):
self.myclient = MyClient()
@mock.patch('the_file.requests.json')
def test_myfunc(self, mock_item):
mock_item.return_value = [
{'itemId': 1},
{'itemId': 2},
]
item_ids = self.myclient.get_item_ids()
self.assertEqual(item_ids, [1, 2])
在我的文件中
import requests
class MyClient(object):
def get_product_info(self):
response = requests.get(PRODUCT_INFO_URL)
return response.json()
我的目标是模拟 get_product_info()
以返回测试中的 return_value
数据。我尝试过模拟 requests.json
和 requests.get.json
,两者都在没有属性上出错,我模拟了 the_file.MyClient.get_product_info
这不会导致错误,但不起作用,它返回真实数据。
我如何模拟这个使用请求库的get_product_info
?
最佳答案
您应该能够只修补 get_product_info()
。
from unittest.mock import patch
class MyClient(object):
def get_product_info(self):
return 'x'
with patch('__main__.MyClient.get_product_info', return_value='z'):
client = MyClient()
info = client.get_product_info()
print('Info is {}'.format(info))
# >> Info is z
只需将 __main__
切换为您的模块名称即可。您可能还会发现patch.object
有用。
关于python - 在 Python 中获得响应后模拟 requests.json,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43401459/