Как издеваться над методом Python OptionParser.error (), который выполняет sys.exit ()?

Я пытаюсь выполнить модульное тестирование кода, который выглядит следующим образом:

def main():
    parser = optparse.OptionParser(description='This tool is cool', prog='cool-tool')
    parser.add_option('--foo', action='store', help='The foo option is self-explanatory')
    options, arguments = parser.parse_args()
    if not options.foo:
        parser.error('--foo option is required')
    print "Your foo is %s." % options.foo
    return 0

if __name__ == '__main__':
   sys.exit(main())

С кодом, который выглядит так:

@patch('optparse.OptionParser')
def test_main_with_missing_p4clientsdir_option(self, mock_optionparser):
    #
    # setup
    #
    optionparser_mock = Mock()
    mock_optionparser.return_value = optionparser_mock
    options_stub = Mock()
    options_stub.foo = None
    optionparser_mock.parse_args.return_value = (options_stub, sentinel.arguments)
    def parser_error_mock(message):
        self.assertEquals(message, '--foo option is required')
        sys.exit(2)
    optionparser_mock.error = parser_error_mock

    #
    # exercise & verify
    #
    self.assertEquals(sut.main(), 2)

Я использую Mock Майкла Фурда и нос для запуска тестов.

Когда я запускаю тест, я получаю:

  File "/Users/dspitzer/Programming/Python/test-optparse-error/tests/sut_tests.py", line 27, in parser_error_mock
    sys.exit(2)
SystemExit: 2

----------------------------------------------------------------------
Ran 1 test in 0.012s

FAILED (errors=1)

Проблема в том, что OptionParser.error выполняет sys.exit (2), и поэтому main (), естественно, полагается на это. Но нос или unittest обнаруживают (ожидаемый) sys.exit (2) и не проходят тест.

Я могу пройти тест, добавив «return 2» под вызов parser.error () в main () и удалив вызов sys.exit () из parser_error_mock (), но мне неприятно изменять тестируемый код, чтобы позвольте тесту пройти. Есть ли лучшее решение?

Обновление: ответ df работает, хотя правильный вызов - self.assertRaises (SystemExit , сут. основная) ».

Это означает, что тест проходит независимо от числа в sys.exit () в parser_error_mock (). Есть ли способ проверить код выхода?

Кстати, тест будет более надежным, если я добавлю:

self.assertEquals(optionparser_mock.method_calls, [('add_option', ('--foo',), {'action': 'store', 'help': 'The foo option is self-explanatory'}), ('parse_args', (), {})])

в конце.

Обновление 2: я могу проверить код выхода, заменив self.assertRaises (SystemExit, sut.main) на:

try:
    sut.main()
except SystemExit, e:
    self.assertEquals(type(e), type(SystemExit()))
    self.assertEquals(e.code, 2)
except Exception, e:
    self.fail('unexpected exception: %s' % e)
else:
    self.fail('SystemExit exception expected')

python unit-testing mocking optparse
person Daryl Spitzer    schedule 08.01.2009    source источник
comment
Первый assertEquals в вашем обновлении 2 не нужен, поскольку строка except выше будет перехватывать только исключения SystemExit.   -  person rbp    schedule 20.04.2010

Ответы (4)


arrow_upward
12
arrow_downward

Будет ли это работать вместо assertEquals?

self.assertRaises(SystemExit, sut.main, 2)

Это должно перехватить SystemExit исключение и предотвратить завершение сценария.

person dF.    schedule 09.01.2009
comment
Я не принял ваш ответ, так как он не совсем правильный. Но поскольку вы дали мне понимание, которое привело меня к ответу (см. Обновления к вопросу), я дам вам пару дней, чтобы отредактировать ваш ответ, а затем я приму. - person Daryl Spitzer; 09.01.2009

arrow_upward
1
arrow_downward

Как отмечалось в моих обновлениях на мой вопрос, мне пришлось изменить ответ dF на:

self.assertRaises(SystemExit, sut.main)

... и я придумал несколько более длинных фрагментов для проверки кода выхода.

[Примечание: я принял свой ответ, но удалю этот ответ и приму dF, если он обновит свой ответ.]

person Daryl Spitzer    schedule 18.01.2009

arrow_upward
0
arrow_downward

Вероятно, в этом вопросе содержится какая-то новая информация:

Java: как проверить методы, вызывающие System.exit ()?

person guerda    schedule 09.01.2009
comment
Я прочитал его быстро и, хотя это та же проблема (в Java), я не нашел здесь ничего применимого (для Python). Ты? - person Daryl Spitzer; 09.01.2009

arrow_upward
0
arrow_downward

Добавьте @raises в вашу тестовую функцию. То есть:

from nose.tools import raises

@raises(SystemExit)
@patch('optparse.OptionParser')
def test_main_with_missing_p4clientsdir_option(self, mock_optionparser):
    # Setup
    ...
    sut.main()
person imolit    schedule 03.12.2012