裝飾器的定義是給一個對象動態(tài)加載功能，就像打游戲時給隊友上buff一樣。一直以來，我對裝飾器用的不多，經(jīng)常會用別的方式搞定，雖然代碼丑一點，但也能用。這次遇到一個特別適合裝飾器的應用場景，就是執(zhí)行單元測試時的環(huán)境配置。

我是用pytest做單元測試，測試入口都是一個個test_打頭的函數(shù)。和unittest不一樣，pytest中并沒有setUp這個方法，雖然有fixture，但讀人家的源碼時也很少看到有人用，這次遇到問題發(fā)現(xiàn)，我靠，就是加個裝飾器的事，可以把setUp和tearDown一起做了，何必多此一舉。

應用場景用個demo舉例，由于生產(chǎn)環(huán)境和測試環(huán)境的不同，在測試環(huán)境中初始化Demo會報錯，比如下面這個模塊。

importsys

classDemo():

def__init__(self):

fail()

deffail():

ifsys.argv[0].split('\\')[-1].find('test')>-1:

raiseEnvironmentError(__name__)

defsuccess():

print(__name__)

定義了一個Demo類，初始化時會調(diào)用fail函數(shù)，這個函數(shù)在pytest環(huán)境下使用時會raise一個EnvironmentError。解決方案就是在這個Demo被調(diào)用時將模塊中的fail函數(shù)替換為success函數(shù)。兩個單元測試用例如下。

importdemo

deftest_demo():

try:

demo.Demo()

exceptExceptionase:

assertisinstance(e,EnvironmentError)

@replace#替換環(huán)境的裝飾器

deftest_replace_demo():

demo.Demo()

其中，replace就是替換環(huán)境用的裝飾器。裝飾器代碼如下。

defreplace(fun):#定義裝飾器，傳入函數(shù)名

_=demo.fail#保存模塊中的fail，以便后面恢復

demo.fail=demo.success#更改

definner():#閉包函數(shù)

fun()

demo.fail=_#恢復模塊中fail函數(shù)

returninner()

用裝飾器配置單元測試環(huán)境真是優(yōu)雅無比，寫完后頓時腰不酸背不痛了呢。。。

裝飾器的難點之一就是閉包(closure)，閉包這個翻譯在中文里沒有很形象的對應關系，這造成了理解障礙。這種例子挺多的，就像單例模式最廣泛的用途并不是提供一個的“單例”，叫“超級全局宇宙唯一變量”比較好。協(xié)程的功能里一點‘協(xié)’的作用都沒有，叫“想在什么時候掛起就在什么適合掛起程”比較好。這樣一想，閉包是不是叫“跨作用域包”或“腳踏兩條船包”或“閉合環(huán)境打包”比較好。

閉包的一個定義是這樣的：

Closuresarefunctionsthatrefertoindependent(free)variables(variablesthatareusedlocally,butdefinedinanenclosingscope).Inotherwords,thesefunctions'remember'theenvironmentinwhichtheywerecreated.

閉包的關鍵能力之一是獲取上級作用域，另一個關鍵點在于python中函數(shù)是一個對象，可以傳來傳去，比如作為返回值。這兩點結(jié)合起來就可以將函數(shù)所處的環(huán)境打包傳到目標區(qū)域。在上面一個例子中inner就是一個閉包函數(shù)，它在這里的作用就是將inner函數(shù)之前的那個區(qū)域，就是第二行和第三行。

下面還有個閉包的例子，可以看到在inner外定義了a，b，在inner中使用了a，在test_closure函數(shù)實例化后可以看看其__closure__方法中的對象，這個__closure__只包含了a，并沒包含b，因為b沒有在inner中使用，被垃圾回收了，而a保留了下來，而a就是由于閉包的特性保留下來的，可以用pdb來看看__closure__中是否保留了b。

deftest_closure():

a,b=1,2

definner():

print(a)

returninner

close=test_closure()

print(close.__closure__[0])#

另外，python中的閉包和javascript中的閉包是一個意思，可能需要實現(xiàn)的功能沒js中那么多，但理解python閉包時可以參考js的教程。

以上內(nèi)容為大家介紹了Python使用裝飾器在執(zhí)行單元測試時配置環(huán)境，希望對大家有所幫助，如果想要了解更多Python相關知識，請關注IT培訓機構(gòu):千鋒教育。