Python Mock模塊原理及使用方法詳解

更新時(shí)間：2020年07月07日 15:38:12 作者：藍(lán) 寅

這篇文章主要介紹了Python Mock模塊原理及使用方法詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

一、mock是什么？

英譯中含義有：虛假的; 不誠(chéng)實(shí)的; 模仿的; 模擬的這個(gè)意思

這個(gè)庫(kù)的主要功能就是模擬一些事務(wù)

官方解釋：Mock是Python中一個(gè)用于支持單元測(cè)試的庫(kù)，它的主要功能是使用mock對(duì)象替代掉指定的Python對(duì)象，以達(dá)到模擬對(duì)象的行為

二、為什么要用mock?

舉例：

假設(shè)你開(kāi)發(fā)一個(gè)項(xiàng)目，里面包含了一個(gè)登錄模塊，登錄模塊需要調(diào)用身份證驗(yàn)證模塊中的認(rèn)證函數(shù)，該認(rèn)證函數(shù)會(huì)進(jìn)行值的返回，然后系統(tǒng)根據(jù)這個(gè)返回值來(lái)做判斷是否能進(jìn)行登錄。但是身份證驗(yàn)證模塊中的認(rèn)證函數(shù)只有在正式上線的系統(tǒng)上才提供。公司內(nèi)部的測(cè)試環(huán)境或者開(kāi)發(fā)環(huán)境上不提供。

如果此時(shí)需要進(jìn)行登錄模塊的業(yè)務(wù)測(cè)試或接口測(cè)試，該如何做？

方法一：

搭建一個(gè)測(cè)試服務(wù)器，在測(cè)試的時(shí)候，讓認(rèn)證函數(shù)和這個(gè)測(cè)試服務(wù)器交互，返回值給登錄模塊；

缺點(diǎn)：

1、測(cè)試服務(wù)器可能不好搭建，或者搭建效率很低；

2、搭建的測(cè)試服務(wù)器可能無(wú)法返回所有可能的值，或者需要大量的工作才能達(dá)到這個(gè)目的。
方法二：

使用mock函數(shù)模擬即可。

三、mock應(yīng)用

3.1 mock安裝

在Python 3.3以前的版本中，需要另外安裝mock模塊，可以使用pip命令來(lái)安裝：

pip install mock

從Python 3.3開(kāi)始，mock模塊已經(jīng)被合并到標(biāo)準(zhǔn)庫(kù)中，被命名為unittest.mock，可以直接import進(jìn)來(lái)使用：

from unittest import mock

備注：目前Python 3.3之后的版本，也可以使用第一種方式應(yīng)用mock

3.2 mock簡(jiǎn)單舉例

from unittest import mock
def add(num1,num2):
  return num1 + num2  # pass

add_value = mock.Mock(return_value=200) # 創(chuàng)建mock對(duì)象
add = add_value # 把mock對(duì)象賦值給add方法

print( add(10,20) )

從上述代碼執(zhí)行可以看出，假使我把a(bǔ)dd()方法的實(shí)現(xiàn)改為pass，暫時(shí)還未實(shí)現(xiàn)，也可以通過(guò)mock先返回所需要的數(shù)據(jù)。

3.3 mock應(yīng)用舉例

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() )

上述代碼是訪問(wèn)我們公司的官網(wǎng)，假設(shè)該官網(wǎng)訪問(wèn)請(qǐng)求還沒(méi)被實(shí)現(xiàn)，此時(shí) visit_newdream() 方法需要 send_request 返回響應(yīng)狀態(tài)碼。這個(gè)時(shí)候可以用到mock。

# 當(dāng)前文件名為 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的簡(jiǎn)單應(yīng)用，由于mock_demo.send_request = success_send 把mock對(duì)象賦值給了方法，使得方法再次調(diào)用的時(shí)候使用了mock的方法完成把模擬返回值傳遞給mock_demo.visit_newdream()方法。

四、小結(jié)