mock總所周知是模擬的意思,我們?cè)谧鼋涌跍y試的時(shí)候有時(shí)候會(huì)發(fā)現(xiàn)部分功能依賴其他業(yè)務(wù)場景或者第三方功能或者是線上數(shù)據(jù)或者業(yè)務(wù)場景過于復(fù)雜(需要大量調(diào)用)的情況,沒有辦法通過接口調(diào)用或者做斷言,這個(gè)時(shí)候就需要mock了

python2中需要pip install mock來導(dǎo)入mock第三方模塊,而python3中被引入到unittest框架中,直接from unittest import mock 就可以導(dǎo)入mock模塊了

在單元測試中只針對(duì)當(dāng)前單元做測試,就是測試當(dāng)前方法或者是當(dāng)前類,但是如果出現(xiàn)這個(gè)方法依賴其他第三方模塊,那不是還要校驗(yàn)其他方法了嗎,那就違背了單元測試這個(gè)概念了,所有將外部依賴組件實(shí)現(xiàn)模擬并替換掉,使得單元測試的焦點(diǎn)放在自己身上

為了演示,用fastapi模擬了一個(gè)tx和ali的接口來調(diào)用作為單元測試,可以看到需要同時(shí)調(diào)用兩個(gè)接口才能完成單元測試

from unittest import TestCase, main, mockimport requestsclass Request(object): def request_tx(self): tx_url = ’http://127.0.0.1:8000/tx’ tx_data = {'a': 1, 'b': 2} tx_result = requests.post(tx_url, json=tx_data) tx_result = tx_result.json() return tx_result def request_ali(self): ali_url = ’http://127.0.0.1:8000/ali’ ali_data = {'c': 1, 'd': 2} ali_result = requests.post(ali_url, json=ali_data) ali_result = ali_result.json() return ali_resultclass Myproject(TestCase): def test_api(self): r = Request() ali_result = r.request_ali() tx_result = r.request_tx() self.assertEqual(int(ali_result[’result’][’all_add_value’]) + int(tx_result[’result’][’all_add_value’]), 6)

’’’@File : mock_third_server.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’from fastapi import FastAPIfrom pydantic import BaseModelimport uvicornapp = FastAPI()def add(a, b): return a + b# json參數(shù)傳參class TX(BaseModel): a: int b: int@app.post(’/tx’)def txApi(u: TX): u.a += 1 u.b -= 1 add_result = add(u.a, u.b) return {’status’: 1, ’result’: {’value_a’: u.a, ’value_b’: u.b, ’all_add_value’: add_result}}class Al(BaseModel): c: int d: int@app.post(’/ali’)def aliApi(u: Al): u.c *= 1 u.d /= 1 add_result = add(u.c, u.d) return {’status’: 1, ’result’: {’value_c’: u.c, ’value_d’: u.d, ’all_add_value’: add_result}}if __name__ == ’__main__’: # 一定要加上這句話才行 uvicorn.run(’mock_third_server:app’, port=8000, debug=True)

可以看到上面的接口是可以調(diào)通的如果txapi因?yàn)橐恍┰虿荒芊祷?ali的接口是通的,如果要實(shí)現(xiàn)ali接口的測試就需要使用mock來模擬tx的接口了

mock方法一:(對(duì)于同一文件下/不同文件下的方法或者類進(jìn)行mock)

’’’@File : mock_third_server.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’from fastapi import FastAPIfrom pydantic import BaseModelimport uvicornapp = FastAPI()def add(a, b): return a + b# json參數(shù)傳參class TX(BaseModel): a: int b: int@app.post(’/tx’)def txApi(u: TX): u.a += 1 u.b -= 1 add_result = add(u.a, u.b) raise('接口異常') # 在這里加了異常 return {’status’: 1, ’result’: {’value_a’: u.a, ’value_b’: u.b, ’all_add_value’: add_result}}class Al(BaseModel): c: int d: int@app.post(’/ali’)def aliApi(u: Al): u.c *= 1 u.d /= 1 add_result = add(u.c, u.d) return {’status’: 1, ’result’: {’value_c’: u.c, ’value_d’: u.d, ’all_add_value’: add_result}}if __name__ == ’__main__’: # 一定要加上這句話才行 uvicorn.run(’mock_third_server:app’, port=8000, debug=True)

’’’@File : local_project_mock_test.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’from unittest import TestCase, main, mockimport requestsclass Request(object): def request_tx(self): tx_url = ’http://127.0.0.1:8000/tx’ tx_data = {'a': 1, 'b': 2} tx_result = requests.post(tx_url, json=tx_data) tx_result = tx_result.json() return tx_result def request_ali(self): ali_url = ’http://127.0.0.1:8000/ali’ ali_data = {'c': 1, 'd': 2} ali_result = requests.post(ali_url, json=ali_data) ali_result = ali_result.json() return ali_resultclass Myproject(TestCase): def test_api(self): r = Request() ali_result = r.request_ali() tx_mock = mock.Mock( return_value={'status': 1, 'result': {'value_a': 2, 'value_b': 1, 'all_add_value': 3}}) r.request_tx = tx_mock tx_result = r.request_tx() self.assertEqual(int(ali_result[’result’][’all_add_value’]) + int(tx_result[’result’][’all_add_value’]), 6)if __name__ == ’__main__’: main()

#以上將request_tx方法mock掉,可以mock掉寫在同一文件的方法

mock方法二:@patch(對(duì)于不同文件下的類進(jìn)行mock)

’’’@File : local_project_mock_test.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’from unittest import TestCase, mainfrom unittest.mock import patchimport mock_request_serverclass Myproject(TestCase): @patch('mock_request_server.Request.request_tx') # @patch裝飾器模擬類對(duì)象 def test_api(self, _mock_tx): _mock_tx.return_value = {’status’: 1, ’result’: {’value_a’: 2, ’value_b’: 1, ’all_add_value’: 3}} r = mock_request_server.Request() ali_result = r.request_ali() tx_result = r.request_tx() print(tx_result) self.assertEqual(int(ali_result[’result’][’all_add_value’]) + int(tx_result[’result’][’all_add_value’]), 6)if __name__ == ’__main__’: main()

’’’@File : mock_request_server.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’import requestsclass Request: def request_tx(self): tx_url = ’http://127.0.0.1:8000/tx’ tx_data = {'a': 1, 'b': 2} tx_result = requests.post(tx_url, json=tx_data) tx_result = tx_result.json() raise(’接口異?！? return tx_result def request_ali(self): ali_url = ’http://127.0.0.1:8000/ali’ ali_data = {'c': 1, 'd': 2} ali_result = requests.post(ali_url, json=ali_data) ali_result = ali_result.json() return ali_result

’’’@File : mock_third_server.py@Copyright : rainbol@Date : 2020/9/28@Desc :’’’from fastapi import FastAPIfrom pydantic import BaseModelimport uvicornapp = FastAPI()def add(a, b): return a + b# json參數(shù)傳參class TX(BaseModel): a: int b: int@app.post(’/tx’)def txApi(u: TX): u.a += 1 u.b -= 1 add_result = add(u.a, u.b) return {’status’: 1, ’result’: {’value_a’: u.a, ’value_b’: u.b, ’all_add_value’: add_result}}class Al(BaseModel): c: int d: int@app.post(’/ali’)def aliApi(u: Al): u.c *= 1 u.d /= 1 add_result = add(u.c, u.d) return {’status’: 1, ’result’: {’value_c’: u.c, ’value_d’: u.d, ’all_add_value’: add_result}}if __name__ == ’__main__’: # 一定要加上這句話才行 uvicorn.run(’mock_third_server:app’, port=8000, debug=True)

以上就是python mock測試的示例的詳細(xì)內(nèi)容，更多關(guān)于python mock測試的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！