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用法,后续更新~