我正在使用 unittest 和 mock 來測(cè)試一個(gè)看起來像這樣的腳本class Hi: def call_other(self): perform some operation sys.exit(1) def f(self): try: res = self.do_something() a = self.something_else(res) except Exception as e: print(e) call_other() print("hi after doing something") -----> (this_print) def process(self) self.f()我的測(cè)試腳本看起來像這樣 class Test_hi(unittest.TestCase) def mock_call_other(self): print("called during error") def test_fail_scenario(): import Hi class here h = Hi() h.process() h.do_something = mock.Mock(retrun_value="resource") h.something_else = mock.Mock(side_effect=Exception('failing on purpose for testing')) h.call_other(side_effect=self.mock_call_other) -----> (this_line)如果我不模擬call_other方法，它將調(diào)用 sys.exit(1) 并且會(huì)在 unittest 運(yùn)行中導(dǎo)致一些問題，所以，我不想call_other在測(cè)試期間調(diào)用 sys.exit(1) 。但是，如果我像上面那樣模擬call_other方法（in this_line），它將簡(jiǎn)單地打印一些東西并繼續(xù)執(zhí)行 method f。意思是，它將執(zhí)行 print 語句（in this_print）這在實(shí)際程序中不應(yīng)該是這種情況，當(dāng)捕獲到異常時(shí)，它將執(zhí)行 sys.exit(1) 并停止程序。當(dāng)捕獲到異常時(shí)，如何使用 unittest 和 mock 實(shí)現(xiàn)相同的目標(biāo)我想停止執(zhí)行此測(cè)試用例并繼續(xù)執(zhí)行下一個(gè)測(cè)試用例。如何做到這一點(diǎn)？請(qǐng)幫忙
1 回答
達(dá)令說
TA貢獻(xiàn)1821條經(jīng)驗(yàn) 獲得超6個(gè)贊
您可以使用unittest' 的功能來斷言您是否期望不需要模擬的異常：
import unittest
import sys
class ToTest:
def foo(self):
raise SystemExit(1)
def bar(self):
sys.exit(1)
def foo_bar(self):
print("This is okay")
return 0
class Test(unittest.TestCase):
def test_1(self):
with self.assertRaises(SystemExit) as cm:
ToTest().foo()
self.assertEqual(cm.exception.code, 1)
def test_2(self):
with self.assertRaises(SystemExit) as cm:
ToTest().bar()
self.assertEqual(cm.exception.code, 1)
def test_3(self):
self.assertEqual(ToTest().foo_bar(), 0)
添加回答
舉報(bào)
0/150
提交
取消