我正在使用 unittest 和 mock 來測試一個看起來像這樣的腳本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()我的測試腳本看起來像這樣 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方法，它將調用 sys.exit(1) 并且會在 unittest 運行中導致一些問題，所以，我不想call_other在測試期間調用 sys.exit(1) 。但是，如果我像上面那樣模擬call_other方法（in this_line），它將簡單地打印一些東西并繼續執行 method f。意思是，它將執行 print 語句（in this_print）這在實際程序中不應該是這種情況，當捕獲到異常時，它將執行 sys.exit(1) 并停止程序。當捕獲到異常時，如何使用 unittest 和 mock 實現相同的目標我想停止執行此測試用例并繼續執行下一個測試用例。如何做到這一點？請幫忙
1 回答
達令說
TA貢獻1821條經驗 獲得超6個贊
您可以使用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)
添加回答
舉報
0/150
提交
取消