在前面的介紹中，我們對unittest進行了分享介紹，那麼在實際的應用中，因為客觀原因需要對失敗，錯誤的測試用例進行重試，所以呢，現有的unittest的框架無法滿足，那麼我們可以去改造下是否能夠滿足呢。本文帶領大家去剖析如何改寫？
首先呢，我們去試著去找下，我們執行時在BSTestRunner、TextTestRunner或者main，都可以執行用例，那麼我們可以看下這些類或者方法裡面如何實現的。
BSTestRunner呼叫方式如下
TextTestRunner的方法是，
使用main方法最後的呼叫也是這個函式。詳細的我們可以看下，首先呼叫這個函式，然後在看實際的呼叫.
最後的呼叫也是這個函式
所以我們就要在裡面的方法去查詢適合我們使用的方法。
在註釋中，我們可以發現在stopTest的方法中可以對其進行改寫。
def stopTest(self, test):
"""Called when the given test has been run"""
self._restoreStdout()
self._mirrorOutput = False
那麼我們應該如何改寫呢，我們梳理下我們的思路。
- 1.傳遞重試次數，預設不需要重試
- 2.在用例執行的錯誤，標記為需要重試
- 3.在該條用例執行完畢後，我們判斷是否需要重試，重試次數是否滿足
- 4.如果需要重試，則儲存最新的從測試結果。
那麼我們開始按照上面的思路進行改造。
程式碼如下
import sys,copy
from io import StringIO as StringIO
TestResult = unittest.TestResult
class MyResult(TestResult):
def __init__(self, verbosity=1, trynum=0):
#預設次數是0
TestResult.__init__(self)
self.outputBuffer = StringIO()
self.stdout0 = None
self.stderr0 = None
self.success_count = 0
self.failure_count = 0
self.error_count = 0
self.verbosity = verbosity
self.trynnum = trynum
self.result = []
self.trys=0#
self.istry=False def startTest(self, test):
TestResult.startTest(self, test)
self.stdout0 = sys.stdout
self.stderr0 = sys.stderr def complete_output(self):
if self.stdout0:
sys.stdout = self.stdout0
sys.stderr = self.stderr0
self.stdout0 = None
self.stderr0 = None
return self.outputBuffer.getvalue() def stopTest(self, test):
#判斷是否要重試
if self.istry is True :
#如果執行的次數小於重試的次數 就重試
if self.trys < self.trynnum :
#刪除最後一個結果
reslut = self.result.pop(-1)
#判斷結果，如果是錯誤就把錯誤的個數減掉
#如果是失敗，就把失敗的次數減掉
if reslut[0] == 1:
self.failure_count -= 1
else:
self.error_count -= 1
sys.stderr.write('{}:用例正在重試中。。。' .format(test.id())+ '\n')
#深copy用例
test = copy.copy(test)
#重試次數增加+1
self.trys += 1
#測試
test(self)
else:
self.istry=False
self.trys =0
self.complete_output() def addSuccess(self, test):
#成功就不要重試
self.istry = False
self.success_count += 1
TestResult.addSuccess(self, test)
output = self.complete_output()
self.result.append((0, test, output, ''))
if self.verbosity > 1:
sys.stderr.write('ok ')
sys.stderr.write(str(test))
sys.stderr.write('\n')
else:
sys.stderr.write('.') def addError(self, test, err):
#重試+1，錯誤次數+1
self.istry = True
self.error_count += 1
TestResult.addError(self, test, err)
_, _exc_str = self.errors[-1]
output = self.complete_output()
self.result.append((2, test, output, _exc_str))
if self.verbosity > 1:
sys.stderr.write('E ')
sys.stderr.write(str(test))
sys.stderr.write('\n')
else:
sys.stderr.write('E') def addFailure(self, test, err):
self.istry = True
TestResult.startTestRun(self)
self.failure_count += 1
TestResult.addFailure(self, test, err)
_, _exc_str = self.failures[-1]
output = self.complete_output()
self.result.append((1, test, output, _exc_str))
if self.verbosity > 1:
sys.stderr.write('F ')
sys.stderr.write(str(test))
sys.stderr.write('\n')
else:
sys.stderr.write('F') def stop(self) -> None:
pass
這樣改造完畢了，我們可以去試試
if __name__ == "__main__":
suitone=suite()
rse=MyResult(trynum=3)
suitone.run(rse)
執行的結果如下：
目前改造滿足我們重試用例的需求，改造完畢。
上面只是一個簡單的改造，滿足了對於失敗的測試用例的重試，其實很簡單，我們有了需求，去根據我們的需求去查詢需要改造的程式碼即可，我們直接繼承原來的類，對需要修改的地方 進行修改，已滿足我們的需求。
歡迎關注我的個人公眾號
