Python测试框架unittest之mock系列（一）- 断言方法、统计方法

最新推荐文章于 2024-06-13 15:14:48 发布
最新推荐文章于 2024-06-13 15:14:48 发布
阅读量1k 收藏 3
点赞数 1
分类专栏： 软件测试 文章标签： 软件测试 接口测试 自动化测试 测试工程师 性能测试
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_53519100/article/details/116807059
版权

1、简介

什么是mock？

mock翻译过来有模拟的意思。这里介绍的mock是辅助单元测试的一个模块。它允许你用模拟对象替换你的系统的部分，并对它们已使用的方式进行断言。

 

什么时候使用mock？

场景1：比如有A和B两个模块，A模块中有调用到B模块的方法，但是B模块中被A模块调用的方法由于一些原因需要被修改，然而我们又不想影响A模块的功能测试，这时候就用到了mock，用来模拟出一个假的B模块。

场景2：有时需要为单元测试的初始设置准备一些其他的资源，但是这些资源又不太经常使用或者使用起来比较麻烦，此时我们就可以定义一个mock对象来模拟一个需要使用的资源，用于代替测试的准备资源。

 

2、安装

2.1、Python 3.3以前的版本

在Python 3.3以前的版本中，mock是一个单独模块，需要单独安装。

安装方式有2种（任选一种即可）。

1、利用pip安装

命令行输入 pip install mock

2、源码安装

下载网址：https://pypi.org/project/mock/

下载完成后，进行解压，例如：mock-2.0.0.tar.gz

图片

如图所示，解压后的文件目录

图片

 

打开命令行，跳转到解压后的目录路径，输入安装命令python setup.py install即可。

之后在代码中直接import进来就可以使用mock了。

import mock

2.2、Python 3.3及更高版本

在Python 3.3及更高版本中，mock已经被集成到了unittest单元测试框架中，所以可以直接使用。

在代码中直接import进来就可以使用mock了。

from unittest import mock

3、基本示例

Mock对象是mock模块中最重要的概念。Mock对象就是mock模块中的一个类的实例，这个类的实例可以用来替换其他的Python对象，来达到模拟的效果。

Mock对象的一般用法：

步骤1：找到你要替换的对象（一个类，或者一个函数，或者一个类实例）。

步骤2：实例化Mock类得到一个mock对象，并且设置这个mock对象的行为（比如被调用的时候返回什么值，被访问成员的时候返回什么值等）。

步骤3：使用这个mock对象替换掉我们想替换的对象，也就是步骤1中确定的对象。

步骤4：之后就可以开始写测试用例，这个时候我们可以保证我们替换掉的对象在测试用例执行的过程中行为和我们预设的一样。

下面开始讲解基本示例：

一、创建Demo.py文件。

创建被测试函数（get、send_request）

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
基本示例：被测试函数
"""
# 引入Requests库
import requests

def get(url):
    # 发起GET请求
    r = requests.get(url)
    # 返回状态码
    return r.status_code

def send_request():
    # 调用get()函数
    return get('https://www.baidu.com/')

 

二、创建MockTest.py文件。

创建TestDemo测试类。

1、不使用mock

1.1、脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
基本示例：测试类
"""
import unittest
from demo import Demo

class TestDemo(unittest.TestCase):

    def test_request(self):
        print(Demo.send_request())
        self.assertEqual(Demo.send_request(), 200)

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

1.2、执行MockTest.py文件，运行结果：

调用Demo.send_request()，期望和实际状态码返回值一样。

图片

 

2、使用mock

2.1、脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
基本示例：测试类
"""
import unittest
from unittest import mock
from demo import Demo

class TestDemo(unittest.TestCase):

    def test_request_mock(self):
        Demo.get = mock.Mock(return_value=404)
        print(Demo.get())
        self.assertEqual(Demo.send_request(),404)

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2.2、执行MockTest.py文件，运行结果：

（1）首先实例化Mock类得到一个mock对象，并且设置这个mock对象的行为（返回值为404）。

（2）使用这个mock对象替换掉我们想替换的对象（Demo.get）。

（3）调用Demo.send_request()，期望和预设值一样（404）。

图片

 

 

4、构造器

 

 

4.1、name

name：是mock对象的唯一标识；用来命名一个mock对象，只是起到标识作用，当你print一个mock对象的时候，可以看到它的name。

 

1、创建MockTest_name.py文件。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数name
"""
from unittest import mock
# name定义了mock对象的唯一标示符
testMock = mock.Mock(name='MyMock')
print(testMock)

 

2、执行MockTest_name.py文件，运行结果：

定义mock对象唯一标识为'MyMock'。

图片

 

4.2、spec

spec：设置mock对象的属性，可以是property或者方法，也可以是其他的列表字符串或者其他的python类。

 

1、创建MockTest_spec.py文件。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数spec
"""
from unittest import mock
# 指定属性组成的list
MyList = ['username','password']
# spec设置mock对象的属性，可以是property或者方法；属性可以是一个列表字符串或者是其他的Python类
testMock = mock.Mock(spec=MyList)
print(testMock)
print(testMock.username)
print(testMock.password)

 

2、执行MockTest_spec.py文件，运行结果：

设置mock两个属性（username、password）

图片

 

4.3、return_value

return_value：设置当这个mock对象被调用的时候，显示出的结果就是return_value的值。

 

创建Demo.py文件（创建被测试类：People类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
被测试类
"""
# People类里有两个成员方法（一个有参数，一个无参数）、一个静态方法
class People:

    def __init__(self):
        self.__age = 20

    def name(self,firstName,lastName):
        return firstName + ' ' + lastName

    def age(self):
        return self.__age

    @staticmethod
    def class_name():
        return People.__name__

 

4.3.1、指定某个值

创建MockTest_return_value.py文件（创建PeopleTest测试类）。

1、不使用mock

1.1、脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数return_value（返回固定值）
"""
from method.Demo import People
import unittest

class PeopleTest(unittest.TestCase):

    def test_age(self):
        # 调用被测试类People()
        p = People()
        print(p.age())
        self.assertEqual(p.age(), 20)

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

1.2、执行MockTest_return_value.py文件，运行结果：

不使用mock，p.age()返回20。

图片

 

2、使用mock

2.1、脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
常用方法：构造器，参数return_value（返回固定值）
"""
from method.Demo import People
from unittest import mock
import unittest

class PeopleTest(unittest.TestCase):

    # 指定某个值
    def test_age(self):
        # 调用被测试类People()
        p = People()
        p.age = mock.Mock(return_value=50)
        print(p.age())
        self.assertEqual(p.age(), 50)

    # 指定某个值
    def test_name(self):
        # 调用被测试类People()
        p = People()
        p.name = mock.Mock(return_value='Hello Mock')
        print(p.name())
        self.assertEqual(p.name(), 'Hello Mock')

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2.2、执行MockTest_return_value.py文件，运行结果：

（1）test_age方法，mock掉age方法，让它返回50。

（2）test_name方法，mock掉name方法，让它返回Hello Mock。

图片

 

4.3.2、指定某个类对象

1、创建MockTest_return_value.py文件（创建PeopleTest测试类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数return_value（返回固定值）
"""
from method.Demo import People
from unittest import mock
import unittest

class PeopleTest(unittest.TestCase):

    # 指定某个类对象
    def test_People(self):
        testMock = mock.Mock(return_value=People())
        print(testMock)
        print(testMock().class_name())
        print(testMock().name('Hello','Mock'))
        print(testMock().age())

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2、执行MockTest_return_value.py文件，运行结果：

指定People类，并执行类里的方法。

图片

 

4.4、side_effect

side_effect：和return_value是相反的，覆盖了return_value，也就是说当这个mock对象被调用的时候，返回的是side_effect的值，而不是return_value。

 

创建Demo.py文件（创建被测试类：People类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
被测试类
"""
# People类里有两个成员方法（一个有参数，一个无参数）、一个静态方法
class People:

    def __init__(self):
        self.__age = 20

    def name(self,firstName,lastName):
        return firstName + ' ' + lastName

    def age(self):
        return self.__age

    @staticmethod
    def class_name():
        return People.__name__

 

4.4.1、依次返回指定值

1、创建MockTest_side_effect.py文件（创建PeopleTest测试类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数side_effect（它给mock分配了可替换的结果，覆盖了return_value）
"""
from method.Demo import People
from unittest import mock
import unittest

class PeopleTest(unittest.TestCase):

    # 依次返回指定值
    def test_age(self):
        # 调用被测试类People()
        p = People()
        MyList = [15,16,17]
        p.age = mock.Mock(return_value=50,side_effect=MyList)
        print(p.age())
        print(p.age())
        print(p.age())
        print(p.age())

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2、执行MockTest_side_effect.py文件，运行结果：

（1）执行前3个p.age()都会返回一个值，即15、16、17。

（2）执行第4个p.age()时，因为返回值没有，则抛出异常。

图片

 

4.4.2、根据参数返回指定值

1、创建MockTest_side_effect.py文件（创建PeopleTest测试类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数side_effect（它给mock分配了可替换的结果，覆盖了return_value）
"""
from method.Demo import People
from unittest import mock
import unittest

class PeopleTest(unittest.TestCase):

    # 根据参数返回指定值
    def test_name(self):
        # 调用被测试类People()
        p = People()
        values = {('a', 'b'): 'Hello Mock', ('1', '2'): 'Hello World'}
        p.name = mock.Mock(side_effect=lambda x, y: values[(x, y)])
        print(p.name('a', 'b'))
        print(p.name('1', '2'))
        self.assertEqual(p.name('a', 'b'), 'Hello Mock')
        self.assertEqual(p.name('1', '2'), 'Hello World') 

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2、执行MockTest_side_effect.py文件，运行结果：

（1）执行p.name('a', 'b')方法对应值为Hello Mock。

（2）执行p.name('1', '2')方法对应值为Hello World。

图片

 

4.4.3、抛出异常

1、创建MockTest_side_effect.py文件（创建PeopleTest测试类）。

脚本代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
构造器：参数side_effect（它给mock分配了可替换的结果，覆盖了return_value）
"""
from method.Demo import People
from unittest import mock
import unittest

class PeopleTest(unittest.TestCase):

    # 抛出异常
    def test_age_error(self):
        # 调用被测试类People()
        p = People()
        p.age = mock.Mock(side_effect=SystemError)
        print(p.age())

if __name__ == '__main__':
    unittest.main(verbosity=2)

 

2、执行MockTest_side_effect.py文件，运行结果：

设置side_effect为SystemError异常，调用p.age()方法就会抛出异常。

图片

如果您觉得文章还不错，请 点赞、分享、在看、收藏 一下，因为这将是我持续输出更多优质文章的最强动力！

在这里推荐一个我自己创建的软件测试交流群，QQ：642830685，群中会不定期的分享软件测试资源，测试面试题以及测试行业资讯，大家可以在群中积极交流技术，还有大佬为你答疑解惑。

程序员二黑⑥
关注
  • 1
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
mock-all-things:使用 mock 进行 python 单元测试的示例
06-16
模拟一切使用 mock 进行 python 单元测试的示例模拟 python 2.7： : （这里也有很多很好的例子） 从图书馆的主页： “mock 是一个用于在 Python 中进行测试的库。它允许您用模拟对象替换被测系统的某些部分，并对...
运行unittest框架的main函数问题解决
小晶的博客
02-14 2042
一、问题现象 if__name__=='__main__': test_data.init_data()#初始化接口测试数据 unittest.main() 写完接口测试用例后运行，但是怎样运行用例都无法通过； 需要先执行test_data.init_data()函数，但是结果直接就运行以test开头的测试用例，运行结果全部失败，最后增加print函数也没有打...
if __name__ == ‘__main__‘: unittest.main()简单粗暴解析
num_61的博客
08-28 6940
写在前面 这段代码不是装逼，也不是非写不可。 if __name__ == '__main__': unittest.main() 分为__name__、__main__、unittest.main()三个点解释 __name__ 三级目录
Python在自动化测试中的高级应用：使用unittest和Mock库
最新发布
人生不怕起点低，就怕没追求
06-13 909
本文将详细介绍Python中的unittest框架和unittest.mock库在自动化测试中的高级应用，并通过代码示例展示其用法。
Python 之模块 module 的 名称 属性__name__说明
yebanliuxing的专栏
07-29 1865
当你打开一个.py文件时,经常会在代码的最下面看到if __name__ == '__main__':,现在就来介 绍一下它的作用.        模块是对象，并且所有的模块都有一个内置属性 __name__。一个模块的 __name__ 的值取决于您如何应用模块。如果 imp
疑问：Pycharm中unittest.main()和if __name__ == '__main__': unittest.main()无法像在Sublime 中那样运行？
Deca_csdn的博客
03-12 797
照着教材写了几个测试用例，分别在Pycarm和Sublime中运行的结果： Pycharm里： 为什么没有找到测试条例呢？？ 把unittest.main()换成if name == ‘main’: unittest就可以了，可是还是没有出现书里说的每执行一个测试条例就会输出一个字符？ 加了一个条例，在unittest.main()的情况下又找不到测试条例了。 把unittest.ma...
单元测试框架 unittest 的运行方法if __name__ == '__main__': unittest.main()
weixin_30454481的博客
07-17 1485
　1. if __name__ == '__main__': unittest.main()2. 　　测试用例实例根据测试的特点分组在一起。unittest为此提供了一个机制：测试套件由unittest'sTestSuiteclass widgetTestSuite = unittest 。TestSuite （） widgetTestSuite 。addTest...
python-unittest-main
07-06
Python的unittest模块是Python内置的一个单元测试框架，用于编写和执行测试用例。在"python-unittest-main"的学习资料中，我们很可能会涉及到以下几个关键知识点： 1. **unittest模块基础**： - `unittest`模块...
python中的测试框架
12-16
它们遵循一系列测试原则，如关注单一功能、独立运行、快速执行、持续测试和可读性。在Python中，有多个测试框架可供选择，每一种都有其特点和优势。 1. **unittest**：作为Python的标准库，unittest与Java的JUnit...
python mock测试的示例
12-16
Python中的Mock测试是一种重要的单元测试技术，用于模拟代码中依赖的外部组件，以便在测试时专注于测试目标函数或类的行为，而不是它们的依赖。在Python 2中，Mock作为一个独立的第三方库，需要通过`pip install ...
Python库 | pytest_mock-3.3.1-py3-none-any.whl
03-23
“pytest_mock”是基于流行的Python测试框架“pytest”的一个扩展插件。pytest是一个强大的单元测试和集成测试框架，它简化了测试代码的编写和执行，使得测试工作更加高效。而pytest_mock则进一步增强了pytest的测试...
unittest小知识（二）- 几种运行方式
weixin_48626846的博客
06-10 1095
unittest的几种运行方式
if __name__ == '__main__'的作用和原理
weixin_45711264的博客
03-13 187
以前困惑了一段时间，网上找了资料，自己验证了下，进一步加深了印象。多动手才是王道。 一、基本作用 一个python文件通常有两种使用方法： 第一是作为脚本直接执行 第二是 import 到其他的 python 脚本中被调用（模块重用）执行。 因此 if name == ‘main’: 的作用就是控制这两种情况执行代码的过程，在 if name == ‘main’: 下的代码只有在第一种情况下（即文...
Python|自动化测试与if __name__=="__main__":
算法与编程之美
03-15 597
欢迎点击「算法与编程之美」↑关注我们！本文首发于微信公众号："算法与编程之美"，欢迎关注，及时了解更多此系列文章。问题描述最近有小伙伴在学习python自动化测试的时候遇...
if函数python作用_if __name__== "__main__" 的意思(作用)python代码复用
weixin_39705931的博客
12-06 528
模块是对象，并且所有的模块都有一个内置属性 __name__。一个模块的 __name__ 的值取决于您如何应用模块。如果 import 一个模块，那么模块__name__ 的值通常为模块文件名，不带路径或者文件扩展名。但是您也可以像一个标准的程序样直接运行模块，在这 种情况下, __name__ 的值将是一个特别缺省"__main__"。/////////////////////////////...
【PyCharm】if __name__ == ‘__main__‘: 其下的代码未执行
weixin_42911781的博客
04-21 656
这次执行方式是Run ‘Unittests in xxx.py’，百度了下说这是PyCharm默认执行Unittests类的单元测试，忽略了if。python脚本中使用import unittest导入unittest进行单元测试，单但执行脚本时并没有执行if。’:的执行，也就没有执行整个xxx.py文件。而Run xx.py相当于执行整个xx.py文件，也就包含if。然后发现这次执行脚本的方式与之前不一样，之前都是run …点击Run选择框中，选则文件xx.py同名的选项。
Python_单元测试
feizuiku0116的博客
08-22 321
一.什么是单元测试？ 单元测试，通俗易懂地讲，就是编写测试来验证某一个模块的功能正确性，一般会指定输入，验证输出是否符合预期。 二.测试实例 import unittest # 将要被测试的排序函数 def sort(arr): l = len(arr) for i in range(0, l): for j in range(i + 1, l): if arr[i] >= arr[j]: tmp = arr[i
测试人必会的Python内置库：unittest.mock（单元测试mock的基础使用）
HB8888888的博客
03-24 3199
1. 为什么需要使用mock unittest.mock是用于在单元测试中模拟和替换指定的对象及行为，以便测试用例更加准确地进行测试运行。例如对于以下代码，想要针对函数func_a写一个简单的单元测试： import unittest def func_c(arg1, arg2): a_dict = {} # 其他代码 return a_dict def func_b(arg3, arg4): b_list = [] a_arg1 = None
utittest和pytest中mock的使用详细介绍
okcross0的博客
04-12 1320
Mock是Python中一个用于支持单元测试的库，它的主要功能是使用mock对象替代掉指定的Python对象，以达到模拟对象的行为。
mock测试没有返回值的方法
05-19
如果你要对一个没有返回值的方法进行Mock测试，你可以使用Python的unittest.mock库中的MagicMock类来模拟该方法的行为。假设你要Mock的方法名称为`my_func()`，它没有返回值，那么你可以这样写Mock测试： ```python from unittest.mock import MagicMock def test_my_func(): # 创建Mock对象 mock_func = MagicMock() # 调用Mock方法 mock_func() # 断言Mock方法已经被调用 assert mock_func.called ``` 在这个测试中，我们创建了一个名为`mock_func`的Mock对象，并调用了它。然后，我们使用`assert`语句来断言该Mock方法是否被调用。如果该方法被调用，`assert mock_func.called`语句会返回True，测试通过。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值