Как обойти "sys.exit ()" в тесте на питоне?

Question

Похоже, что тестирование Python завершится при обнаружении "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 framework.

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