python mock模塊使用(一)
什麽是mock
unittest.mock是一個用於在Python中進行單元測試的庫,Mock翻譯過來就是模擬的意思,顧名思義這個庫的主要功能是模擬一些東西。
它的主要功能是使用mock對象替代掉指定的Python對象,以達到模擬對象的行為。
學過python自動化的對unittest並不陌生,unittest其實是單元測試框架,
但對於單元測試,估計很多小夥伴都不懂,單元測試才是自動化測試的至高境界,其中mock是單元測試的脊髓所在
mock能做什麽:
-
1.前後端聯調,如果你是一個前端頁面開發,現在需要開發一個功能:
下一個訂單,支付頁面的接口,根據支付結果,支付成功,展示支付成功頁,支付失敗,展示支付失敗頁。
要完成此功能,你需要調用後端的接口,根據返回給你的結果,來展示不同的頁面。此時後端接口還沒開發好,
作為一個前端開發總不能等別人開發好了,你再開發,那你只有加班的命了。
為了同步開發完成任務,此時,你可以根據接口文檔的規定,把接口的地址和入參傳過去,然後自己mock接口的不同返回界面,來完成前端的開發任務 -
2.單元測試,單元測試的目的是測試某個小小單元的功能,但現實中開發的函數或方法都是有依賴關系的,比如b函數的參數,需要調用a函數的返回結果,但是我前面已經測試a函數了
這種情況下,就不需要再測一次a函數了,此時就可以用mock模塊來模擬調用這部分內容,並給出返回結果 -
3.第三方接口依賴,在做接口自動化的時候,有時候需要調用第三方的接口,但是別人公司的接口服務不受你的控制,有可能別人提供的測試環境今天服務給你開著,別人就關掉了,
給自動化接口測試帶來很多的麻煩,此時就可以自己寫一個mock-server來模擬接口的返回數據
mock環境準備
1.python2.x的版本,mock是一個獨立的模塊,需要用pip安裝
pip install -U mock
3.從Python 3.3以後的版本mock已經合並到unittest模塊中了,是unittest單元測試的一部分,直接導入過來就行
from unittest import mock
依賴關系
1.如下場景:支付是一個獨立的接口,由其它開發提供,根據支付的接口返回狀態去顯示失敗,還是成功,這個是你需要實現的功能
也就是說你寫一個b功能,你的同事寫一個a功能,你的b功能需要根據a功能的結果去判斷,然後實現對應的功能。這就是存在依賴關系,你同事開發的進度你是無法控制的
你要是等他開發完了,你再開發,那你就坐等加班吧.
2.以下是自己寫的 zhifu_statues()函數功能,大概設計如下,保存為temple.py文件
# 保存為temple.py
# coding:utf-8
# 作者:上海-悠悠 QQ交流群:588402570
def zhifu():
‘‘‘假設這裏是一個支付的功能,未開發完
支付成功返回:{"result": "success", "reason":"null"}
支付失敗返回:{"result": "fail", "reason":"余額不足"}
reason返回失敗原因
‘‘‘
pass
def zhifu_statues():
‘‘‘根據支付的結果success or fail,判斷跳轉到對應頁面‘‘‘
result = zhifu()
print(result)
try:
if result["result"] == "success":
return "支付成功"
elif result["result"] == "fail":
print("失敗原因:%s" % result["reason"])
return "支付失敗"
else:
return "未知錯誤異常"
except:
return "Error, 服務端返回異常!"
3.單元測試用例設計
# coding:utf-8
from unittest import mock
import unittest
import temple
# 作者:上海-悠悠 QQ交流群:588402570
class Test_zhifu_statues(unittest.TestCase):
‘‘‘單元測試用例‘‘‘
def test_01(self):
‘‘‘測試支付成功場景‘‘‘
# mock一個支付成功的數據
temple.zhifu = mock.Mock(return_value={"result": "success", "reason":"null"})
# 根據支付結果測試頁面跳轉
statues = temple.zhifu_statues()
print(statues)
self.assertEqual(statues, "支付成功")
def test_02(self):
‘‘‘測試支付失敗場景‘‘‘
# mock一個支付成功的數據
temple.zhifu = mock.Mock(return_value={"result": "fail", "reason": "余額不足"})
# 根據支付結果測試頁面跳轉
statues = temple.zhifu_statues()
self.assertEqual(statues, "支付失敗")
if __name__ == "__main__":
unittest.main()
4.單元測試是為了保證函數的每個分支都測到,以上zhifu_statues()函數有四個分支,也就是說要寫四個案例,這裏我只寫了2個,還有2個案例,你們可以自己練習下
python mock模塊使用(一)