python unittest 之mock学习笔记

mock的详细用法(英文介绍):
本文先对函数的mock方法进行演示。
假设有文件fun1和fun2,fun2中的函数调用了fun1中的函数。利用mock方法生成fun1中函数的一个fake返回值,在此基础上,对fun2中的函数进行单元测试。

代码如下:
fun1文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# fun1.py
def flag():
    return 1

fun2文件

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#  fun2.py
import fun1
def adds(a,b,c):
    if fun1.flag():
        return (a+b+c)
    else:
        return (a+b)

对fun2利用mock测试文件:

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# mockTest.py

from fun2 import adds

import mock
import unittest

class AddTestCase(unittest.TestCase):
    @mock.patch('fun2.fun1')
    def test_adds(self, mock_fun1):
        # 设置fun2中调用fun1中的flag为0
        # 则调用fun2中的函数adds返回a+b
        print ("设置fun2中调用fun1中的flag为0")
        print ("则调用fun2中的函数adds返回前两个参数的和")
        mock_fun1.flag.return_value = 0
        num = adds(1,2,3)
        self.assertEqual(num,3)

        # 设置fun2中调用fun1中的flag为1
        # 则调用fun2中的函数adds返回a+b+c
        print ("设置fun2中调用fun1中的flag为1")
        print ("则调用fun2中的函数adds返回三个的和")
        mock_fun1.flag.return_value = 1
        num = adds(1, 2, 3)
        self.assertEqual(num, 6)

关于类的mock用法,后续更新~

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值