python中mock的常規使用


一、首先mock是什么?

  • 英譯中含義有:虛假的; 不誠實的; 模仿的; 模擬的 這個意思
  • 這個庫的主要功能就是模擬一些事務
  • 官方解釋:Mock是Python中一個用於支持單元測試的庫,它的主要功能是使用mock對象替代掉指定的Python對象,以達到模擬對象的行為

二、為什么要用mock?

舉例:
假設你開發一個項目,里面包含了一個登錄模塊,登錄模塊需要調用身份證驗證模塊中的認證函數,該認證函數會進行值的返回,然后系統根據這個返回值來做判斷是否能進行登錄。但是身份證驗證模塊中的認證函數只有在正式上線的系統上才提供。公司內部的測試環境或者開發環境上不提供。

如果此時需要進行登錄模塊的業務測試或接口測試,該如何做?

一:

  • 搭建一個測試服務器,在測試的時候,讓認證函數和這個測試服務器交互,返回值給登錄模塊;
  • 缺點:1、測試服務器可能不好搭建,或者搭建效率很低;2、搭建的測試服務器可能無法返回所有可能的值,或者需要大量的工作才能達到這個目的。

二:

  • 在此使用mock函數模擬即可。

三、mock應用

1 mock安裝

  • 在Python 3.3以前的版本中,需要另外安裝mock模塊,可以使用pip命令來安裝:
pip install mock 
  • 從Python 3.3開始,mock模塊已經被合並到標准庫中,被命名為unittest.mock,可以直接import進來使用:
from unittest import mock 

備注:目前Python 3.3之后的版本,也可以使用第一種方式應用mock

2 mock簡單舉例

from unittest import mock def add(num1,num2): return num1 + num2 # pass add_value = mock.Mock(return_value=200) # 創建mock對象 add = add_value # 把mock對象賦值給add方法 print( add(10,20) ) 

從上述代碼執行可以看出,假使我把add()方法的實現改為pass,暫時還未實現,也可以通過mock先返回所需要的數據。

3.3 mock應用舉例

import requests def send_request(url): r = requests.get(url) return r.status_code def visit_newdream(): return send_request("http://www.hnxmxit.com/") print( visit_newdream() ) 

上述代碼是訪問我們公司的官網,假設該官網訪問請求還沒被實現,此時 visit_newdream() 方法需要 send_request 返回響應狀態碼。這個時候可以用到mock。

# 當前文件名為 mock_demo.py import unittest from unittest import mock import requests import mock_demo def send_request(url): r = requests.get(url) return r.status_code def visit_newdream(): return send_request("http://www.hnxmxit.com/") class TestClient(unittest.TestCase): def test_success_request(self): success_send = mock.Mock(return_value='200') mock_demo.send_request = success_send self.assertEqual(mock_demo.visit_newdream(), '200') if __name__=="__main__": unittest.main() 

上述代碼完成了mock的簡單應用,由於mock_demo.send_request = success_send 把mock對象賦值給了方法,使得方法再次調用的時候使用了mock的方法完成把模擬返回值傳遞給mock_demo.visit_newdream()方法。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM