需求:
在某种情况下,需要用不同的参数组合测试同样的行为,你希望从test case的执行结果上知道在测试什么,而不是单单得到一个大的 test case;此时如果仅仅写一个test case并用内嵌循环来进行,那么其中一个除了错误,很难从测试结果里边看出来。
问题
问题的关键在于是否有办法根据输入参数的不同组合产生出对应的test case;譬如你有10组数据,那么得到10个test case,当然不适用纯手工的方式写那么多个test_成员函数。
分析
一种可能的思路是不利用unittest.TestCase这个类框中的test_成员函数的方法,而是自己写runTest这个成员函数,那样会有一些额外的工作,而且看起来不是那么“智能”。那该如何让框架自动调用testcase呢?
思路
利用setattr来自动为已有的TestCase类添加成员函数
为了使这个方法凑效,需要用类的static method来生成decorate类的成员函数,并使该函数返回一个test函数对象出去
在某个地方注册这个添加test成员函数的调用(只需要在实际执行前就可以,可以放在模块中自动执行亦可以手动调用)
# -*- coding: utf-8 -*-
import unittest
class MyTestCase(unittest.TestCase):
def setUp(self):
pass
def action(self, arg1, arg2):
pass
def tearDown(self):
pass
@staticmethod
def getTestFunc(arg1, arg2):
def func(self):
self.action(arg1, arg2)
return func
def generate_cases():
data = [('arg11', 'arg12'), ('arg21', 'arg22'), ('arg31', 'arg32')]
for args in data:
setattr(MyTestCase, 'test_func_%s_%s' % (args[0], args[1]),
MyTestCase.getTestFunc(*args))
generate_cases()
if __name__ == '__main__':
unittest.main()
转载:https://my.oschina.net/u/2474096/blog/877006