python mock_Python中使用mock.Mock()进行mock测试
先來(lái)一個(gè)簡(jiǎn)單的例子：
importrequestsfrom unittest importmockdefrequest_lemonfix():""":return:"""res= requests.get('http://www.lemonfix.com')return res.status_code.encode('utf-8')if __name__ == '__main__':
request_lemonfix= mock.Mock(return_value="這里會(huì)顯示論壇主頁(yè)")print(request_lemonfix())
然后再來(lái)一個(gè)實(shí)用的例子：第三方支付接口目前不能調(diào)用情況下 使用mock完成測(cè)試
importrequestsclassPayment:"""定義第三方支付類"""
defauthe(self, card_num, amount):"""請(qǐng)求第三方支付接口，并返回響應(yīng)碼
:param card_num: 卡號(hào)
:param amount: 金額
:return: 返回狀態(tài)碼200 代表支付成功，500 代表支付異常"""url= "https://www.dd.com" #第三方的url
data = {"card_num": card_num, "amount": amount}
response= requests.post(url, data=data)returnresponse.status_codedefpay(self, user_id, card_num, amount):"""支付
:param user_id: 用戶id
:param card_num: 卡號(hào)
:param amount: 金額
:return:"""
try:
status_code=self.authe(card_num, amount)exceptTimeoutError:
status_code=self.authe(card_num, amount)if status_code == 200:print("支付成功")return "success"
if status_code == 500:print("支付失敗")return "fail"
下面來(lái)看測(cè)試類：
import unittest
from payment import Payment
from unittest import mock
class PaymentTest(unittest.TestCase):
"""
測(cè)試支付接口
"""
def setUp(self):
self.payment = Payment()
def test_success(self):
"""
測(cè)試支付成功
:return:
"""
self.payment.authe = mock.Mock(return_value=200)
res = self.payment.pay(user_id=10001, card_num="123444", amount=1000)
self.assertEqual("success", res)
def test_fail(self):
"""
測(cè)試支付成功
:return:
"""
self.payment.authe = mock.Mock(return_value=500)
res = self.payment.pay(user_id=10002, card_num="1233444", amount=1000)
self.assertEqual("fail", res)
def test_retry_success(self):
"""
測(cè)試支付失敗重試在成功
side_effect 可以為序列類型 異常類型，對(duì)象
如果為序列類型 每次調(diào)用mock對(duì)象，會(huì)依次將side effcet中的元素返回
:return:
"""
self.payment.authe = mock.Mock(side_effect=[TimeoutError, 200])
res = self.payment.pay(user_id=10003, card_num="1234444", amount=1000)
self.assertEqual("success", res)
if __name__ == '__main__':
unittest.main()
總結(jié)
以上是生活随笔為你收集整理的python mock_Python中使用mock.Mock()进行mock测试的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇： 刺激战场金币怎么使用
- 下一篇： python新闻爬虫教程_python简