脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → python mock测试

python mock测试的示例

 更新时间：2020年10月19日 10:11:46   作者：RainBol  
这篇文章主要介绍了python mock测试的示例，帮助大家利用python进行模拟接口测试，感兴趣的朋友可以了解下

mock总所周知是模拟的意思,我们在做接口测试的时候有时候会发现部分功能依赖其他业务场景或者第三方功能或者是线上数据或者业务场景过于复杂(需要大量调用)的情况,没有办法通过接口调用或者做断言,这个时候就需要mock了

python2中需要pip install mock来导入mock第三方模块,而python3中被引入到unittest框架中,直接from unittest import mock 就可以导入mock模块了

在单元测试中只针对当前单元做测试,就是测试当前方法或者是当前类,但是如果出现这个方法依赖其他第三方模块,那不是还要校验其他方法了吗,那就违背了单元测试这个概念了,所有将外部依赖组件实现模拟并替换掉,使得单元测试的焦点放在自己身上

为了演示,用fastapi模拟了一个tx和ali的接口来调用作为单元测试,可以看到需要同时调用两个接口才能完成单元测试

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参数传参
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)

可以看到上面的接口是可以调通的如果txapi因为一些原因不能返回,ali的接口是通的,如果要实现ali接口的测试就需要使用mock来模拟tx的接口了

mock方法一:(对于同一文件下/不同文件下的方法或者类进行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参数传参
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掉写在同一文件的方法

mock方法二:@patch(对于不同文件下的类进行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装饰器模拟类对象
  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参数传参
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测试的示例的详细内容，更多关于python mock测试的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • pytorch中tensor的合并与截取方法

    pytorch中tensor的合并与截取方法

    今天小编就为大家分享一篇pytorch中tensor的合并与截取方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07
  • Python人工智能语音合成实现案例详解

    Python人工智能语音合成实现案例详解

    这篇文章主要为大家介绍了Python人工智能语音合成实现案例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
    2023-03-03
  • Python实战之实现简单的名片管理系统

    Python实战之实现简单的名片管理系统

    这篇文章主要介绍了Python实战之实现简单的名片管理系统,文中有非常详细的代码示例,对正在学习python的小伙伴们有非常好的帮助,需要的朋友可以参考下
    2021-04-04
  • Python中请使用isinstance()判断变量类型

    Python中请使用isinstance()判断变量类型

    这篇文章主要介绍了Python中请使用isinstance()判断变量类型,本文先是给出了isinstance函数判断变量类型的例子,并对isinstance 和 type的区别做了讲解,需要的朋友可以参考下
    2014-08-08
  • python opencv鼠标事件实现画框圈定目标获取坐标信息

    python opencv鼠标事件实现画框圈定目标获取坐标信息

    这篇文章主要为大家详细介绍了python opencv鼠标事件实现画框圈定目标，获取坐标信息，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2019-08-08
  • Django 自定义权限管理系统详解(通过中间件认证)

    Django 自定义权限管理系统详解(通过中间件认证)

    这篇文章主要介绍了Django 自定义权限管理系统详解(通过中间件认证)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • Python实现CART决策树算法及详细注释

    Python实现CART决策树算法及详细注释

    CART算法是一种树构建算法，既可以用于分类任务，又可以用于回归，本文仅讨论基本的CART分类决策树构建，不讨论回归树和剪枝等问题，感兴趣的朋友跟随小编一起看看吧
    2021-10-10
  • python中f字符串f-string用法详解

    python中f字符串f-string用法详解

    f-string用大括号{}表示被替换字段,其中直接填入替换内容,本文给大家介绍python中f字符串f-string用法详解,感兴趣的朋友一起看看吧
    2023-10-10
  • 用Python编写个解释器实现方法接受

    用Python编写个解释器实现方法接受

    计算机只能理解机器码。归根结底，编程语言只是一串文字，目的是为了让人类更容易编写他们想让计算机做的事情。真正的魔法是由编译器和解释器完成，它们弥合了两者之间的差距。解释器逐行读取代码并将其转换为机器码
    2023-01-01
  • Python开发之QT解决无边框界面拖动卡屏问题(附带源码)

    Python开发之QT解决无边框界面拖动卡屏问题(附带源码)

    朋友在学习QT的过程中，都会遇到各种问题，今天就QT无边框拖动花屏问题给大家详细介绍，究竟该如何解决呢，下面通过实例代码和图文相结合给大家详细介绍，需要的朋友参考下吧
    2021-05-05

最新评论