前言

用 python 做過自動化的小伙伴，大多數都應該使用過 ddt 這個模塊，不可否認 ddt 這個模塊確實挺好用，可以自動根據用例資料，來生成測驗用例，能夠很方便的將測驗資料和測驗用例執行的邏輯進行分離，接下來就帶大家一起自己，手把手擼出一個 ddt，

1、DDT 的實作原理

首先我們來看一下 ddt 的基本使用：

ddt 在使用時非常簡潔，也就是兩個裝飾器，@ddt 這個裝飾器裝飾測驗類，@data 這個裝飾器裝飾器用例方法并傳入測驗資料，這兩個裝飾器實作的效果就是根據傳入的用例資料自動生成用例，具體是怎么實作的呢？其實實作的思路也特別的簡單，也就兩個步驟：

第一步：把傳進來的用例資料保存起來

第二步：遍歷用例資料，每遍歷一條資料 就動態的給測驗類添加一個用例方法，

ddt 中的兩個裝飾器其實實作的就是這么兩個步驟：

@data:做的是第一步將傳入測驗資料保存起來;

@ddt 做的是第二步，遍歷用例資料，給測驗類動態添加用例方法，

2、data 裝飾器的實作

前面我們說到 data 這個裝飾器，做的事情是將用例資料保存起來，那么如何保存呢？其實最簡單的方式就是 保存被裝飾的這個用例方法的屬性，接下來我們來具體實作：

先看一個 ddt 使用的案例

  @ddt
class TestLogin(unittest.TestCase):

    @data(11,22)
    def test_login(self, item):
    	pass

了解過裝飾器裝飾器原理的小伙伴，應該都知道上面@data(11,22) 這行代碼執行的效果等同于

  test_login = data(11,22)(test_login)

接下來我們來分析一下上面這行代碼，首先是呼叫 data 這個裝飾器函式，把用例資料 11,22 當成引數傳入進去，然后回傳一個可呼叫物件(函式)，再次呼叫回傳的函式并把用例方法傳入進去，明確了呼叫的流程那么我們就可以結合之前的需求去定義 data 這個裝飾器函式了，具體實作如下：

  def data(*args):
    def wrapper(func):
        setattr(func, "PARAMS", args)
        return func
    return wrapper

代碼解讀：

前面的案例在使用 data 時，執行的 test_login = data(11,22)(test_login) 先呼叫 data 傳入的 11,22 通過不定長引數 args 接收，然后回傳嵌套的函式 wrapper 然后呼叫回傳的 wrapper 函式，傳入被裝飾的 test_login 方法 在 wrapper 函式中我們把用例資料保存為 test_login 這個方法的 PARAMS 屬性，再把 test_login 回傳 到此為止，data 這個裝飾器我們就實作用例資料的保存

3、ddt 裝飾器的實作

通過 data 這個裝飾器我們實作了用例資料保存之后，我們接下來實作 ddt 這個裝飾器，根據用例資料生成測驗用例，前面的案例 @ddt 裝飾測驗類的時候，實際上執行的效果等同于下面的代碼

  TestLogin = ddt(TestLogin)

這行代碼就是把被裝飾器的類傳入到 ddt 這個裝飾器函式中，再把回傳值賦值給 TestLogin，之前我們分析的時候說了 ddt 這個裝飾器做的事情是遍歷用例資料，動態的給測驗類添加用例方法，接下來我們就來實作 ddt 這個裝飾器內部的邏輯，

  def ddt(cls):
    for name, func in list(cls.__dict__.items()):
        if hasattr(func, "PARAMS"):
            for index, case_data in enumerate(getattr(func, "PARAMS")):
                new_test_name ="{}_{}".format(name,index)
                setattr(cls, new_test_name, func)
            else:
                delattr(cls, name)
    return cls

代碼解讀：

ddt 函式內部邏輯說明： 1、呼叫 ddt 這個函式時會把測驗類當成引數傳入進來， 2、然后通過 cls.__dict__ 獲取測驗的所有屬性和方法，進行遍歷 3、判斷變數出來的屬性或方法 有沒有 PARAMS 這個屬性， 4、如果有，則說明這個方法用 data 裝飾器裝飾過并傳入了用例資料， 5、通過 getattr(func, "PARAMS")獲取所有的用例資料，進行遍歷， 6、每遍歷出來一組用例資料，生產一個用例方法名， 再動態的給測驗類添加一個用例方法， 7、遍歷完所有用例資料之后，洗掉測驗類原來定義的測驗方法 8、最后回傳測驗類

當目前為止 ddt 和 data 這兩個裝飾器函式的基本功能實作了，可以自動根據用例資料生成測驗用例了，接下來我們寫個測驗類來檢查一下

  # 定義裝飾器函式data
def data(*args):
    def wrapper(func):
        setattr(func, "PARAMS", args)
        return func

    return wrapper

# 定義裝飾器函式ddt
def ddt(cls):
    for name, func in list(cls.__dict__.items()):
        if hasattr(func, "PARAMS"):
            for index, case_data in enumerate(getattr(func, "PARAMS")):
                new_test_name = "{}_{}".format(name, index)
                setattr(cls, new_test_name, func)
            else:
                delattr(cls, name)
    return cls


import unittest

# 撰寫測驗類
@ddt
class TestDome(unittest.TestCase):
    @data(11, 22, 33, 44)
    def test_demo(self):
        pass

運行上述用例，我們就會發現執行了四條用例，根據用例資料生成用例的功能就已經實作了，

4、解決用例引數傳遞的問題

雖然上面基本的功能已經實作了，但是還存在一個問題，用例的資料沒有傳遞到用例方法中，那么用例資料傳遞怎么實作了，我們可以通過一個閉包函式對用例方法進行修，從而實作在呼叫用例方法的時候，把用例測驗當成引數傳遞進去，修改原有用例方法的函式代碼如下

  from functools import wraps

def update_test_func(test_func,case_data):
    @wraps(test_func)
    def wrapper(self):
        return test_func(self, case_data)
    return wrapper

代碼解讀：

上面我們定義了一個叫做 update_test_func 的閉包函式 閉包函式接收兩個引數：test_func(接收用例方法),case_data(接收用例資料) 閉包函式回傳一個嵌套函式，嵌套函式內部呼叫原來的用例方法，并傳入測驗資料 嵌套函式在定義時，使用了 functools 模塊中的裝飾器 wraps 來裝飾，它可以讓 wrapper 這個嵌套函式具有 test_func 這個用例函式的相關屬性，

下面我們回到前面寫的 ddt 這個函式中，在給測驗類添加用例之前，呼叫 update_test_func 方法對用例方法進行修改，

  def ddt(cls):
    for name, func in list(cls.__dict__.items()):
        if hasattr(func, "PARAMS"):
            for index, case_data in enumerate(getattr(func, "PARAMS")):
                # 生成一個用例方法名
                new_test_name = "{}_{}".format(name, index)
                # 修改原有的測驗方法，設定用例資料為測驗方法的引數
                test_func = update_test_func(func,case_data)
                setattr(cls, new_test_name, test_func)
            else:
                delattr(cls, name)
    return cls

通過加上這一步之后，我們在測驗類中 動態給測驗類添加的測驗方法，其實指向的全部是 update_test_func 里面定義的 wrapper 函式，在執行測驗用的時候實際上也是執行的 wrapper 函式，而在 wrapper 函式內部，我們呼叫了原來定義的測驗方法，并將用例資料傳入了進去，到此為止 ddt 的功能我們就完全實作了，

下面是一個完整的案例，大家可以復制過去運行，也可以自己去寫一遍，還可以根據自己的一些需求進行自定義的擴展，

完整案例

from functools import wraps
import unittest
# --------ddt的實作--------
def data(*args):
    def wrapper(func):
        setattr(func, "PARAMS", args)
        return func
    return wrapper
def update_test_func(test_func, case_data):
    @wraps(test_func)
    def wrapper(self):
        return test_func(self, case_data)
    return wrapper
def ddt(cls):
    for name, func in list(cls.__dict__.items()):
        if hasattr(func, "PARAMS"):
            for index, case_data in enumerate(getattr(func, "PARAMS")):
                # 生成一個用例方法名
                new_test_name = "{}_{}".format(name, index)
                # 修改原有的測驗方法，設定用例資料為測驗方法的引數
                test_func = update_test_func(func, case_data)
                setattr(cls, new_test_name, test_func)
            else:
                delattr(cls, name)
    return cls
# --------測驗用例撰寫--------
@ddt
class TestDome(unittest.TestCase):
    @data(11, 22, 33, 44)
    def test_demo(self, data):
        assert data < 40
#---------用例執行-----------
unittest.main()

標籤：其他

上一篇：jmeter使用Plugins插件進行服務器性能監控

下一篇：? 實戰系列 ? 1?? 微信小程式自動化測驗實踐（附 Python 原始碼）??