Unittest篇之数据驱动ddt

引入

在进行单元测试的时候我们可能会遇到这种情况：某条用例我想执行很多遍，但是每次执行传入的参数不同。没错，我们在测试登录功能的时候就很可能出现这种需求。这就引入了我们今天的ddt（数据驱动）。

ddt安装

ddt是python的扩展库，我们可以使用pip来进行安装，命令如下：

pip install ddt

ddt使用

@ddt.ddt 和@ddt.data()

话不多说，先来看代码：

#coding=utf-8

import unittest
import ddt

messages=("1","2","3")
@ddt.ddt
class TestDDT(unittest.TestCase):

    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")
    @ddt.data(*messages)
    def test_print(self,message1):
        print(message1)

if __name__=="__main__":
    unittest.main()

在上面这段代码中， 我们先是导入了ddt包；接着定义了一组数据，也就是我们要传的参数元组；然后我们在测试类上加了@ddt.ddt的这样一个标签；然后我们正常写我们的测试函数，注意在测试函数上我们又加了@ddt.data()这样一个标签，括号里面写的是我们刚刚提到的参数元组（别忘了前面的*号）；最后是我们的运行函数。具体运行情况如下图：就这样我们通过写一个测试用例，传入不同的参数进行了多次执行。

当然，你也可以不定义参数元组或列表，直接传入，执行效果是一样的：

#coding=utf-8

import unittest
import ddt

@ddt.ddt
class TestDDT(unittest.TestCase):

    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")
    @ddt.data("1","2","3")
    def test_print(self,message1):
        print(message1)

if __name__=="__main__":
    unittest.main()

@ddt.unpack

当我们传入的参数不只是一个，而是两个甚至多个时候，我们需要用到另外一个标签，我们在@ddt.data()标签下又加了@ddt.unpack这个标签，如果不加这个标签执行将会报错，如下代码：

#coding=utf-8

import unittest
import ddt

messages=[("you","你"),("me","我"),("he","他")]
@ddt.ddt
class TestDDT(unittest.TestCase):

    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")
    @ddt.data(*messages)
    @ddt.unpack
    def test_print(self,message1,message2,):
        print(message1,"------",message2)

if __name__=="__main__":
    unittest.main()

运行结果如图：

@ddt.file()

我们通过在代码中的定义数据列表的方式来实现数据驱动，以后维护起来不方便。我们当然可以通过在Excel表格中存储数据然后解析生成列表的方式来数据分离。这里我们介绍另一种方式，就是@ddt.file()标签。这个方法中我们可以处理yaml类型的文件(文件后缀名是".yml"或".yaml"的时候)，也可以处理json文件(文件后缀名不是".yml"或".yaml"的时候)。
JSON文件：我们先建立一个JSON格式的文件：

[
  "你",
  "我",
  "他"
]

我们的测试代码如下：

#coding=utf-8

import unittest
import ddt


@ddt.ddt
class TestDDT(unittest.TestCase):

    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")

    @ddt.file_data("testJson.json")
    def test_print(self,message1):
        print(message1)

if __name__=="__main__":
    unittest.main()

yaml文件：我们先建立一个yaml文件，关于".yaml"文件的格式可以参考yaml编写参考（感谢该作者），我们在使用yaml文件时需要安装PyYAML扩展模块，使用pip命令安装即可。

-
  "你"
-
  "我"
-
  "他"

我们的测试代码如下：

#coding=utf-8

import unittest
import ddt


@ddt.ddt
class TestDDT(unittest.TestCase):

    def setUp(self):
        print("start")
    def tearDown(self):
        print("end")

    @ddt.file_data("testYaml.yaml")
    def test_print(self,message1):
        print(message1)

if __name__=="__main__":
    unittest.main()

两个方式的运行结果均如下图：

参考：
https://www.cnblogs.com/yoyoketang/p/6706490.html(感谢作者)
https://www.cnblogs.com/chendai21/p/8675982.html（感谢作者）
https://www.cnblogs.com/miniren/p/7099187.html（感谢作者）