偶看unittest中,发现一个很好用的功能函数doCleanups,看看官方是怎么解释的:
doCleanups()This method is called unconditionally after tearDown(), or after setUp() if setUp() raises an exception.It is responsible for calling all the cleanup functions added by addCleanup(). If you need cleanup functions to be called prior to tearDown() then you can call doCleanups() yourself.doCleanups() pops methods off the stack of cleanup functions one at a time, so it can be called at any time.
该功能函数在2.7之后就支持了
来来来,简单用中文说明一下吧,大概意思如下:
无条件的执行改函数,在tearnDown()之后或者在setUp()之后即使setUp失败的情况下也会执行 与addCleanup搭配使用
看完后大家都明白了吧,平时我们使用tearDown函数时,当setUp运行失败时,tearDown就不执行了,所以就会有遗留资源的存在,正好doCleanups帮我们解决了这个困扰,不用再写try...except....finally了。
下面,用个实例来看看doCleanups怎么运行的
#coding:utf-8'''Created on 2016年8月31日@author: zq'''import unittestclass my(unittest.TestCase): def a(self): print "aaaa" def setUp(self): print "setUp" raise IOError,"errorororororo" #这里特意让setUp产生错误 def test_1(self): '''i dont konw''' print "test_1" def tearDown(self): print 'this tearDown' def doCleanups(self): print "this is cleanups" def test_2(self): print "test_2" @classmethod def tearDownClass(cls): print "teardownClass...." if __name__=="__main__": test=unittest.TestSuite() test.addTest(my('test_1')) test.addTest(my('test_2')) runner=unittest.TextTestRunner() runner.run(test)
我们运行后看看结果:
setUpthis is cleanupssetUpthis is cleanupsteardownClass....EE
看看,从运行结果中可以看出。即使setUp出现错误的情况下,doCleanups还是运行了。