python mock测试的示例


Posted in Python onOctober 19, 2020

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测试的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
详解Python中的各种函数的使用
May 24 Python
使用Python对SQLite数据库操作
Apr 06 Python
tensorflow实现softma识别MNIST
Mar 12 Python
Python callable()函数用法实例分析
Mar 17 Python
Python使用numpy模块创建数组操作示例
Jun 20 Python
详解多线程Django程序耗尽数据库连接的问题
Oct 08 Python
windows10下安装TensorFlow Object Detection API的步骤
Jun 13 Python
Python+OpenCV+pyQt5录制双目摄像头视频的实例
Jun 28 Python
Python模块相关知识点小结
Mar 09 Python
tensorflow 20:搭网络,导出模型,运行模型的实例
May 26 Python
利用Python判断整数是否是回文数的3种方法总结
Jul 07 Python
python scrapy简单模拟登录的代码分析
Jul 21 Python
python 提高开发效率的5个小技巧
Oct 19 #Python
python 利用toapi库自动生成api
Oct 19 #Python
协程Python 中实现多任务耗资源最小的方式
Oct 19 #Python
python爬取音频下载的示例代码
Oct 19 #Python
Python爬虫教程知识点总结
Oct 19 #Python
自定义Django_rest_framework_jwt登陆错误返回的解决
Oct 18 #Python
如何利用python读取micaps文件详解
Oct 18 #Python
You might like
基于jquery的时间段实现代码
2012/08/02 Javascript
js冒泡法和数组转换成字符串示例代码
2013/08/14 Javascript
分享Javascript中最常用的55个经典小技巧
2013/11/29 Javascript
jQuery中offsetParent()方法用法实例
2015/01/19 Javascript
浅谈javascript 归并方法
2015/01/21 Javascript
JavaScript事件委托实例分析
2015/05/26 Javascript
JavaScript html5 canvas绘制时钟效果(二)
2016/03/27 Javascript
JS与HTML结合使用marquee标签实现无缝滚动效果代码
2016/07/05 Javascript
关于javascript原型的修改与重写(覆盖)差别详解
2016/08/31 Javascript
jQuery基于排序功能实现上移、下移的方法
2016/11/26 Javascript
js格式化时间的简单实例
2016/11/27 Javascript
Vue.js实战之组件的进阶
2017/04/04 Javascript
javascript闭包功能与用法实例分析
2017/04/06 Javascript
Express+Nodejs 下的登录拦截实现代码
2017/07/01 NodeJs
Vue内部渲染视图的方法
2019/09/02 Javascript
JavaScript实现表单验证功能
2020/12/09 Javascript
微信小程序学习之自定义滚动弹窗
2020/12/20 Javascript
[03:43]2014DOTA2西雅图国际邀请赛 newbee战队巡礼
2014/07/07 DOTA
python中lambda函数 list comprehension 和 zip函数使用指南
2014/09/28 Python
Python利用IPython提高开发效率
2016/08/10 Python
使用python绘制常用的图表
2016/08/27 Python
Python之reload流程实例代码解析
2018/01/29 Python
keras中模型训练class_weight,sample_weight区别说明
2020/05/23 Python
Python高并发解决方案实现过程详解
2020/07/31 Python
利用纯html5绘制出来的一款非常漂亮的时钟
2015/01/04 HTML / CSS
澳大利亚最大的女装零售商:Millers
2017/09/10 全球购物
PHP如何调用MYSQL存储过程
2014/05/30 面试题
ddl,dml和dcl的含义
2016/05/08 面试题
实习自我鉴定模板
2013/09/28 职场文书
八一建军节活动方案
2014/02/10 职场文书
十八届三中全会学习方案
2014/02/16 职场文书
大学毕业典礼演讲稿
2014/09/09 职场文书
2015届本科毕业生自我鉴定
2014/09/27 职场文书
会议主持词通用版
2019/04/02 职场文书
六年级作文之预言作文
2019/10/25 职场文书
你知道Java Spring的两种事务吗
2022/03/16 Java/Android