Bootstrap

unittest中的数据驱动ddt

unittest中的ddt(‌Data-Driven Tests,‌数据驱动测试)‌的作用是实现测试数据与测试脚本的分离,‌通过ddt将测试数据加载到脚本中,‌使得一组数据对应一个测试用例,‌用例自动加载生成。‌

ddt是一种数据驱动测试框架,‌它允许开发者将测试数据与测试脚本分离,‌通过这种方式,‌可以更灵活地管理和运行测试用例。‌使用ddt的好处在于,‌它能够提高测试用例的可维护性和可读性,‌同时减少重复代码,‌使得测试数据和测试逻辑更加清晰。‌通过ddt和unittest框架的结合,‌可以实现测试用例脚本的编写、‌测试执行控制以及测试数据的批量加载,‌从而完成不同接口测试用例的批量执行和覆盖不同的测试场景。‌

@ddt:类的装饰器,继承的是TestCase类
@data():@data装饰符可以把参数当成测试数据,参数可以是单个值、列表、元祖、字典这些类型,用于输入测试数据
@unpack:分解数据标志,主要是把元祖和列表解析成多个参数
@file_data():输入文件,如json或者yaml类型文件

ddt的安装

pip intsall ddt

下面我们以百度搜索为例,来看看ddt的用法

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


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

    def baidu_search(self, search_key):
        self.driver.get(self.base_url)
        self.driver.find_element(by='id', value="kw").send_keys(search_key)
        self.driver.find_element(by='id', value="su").click()
        sleep(10)

    #参数化使用方法一
    @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):
        cls.driver.quit()



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


执行结果如下:
在这里插入图片描述

;