自动化测试：使用DDT

DDT（Data-Driven Tests）是针对 unittest 单元测试框架设计的扩展库。允许使用不同
的测试数据来运行一个测试用例，并将其展示为多个测试用例。
GitHub 地址：https://github.com/datadriventests/ddt。
DDT 支持 pip 安装。
安装ddt:pip install ddt

import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack

@ddt
class TestBaidu(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://www.baidu.com"

    def baidu_search(self,search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        sleep(3)

    #参数化使用方式一--列表
    @data(["case1","selenium"],["case2","ddt"],["case3","python"])
    @unpack
    def test_search1(self,case,search_key):
        print("第一组测试用例：", case)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + "_百度搜索")

    #参数化使用方式二--元组
    @data(("case1","selenium"),("case2","ddt"),("case3","python"))
    @unpack
    def test_search2(self,case,search_key):
        print("第二组测试用例：",case)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + "_百度搜索")

    #参数化使用方式三--字典
    @data({"search_key":"selenium"},{"search_key":"ddt"},{"search_key":"python"})
    @unpack
    def test_search3(self,search_key):
        print("第三组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    @classmethod
    def tearDownClass(cls):
        sleep(10)
        cls.driver.quit()

if __name__ == '__main__':
    unittest.main(verbosity=2)

使用 DDT 需要注意以下几点。
首先，测试类需要通过@ddt 装饰器进行装饰。
其次，DDT 提供了不同形式的参数化。这里列举了三组参数化，第一组为列表，第二
组为元组，第三组为字典。需要注意的是，字典的 key 与测试方法的参数要保持一致。
执行结果如下。

image.png

DDT 同样支持数据文件的参数化。它封装了数据文件的读取，让我们更专注于数据文
件中的内容，以及在测试用例中的使用，而不需要关心数据文件是如何被读取进来的。
1、使用json参数化，脚本同级目录下新建ddt_data_file.json文件

{
    "case1":{"search_key":"python"},
    "case2":{"search_key":"ddt"},
    "case3":{"search_key":"selenium"}
}

脚本中使用方式为：

    #参数化使用方式四-json数据文件参数化
    @file_data('ddt_data_file.json')
    def test_search4(self,search_key):
        print("第四组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

2、使用yaml，脚本同级目录下新建ddt_data_file.yaml文件

case1:
- search_key : "python"
case2:
- search_key : "ddt"
case3:
- search_key : "selenium"

脚本中使用方式为：

    #参数化使用方式五-yaml数据文件参数化
    @file_data('ddt_data_file.yaml')
    def test_search5(self,case):
        search_key = case[0]["search_key"]
        print("第五组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

这里的取值与上面的 JSON 文件有所不同，因为每一条用例都被解析为[{'search_key':
'python'}]，所以要想取到搜索关键字，则需要通过 case[0]["search_key"]的方式获取。
所以完整的脚本为：

import unittest
from time import sleep
from selenium import webdriver
from ddt import ddt,data,file_data,unpack

@ddt
class TestBaidu(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://www.baidu.com"

    def baidu_search(self,search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        sleep(3)

    #参数化使用方式一--列表
    @data(["case1","selenium"],["case2","ddt"],["case3","python"])
    @unpack
    def test_search1(self,case,search_key):
        print("第一组测试用例：", case)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + "_百度搜索")

    #参数化使用方式二--元组
    @data(("case1","selenium"),("case2","ddt"),("case3","python"))
    @unpack
    def test_search2(self,case,search_key):
        print("第二组测试用例：",case)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key + "_百度搜索")

    #参数化使用方式三--字典
    @data({"search_key":"selenium"},{"search_key":"ddt"},{"search_key":"python"})
    @unpack
    def test_search3(self,search_key):
        print("第三组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    #参数化使用方式四-json数据文件参数化
    @file_data('ddt_data_file.json')
    def test_search4(self,search_key):
        print("第四组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    #参数化使用方式五-yaml数据文件参数化
    @file_data('ddt_data_file.yaml')
    def test_search5(self,case):
        search_key = case[0]["search_key"]
        print("第五组测试用例：",search_key)
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    @classmethod
    def tearDownClass(cls):
        sleep(10)
        cls.driver.quit()

if __name__ == '__main__':
    unittest.main(verbosity=2)

                                                    --来源于虫师的《Selenium3自动化测试实战——基于Python语言》