python技术栈之单元测试中mock的使用

什么是mock?

mock测试就是在测试过程中,对于某些不容易构造或者不容易获取的对象,用一个虚拟的对象来创建以便测试的测试方法。

mock的作用

特别是开发过程中上下游未完成的工序导致当前无法测试,需要虚拟某些特定对象以便测试。

unittest是python内置的单元测试库,在做接口测试时,如果开发的接口未开发出来,我们如果想要测试接口联调,又不能干等着,这时可以使用unittest.mock模拟接口返回,进行接口测试。

举个栗子

示例:1、新增登录和获取个人信息两个接口,分别由A和B开发。2、A登录接口尚未开发完成,B获取个人信息接口已经开发完成。3、已知登录接口返回3种状态:登录成功,登录失败,登录异常。

场景源码:case.py

python 复制代码
# -*-coding:utf-8 -*-

def login():
    # 登录接口,尚未开发完成
    # 登录成功返回:{"result": "success", "message": "登录成功"}
    # 登录失败返回:{"result": "fail", "message":"账号或密码错误"}
    # 登录异常返回:{"code": "404", "message": "找不到页面"}
    # message返回失败原因

    pass


def get_user_info():
    # 根据登录的结果success or fail,判断跳转到对应页面

    result = login()
    print(result)
    try:
        if result["result"] == "success":
            return "登录成功"
        elif result["result"] == "fail":
            return "登录失败"
        else:
            return "未知失败"
    except:
        return "服务端异常"

单元测试用例设计:

python 复制代码
# -*-coding:utf-8 -*-
from unittest import mock
import unittest
import case


class TestLogin(unittest.TestCase):
    # 单元测试用例

    def test_login_success(self):
        # 测试登录成功场景
        # mock一个支付成功的数据

        case.login = mock.Mock(return_value={"result": "success", "message": "登录成功"})
        # 根据支付结果测试页面跳转
        statues = case.get_user_info()
        self.assertEqual(statues, "登录成功")

    def test_login_fail(self):
        # 测试登录失败场景
        # mock一个登录失败的数据

        case.login = mock.Mock(return_value={"result": "fail", "message": "账号或密码错误"})
        # 根据登录结果测试页面跳转
        statues = case.get_user_info()
        self.assertEqual(statues, "登录失败")

    def test_login_fail2(self):
        # 测试登录失败场景
        # mock一个登录失败的数据

        case.login = mock.Mock(return_value={"result": "", "message": "账号或者密码为空"})
        # 根据登录结果测试页面跳转
        statues = case.get_user_info()
        self.assertEqual(statues, "未知失败")

    def test_login_error(self):
        # 测试登录异常场景
        # mock一个登录异常的数据

        case.login = mock.Mock(return_value={"code": "404", "message": "找不到页面"})
        # 根据登录结果测试页面跳转
        statues = case.get_user_info()
        self.assertEqual(statues, "服务端异常")


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

执行unittest单元测试用例:

最后感谢每一个认真阅读我文章的人,礼尚往来总是要有的,这些资料,对于【软件测试】的朋友来说应该是最全面最完整的备战仓库,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走:

这些资料,对于【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴上万个测试工程师们走过最艰难的路程,希望也能帮助到你!

相关推荐
小龙在山东30 分钟前
Python 包管理工具 uv
windows·python·uv
weixin_307779131 小时前
批量OCR的GitHub项目
python·github·ocr
孤狼warrior2 小时前
灰色预测模型
人工智能·python·算法·数学建模
神仙别闹2 小时前
基于Python实现LSTM对股票走势的预测
开发语言·python·lstm
机器学习之心2 小时前
小波增强型KAN网络 + SHAP可解释性分析(Pytorch实现)
人工智能·pytorch·python·kan网络
JavaEdge在掘金2 小时前
MySQL 8.0 的隐藏索引:索引管理的利器,还是性能陷阱?
python
宇钶宇夕3 小时前
EPLAN 电气制图:建立自己的部件库,添加部件-加SQL Server安装教程(三)上
运维·服务器·数据库·程序人生·自动化
站大爷IP3 小时前
Python办公自动化实战:手把手教你打造智能邮件发送工具
python
chao_7893 小时前
回溯题解——子集【LeetCode】二进制枚举法
开发语言·数据结构·python·算法·leetcode
zdw3 小时前
fit parse解析佳明.fit 运动数据,模仿zwift数据展示
python