如何解決python nosetest中的「sys.exit（）」問題？

Question

python nosetest在遇到「sys.exit（）」時會退出，並且這個內置模擬不起作用。感謝您的建議。

Answer 1

您可以嘗試捕獲SystemExit異常。當有人打電話給sys.exit()時，會引發這個問題。

with self.assertRaises(SystemExit): 
    myFunctionThatSometimesCallsSysExit() 

Answer 2

import sys 
sys.exit = lambda *x: None 

請記住，程序可能相當不希望sys.exit()後繼續，所以修補它可能沒有實際的幫助......

Answer 3

如果您使用mock修補sys.exit，你可能會錯誤地修補它。我

這個小測試工作正常：

import sys 
from mock import patch 

def myfunction(): 
    sys.exit(1) 

def test_myfunction(): 
    with patch('foo.sys.exit') as exit_mock: 
     myfunction() 
     assert exit_mock.called 

調用時：

nosetests foo.py 

輸出：

. 
---------------------------------------------------------------------- 
Ran 1 test in 0.001s 

OK 

Answer 4

這是unittest框架的例子。

with self.assertRaises(SystemExit) as cm: 
    my_function_that_uses_sys_exit() 
self.assertEqual(cm.exception.code, expected_return_code)