ddt源码分析

前置知识点    

相信做自动化测试的小伙伴们都知道数据驱动的好处,但是要知道怎么用之外 还是要了解下它是怎么工作的,了解了之后就可以改造ddt以更适合自己的需要

装饰器

首先要了解装饰器,因为ddt本身就是一个装饰器,它装饰了你的测试类和测试方法。装饰器有啥用呢,比如你封装了selenium的一些定位的方法,在执行的过程中打印出每个方法的传入参数,在每个方法的第一行都加上logging?太繁琐了吧,这个时候写一个打印日志的装饰器就可以搞定这个问题:如下:

装饰器:

```

def logging_wrapper(func):

def wrapper(self, *args, **kwargs):

if args:

logging.debug('请求参数:{}‘.format(args[0]))

return func(self, *args, **kwargs)

return wrapper

class MyTest():

@logging_wrapper

def func(self, a,b):

        pass

```


  当然还可以在里面加上捕获异常等等,可以理解为就是把函数当作参数传给了另一个函数。我们只需要知道怎么用和它的大致原理,不必钻研太深,如果要再多了解一些,有很多文章都写的比较好,就不赘述了。

反射

第二个知识点就是getattr setattr和delattr的使用了,众所周知,在python中 一切皆对象;那么获取 新增/修改和删除对象的属性怎么做呢,这三个内置函数就可以搞定。

```

class MyTest():

    def test1(self):

        pass

```

```

# 获取属性:

a=getattr(MyTest, 'test1')

# 新增属性:

setattr(MyTest, 'b', 'abc')

# 删除属性

delattr(MyTest, 'b)

```

另外结合__import__和inspect可以动态导入测试类或者业务的封装方法,多操作步骤的顺序执行;具体的后面再说

源码分析

ddt就一个py文件,先找到入口ddt函数,

```

def ddt(cls):

for name, funcin list(cls.__dict__.items()):

    if hasattr(func, DATA_ATTR):

        for i, v in enumerate(getattr(func, DATA_ATTR)):

            test_name= mk_test_name(name, getattr(v, "__name__", v), i)

            test_docstring= getattr(v, "__doc__", None)

            if hasattr(func, UNPACK_ATTR):

                if isinstance(v, tuple) or isinstance(v, list):

                    add_test(cls, test_name, test_docstring, func, *v)

                else:

                    # unpack dictionary

                    add_test(cls, test_name, test_docstring, func, **v)

            else:

                add_test(cls, test_name, test_docstring, func, v)

        delattr(cls, name)

    elif hasattr(func, FILE_ATTR):

        file_attr= getattr(func, FILE_ATTR)

        process_file_data(cls, name, func, file_attr)

        delattr(cls, name)

return cls

```

第一步:循环获取cls的所有方法,判断方法是否有DATA_ATTR或者FILE_ATTR属性;看过ddt是怎么使用,数据驱动的数据有两种形式:列表和json/yaml文件。而且会排除没有被装饰的方法

第二步:有DATA_ATTR或者FILE_ATTR属性的方法,会用被装饰的方法的name加index之类拼接出新的测试方法的name,即mk_test_name就是生成name的

第三步:add_test给cls增加新的测试方法,看下add_test的定义,使用了一个feed_data来实现增加测试方法,假如数据是五份测试数据,就会依次增加五个测试方法

第四部:删除原始的测试方法

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,968评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,601评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,220评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,416评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,425评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,144评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,432评论 3 401
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,088评论 0 261
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,586评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,028评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,137评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,783评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,343评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,333评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,559评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,595评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,901评论 2 345

推荐阅读更多精彩内容

  • 写在前面的话 代码中的# > 表示的是输出结果 输入 使用input()函数 用法 注意input函数输出的均是字...
    FlyingLittlePG阅读 2,734评论 0 8
  • 参考文献: 《大话设计模式》——吴强 《Python设计模式》——pythontip.com 《23种设计模式》—...
    梁林張斌阅读 2,257评论 0 4
  • 高阶函数:将函数作为参数 sortted()它还可以接收一个key函数来实现自定义的排序,reversec参数可反...
    royal_47a2阅读 679评论 0 0
  • 字符串格式化调用方法 —— format 通过创建字符串模板,利用format函数,替代相应的值。 可以通过绝对位...
    plutoese阅读 1,505评论 0 47
  • 那天爸爸取回蚕宝宝,拿回了家,我迫不及待的打开。 ...
    五二班韩馨然阅读 415评论 0 0