python mock測(cè)試的示例

發(fā)布時(shí)間：2020-10-25 18:27:44 來(lái)源：腳本之家閱讀：171 作者：RainBol 欄目：開(kāi)發(fā)技術(shù)

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

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

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

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

from unittest import TestCase, main, mock
import requests


class 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_result


class 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 FastAPI
from pydantic import BaseModel
import uvicorn

app = 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接口的測(cè)試就需要使用mock來(lái)模擬tx的接口了

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

'''
@File    : mock_third_server.py
@Copyright : rainbol
@Date    : 2020/9/28
@Desc    :
'''

from fastapi import FastAPI
from pydantic import BaseModel
import uvicorn

app = 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, mock
import requests


class 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_result


class 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掉寫(xiě)在同一文件的方法

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

'''
@File    : local_project_mock_test.py
@Copyright : rainbol
@Date    : 2020/9/28
@Desc    :
'''

from unittest import TestCase, main
from unittest.mock import patch
import mock_request_server


class Myproject(TestCase):
  @patch("mock_request_server.Request.request_tx") # @patch裝飾器模擬類(lèi)對(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 requests


class 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 FastAPI
from pydantic import BaseModel
import uvicorn

app = 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測(cè)試的示例的詳細(xì)內(nèi)容，更多關(guān)于python mock測(cè)試的資料請(qǐng)關(guān)注億速云其它相關(guān)文章！

向AI問(wèn)一下細(xì)節(jié)

python mock測(cè)試的示例

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽