当前位置：首页 > news >正文

UnitTest笔记：拓展库DDT的使用

news 2025/7/4 19:14:27

DDT (Data-Drivers- Tests) 允许使用不同的测试数据运行同一个测试用例，展示为不同的测试用例。

第一步： pip安装 ddt

第二步：创建test_baidu_ddt.py

1. 测试类要使用@ddt 修饰

2. 不同形式的参数化：列表，字典（key 要与测试方法保持一致），元祖

import unittest
from selenium import webdriver
from time import sleep
from ddt import ddt, data, file_data, unpack@ddt
class TestBaidu(unittest.TestCase):@classmethoddef 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"])@unpackdef test_search1(self, case, search_key):print("第一组测试用例：", case)self.baidu_search(search_key)print("self.driver.title:", self.driver.title)self.assertEqual(self.driver.title, search_key + "_百度搜索")# 参数化使用方式二-------------------- 元组@data(("case1", "selenium"), ("case2", "ddt"), ("case3", "python"))@unpackdef test_search2(self, case, search_key):print("第二组测试用例：", case)self.baidu_search(search_key)self.assertEqual(self.driver.title, search_key + "_百度搜索")print("self.driver.title:", self.driver.title)# 参数化使用方式三---------------------  字典@data({"search_key": "selenium"},{"search_key": "ddt"},{"search_key": "python"})@unpackdef test_search3(self, search_key):print("第三组测试用例：", search_key)self.baidu_search(search_key)self.assertEqual(self.driver.title, search_key + "_百度搜索")print("self.driver.title:", self.driver.title)# 参数化读取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 + "_百度搜索")print("self.driver.title:", self.driver.title)# 参数化读取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 + "_百度搜索")print("self.driver.title:", self.driver.title)@classmethoddef tearDownClass(cls):cls.driver.quit()if __name__ == '__main__':unittest.main(verbosity=2)  #日志等级2

3.支持数据文件读取

3.1json 文件读取

数据文件：test_case/ddt_data_file.json @file_data('ddt_data_file.json') 读取文件要与test_baidu_ddt.py 相同否则需要加上路径

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

3.2 yaml 文件读取：ddt_data_file.yaml

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

每个数据会被解析为[{'search_key':'python'}] 获取值case[0]["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 + "_百度搜索")print("self.driver.title:", self.driver.title)

第三步：创建 unittestdemo1\baidu_ddt_run_tests.py

HTMLTestRunner

import time
import unittest
from HTMLTestRunner import HTMLTestRunner# # 把测试报告作为附件发送到指定邮箱。
# def send_mail(report):
#     yag = yagmail.SMTP(user="1164512033@qq.com",
#                        password="zjx3601296",
#                        host='mail.qq.com')
#     subject = "标题，自动化测试报告"
#     contents = "正文，请查看附件。"
#     yag.send('1164512033@qq.com', subject, contents, report)
#     print('email has send out !')if __name__ == '__main__':# 定义测试用例的目录为当前目录print("----__main__-------")test_dir = './test_case'  # 测试用例的路径suit = unittest.defaultTestLoader.discover(test_dir, pattern='test_baidu_dtt.py')print("suit===========",suit)# 取当前日期时间now_time = time.strftime("%Y-%m-%d %H_%M_%S")html_report = './test_report/' + now_time + 'result.html'fp = open(html_report, 'wb')# 调用HTMLTestRunner，运行测试用例runner = HTMLTestRunner(stream=fp,title="baidu_ddt测试报告",description="运行环境：Windows 10, Chrome浏览器")runner.run(suit)print("run -------------------------------------suit")fp.close()#send_mail(html_report)  # 发送报告

运行结果： baidu_ddt_run_tests.py 不能以test 开头