寫接口case時(shí)，有時(shí)需要對(duì)cae做一些共性的操作，最典型的場(chǎng)景如：獲取case執(zhí)行時(shí)間、打印log等。

有沒(méi)有一種辦法來(lái)集中處理共性操作從而避免在每個(gè)case中都寫相同的代碼（如：每個(gè)case都需要寫相同的獲取執(zhí)行時(shí)間的代碼）呢？

Python的裝飾器可以做到這一點(diǎn)。

可以這樣理解裝飾器，裝飾器運(yùn)用閉包對(duì)目標(biāo)函數(shù)進(jìn)行裝飾（目標(biāo)函數(shù)作為閉包外部函數(shù)的引用），即在執(zhí)行目標(biāo)函數(shù)之前、之后執(zhí)行一些指定的代碼來(lái)完成想要的業(yè)務(wù)邏輯。

概念看起來(lái)總是晦澀，直接上語(yǔ)法吧。

第一步，先看一下閉包的語(yǔ)法：

def outer(a):
    b = 10
    # inner是內(nèi)函數(shù)
    def inner():
        # 在內(nèi)函數(shù)中 用到了外函數(shù)的臨時(shí)變量
        print(a + b)
    # 外函數(shù)的返回值是內(nèi)函數(shù)的引用
    return inner

第二步，再來(lái)看一下閉包的裝飾器版本的語(yǔ)法：

和上面閉包的例子只有一個(gè)區(qū)別：閉包中外部函數(shù)的引用是一個(gè)整數(shù)a，而此時(shí)是一個(gè)函數(shù)的引用（函數(shù)的引用也就是函數(shù)名）。

# 裝飾器用到閉包原理：外函數(shù)內(nèi)部定義了一個(gè)內(nèi)函數(shù)，內(nèi)函數(shù)使用外函數(shù)的局部變量，并且外函數(shù)返回了內(nèi)函數(shù)的引用
def outer(target):   # 裝飾器函數(shù) 傳入一個(gè)想對(duì)其裝飾的目標(biāo)函數(shù)的 引用，將在內(nèi)函數(shù)中使用。
　　 b = 10
    c = 1
    d = 5
    # inner是內(nèi)函數(shù)
    def inner():
        print(b + c)
        # 外部函數(shù)傳入的參數(shù)target，就是希望裝飾的目標(biāo)函數(shù)的引用
        target()   # 這里實(shí)際上執(zhí)行了目標(biāo)函數(shù)，想對(duì)這個(gè)函數(shù)進(jìn)行裝飾，所以在該函數(shù)執(zhí)行之前和之后進(jìn)行一番操作，具體什么操作看業(yè)務(wù)邏輯
        print(c + d)
    # 外函數(shù)的返回值是內(nèi)函數(shù)的引用
    return inner

注意：target只是函數(shù)的一個(gè)引用（引用指向函數(shù)在內(nèi)存中的位置），不會(huì)執(zhí)行。帶（）時(shí)（ target() ）才會(huì)執(zhí)行該函數(shù)。

最后一步，再看一下裝飾器的語(yǔ)法：

@decorator
    def test_01():

所以可以總結(jié)出：裝飾器decorator是閉包的外部函數(shù)，即 outer() （裝飾器是一個(gè)函數(shù)，即閉包的外部函數(shù)），被裝飾函數(shù)test_01是閉包傳入的參數(shù)，即target。

舉個(gè)例子：

以統(tǒng)計(jì)各接口請(qǐng)求耗時(shí)為例。

裝飾器（decorat.py）：

import time
def time_consume(func):
    def inner():
        time_start = time.time()  # 目標(biāo)函數(shù)開(kāi)始之前取一下時(shí)間
        print("\n接口請(qǐng)求前的時(shí)間是", time_start)
        func()
        time_end = time.time()  # 目標(biāo)函數(shù)結(jié)束之后取一下時(shí)間
        print("接口請(qǐng)求后的時(shí)間是", time_end)
        t = time_end - time_start  # 計(jì)算目標(biāo)函數(shù)執(zhí)行花了多長(zhǎng)時(shí)間
        print("接口耗時(shí):", t)
    return inner

接口（test_case.py）：

import requests
import decorat
@decorat.time_consume
def test_demo():
    res = requests.get("https://www.baidu.com")
    assert res.status_code == 200

效果：

再來(lái)總結(jié)下這個(gè)例子的整個(gè)過(guò)程：

'''
@decorat.time_consume實(shí)際上執(zhí)行的是: test_demo = @decorat.time_consume(test_demo)
因?yàn)榫幊陶Z(yǔ)言都是從右向左來(lái)解析執(zhí)行的，那么這句代碼會(huì)發(fā)生的事情是：

    1 、把目標(biāo)函數(shù)test_demo(是一個(gè)變量名，里面存的是目標(biāo)函數(shù)的引用) 傳入time_consume函數(shù)，被參數(shù)func接收，這時(shí)func也是目標(biāo)函數(shù)的引用 func和test_demo指向同一個(gè)函數(shù)對(duì)象    
    2 、time_consume函數(shù)定義了內(nèi)部函數(shù)inner，在inner里調(diào)用func，
    這用到閉包的原理（閉包原理：外函數(shù)結(jié)束的時(shí)會(huì)把自身的引用綁定給內(nèi)函數(shù)），外函數(shù)結(jié)束的時(shí)候會(huì)把func綁定給內(nèi)函數(shù)，供內(nèi)函數(shù)來(lái)使用
    3、 外函數(shù)結(jié)束的時(shí)候把自己創(chuàng)建的內(nèi)函數(shù)的引用inner返回給test_demo接收，
     這時(shí)test_demo已經(jīng)不是原來(lái)編寫的目標(biāo)函數(shù)了，test_demo可以理解成是一個(gè)inner函數(shù)的實(shí)例對(duì)象，再執(zhí)行test_demo() 的時(shí)候?qū)嶋H上執(zhí)行了inner()的一個(gè)對(duì)象
    4、 再執(zhí)行test_demo() 的時(shí)候 實(shí)際上執(zhí)行了inner() ：
     先執(zhí)行取時(shí)間，打印
     之后執(zhí)行func()，才是執(zhí)行目標(biāo)函數(shù)，即執(zhí)行test_demo()本身
     最后再次取時(shí)間，打印結(jié)果

大白話版本：

其實(shí)就一句話：
被裝飾函數(shù)作為裝飾器外部函數(shù)的參數(shù)傳入，在裝飾器的內(nèi)部函數(shù)中執(zhí)行被裝飾函數(shù)，并外加其他的代碼片段A，
這樣被裝飾函數(shù)除了具備自身的邏輯外，也擁有了裝飾器內(nèi)部函數(shù)中代碼片段A的邏輯。使得無(wú)需修改被裝飾函數(shù)，
就增強(qiáng)了被裝飾函數(shù)的功能。

再來(lái)看兩種情景。

第一個(gè)：被裝飾函數(shù)有參數(shù)

一般接口測(cè)試的test_case不會(huì)想上面例子中提到的是一個(gè)函數(shù)，而是作為一個(gè)類的方法出現(xiàn)的，比如：

運(yùn)行報(bào)錯(cuò)了，報(bào)錯(cuò)日志的意思是inner()需要0個(gè)入?yún)ⅲ潜粋魅肓?個(gè)。通過(guò)該報(bào)錯(cuò)證明了上面提到的這個(gè)結(jié)論

原因是test_demo()有參數(shù)self，而inner()沒(méi)有定義入?yún)?。怎么解決呢，給inner()定義一個(gè)可變?nèi)雲(yún)?？先?lái)看第二個(gè)問(wèn)題，最后一起來(lái)證明我們的推測(cè)吧。

第二個(gè)：被裝飾函數(shù)有返回值

問(wèn)題出現(xiàn)了，返回值打印出來(lái)是None，因?yàn)閕nner()里沒(méi)有變量去接收test_demo的返回值并返回嗎？帶著第一個(gè)問(wèn)題的推測(cè)，一起來(lái)改下代碼。

做2處改動(dòng)：

1、inner()定義可變?nèi)雲(yún)?/p>

2、inner()里定義變量去接收test_demo的返回值并return該變量

test_case作為一個(gè)類的方法出現(xiàn)的問(wèn)題解決了。

返回值也能被正常打印了。

改動(dòng)后的裝飾器可以作為一個(gè)定義裝飾器的通用模板，基本可以給各種各樣的函數(shù)來(lái)裝飾了。

def decorat_demo(func):
    def inner(*args, **kwargs):   # inner()接收可變參數(shù)
        # any code before # 定義目標(biāo)函數(shù)前的操作
        # 調(diào)用目標(biāo)函數(shù)
        res = func(*args, **kwargs)  # 定義變量接收目標(biāo)函數(shù)返回值
        # any code after # 定義目標(biāo)函數(shù)后的操作
        return res     # 返回目標(biāo)函數(shù)返回值
    return inner

到此這篇關(guān)于Python裝飾器使用接口測(cè)試的步驟的文章就介紹到這了,更多相關(guān)python裝飾器接口測(cè)試內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論