我正在尝试对一些看起来像这样的代码进行单元测试:
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)
我正在使用Michael Foord 的 Mock和 nose 来运行测试。
当我运行测试时,我得到:
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() 自然依赖于此。但是鼻子或单元测试检测到(预期的) sys.exit(2) 并且测试失败。
我可以通过在 main() 中的 parser.error() 调用下添加“return 2”并从 parser_error_mock() 中删除 sys.exit() 调用来使测试通过,但我发现将被测代码修改为允许测试通过。有更好的解决方案吗?
更新:df的答案有效,尽管正确的调用是“self.assertRaises(SystemExit, sut.main)”。
这意味着测试通过了 parser_error_mock() 中 sys.exit() 中的任何数字。有什么方法可以测试退出代码吗?
顺便说一句,如果我添加,测试会更加健壮:
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')