文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

python自动化测试中装饰器@ddt和@data源码解析

2023-10-08 08:34

关注

在这里插入图片描述

一、使用ddt和data装饰器的大致框架如下,每个test_开头的方法,代表一条测试用例

from ddt import ddt,dataimport unittesttest_datas=[    {'id':1,'title':'测试用例1'},    {'id':2,'title':'测试用例2'},    {'id':3,'title':'测试用例3'},    {'id':4,'title':'测试用例4'}]@ddtclass TestDemo(unittest.TestCase):    @data(*test_datas)    def test_demo1(self,item):        print('测试用例执行',item)

unittest中的测试用例:

测试类中每一个test开头的方法就是一条测试用例

ddt根据用例数据生成测试用例的思路:

1、利用data装饰器:传入测试数据,在装饰器中将测试数据保存起来
2、ddt这个装饰器:遍历测试数据,每遍历出一条数据,往测试类中添加一个test开头的方法
setattr(类,方法名,方法)

二、给类动态的增加方法

案例1

setattr(对象/类,属性名/方法名,属性值/方法)

特别注意:

给类动态增加方法一定要加self

class Demo:    def test_1(self):        print("这个是方法test_1")def kobe(self,item):    print("kobe-----执行了",item)datas=[2,8,23,22,24]#根据数据动态给测试类中增加5个方法for i in datas:    name='test_1_{}'.format(i)    #给类动态增加方法    setattr(Demo,name,kobe)print(Demo.__dict__)

在这里插入图片描述

案例2:调用动态执行的5个方法,执行结果都为kobe-----执行了 24,有bug

class Demo:    def test_1(self):        print("这个是方法test_1")def kobe(self,item):    print("kobe-----执行了",item)datas=[2,8,23,22,24]#根据数据动态给测试类中增加5个方法for i in datas:    name='test_1_{}'.format(i)    def wrapper(self):        kobe(self,i)    #给类动态增加方法    setattr(Demo,name,wrapper)#print(Demo.__dict__)Demo().test_1_2()Demo().test_1_8()Demo().test_1_22()Demo().test_1_23()Demo().test_1_24()

执行结果:
kobe-----执行了 24
kobe-----执行了 24
kobe-----执行了 24
kobe-----执行了 24
kobe-----执行了 24

原因分析

在这里插入图片描述

案例3:解决案例2的bug

定义闭包create_method:进行数据锁定,锁定的是datas=[2,8,23,22,24]

class Demo:    def test_1(self):        print("这个是方法test_1")def kobe(self,item):    print("kobe-----执行了",item)datas=[2,8,23,22,24]#todo 使用闭包进行数据锁定def create_method(i):    def wrapper(self):        kobe(self,i)    return wrapper#根据数据动态给测试类中增加5个方法for i in datas:    name='test_1_{}'.format(i)    wrapper=create_method(i)    #给类动态增加方法    setattr(Demo,name,wrapper)Demo().test_1_2()Demo().test_1_8()Demo().test_1_22()Demo().test_1_23()Demo().test_1_24()

三、ddt和data的源码解析

from ddt import ddt,dataimport unittesttest_datas=[    {'id':1,'title':'测试用例1'},    {'id':2,'title':'测试用例2'},    {'id':3,'title':'测试用例3'},    {'id':4,'title':'测试用例4'}]def ddt(cls):    '''遍历测试数据,给类动态添加方法'''    #如何通过类获取方法?    #res=cls.__dict__    #print('测试类的方法和属性字典',res)    for name,method in list(cls.__dict__.items()):        #遍历出来的属性值(方法)是否拥有datas属性(测试数据)        if hasattr(method,'datas'):            #获取方法中保存的测试数据            datas=getattr(method,'datas')            #遍历测试数据            for index,value in enumerate(datas):                print("数据:",value)                #给测试类动态添加用例                method_name='{}_{}'.format(name,index+1)                print('方法名',method_name)#给类动态的增加方法                def wrapper(self):                    method(self, value)                #todo 给测试类动态添加一个测试方法                setattr(cls,method_name,wrapper)    return clsdef data(*args):    '''将测试数据保存为测试方法的属性'''    #*args接收到的是data装饰器传递进来的数据    def wrapper(func):        #func接收的是data装饰的函数        func.datas=args        return func    return wrapper@ddtclass TestDemo():    @data(*test_datas)      #test_demo1=data(*test_datas)(test_demo1)    def test_demo1(self,item):        print('测试用例执行',item)#print(TestDemo.test_demo1.__dict__)

这样写的话有bug
在这里插入图片描述
原因:
在这里插入图片描述

解决:

采用闭包进行数据锁定,锁定value和method

def create_test_method(method,value):    def wrapper(self):        method(self, value)    return wrapper
from ddt import ddt,dataimport unittesttest_datas=[    {'id':1,'title':'测试用例1'},    {'id':2,'title':'测试用例2'},    {'id':3,'title':'测试用例3'},    {'id':4,'title':'测试用例4'}]def create_test_method(method,value):    def wrapper(self):        method(self, value)    return wrapperdef ddt(cls):    #todo @ddt这个装饰器:遍历测试数据,每遍历出一条数据,往测试类中添加一个test开头的方法    #setattr(类,方法名,方法)    res=list(cls.__dict__.items())    print(res)    for name,method in res:        print(name,method)        if hasattr(method,'datas'):            #如果有datas属性,获取方法中保存的datas            datas=getattr(method,'datas')            #遍历测试数据            for index,value in enumerate(datas):                print('测试数据:',value)                #给测试类动态的增加测试用例                method_name='{}_{}'.format(name,index+1)                print('方法:',method_name,method)                #todo 给类动态的增加方法,最终希望执行def test_demo1(self,item):这个方法的                #test_method=method                #但是item需要自己传,但是unittest是不需要传递参数的    # def wrapper(self):                #     method(self,value)                wrapper=create_test_method(method, value)                # todo 给测试类动态添加一个测试方法                setattr(cls, method_name, wrapper)            else:                delattr(cls,name)    return clsdef data(*args):    # *args为给装饰器传递的参数test_datas    def wrapper(func):        # func为被装饰器装饰的函数test_demo1        #todo @data装饰器的作用是保存测试数据,将测试数据存放到函数属性中        func.datas = test_datas        return func    return wrapper@ddtclass TestDemo(unittest.TestCase):    @data(*test_datas)      #test_demo1=data(*test_datas)(test_demo1)    def test_demo1(self,item):        print('测试用例执行',item)

分部解析代码

@data(*test_datas)    def test_demo1(self,item):    print('测试用例执行',item)

1、上面3行代码可以写成如下:

@data(*test_datas):可以表示为test_demo1=data(*test_datas)(test_demo1)

2、输出属性(方法)名称和属性值

for name,method in list(cls.__dict__.items())

在这里插入图片描述

3、将遍历出来的属性名(方法)判断是否包含datas属性,如果有datas属性,获取方法中保存的datas

if hasattr(method,'datas'):datas=getattr(method,'datas')

在这里插入图片描述

来源地址:https://blog.csdn.net/YZL40514131/article/details/126679826

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯