python mock模块使用

Posted xiaohuhu

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python mock模块使用相关的知识,希望对你有一定的参考价值。

什么是mock

unittest.mock是一个用于在Python中进行单元测试的库,Mock翻译过来就是模拟的意思,顾名思义这个库的主要功能是模拟一些东西。
它的主要功能是使用mock对象替代掉指定的Python对象,以达到模拟对象的行为。

学过python自动化的对unittest并不陌生,unittest其实是单元测试框架,
但对于单元测试,估计很多小伙伴都不懂,单元测试才是自动化测试的至高境界,其中mock是单元测试的脊髓所在

mock能做什么:

  • 1.前后端联调,如果你是一个前端页面开发,现在需要开发一个功能:
    下一个订单,支付页面的接口,根据支付结果,支付成功,展示支付成功页,支付失败,展示支付失败页。
    要完成此功能,你需要调用后端的接口,根据返回给你的结果,来展示不同的页面。此时后端接口还没开发好,
    作为一个前端开发总不能等别人开发好了,你再开发,那你只有加班的命了。
    为了同步开发完成任务,此时,你可以根据接口文档的规定,把接口的地址和入参传过去,然后自己mock接口的不同返回界面,来完成前端的开发任务

  • 2.单元测试,单元测试的目的是测试某个小小单元的功能,但现实中开发的函数或方法都是有依赖关系的,比如b函数的参数,需要调用a函数的返回结果,但是我前面已经测试a函数了
    这种情况下,就不需要再测一次a函数了,此时就可以用mock模块来模拟调用这部分内容,并给出返回结果

  • 3.第三方接口依赖,在做接口自动化的时候,有时候需要调用第三方的接口,但是别人公司的接口服务不受你的控制,有可能别人提供的测试环境今天服务给你开着,别人就关掉了,
    给自动化接口测试带来很多的麻烦,此时就可以自己写一个mock-server来模拟接口的返回数据

mock环境准备

1.python2.x的版本,mock是一个独立的模块,需要用pip安装

pip install -U mock

3.从Python 3.3以后的版本mock已经合并到unittest模块中了,是unittest单元测试的一部分,直接导入过来就行

from unittest import mock

依赖关系

1.如下场景:支付是一个独立的接口,由其它开发提供,根据支付的接口返回状态去显示失败,还是成功,这个是你需要实现的功能

也就是说你写一个b功能,你的同事写一个a功能,你的b功能需要根据a功能的结果去判断,然后实现对应的功能。这就是存在依赖关系,你同事开发的进度你是无法控制的
你要是等他开发完了,你再开发,那你就坐等加班吧.

2.以下是自己写的 zhifu_statues()函数功能,大概设计如下,保存为temple.py文件

# 保存为temple.py

# coding:utf-8
# 作者:上海-悠悠 QQ交流群:588402570

def zhifu():
    ‘‘‘假设这里是一个支付的功能,未开发完
    支付成功返回:{"result": "success", "reason":"null"}
    支付失败返回:{"result": "fail", "reason":"余额不足"}
    reason返回失败原因
    ‘‘‘
    pass

def zhifu_statues():
    ‘‘‘根据支付的结果success or fail,判断跳转到对应页面‘‘‘
    result = zhifu()
    print(result)
    try:
        if result["result"] == "success":
            return "支付成功"
        elif result["result"] == "fail":
            print("失败原因:%s" % result["reason"])
            return "支付失败"
        else:
            return "未知错误异常"
    except:
        return "Error, 服务端返回异常!"

3.单元测试用例设计

# coding:utf-8
from unittest import mock
import unittest
import temple
# 作者:上海-悠悠 QQ交流群:588402570

class Test_zhifu_statues(unittest.TestCase):
    ‘‘‘单元测试用例‘‘‘
    def test_01(self):
        ‘‘‘测试支付成功场景‘‘‘
        # mock一个支付成功的数据
        temple.zhifu = mock.Mock(return_value={"result": "success", "reason":"null"})
        # 根据支付结果测试页面跳转
        statues = temple.zhifu_statues()
        print(statues)
        self.assertEqual(statues, "支付成功")

    def test_02(self):
        ‘‘‘测试支付失败场景‘‘‘
        # mock一个支付成功的数据
        temple.zhifu = mock.Mock(return_value={"result": "fail", "reason": "余额不足"})
        # 根据支付结果测试页面跳转
        statues = temple.zhifu_statues()
        self.assertEqual(statues, "支付失败")

if __name__ == "__main__":
    unittest.main()

4.单元测试是为了保证函数的每个分支都测到,以上zhifu_statues()函数有四个分支,也就是说要写四个案例,这里我只写了2个,还有2个案例,你们可以自己练习下










以上是关于python mock模块使用的主要内容,如果未能解决你的问题,请参考以下文章

Python——单元测试中mock原理和使用

Python学习:如何试用pytest-mock

Python学习:如何试用pytest-mock

Python学习:如何试用pytest-mock

Python学习:如何试用pytest-mock

mock基本使用