python测试框架unittest数据驱动插件ddt源代码分析

unittest需要第三方插件ddt来实现数据驱动。数据可以是列表、字典或者JSON文件。

>pip install ddt进行安装

下面是一个简单的使用举例。

import unittest

import ddt

@ddt.ddt

class TestAdd(unittest.TestCase):

    @ddt.data([1,2,3],[3,4,7])

    @ddt.unpack

    def test_add(self,a,b,c):

        assert a+b==c

则分别使用提供的两组数据,执行两个test。

提供多少组数据,就创建多少个测试方法。这里创建的测试方法是test_add_1__1__2__3_ 和

test_add_2__3__4__7_ (__main__.TestAdd)。

@ddt.file_data('data.json')实现读取json数据,内容可以是列表或者字典。

 

excel数据驱动开发思路:

将读取出的excle文件中的数据以双重列表的方式返回,作为@ddt.data()的参数。

比如:实现一个函数,get_data_from_sheet(excel_file_path,sheet_name)。以双重列表的方式返回data。@ddt.data(*get_data_from_sheet(excel_file_path,sheet_name))使用数据。

 

ddt源代码分析:

下面以处理列表数据为例来分析,处理字典数据和json数据文件的主要流程是相同的。

ddt(arg=None, **kwargs)中对于每一组数据,执行test_name = mk_test_name(name,getattr(v, "__name__", v),i,fmt_test_name)构造新test函数的名字,其中name是原test函数的名字,v是一组数据,i是这组数据的index,fmt_test_name指定新的test函数的名字的格式。

然后执行add_test(cls,test_name,test_data_docstring,func,*v)基于原test函数func生成将新的test函数test_name,加入原testcase类cls,test_data_docstring 是新的test函数的docstring,v是对应的那组数据。

最后执行delattr(cls, name),删除原test函数。原test函数已经通过各组数据具体化成了新的test函数,所以已经不需要它了。留着它,执行时会抛出异常,TypeError: test_add() missing 3 required positional arguments: 'a', 'b', and 'c'。

 

下面来重点看一下生成新test函数的逻辑。

add_test(cls, test_name, test_docstring, func, *args, **kwargs)中setattr(cls, test_name, feed_data(func, test_name, test_docstring, *args, **kwargs))。

feed_data(func, new_name, test_data_docstring, *args, **kwargs)中

    @wraps(func)

    def wrapper(self):

        return func(self, *args, **kwargs)

    wrapper.__name__ = new_name

    wrapper.__wrapped__ = func

    wrapper.__doc__ = test_data_docstring

    return wrapper

在原来的test函数func的基础上使用具体参数值,生成新的test函数,名字是new_name。然后将这个函数作为一个方法添加到原测试用例类cls中。

为了方便大家理解,下面我把生成新test函数的逻辑提取出来,写出下面的简化实现,以及对其使用的举例。

#下面两个函数生成新的test函数

def add_test(cls, test_name, func, *args):

    setattr(cls, test_name, feed_data(func, test_name, *args))

def feed_data(func, new_name, *args):

    def wrapper(self):

        return func(self, *args)

    wrapper.__name__ = new_name

    return wrapper

#下面是使用举例

import unittest

class TestAdd(unittest.TestCase):

    def test_add(self,a,b,c):

        self.assertEqual(a+b,c)

add_test(TestAdd,"test_add_1_1_2_3",TestAdd.test_add,1,2,3)

add_test(TestAdd,"test_add_1_3_4_7",TestAdd.test_add,3,4,7)

delattr(TestAdd,"test_add")

unittest.main(verbosity=2)

'''

运行结果:

Total number of cases:2

test_add_1_1_2_3 (__main__.TestAdd) 1 ... ok

test_add_1_3_4_7 (__main__.TestAdd) 2 ... ok

----------------------------------------------------------------------

Ran 2 tests in 0.000s

OK

'''

你可能感兴趣的:(Python)