Python使用裝飾器在執(zhí)行單元測(cè)試時(shí)配置環(huán)境

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

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

應(yīng)用場(chǎng)景用個(gè)demo舉例，由于生產(chǎn)環(huán)境和測(cè)試環(huán)境的不同，在測(cè)試環(huán)境中初始化Demo會(huì)報(bào)錯(cuò)，比如下面這個(gè)模塊。

importsys

classDemo():

def__init__(self):

fail()

deffail():

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

raiseEnvironmentError(__name__)

defsuccess():

print(__name__)

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

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，以便后面恢復(fù)

demo.fail=demo.success#更改

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

fun()

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

returninner()

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

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

閉包的一個(gè)定義是這樣的：

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

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

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

deftest_closure():

a,b=1,2

definner():

print(a)

returninner

close=test_closure()

print(close.__closure__[0])#

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

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