python—unittest—数据驱动详细讲解(ddt)

小半 • 2023年1月12日下午2:48 • 技术随笔 • 阅读 229

数据驱动ddt

数据驱动ddt可以实现测试数据与测试脚本的分离，通过ddt来将测试数据加载到脚本中。采用数据驱动设计模式使一组数据对应一个测试用例，用例自动加载生成

ddt基础

pip install ddt

测试数据为嵌套字典的列表
测试类前加修饰@ddt
测试用例前加修饰@data()
运行后用例会自动加载成多个单独的用例

代码案例

import unittest
from ddt import ddt, data

#测试用例
cases = [
        {'title': '登录成功', 'expected': {'code': 200, 'msg': '登录成功'}, 'data': ('kobe', '666')},
        {'title': '登录失败', 'expected': {'code': 201, 'msg': '用户名或者密码不正确'}, 'data': ('kobe', '888')},
        {'title': '用户名不能为空', 'expected': {'code': 201, 'msg': '用户名不能为空'}, 'data': ('', '666')},
        {'title': '密码不能为空', 'expected': {'code': 201, 'msg': '密码不能为空'}, 'data': ('kobe', '')},
        {'title': '用户名和密码不能为空', 'expected': {'code': 201, 'msg': '用户名和密码不能为空'}, 'data': ('', '')},
    ]

@ddt
class LoginTestCase(unittest.TestCase):

	@classmethod
    def setUpClass(cls) -> None:
        print('开始')

    def setUp(self) -> None:
        print('开始执行用例')

    @data(*cases)
    def test_login(self, case):
        print(case)
        self.assertEqual(case['expected'], self.login(case['data'][0], case['data'][1]))

    def login(self, username, password):
        if username == 'kobe' and password == '666':
            return {'code': 200, 'msg': '登录成功'}

        if username == 'kobe' and username != '' and password != '666' and password != '':
            return {'code': 201, 'msg': '用户名或者密码不正确'}

        if username == 'kobe' and password == '':
            return {'code': 201, 'msg': '密码不能为空'}

        if username == '' and password == '666':
            return {'code': 201, 'msg': '用户名不能为空'}

        if username == '' and password == '':
            return {'code': 201, 'msg': '用户名和密码不能为空'}

    def tearDown(self) -> None:
        print('用例执行完毕')
        
    @classmethod
    def tearDownClass(cls) -> None:
        print('结束')


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

case依次为，每一条case表示一条用例

{'title': '登录成功', 'expected': {'code': 200, 'msg': '登录成功'}, 'data': ('kobe', '666')},
{'title': '登录失败', 'expected': {'code': 201, 'msg': '用户名或者密码不正确'}, 'data': ('kobe', '888')},
{'title': '用户名不能为空', 'expected': {'code': 201, 'msg': '用户名不能为空'}, 'data': ('', '666')},
{'title': '密码不能为空', 'expected': {'code': 201, 'msg': '密码不能为空'}, 'data': ('kobe', '')},
{'title': '用户名和密码不能为空', 'expected': {'code': 201, 'msg': '用户名和密码不能为空'}, 'data': ('', '')},